UncountablePythonSDK 0.0.14__py3-none-any.whl → 0.0.16__py3-none-any.whl

{UncountablePythonSDK-0.0.14.dist-info → UncountablePythonSDK-0.0.16.dist-info}/METADATA

@@ -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.dist-info → UncountablePythonSDK-0.0.16.dist-info}/RECORD

@@ -71,7 +71,7 @@ uncountable/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/core/__init__.py,sha256=GCXRRhhYv-yKzqKjAij7GcNiOr90BBbgdhhpusWtmUc,100
 uncountable/core/client.py,sha256=jcJBGwe31fOqfGiUJzW_KMcfE0hax55jgVVliALBNLQ,4160
 uncountable/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/construct_client.py,sha256=fRIoxkK6HUf2HVAJkFieMvk4fAGOCgFD9y99brgXWdQ,912
+uncountable/integration/construct_client.py,sha256=r6M5pnIO0fKcjf5d_AREPtWZ6AkWgcjkdu_jHQEYlT8,1084
 uncountable/integration/cron.py,sha256=TIPqMPMSMtMJTu4aXwLf6QY-OLrpmyITLDp48UIr4Ok,919
 uncountable/integration/entrypoint.py,sha256=sPtUeKJEivf0iqY-H4LHWhyeUHMZK_U4CrDj1Kw8Ld0,1395
 uncountable/integration/job.py,sha256=5QuCjeze9vtEirntP9P9hDD3LVrBn5pApDWh2IZeEiY,817
@@ -80,7 +80,7 @@ uncountable/integration/types.py,sha256=n9idu2_qHOA5CQdE6NK8HS6aZ8ugTZKTfTTnm4sy
 uncountable/integration/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/db/connect.py,sha256=iI9e8a2hfbFP-dvH0MGLsrG-RpM0dHKCL-oCLkah9hs,181
 uncountable/integration/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-uncountable/integration/executors/script_executor.py,sha256=l_mniGlS9Ni6FdmspN7bvMWVocOaXKCo8FM9SFW8n8c,600
+uncountable/integration/executors/script_executor.py,sha256=6oMPAFe0PUdqt76e8jMi4vXszGVsVHLULob7Qbl3o38,816
 uncountable/types/__init__.py,sha256=SbBDFhHu5iqnvnYjWCBOkjg3_mXuaBHEGfnq021wfHw,3873
 uncountable/types/base.py,sha256=7yMC3sCQ13wd92896Ga2XbZffCqq98SKv09rFGb1ulg,2682
 uncountable/types/calculations.py,sha256=16J-KKMp-I8ZQUkYNmKCHfAn6DGb99cFinALcDIdGHY,562
@@ -139,7 +139,7 @@ uncountable/types/api/recipes/get_recipe_output_metadata.py,sha256=L9s2ykPP4pd02
 uncountable/types/api/recipes/get_recipes_data.py,sha256=dOKokz6rJp3AiqNrF8rAZFlmJSs3ejdNIJhwKw0Utr0,5317
 uncountable/types/api/recipes/set_recipe_inputs.py,sha256=sHEwPocBucWRnnoX7nbNaFqdflxFkqdjuVydNezqluY,1326
 uncountable/types/api/recipes/set_recipe_outputs.py,sha256=QYq39TNchQ80ET1C77OE9fwhbu_HmIoEDmrQJHkkCu0,1609
-UncountablePythonSDK-0.0.14.dist-info/METADATA,sha256=K3rzjPqELhDb7eb2I8e9qI2B3EYiDfA_ke3XHb2vsCg,1298
-UncountablePythonSDK-0.0.14.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-UncountablePythonSDK-0.0.14.dist-info/top_level.txt,sha256=8ga1DCSWt4Uc_XJ5cR0QrDQuyoeo2uoSzaAJdQT2KBo,36
-UncountablePythonSDK-0.0.14.dist-info/RECORD,,
+UncountablePythonSDK-0.0.16.dist-info/METADATA,sha256=XEMklRrImLVb1W7XuCzBmUq-TS3GyYMs4tQc-mfCQb0,1298
+UncountablePythonSDK-0.0.16.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+UncountablePythonSDK-0.0.16.dist-info/top_level.txt,sha256=8ga1DCSWt4Uc_XJ5cR0QrDQuyoeo2uoSzaAJdQT2KBo,36
+UncountablePythonSDK-0.0.16.dist-info/RECORD,,

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"
             )
 

uncountable/integration/executors/script_executor.py

@@ -1,18 +1,19 @@
 
-
+import os
 import importlib
 import inspect
 from uncountable.integration.job import Job
-from uncountable.integration.types import JobExecutorScript
+from uncountable.integration.types import JobExecutorScript, ProfileMetadata
 
 
-def resolve_script_executor(executor: JobExecutorScript) -> type[Job]:
-    job_module = importlib.import_module(executor.import_path)
+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 Job in job_class.__bases__:
+        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}"
+    ), f"expected exactly one job class in {executor.import_path}, found {len(found_jobs)}"
     return found_jobs[0]