UncountablePythonSDK 0.0.52__py3-none-any.whl → 0.0.131__py3-none-any.whl

uncountable/types/async_batch_t.py

@@ -1,5 +1,4 @@
 # DO NOT MODIFY -- This file is generated by type_spec
-# flake8: noqa: F821
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
@@ -7,16 +6,19 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from pkgs.strenum_compat import StrEnum
+from enum import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
 from . import base_t
+from . import entity_t
 
 __all__: list[str] = [
     "AsyncBatchActionReturn",
     "AsyncBatchRequest",
     "AsyncBatchRequestPath",
+    "AsyncBatchResultValue",
     "QueuedAsyncBatchRequest",
+    "SavedAsyncBatchActionReturn",
 ]
 
 
@@ -32,29 +34,74 @@ class AsyncBatchRequestPath(StrEnum):
     ASSOCIATE_EQUIPMENT_INPUT = "equipment/associate_equipment_input"
     INVOKE_UPLOADER = "uploader/invoke_uploader"
     ASSOCIATE_RECIPE_AS_INPUT = "recipes/associate_recipe_as_input"
+    CLEAR_RECIPE_OUTPUTS = "recipes/clear_recipe_outputs"
+    GRANT_ENTITY_PERMISSIONS = "entity/grant_entity_permissions"
+    SET_ENTITY_FIELD_VALUES = "entity/set_entity_field_values"
+    ASSOCIATE_RECIPE_AS_LOT = "recipes/associate_recipe_as_lot"
+    CREATE_OR_UPDATE_ENTITY = "entity/create_or_update_entity"
+    ADD_TIME_SERIES_DATA = "recipes/add_time_series_data"
+    LOOKUP_ENTITY = "entity/lookup_entity"
+    CREATE_RECIPE_LINK = "recipe_links/create_recipe_link"
+    UPSERT_CONDITION_MATCH = "condition_parameters/upsert_condition_match"
+    COMPLETE_ASYNC_UPLOAD = "runsheet/complete_async_upload"
+    CREATE_MIX_ORDER = "recipes/create_mix_order"
+    PUSH_NOTIFICATION = "integrations/push_notification"
+    COMPLETE_ASYNC_PARSE = "uploader/complete_async_parse"
+    SET_RECIPE_TOTAL = "recipes/set_recipe_total"
+    TRANSITION_ENTITY_PHASE = "entity/transition_entity_phase"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.async_batch.AsyncBatchRequest",
     unconverted_values={"data"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AsyncBatchRequest:
     path: AsyncBatchRequestPath
     data: base_t.JsonValue
     batch_reference: str
-    depends_on: typing.Optional[list[str]] = None
+    depends_on: list[str] | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclasses.dataclass(kw_only=True)
+@serial_class(
+    named_type_path="sdk.async_batch.AsyncBatchResultValue",
+    unconverted_values={"value"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class AsyncBatchResultValue:
+    data_key: str
+    value: base_t.JsonValue
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.async_batch.AsyncBatchActionReturn",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class AsyncBatchActionReturn:
     modification_made: bool
-    result_id: typing.Optional[base_t.ObjectId] = None
+    result_id: base_t.ObjectId | None = None
+    entity: entity_t.Entity | None = None
+    result_values: list[AsyncBatchResultValue] | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclasses.dataclass(kw_only=True, frozen=True, eq=True)
+@serial_class(
+    named_type_path="sdk.async_batch.SavedAsyncBatchActionReturn",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class SavedAsyncBatchActionReturn:
+    identifier: str
+    result_data: AsyncBatchActionReturn
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.async_batch.QueuedAsyncBatchRequest",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True, frozen=True, eq=True)  # type: ignore[literal-required]
 class QueuedAsyncBatchRequest:
     path: AsyncBatchRequestPath
     batch_reference: str

uncountable/types/async_jobs.py

@@ -0,0 +1,8 @@
+# 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 .async_jobs_t import AsyncJobType as AsyncJobType
+from .async_jobs_t import AsyncJobStatus as AsyncJobStatus
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/async_jobs_t.py

@@ -0,0 +1,52 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# 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 enum import StrEnum
+from pkgs.serialization import serial_string_enum
+
+__all__: list[str] = [
+    "ASYNC_JOB_STATUS_FIELD_REF_NAME",
+    "ASYNC_JOB_TYPE_FIELD_REF_NAME",
+    "AsyncJobStatus",
+    "AsyncJobType",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_string_enum(
+    labels={
+        "async_batch_load": "Async Batch Load",
+        "integration_server_run": "Integration Server Run",
+    },
+)
+class AsyncJobType(StrEnum):
+    ASYNC_BATCH_LOAD = "async_batch_load"
+    INTEGRATION_SERVER_RUN = "integration_server_run"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_string_enum(
+    labels={
+        "in_progress": "In Progress",
+        "completed": "Completed",
+        "error": "Error",
+    },
+)
+class AsyncJobStatus(StrEnum):
+    IN_PROGRESS = "in_progress"
+    COMPLETED = "completed"
+    ERROR = "error"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+ASYNC_JOB_TYPE_FIELD_REF_NAME = "core_async_job_jobType"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+ASYNC_JOB_STATUS_FIELD_REF_NAME = "core_async_job_status"
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/auth_retrieval.py

@@ -0,0 +1,11 @@
+# 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 .auth_retrieval_t import AuthRetrievalType as AuthRetrievalType
+from .auth_retrieval_t import AuthRetrievalBase as AuthRetrievalBase
+from .auth_retrieval_t import AuthRetrievalOAuth as AuthRetrievalOAuth
+from .auth_retrieval_t import AuthRetrievalBasic as AuthRetrievalBasic
+from .auth_retrieval_t import AuthRetrieval as AuthRetrieval
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/auth_retrieval_t.py

@@ -0,0 +1,75 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# 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 enum import StrEnum
+import dataclasses
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+from . import base_t
+from . import secret_retrieval_t
+
+__all__: list[str] = [
+    "AuthRetrieval",
+    "AuthRetrievalBase",
+    "AuthRetrievalBasic",
+    "AuthRetrievalOAuth",
+    "AuthRetrievalType",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class AuthRetrievalType(StrEnum):
+    OAUTH = "oauth"
+    BASIC = "basic"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.auth_retrieval.AuthRetrievalBase",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class AuthRetrievalBase:
+    type: AuthRetrievalType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.auth_retrieval.AuthRetrievalOAuth",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class AuthRetrievalOAuth(AuthRetrievalBase):
+    type: typing.Literal[AuthRetrievalType.OAUTH] = AuthRetrievalType.OAUTH
+    refresh_token_secret: secret_retrieval_t.SecretRetrieval
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.auth_retrieval.AuthRetrievalBasic",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class AuthRetrievalBasic(AuthRetrievalBase):
+    type: typing.Literal[AuthRetrievalType.BASIC] = AuthRetrievalType.BASIC
+    api_id_secret: secret_retrieval_t.SecretRetrieval
+    api_key_secret: secret_retrieval_t.SecretRetrieval
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+AuthRetrieval = typing.Annotated[
+    AuthRetrievalOAuth | AuthRetrievalBasic,
+    serial_union_annotation(
+    named_type_path="sdk.auth_retrieval.AuthRetrieval",
+        discriminator="type",
+        discriminator_map={
+            "oauth": AuthRetrievalOAuth,
+            "basic": AuthRetrievalBasic,
+        },
+    ),
+]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/base.py

@@ -1,4 +1,3 @@
-# flake8: noqa: F821
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file

uncountable/types/base_t.py

@@ -1,5 +1,4 @@
 # DO NOT MODIFY -- This file is generated by type_spec
-# flake8: noqa: F821
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
@@ -39,10 +38,16 @@ PureJsonScalar = Union[str, float, bool, None]
 # Regular expressions for identifying ref names and IDs. Ref names should be
 # using this regular expression as a constriant in the database.
 REF_NAME_REGEX = r"^[a-zA-Z0-9_/-]+$"
-# Does not support nonpositive integers at the moment.
-ID_REGEX = r"[1-9][0-9]{0,20}"
+REF_NAME_STRICT_REGEX_STRING = "^[a-zA-Z_][a-zA-Z0-9_]*$"
+REF_NAME_STRICT_REGEX = rf"{REF_NAME_STRICT_REGEX_STRING}"
+# Ids matching a strict integer number are converted to integers
+ID_REGEX = r"-?[1-9][0-9]{0,20}"
 
 
+# ENABLE_SLOTS should be removed after slots have been tested locally
+import os
+ENABLE_SLOTS = os.environ.get("UNC_ENABLE_DATACLASS_SLOTS") == "true"
+
 if TYPE_CHECKING:
     JsonValue = Union[JsonScalar, Mapping[str, "JsonValue"], Sequence["JsonValue"]]
     ExtJsonValue = JsonValue
@@ -69,16 +74,13 @@ def is_pure_json_value(value: ExtJsonValue) -> bool:
         return True
 
     if isinstance(value, list):
-        for item in value:
-            if not is_pure_json_value(item):
-                return False
-        return True
+        return all(is_pure_json_value(item) for item in value)
 
     if isinstance(value, dict):
-        for key, item in value.items():
-            if not is_pure_json_value(key) or not is_pure_json_value(item):
-                return False
-        return True
+        return all(
+            is_pure_json_value(key) and is_pure_json_value(item)
+            for key, item in value.items()
+        )
 
     return False
 # === END section from base.prepart.py ===

uncountable/types/calculations.py

@@ -1,4 +1,3 @@
-# flake8: noqa: F821
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file

uncountable/types/calculations_t.py

@@ -1,5 +1,4 @@
 # DO NOT MODIFY -- This file is generated by type_spec
-# flake8: noqa: F821
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
@@ -8,6 +7,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
+from pkgs.serialization import serial_class
 from . import base_t
 
 __all__: list[str] = [
@@ -16,7 +16,10 @@ __all__: list[str] = [
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclasses.dataclass(kw_only=True)
+@serial_class(
+    named_type_path="sdk.calculations.Calculation",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Calculation:
     id: base_t.ObjectId
     name: str

uncountable/types/chemical_structure.py

@@ -1,4 +1,3 @@
-# flake8: noqa: F821
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file

uncountable/types/chemical_structure_t.py

@@ -1,5 +1,4 @@
 # DO NOT MODIFY -- This file is generated by type_spec
-# flake8: noqa: F821
 # ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
@@ -9,6 +8,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 import dataclasses
 from pkgs.serialization import serial_class
+from . import base_t
 
 __all__: list[str] = [
     "ChemicalStructure",
@@ -17,11 +17,12 @@ __all__: list[str] = [
 
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
+    named_type_path="sdk.chemical_structure.ChemicalStructure",
     to_string_values={"molecular_weight"},
 )
-@dataclasses.dataclass(kw_only=True)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class ChemicalStructure:
-    ketcher_file: typing.Optional[str] = None
-    gross_formula: typing.Optional[str] = None
-    molecular_weight: typing.Optional[Decimal] = None
+    ketcher_file: str | None = None
+    gross_formula: str | None = None
+    molecular_weight: Decimal | None = None
 # DO NOT MODIFY -- This file is generated by type_spec