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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: orca-python
-Version: 0.7.4
+Version: 0.7.6
 Summary: Python SDK for the Predixus Orca product
 Author: Frederick Mannings
 Author-email: contact@predixus.com
@@ -81,10 +81,10 @@ proc.Start()
 Then run your python file to register it with orca-core:
 
 ```bash
- ORCASERVER=grpc://localhost:32770 HOST=172.18.0.1 python main.py
+ ORCA_CORE=grpc://localhost:32770 PROCESSOR_ADDRESS=172.18.0.1 PROCESSOR_PORT=8080 python main.py
 ```
 
-Replace the contents of `ORCASERVER` and `HOST` with the output of `orca status`.
+Replace the contents of `ORCA_CORE`, `PROCESSOR_ADDRESS` and `PROCESSOR_PORT` with the output of `orca status`.
 
 6. Emit a window to orcacore
    TBD

{orca_python-0.7.4 → orca_python-0.7.6}/README.md

@@ -65,10 +65,10 @@ proc.Start()
 Then run your python file to register it with orca-core:
 
 ```bash
- ORCASERVER=grpc://localhost:32770 HOST=172.18.0.1 python main.py
+ ORCA_CORE=grpc://localhost:32770 PROCESSOR_ADDRESS=172.18.0.1 PROCESSOR_PORT=8080 python main.py
 ```
 
-Replace the contents of `ORCASERVER` and `HOST` with the output of `orca status`.
+Replace the contents of `ORCA_CORE`, `PROCESSOR_ADDRESS` and `PROCESSOR_PORT` with the output of `orca status`.
 
 6. Emit a window to orcacore
    TBD

orca_python-0.7.6/orca_python/envs.py

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

{orca_python-0.7.4 → 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.4 → orca_python-0.7.6}/pyproject.toml

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

orca_python-0.7.4/orca_python/envs.py

@@ -1,24 +0,0 @@
-import os
-from typing import Tuple
-
-from orca_python.exceptions import MissingDependency
-
-
-def getenvs() -> Tuple[str, ...]:
-    orcaserver = os.getenv("ORCASERVER", "")
-    if orcaserver == "":
-        MissingDependency("ORCASERVER is required")
-    orcaserver = orcaserver.lstrip("grpc://")
-
-    port = os.getenv("PORT", "")
-    if port == "":
-        MissingDependency("PORT required")
-
-    host = os.getenv("HOST", "")
-    if host == "":
-        MissingDependency("HOST is required")
-
-    return orcaserver, port, host
-
-
-ORCASERVER, PORT, HOST = getenvs()