micromegas 0.1.6__tar.gz → 0.1.8__tar.gz

{micromegas-0.1.6 → micromegas-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: micromegas
-Version: 0.1.6
+Version: 0.1.8
 Summary: Python analytics client for https://github.com/madesroches/micromegas/
 Author: Marc-Antoine Desroches
 Author-email: madesroches@gmail.com
@@ -16,6 +16,7 @@ Requires-Dist: protobuf (>=5.27.1,<6.0.0)
 Requires-Dist: pyarrow (>=16.0.0,<17.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
 Requires-Dist: tabulate (>=0.9.0,<0.10.0)
+Requires-Dist: tqdm (>=4.66.5,<5.0.0)
 Description-Content-Type: text/markdown
 
 # Micromegas

{micromegas-0.1.6 → micromegas-0.1.8}/micromegas/client.py

@@ -15,6 +15,13 @@ class Client:
             headers=self.headers,
         )
 
+    def find_stream(self, stream_id):
+        return request.request(
+            self.analytics_base_url + "find_stream",
+            {"stream_id": stream_id},
+            headers=self.headers,
+        )
+    
     def query_processes(self, begin, end, limit):
         return request.request(
             self.analytics_base_url + "query_processes",
@@ -145,19 +152,7 @@ class Client:
             except cbor2.CBORDecodeEOF:
                 break
 
-    def create_or_update_partitions(
-        self, view_set_name, view_instance_id, begin, end, partition_delta_seconds
-    ):
-        args = {
-            "view_set_name": view_set_name,
-            "view_instance_id": view_instance_id,
-            "begin": time.format_datetime(begin),
-            "end": time.format_datetime(end),
-            "partition_delta_seconds": partition_delta_seconds,
-        }
-        self.__stream_request("create_or_update_partitions", args)
-
-    def merge_partitions(
+    def materialize_partitions(
         self, view_set_name, view_instance_id, begin, end, partition_delta_seconds
     ):
         args = {
@@ -167,7 +162,7 @@ class Client:
             "end": time.format_datetime(end),
             "partition_delta_seconds": partition_delta_seconds,
         }
-        self.__stream_request("merge_partitions", args)
+        self.__stream_request("materialize_partitions", args)
 
     def retire_partitions(
         self, view_set_name, view_instance_id, begin, end, partition_delta_seconds

{micromegas-0.1.6 → micromegas-0.1.8}/micromegas/time.py

@@ -4,15 +4,17 @@ import re
 
 def format_datetime(value):
     nonetype = type(None)
-    match type(value):
-        case datetime.datetime:
-            if value.tzinfo is None:
-                raise RuntimeError("datetime needs a valid time zone")
-            return value.isoformat()
-        case pandas.Timestamp:
-            return value.isoformat()
-        case nonetype:
-            return None
+    value_type = type(value)
+    if value_type == datetime.datetime:
+        if value.tzinfo is None:
+            raise RuntimeError("datetime needs a valid time zone")
+        return value.isoformat()
+    elif value_type == pandas.Timestamp:
+        return value.isoformat()
+    elif value_type == str:
+        return format_datetime(datetime.datetime.fromisoformat(value))
+    elif value_type == type(None):
+        return None
     raise RuntimeError("value of unknown type in format_datetime")
 
 def parse_time_delta(user_string):

{micromegas-0.1.6 → micromegas-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "micromegas"
-version = "0.1.6"
+version = "0.1.8"
 description = "Python analytics client for https://github.com/madesroches/micromegas/"
 authors = ["Marc-Antoine Desroches <madesroches@gmail.com>"]
 readme = "README.md"
@@ -14,6 +14,7 @@ pyarrow = "^16.0.0"
 python = "^3.10"
 requests = "^2.31.0"
 tabulate = "^0.9.0"
+tqdm = "^4.66.5"
 
 
 [tool.poetry.group.test.dependencies]

micromegas-0.1.6/micromegas/thirdparty/perfetto/protos/perfetto/config/test_config_pb2.py

@@ -1,38 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: protos/perfetto/config/test_config.proto
-# Protobuf Python Version: 5.27.1
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    5,
-    27,
-    1,
-    '',
-    'protos/perfetto/config/test_config.proto'
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(protos/perfetto/config/test_config.proto\x12\x0fperfetto.protos\"\x8d\x04\n\nTestConfig\x12\x15\n\rmessage_count\x18\x01 \x01(\r\x12\x1f\n\x17max_messages_per_second\x18\x02 \x01(\r\x12\x0c\n\x04seed\x18\x03 \x01(\r\x12\x14\n\x0cmessage_size\x18\x04 \x01(\r\x12\x1e\n\x16send_batch_on_register\x18\x05 \x01(\x08\x12=\n\x0c\x64ummy_fields\x18\x06 \x01(\x0b\x32\'.perfetto.protos.TestConfig.DummyFields\x1a\xc3\x02\n\x0b\x44ummyFields\x12\x14\n\x0c\x66ield_uint32\x18\x01 \x01(\r\x12\x13\n\x0b\x66ield_int32\x18\x02 \x01(\x05\x12\x14\n\x0c\x66ield_uint64\x18\x03 \x01(\x04\x12\x13\n\x0b\x66ield_int64\x18\x04 \x01(\x03\x12\x15\n\rfield_fixed64\x18\x05 \x01(\x06\x12\x16\n\x0e\x66ield_sfixed64\x18\x06 \x01(\x10\x12\x15\n\rfield_fixed32\x18\x07 \x01(\x07\x12\x16\n\x0e\x66ield_sfixed32\x18\x08 \x01(\x0f\x12\x14\n\x0c\x66ield_double\x18\t \x01(\x01\x12\x13\n\x0b\x66ield_float\x18\n \x01(\x02\x12\x14\n\x0c\x66ield_sint64\x18\x0b \x01(\x12\x12\x14\n\x0c\x66ield_sint32\x18\x0c \x01(\x11\x12\x14\n\x0c\x66ield_string\x18\r \x01(\t\x12\x13\n\x0b\x66ield_bytes\x18\x0e \x01(\x0c')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'protos.perfetto.config.test_config_pb2', _globals)
-if not _descriptor._USE_C_DESCRIPTORS:
-  DESCRIPTOR._loaded_options = None
-  _globals['_TESTCONFIG']._serialized_start=62
-  _globals['_TESTCONFIG']._serialized_end=587
-  _globals['_TESTCONFIG_DUMMYFIELDS']._serialized_start=264
-  _globals['_TESTCONFIG_DUMMYFIELDS']._serialized_end=587
-# @@protoc_insertion_point(module_scope)

micromegas-0.1.6/micromegas/thirdparty/perfetto/protos/perfetto/trace/test_event_pb2.py

@@ -1,39 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# NO CHECKED-IN PROTOBUF GENCODE
-# source: protos/perfetto/trace/test_event.proto
-# Protobuf Python Version: 5.27.1
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import runtime_version as _runtime_version
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-_runtime_version.ValidateProtobufRuntimeVersion(
-    _runtime_version.Domain.PUBLIC,
-    5,
-    27,
-    1,
-    '',
-    'protos/perfetto/trace/test_event.proto'
-)
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from protos.perfetto.trace.track_event import debug_annotation_pb2 as protos_dot_perfetto_dot_trace_dot_track__event_dot_debug__annotation__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&protos/perfetto/trace/test_event.proto\x12\x0fperfetto.protos\x1a\x38protos/perfetto/trace/track_event/debug_annotation.proto\"\xfb\x02\n\tTestEvent\x12\x0b\n\x03str\x18\x01 \x01(\t\x12\x11\n\tseq_value\x18\x02 \x01(\r\x12\x0f\n\x07\x63ounter\x18\x03 \x01(\x04\x12\x0f\n\x07is_last\x18\x04 \x01(\x08\x12\x37\n\x07payload\x18\x05 \x01(\x0b\x32&.perfetto.protos.TestEvent.TestPayload\x1a\xf2\x01\n\x0bTestPayload\x12\x0b\n\x03str\x18\x01 \x03(\t\x12\x36\n\x06nested\x18\x02 \x03(\x0b\x32&.perfetto.protos.TestEvent.TestPayload\x12\x15\n\rsingle_string\x18\x04 \x01(\t\x12\x12\n\nsingle_int\x18\x05 \x01(\x05\x12\x15\n\rrepeated_ints\x18\x06 \x03(\x05\x12\x1f\n\x17remaining_nesting_depth\x18\x03 \x01(\r\x12;\n\x11\x64\x65\x62ug_annotations\x18\x07 \x03(\x0b\x32 .perfetto.protos.DebugAnnotation')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'protos.perfetto.trace.test_event_pb2', _globals)
-if not _descriptor._USE_C_DESCRIPTORS:
-  DESCRIPTOR._loaded_options = None
-  _globals['_TESTEVENT']._serialized_start=118
-  _globals['_TESTEVENT']._serialized_end=497
-  _globals['_TESTEVENT_TESTPAYLOAD']._serialized_start=255
-  _globals['_TESTEVENT_TESTPAYLOAD']._serialized_end=497
-# @@protoc_insertion_point(module_scope)