micromegas 0.1.0__tar.gz → 0.1.1__tar.gz

{micromegas-0.1.0 → micromegas-0.1.1}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: micromegas
-Version: 0.1.0
-Summary: 
+Version: 0.1.1
+Summary: Python analytics client for https://github.com/madesroches/micromegas/
 Author: Marc-Antoine Desroches
 Author-email: madesroches@gmail.com
 Requires-Python: >=3.10,<4.0

{micromegas-0.1.0 → micromegas-0.1.1}/micromegas/__init__.py

@@ -1 +1,2 @@
 from . import request
+from . import client

micromegas-0.1.1/micromegas/client.py

@@ -0,0 +1,56 @@
+from . import request
+
+
+class Client:
+    def __init__(self, base_url, headers={}):
+        self.analytics_base_url = base_url + "analytics/"
+        self.headers = headers
+
+    def query_processes(self, begin, end, limit):
+        return request.request(
+            self.analytics_base_url + "query_processes",
+            {"begin": begin.isoformat(), "end": end.isoformat(), "limit": limit},
+            headers=self.headers,
+        )
+
+    def query_streams(self, begin, end, limit, tag_filter=None):
+        args = {
+            "begin": begin.isoformat(),
+            "end": end.isoformat(),
+            "limit": limit,
+        }
+
+        if tag_filter is not None:
+            args["tag_filter"] = tag_filter
+
+        return request.request(
+            self.analytics_base_url + "query_streams",
+            args,
+            headers=self.headers,
+        )
+
+    def query_blocks(self, begin, end, limit, stream_id):
+        args = {
+            "begin": begin.isoformat(),
+            "end": end.isoformat(),
+            "limit": limit,
+            "stream_id": stream_id,
+        }
+
+        return request.request(
+            self.analytics_base_url + "query_blocks",
+            args,
+            headers=self.headers,
+        )
+
+    def query_spans(self, begin, end, limit, stream_id):
+        return request.request(
+            self.analytics_base_url + "query_spans",
+            {
+                "begin": begin.isoformat(),
+                "end": end.isoformat(),
+                "limit": limit,
+                "stream_id": stream_id,
+            },
+            headers=self.headers,
+        )

{micromegas-0.1.0 → micromegas-0.1.1}/micromegas/request.py

@@ -3,9 +3,11 @@ import io
 import pyarrow.parquet as pq
 import requests
 
-def request(url, args):
+
+def request(url, args, headers={}):
     response = requests.post(
         url,
+        headers=headers,
         data=cbor2.dumps(args),
     )
     if response.status_code != 200:

{micromegas-0.1.0 → micromegas-0.1.1}/pyproject.toml

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