orca-python 0.7.0__py3-none-any.whl → 0.7.2__py3-none-any.whl

orca_python/exceptions.py

@@ -10,6 +10,10 @@ class InvalidAlgorithmReturnType(BaseOrcaException):
     """Raised when the return type of an algorithm is not valid"""
 
 
+class InvalidWindowArgument(BaseOrcaException):
+    """Raised when an argument to the Window class is not valid"""
+
+
 class InvalidDependency(BaseOrcaException):
     """Raised when a dependency is invalid"""
 

orca_python/main.py

@@ -10,8 +10,11 @@ import re
 import sys
 import asyncio
 import logging
+import datetime as dt
 import traceback
 
+from google.protobuf import json_format, timestamp_pb2
+
 logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
@@ -48,6 +51,7 @@ from service_pb2_grpc import OrcaProcessorServicer
 from orca_python import envs
 from orca_python.exceptions import (
     InvalidDependency,
+    InvalidWindowArgument,
     InvalidAlgorithmArgument,
     InvalidAlgorithmReturnType,
 )
@@ -133,11 +137,41 @@ class NoneResult:
 returnResult = StructResult | ArrayResult | ValueResult | NoneResult
 
 
+@dataclass
+class Window:
+    time_from: dt.datetime
+    time_to: dt.datetime
+    name: str
+    version: str
+    origin: str
+    metadata: Dict[str, Any] = field(default_factory=dict)
+
+
 @dataclass
 class ExecutionParams:
-    window: pb.Window
+    window: Window
     dependencies: Optional[Iterable[pb.AlgorithmResult]] = None
 
+    def __init__(
+        self,
+        window: Window | pb.Window,
+        dependencies: Optional[Iterable[pb.AlgorithmResult]] = None,
+    ):
+        if isinstance(window, Window):
+            self.window = window
+        elif isinstance(window, pb.Window):
+            self.window = Window(
+                time_from=window.time_from.ToDatetime(),
+                time_to=window.time_to.ToDatetime(),
+                name=window.window_type_name,
+                version=window.window_type_version,
+                origin=window.origin,
+                metadata=json_format.MessageToDict(window.metadata),
+            )
+        else:
+            raise InvalidWindowArgument(f"window of type {type(window)} not handled")
+        self.dependencies = dependencies
+
 
 class AlgorithmFn(Protocol):
     def __call__(
@@ -148,16 +182,6 @@ class AlgorithmFn(Protocol):
 T = TypeVar("T", bound=AlgorithmFn)
 
 
-@dataclass
-class Window:
-    time_from: int
-    time_to: int
-    name: str
-    version: str
-    origin: str
-    metadata: Dict[str, Any] = field(default_factory=dict)
-
-
 def EmitWindow(window: Window) -> None:
     """
     Emits a window to Orca-core.
@@ -167,9 +191,15 @@ def EmitWindow(window: Window) -> None:
     """
     LOGGER.info(f"Emitting window: {window}")
 
+    _time_from = timestamp_pb2.Timestamp()
+    _time_from.FromDatetime(window.time_from)
+
+    _time_to = timestamp_pb2.Timestamp()
+    _time_to.FromDatetime(window.time_to)
+
     window_pb = pb.Window()
-    window_pb.time_to = window.time_to
-    window_pb.time_from = window.time_from
+    window_pb.time_to = _time_from
+    window_pb.time_from = _time_to
     window_pb.window_type_name = window.name
     window_pb.window_type_version = window.version
     window_pb.origin = window.origin

{orca_python-0.7.0.dist-info → orca_python-0.7.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: orca-python
-Version: 0.7.0
+Version: 0.7.2
 Summary: Python SDK for the Predixus Orca product
 Author: Frederick Mannings
 Author-email: contact@predixus.com

{orca_python-0.7.0.dist-info → orca_python-0.7.2.dist-info}/RECORD

@@ -8,10 +8,10 @@ vendor/validate_pb2.pyi,sha256=Qyebm_XUdrvo8As3XovHu7vVsHXYJtTdP3bYUmQEecs,30245
 vendor/validate_pb2_grpc.py,sha256=qIlTS0MGHE7myQ7tCMbS-pPqA9FMZl_Hn9Mh_uq2o9o,896
 orca_python/__init__.py,sha256=tHVOEF69iiLzbymMTuEKdbGSqeaAzpc-BiZY7VY1Gz4,362
 orca_python/envs.py,sha256=zSuukNSpEw78VpSSIDAA9KRAurWedQMwx1qyQg9f8pk,560
-orca_python/exceptions.py,sha256=WjkKGQvmS8Z7L63rIROnLFjA-8sGX5R9714xgD1j_7E,512
-orca_python/main.py,sha256=3nRxUNOqU4IXuQYkyjQVFLdsgAwemFWScCwuE6RKxs0,29408
+orca_python/exceptions.py,sha256=el7a5VUd21nNCTSTgVn5T9gyULncrQgKyFQghWmApnw,629
+orca_python/main.py,sha256=0o4ZzVpwKUsNk5nGdp0aBI3N-Q7_JkK62UErjBhftlc,30453
 orca_python/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-orca_python-0.7.0.dist-info/LICENSE,sha256=sxb8X8qhbZ_JaCBFmoIriJzA-jelKmh86sAKQsIRN_I,1062
-orca_python-0.7.0.dist-info/METADATA,sha256=o-6F9dfhcXJ5Oa4ivf4iMddUd_0i_ZqoC5rPdqbdTxU,3019
-orca_python-0.7.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-orca_python-0.7.0.dist-info/RECORD,,
+orca_python-0.7.2.dist-info/LICENSE,sha256=sxb8X8qhbZ_JaCBFmoIriJzA-jelKmh86sAKQsIRN_I,1062
+orca_python-0.7.2.dist-info/METADATA,sha256=Eg_nxSuf7bc9F6p5pFFNr-44UldKOgPeZhe8ZWxJZ0M,3019
+orca_python-0.7.2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+orca_python-0.7.2.dist-info/RECORD,,