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

uncountable/types/job_definition.py

@@ -0,0 +1,26 @@
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .job_definition_t import JobDefinitionType as JobDefinitionType
+from .job_definition_t import JobExecutorType as JobExecutorType
+from .job_definition_t import AuthRetrievalType as AuthRetrievalType
+from .job_definition_t import GenericUploadDataSourceType as GenericUploadDataSourceType
+from .job_definition_t import JobExecutorBase as JobExecutorBase
+from .job_definition_t import JobExecutorScript as JobExecutorScript
+from .job_definition_t import GenericUploadDataSourceBase as GenericUploadDataSourceBase
+from .job_definition_t import GenericUploadDataSourceSFTP as GenericUploadDataSourceSFTP
+from .job_definition_t import GenericUploadDataSource as GenericUploadDataSource
+from .job_definition_t import JobExecutorGenericUpload as JobExecutorGenericUpload
+from .job_definition_t import JobExecutor as JobExecutor
+from .job_definition_t import JobDefinitionBase as JobDefinitionBase
+from .job_definition_t import CronJobDefinition as CronJobDefinition
+from .job_definition_t import JobDefinition as JobDefinition
+from .job_definition_t import AuthRetrievalBase as AuthRetrievalBase
+from .job_definition_t import AuthRetrievalEnv as AuthRetrievalEnv
+from .job_definition_t import AuthRetrieval as AuthRetrieval
+from .job_definition_t import ProfileDefinition as ProfileDefinition
+from .job_definition_t import ProfileMetadata as ProfileMetadata
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/job_definition_t.py

@@ -0,0 +1,203 @@
+# 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
+from . import generic_upload_t
+from . import secret_retrieval_t
+
+__all__: list[str] = [
+    "AuthRetrieval",
+    "AuthRetrievalBase",
+    "AuthRetrievalEnv",
+    "AuthRetrievalType",
+    "CronJobDefinition",
+    "GenericUploadDataSource",
+    "GenericUploadDataSourceBase",
+    "GenericUploadDataSourceSFTP",
+    "GenericUploadDataSourceType",
+    "JobDefinition",
+    "JobDefinitionBase",
+    "JobDefinitionType",
+    "JobExecutor",
+    "JobExecutorBase",
+    "JobExecutorGenericUpload",
+    "JobExecutorScript",
+    "JobExecutorType",
+    "ProfileDefinition",
+    "ProfileMetadata",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class JobDefinitionType(StrEnum):
+    CRON = "cron"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class JobExecutorType(StrEnum):
+    SCRIPT = "script"
+    GENERIC_UPLOAD = "generic_upload"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class AuthRetrievalType(StrEnum):
+    ENV = "env"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class GenericUploadDataSourceType(StrEnum):
+    SFTP = "sftp"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class JobExecutorBase:
+    type: JobExecutorType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class JobExecutorScript(JobExecutorBase):
+    type: typing.Literal[JobExecutorType.SCRIPT] = JobExecutorType.SCRIPT
+    import_path: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class GenericUploadDataSourceBase:
+    type: GenericUploadDataSourceType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class GenericUploadDataSourceSFTP(GenericUploadDataSourceBase):
+    type: typing.Literal[GenericUploadDataSourceType.SFTP] = GenericUploadDataSourceType.SFTP
+    host: str
+    username: str
+    pem_secret: secret_retrieval_t.SecretRetrieval
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+GenericUploadDataSource = typing.Annotated[
+    typing.Union[GenericUploadDataSourceSFTP],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "sftp": GenericUploadDataSourceSFTP,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class JobExecutorGenericUpload(JobExecutorBase):
+    type: typing.Literal[JobExecutorType.GENERIC_UPLOAD] = JobExecutorType.GENERIC_UPLOAD
+    data_source: GenericUploadDataSource
+    upload_strategy: generic_upload_t.GenericUploadStrategy
+    remote_directories: list[generic_upload_t.GenericRemoteDirectoryScope]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+JobExecutor = typing.Annotated[
+    typing.Union[JobExecutorScript, JobExecutorGenericUpload],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "script": JobExecutorScript,
+            "generic_upload": JobExecutorGenericUpload,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class JobDefinitionBase:
+    id: str
+    name: str
+    executor: JobExecutor
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class CronJobDefinition(JobDefinitionBase):
+    type: typing.Literal[JobDefinitionType.CRON] = JobDefinitionType.CRON
+    cron_spec: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+JobDefinition = typing.Annotated[
+    typing.Union[CronJobDefinition],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "cron": CronJobDefinition,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class AuthRetrievalBase:
+    type: AuthRetrievalType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class AuthRetrievalEnv(AuthRetrievalBase):
+    type: typing.Literal[AuthRetrievalType.ENV] = AuthRetrievalType.ENV
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+AuthRetrieval = typing.Annotated[
+    typing.Union[AuthRetrievalEnv],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "env": AuthRetrievalEnv,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class ProfileDefinition:
+    auth_retrieval: AuthRetrieval
+    base_url: str
+    jobs: list[JobDefinition]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class ProfileMetadata:
+    name: str
+    base_url: str
+    auth_retrieval: AuthRetrieval
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/outputs_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
 from . import units_t
 
@@ -17,7 +17,7 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleOutput:
     output_id: base_t.ObjectId
     name: str

uncountable/types/phases_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 Phase:
     id: base_t.ObjectId
     name: str

uncountable/types/recipe_identifiers_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 pkgs.serialization import serial_class
 from . import base_t
 
@@ -24,7 +24,7 @@ __all__: list[str] = [
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeIdentifierEditableName:
     type: typing.Literal["name"] = "name"
     editable_name: str
@@ -35,7 +35,7 @@ class RecipeIdentifierEditableName:
     to_string_values={"quantity_dec"},
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeIdentifierMetadata:
     type: typing.Literal["metadata"] = "metadata"
     recipe_metadata_id: base_t.ObjectId
@@ -47,7 +47,7 @@ class RecipeIdentifierMetadata:
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeIdentifierMaterialFamily:
     type: typing.Literal["material_family_id"] = "material_family_id"
     material_family_id: base_t.ObjectId

uncountable/types/recipe_links_t.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_string_enum
 from . import base_t
 
@@ -41,7 +41,7 @@ class RecipeLinkType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True, frozen=True, eq=True)
+@dataclasses.dataclass(kw_only=True, frozen=True, eq=True)
 class RecipeLink:
     recipe_id_from: base_t.ObjectId
     recipe_id_to: base_t.ObjectId

uncountable/types/recipe_metadata_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 pkgs.serialization import serial_class
 from . import base_t
 
@@ -23,7 +23,7 @@ __all__: list[str] = [
     unconverted_values={"value_json"},
     to_string_values={"value_numeric"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class MetadataValue:
     metadata_id: base_t.ObjectId
     value_numeric: typing.Optional[Decimal] = None
@@ -32,7 +32,7 @@ class MetadataValue:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class SimpleRecipeMetadataField:
     metadata_id: base_t.ObjectId
     name: str
@@ -44,7 +44,7 @@ class SimpleRecipeMetadataField:
     unconverted_values={"quantity_json"},
     to_string_values={"quantity_dec"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeMetadata:
     metadata_id: base_t.ObjectId
     quantity_dec: typing.Optional[Decimal] = None

uncountable/types/recipe_output_metadata_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 RecipeOutputMetadataField:
     id: base_t.ObjectId
     name: str

uncountable/types/recipe_tags_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 SimpleRecipeTag:
     recipe_tag_id: base_t.ObjectId
     name: str

uncountable/types/recipe_workflow_steps_t.py

@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
-from dataclasses import dataclass
+import dataclasses
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_union_annotation
 from . import identifier_t
@@ -32,7 +32,7 @@ class RecipeWorkflowStepIdentifierType(StrEnum):
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeWorkflowStepIdentifierBase:
     type: RecipeWorkflowStepIdentifierType
 
@@ -41,7 +41,7 @@ class RecipeWorkflowStepIdentifierBase:
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeWorkflowStepIdentifierDefault(RecipeWorkflowStepIdentifierBase):
     type: typing.Literal[RecipeWorkflowStepIdentifierType.DEFAULT] = RecipeWorkflowStepIdentifierType.DEFAULT
 
@@ -56,7 +56,7 @@ class RecipeWorkflowStepPosition(StrEnum):
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeWorkflowStepIdentifierWorkflowStep(RecipeWorkflowStepIdentifierBase):
     type: typing.Literal[RecipeWorkflowStepIdentifierType.WORKFLOW_STEP] = RecipeWorkflowStepIdentifierType.WORKFLOW_STEP
     workflow_step_key: identifier_t.IdentifierKey
@@ -67,7 +67,7 @@ class RecipeWorkflowStepIdentifierWorkflowStep(RecipeWorkflowStepIdentifierBase)
 @serial_class(
     parse_require={"type"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeWorkflowStepIdentifierKey:
     type: typing.Literal[RecipeWorkflowStepIdentifierType.IDENTIFIER_KEY] = RecipeWorkflowStepIdentifierType.IDENTIFIER_KEY
     recipe_workflow_step_key: identifier_t.IdentifierKey

uncountable/types/recipes_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
 
 __all__: list[str] = [
     "RecipeAttributeFormatting",
@@ -15,7 +15,7 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class RecipeAttributeFormatting:
     background_color: str
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/response_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 pkgs.serialization import serial_class
 
 __all__: list[str] = [
@@ -19,7 +19,7 @@ __all__: list[str] = [
 @serial_class(
     parse_require={"status"},
 )
-@dataclass(kw_only=True)
+@dataclasses.dataclass(kw_only=True)
 class Response:
     status: typing.Literal["ok"] = "ok"
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/secret_retrieval.py

@@ -0,0 +1,12 @@
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .secret_retrieval_t import SecretRetrievalBase as SecretRetrievalBase
+from .secret_retrieval_t import SecretRetrievalType as SecretRetrievalType
+from .secret_retrieval_t import SecretRetrievalEnv as SecretRetrievalEnv
+from .secret_retrieval_t import SecretRetrievalAWS as SecretRetrievalAWS
+from .secret_retrieval_t import SecretRetrieval as SecretRetrieval
+# DO NOT MODIFY -- This file is generated by type_spec

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