micromegas 0.3.0__tar.gz → 0.3.2__tar.gz

{micromegas-0.3.0 → micromegas-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: micromegas
-Version: 0.3.0
+Version: 0.3.2
 Summary: Python analytics client for https://github.com/madesroches/micromegas/
 Author: Marc-Antoine Desroches
 Author-email: madesroches@gmail.com
@@ -27,20 +27,22 @@ Python analytics client for https://github.com/madesroches/micromegas/
 
 ## Example usage
 
-Query the 2 most recent log entries from the analytics service
+Query the 2 most recent log entries from the flightsql service
 
 ```python
 import datetime
 import pandas as pd
 import micromegas
+import grpc
 
-BASE_URL = "http://localhost:8082/"
-client = micromegas.client.Client(BASE_URL)
+host_port = "localhost:50051"
+channel_cred = grpc.local_channel_credentials()
+client = micromegas.flightsql.client.FlightSQLClient(host_port, channel_cred)
 sql = """
 SELECT time, process_id, level, target, msg
 FROM log_entries
 WHERE level <= 4
-AND exe LIKE '%analytics%'
+AND exe LIKE '%flight%'
 ORDER BY time DESC
 LIMIT 2
 """

{micromegas-0.3.0 → micromegas-0.3.2}/README.md

@@ -4,20 +4,22 @@ Python analytics client for https://github.com/madesroches/micromegas/
 
 ## Example usage
 
-Query the 2 most recent log entries from the analytics service
+Query the 2 most recent log entries from the flightsql service
 
 ```python
 import datetime
 import pandas as pd
 import micromegas
+import grpc
 
-BASE_URL = "http://localhost:8082/"
-client = micromegas.client.Client(BASE_URL)
+host_port = "localhost:50051"
+channel_cred = grpc.local_channel_credentials()
+client = micromegas.flightsql.client.FlightSQLClient(host_port, channel_cred)
 sql = """
 SELECT time, process_id, level, target, msg
 FROM log_entries
 WHERE level <= 4
-AND exe LIKE '%analytics%'
+AND exe LIKE '%flight%'
 ORDER BY time DESC
 LIMIT 2
 """

{micromegas-0.3.0 → micromegas-0.3.2}/micromegas/flightsql/client.py

@@ -238,14 +238,13 @@ class FlightSQLClient:
                 print(row["time"], row["msg"])
 
     def materialize_partitions(
-        self, view_set_name, view_instance_id, begin, end, partition_delta_seconds
+        self, view_set_name, begin, end, partition_delta_seconds
     ):
         sql = """
           SELECT time, msg
-          FROM materialize_partitions('{view_set_name}', '{view_instance_id}', '{begin}', '{end}', {partition_delta_seconds})
+          FROM materialize_partitions('{view_set_name}', '{begin}', '{end}', {partition_delta_seconds})
         """.format(
             view_set_name=view_set_name,
-            view_instance_id=view_instance_id,
             begin=begin.isoformat(),
             end=end.isoformat(),
             partition_delta_seconds=partition_delta_seconds,
@@ -253,3 +252,56 @@ class FlightSQLClient:
         for rb in self.query_stream(sql):
             for index, row in rb.to_pandas().iterrows():
                 print(row["time"], row["msg"])
+
+    def find_process(self, process_id):
+        sql = """
+            SELECT *
+            FROM processes
+            WHERE process_id='{process_id}';
+            """.format(
+            process_id=process_id
+        )
+        return self.query(sql)
+
+    def query_streams(self, begin, end, limit, process_id=None, tag_filter=None):
+        conditions = []
+        if process_id is not None:
+            conditions.append("process_id='{process_id}'".format(process_id=process_id))
+        if tag_filter is not None:
+            conditions.append(
+                "(array_position(tags, '{tag}') is not NULL)".format(tag=tag_filter)
+            )
+        where = ""
+        if len(conditions) > 0:
+            where = "WHERE " + " AND ".join(conditions)
+        sql = """
+            SELECT *
+            FROM streams
+            {where}
+            LIMIT {limit};
+            """.format(
+            where=where, limit=limit
+        )
+        return self.query(sql, begin, end)
+
+    def query_blocks(self, begin, end, limit, stream_id):
+        sql = """
+            SELECT *
+            FROM blocks
+            WHERE stream_id='{stream_id}'
+            LIMIT {limit};
+            """.format(
+                limit=limit,stream_id=stream_id
+        )
+        return self.query(sql, begin, end)
+
+    def query_spans(self, begin, end, limit, stream_id):
+        sql = """
+            SELECT *
+            FROM view_instance('thread_spans', '{stream_id}')
+            LIMIT {limit};
+            """.format(
+                limit=limit,stream_id=stream_id
+        )
+        return self.query(sql, begin, end)
+    

{micromegas-0.3.0 → micromegas-0.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "micromegas"
-version = "0.3.0"
+version = "0.3.2"
 description = "Python analytics client for https://github.com/madesroches/micromegas/"
 authors = ["Marc-Antoine Desroches <madesroches@gmail.com>"]
 readme = "README.md"