UncountablePythonSDK 0.0.14__tar.gz → 0.0.16__tar.gz

{UncountablePythonSDK-0.0.14 → UncountablePythonSDK-0.0.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.14
+Version: 0.0.16
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{UncountablePythonSDK-0.0.14 → UncountablePythonSDK-0.0.16}/UncountablePythonSDK.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.14
+Version: 0.0.16
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{UncountablePythonSDK-0.0.14 → UncountablePythonSDK-0.0.16}/uncountable/integration/construct_client.py

@@ -8,13 +8,18 @@ from uncountable.integration.types import (
 )
 
 
-def construct_uncountable_client(
-    profile_meta: ProfileMetadata
-) -> Client:
+def _get_env_var(name: str) -> str:
+    value = os.getenv(name)
+    if value is None:
+        raise Exception(f"environment variable {name} is missing")
+    return value
+
+
+def construct_uncountable_client(profile_meta: ProfileMetadata) -> Client:
     match profile_meta.auth_retrieval:
         case AuthRetrievalEnv():
-            api_id = os.getenv(f"UNC_PROFILE_{profile_meta.name.upper()}_API_ID")
-            api_secret_key = os.getenv(
+            api_id = _get_env_var(f"UNC_PROFILE_{profile_meta.name.upper()}_API_ID")
+            api_secret_key = _get_env_var(
                 f"UNC_PROFILE_{profile_meta.name.upper()}_API_SECRET_KEY"
             )
 

UncountablePythonSDK-0.0.16/uncountable/integration/executors/script_executor.py

@@ -0,0 +1,19 @@
+
+import os
+import importlib
+import inspect
+from uncountable.integration.job import Job
+from uncountable.integration.types import JobExecutorScript, ProfileMetadata
+
+
+def resolve_script_executor(executor: JobExecutorScript, profile_metadata: ProfileMetadata) -> type[Job]:
+    job_module_path = ".".join([os.environ["UNC_PROFILES_MODULE"], profile_metadata.name, executor.import_path])
+    job_module = importlib.import_module(job_module_path)
+    found_jobs: list[type[Job]] = []
+    for _, job_class in inspect.getmembers(job_module, inspect.isclass):
+        if getattr(job_class, "_unc_job_registered", False):
+            found_jobs.append(job_class())
+    assert (
+        len(found_jobs) == 1
+    ), f"expected exactly one job class in {executor.import_path}, found {len(found_jobs)}"
+    return found_jobs[0]

UncountablePythonSDK-0.0.14/uncountable/integration/executors/script_executor.py

@@ -1,18 +0,0 @@
-
-
-import importlib
-import inspect
-from uncountable.integration.job import Job
-from uncountable.integration.types import JobExecutorScript
-
-
-def resolve_script_executor(executor: JobExecutorScript) -> type[Job]:
-    job_module = importlib.import_module(executor.import_path)
-    found_jobs: list[type[Job]] = []
-    for _, job_class in inspect.getmembers(job_module, inspect.isclass):
-        if Job in job_class.__bases__:
-            found_jobs.append(job_class())
-    assert (
-        len(found_jobs) == 1
-    ), f"expected exactly one job class in {executor.import_path}"
-    return found_jobs[0]