orca-python 0.7.5__tar.gz → 0.7.6__tar.gz

{orca_python-0.7.5 → orca_python-0.7.6}/PKG-INFO

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

{orca_python-0.7.5 → orca_python-0.7.6}/orca_python/envs.py

@@ -4,7 +4,7 @@ from typing import Tuple
 from orca_python.exceptions import MissingDependency
 
 
-def getenvs() -> Tuple[str, ...]:
+def getenvs() -> Tuple[bool, str, str, str]:
     orcaserver = os.getenv("ORCA_CORE", "")
     if orcaserver == "":
         raise MissingDependency("ORCA_CORE is required")
@@ -18,7 +18,13 @@ def getenvs() -> Tuple[str, ...]:
     if host == "":
         raise MissingDependency("PROCESSOR_ADDRESS is required")
 
-    return orcaserver, port, host
+    env = os.getenv("ENV", "")
+    if env == "production":
+        is_production = True
+    else:
+        is_production = False
 
+    return is_production, orcaserver, port, host
 
-ORCASERVER, PORT, HOST = getenvs()
+
+is_production, ORCASERVER, PORT, HOST = getenvs()

{orca_python-0.7.5 → orca_python-0.7.6}/orca_python/main.py

@@ -209,10 +209,20 @@ def EmitWindow(window: Window) -> None:
     json_format.ParseDict(window.metadata, struct_value)
     window_pb.metadata = struct_value
 
-    with grpc.insecure_channel(envs.ORCASERVER) as channel:
-        stub = service_pb2_grpc.OrcaCoreStub(channel)
-        response = stub.EmitWindow(window_pb)
-        LOGGER.info(f"Window emitted: {response}")
+    if envs.is_production:
+        # secure channel with TLS
+        with grpc.secure_channel(
+            envs.ORCASERVER, grpc.ssl_channel_credentials()
+        ) as channel:
+            stub = service_pb2_grpc.OrcaCoreStub(channel)
+            response = stub.EmitWindow(window_pb)
+            LOGGER.info(f"Window emitted: {response}")
+    else:
+        # insecure channel for local development
+        with grpc.insecure_channel(envs.ORCASERVER) as channel:
+            stub = service_pb2_grpc.OrcaCoreStub(channel)
+            response = stub.EmitWindow(window_pb)
+            LOGGER.info(f"Window emitted: {response}")
 
 
 @dataclass
@@ -641,10 +651,20 @@ class Processor(OrcaProcessorServicer):  # type: ignore
                     dep_msg.processor_name = dep.processor
                     dep_msg.processor_runtime = dep.runtime
 
-        with grpc.insecure_channel(envs.ORCASERVER) as channel:
-            stub = service_pb2_grpc.OrcaCoreStub(channel)
-            response = stub.RegisterProcessor(registration_request)
-            LOGGER.info(f"Algorithm registration response recieved: {response}")
+        if envs.is_production:
+            # secure channel with TLS
+            with grpc.secure_channel(
+                envs.ORCASERVER, grpc.ssl_channel_credentials()
+            ) as channel:
+                stub = service_pb2_grpc.OrcaCoreStub(channel)
+                response = stub.RegisterProcessor(registration_request)
+                LOGGER.info(f"Algorithm registration response received: {response}")
+        else:
+            # insecure channel for local development
+            with grpc.insecure_channel(envs.ORCASERVER) as channel:
+                stub = service_pb2_grpc.OrcaCoreStub(channel)
+                response = stub.RegisterProcessor(registration_request)
+                LOGGER.info(f"Algorithm registration response received: {response}")
 
     def Start(self) -> None:
         """

{orca_python-0.7.5 → orca_python-0.7.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "orca-python"
-version = "0.7.5"
+version = "0.7.6"
 description = "Python SDK for the Predixus Orca product"
 authors = [
   "Frederick Mannings <contact@predixus.com>"