UncountablePythonSDK 0.0.40__py3-none-any.whl → 0.0.42__py3-none-any.whl

uncountable/types/secret_retrieval_t.py

@@ -0,0 +1,69 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from pkgs.strenum_compat import StrEnum
+import dataclasses
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+
+__all__: list[str] = [
+    "SecretRetrieval",
+    "SecretRetrievalAWS",
+    "SecretRetrievalBase",
+    "SecretRetrievalEnv",
+    "SecretRetrievalType",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class SecretRetrievalBase:
+    type: SecretRetrievalType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class SecretRetrievalType(StrEnum):
+    ENV = "env"
+    AWS = "aws"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class SecretRetrievalEnv(SecretRetrievalBase):
+    type: typing.Literal[SecretRetrievalType.ENV] = SecretRetrievalType.ENV
+    env_key: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class SecretRetrievalAWS(SecretRetrievalBase):
+    type: typing.Literal[SecretRetrievalType.AWS] = SecretRetrievalType.AWS
+    secret_name: str
+    region: str
+    sub_key: typing.Optional[str] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+SecretRetrieval = typing.Annotated[
+    typing.Union[SecretRetrievalEnv, SecretRetrievalAWS],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "env": SecretRetrievalEnv,
+            "aws": SecretRetrievalAWS,
+        },
+    ),
+]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/units_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from . import base_t
 
 __all__: list[str] = [
@@ -16,7 +16,7 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleUnit:
     unit_id: base_t.ObjectId
     name: str

uncountable/types/users_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from . import base_t
 
 __all__: list[str] = [
@@ -16,7 +16,7 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleUser:
     user_id: base_t.ObjectId
     display_name: str

uncountable/types/workflows_t.py

@@ -7,7 +7,7 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from dataclasses import dataclass
+import dataclasses
 from . import base_t
 
 __all__: list[str] = [
@@ -17,14 +17,14 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleWorkflowStep:
     workflow_step_id: base_t.ObjectId
     name: typing.Optional[str]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleWorkflow:
     workflow_id: base_t.ObjectId
     name: str

uncountable/integration/types.py

@@ -1,89 +0,0 @@
-# TODO: move to type spec
-
-
-from dataclasses import dataclass
-from enum import StrEnum
-from typing import Literal
-
-
-class JobDefinitionType(StrEnum):
-    CRON = "cron"
-    # also imagine other job types like webhook-triggered or sftp-triggered,
-    # manual, etc
-
-
-class JobExecutorType(StrEnum):
-    SCRIPT = "script"
-    # also imagine builtin executors like 'sftp_sync'
-
-
-class AuthRetrievalType(StrEnum):
-    ENV = "env"
-    # also imagine secrets manager, keyvault, etc
-
-
-@dataclass
-class JobExecutorBase:
-    type: JobExecutorType
-
-
-@dataclass
-class JobExecutorScript(JobExecutorBase):
-    type: Literal[JobExecutorType.SCRIPT]
-    import_path: str
-
-
-JobExecutor = JobExecutorScript
-
-
-@dataclass
-class JobDefinitionBase:
-    id: str
-    name: str
-
-
-@dataclass
-class CronJobDefinition(JobDefinitionBase):
-    type: Literal[JobDefinitionType.CRON]
-    cron_spec: str
-    # Here we assert that the executor has to be a script, but we could add
-    # other builtin executor types that cron jobs can support later
-    executor: JobExecutorScript
-
-
-JobDefinition = CronJobDefinition
-
-
-@dataclass
-class AuthRetrievalBase:
-    type: AuthRetrievalType
-
-
-@dataclass
-class AuthRetrievalEnv:
-    # We don't really need any extra info here, we can enforce that the auth
-    # keys are named like UNC_PROFILE_{profile name}_API_SECRET_KEY. For
-    # supporting pulling secrets from secrets manager etc it will be nice to
-    # use dataclass properties to get the secret name, region etc.
-    type: Literal[AuthRetrievalType.ENV]
-
-
-AuthRetrieval = AuthRetrievalEnv
-
-
-@dataclass
-class ProfileDefinition:
-    # profile name (expected to be something like customer_name) will be
-    # obtained from the folder name instead of specified here. Forces jobs to
-    # be organized nicely in folders that separate their identities.
-    auth_retrieval: AuthRetrieval
-    base_url: str
-    jobs: list[JobDefinition]
-
-
-@dataclass
-class ProfileMetadata:
-    # supplied by inspecting the folder name that the profile is in
-    name: str
-    base_url: str
-    auth_retrieval: AuthRetrieval