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

uncountable/types/api/recipes/create_recipes.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,8 +8,8 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import base as base_t
-from ... import recipe_metadata as recipe_metadata_t
+from ... import base_t
+from ... import recipe_metadata_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/disassociate_recipe_as_input.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,8 +8,8 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import async_batch as async_batch_t
-from ... import identifier as identifier_t
+from ... import async_batch_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/edit_recipe_inputs.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -11,9 +11,9 @@ from pkgs.strenum_compat import StrEnum
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_union_annotation
-from ... import identifier as identifier_t
-from ... import recipe_inputs as recipe_inputs_t
-from ... import recipe_workflow_steps as recipe_workflow_steps_t
+from ... import identifier_t
+from ... import recipe_inputs_t
+from ... import recipe_workflow_steps_t
 
 __all__: list[str] = [
     "AnnotationEdit",

uncountable/types/api/recipes/get_curve.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,8 +8,8 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import base as base_t
-from ... import curves as curves_t
+from ... import base_t
+from ... import curves_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/get_recipe_calculations.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -9,8 +9,8 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
-from ... import base as base_t
-from ... import calculations as calculations_t
+from ... import base_t
+from ... import calculations_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/get_recipe_links.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,8 +8,8 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import base as base_t
-from ... import recipe_links as recipe_links_t
+from ... import base_t
+from ... import recipe_links_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/get_recipe_names.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import base as base_t
+from ... import base_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/get_recipe_output_metadata.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -9,8 +9,8 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
-from ... import base as base_t
-from ... import recipe_output_metadata as recipe_output_metadata_t
+from ... import base_t
+from ... import recipe_output_metadata_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/get_recipes_data.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -9,13 +9,13 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
-from ... import base as base_t
-from ... import experiment_groups as experiment_groups_t
-from ... import inputs as inputs_t
-from ... import outputs as outputs_t
-from ... import recipe_metadata as recipe_metadata_t
-from ... import recipe_tags as recipe_tags_t
-from ... import workflows as workflows_t
+from ... import base_t
+from ... import experiment_groups_t
+from ... import inputs_t
+from ... import outputs_t
+from ... import recipe_metadata_t
+from ... import recipe_tags_t
+from ... import workflows_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/lock_recipes.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -9,7 +9,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
 from dataclasses import dataclass
-from ... import identifier as identifier_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/remove_recipe_from_project.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import identifier as identifier_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/set_recipe_inputs.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -9,7 +9,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
-from ... import base as base_t
+from ... import base_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/set_recipe_metadata.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,8 +8,8 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import identifier as identifier_t
-from ... import recipe_metadata as recipe_metadata_t
+from ... import identifier_t
+from ... import recipe_metadata_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/set_recipe_output_annotations.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -11,8 +11,8 @@ from pkgs.strenum_compat import StrEnum
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_union_annotation
-from ... import base as base_t
-from ... import identifier as identifier_t
+from ... import base_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "AnnotationEdit",

uncountable/types/api/recipes/set_recipe_outputs.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -9,9 +9,9 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
-from ... import base as base_t
-from ... import recipes as recipes_t
-from ... import response as response_t
+from ... import base_t
+from ... import recipes_t
+from ... import response_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/set_recipe_tags.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -10,8 +10,8 @@ from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
 from dataclasses import dataclass
 from pkgs.serialization import serial_class
-from ... import base as base_t
-from ... import identifier as identifier_t
+from ... import base_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/unarchive_recipes.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,7 +8,7 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import identifier as identifier_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/recipes/unlock_recipes.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -9,7 +9,7 @@ import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from pkgs.strenum_compat import StrEnum
 from dataclasses import dataclass
-from ... import identifier as identifier_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/api/triggers/run_trigger.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 from __future__ import annotations
@@ -8,8 +8,8 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from ... import entity as entity_t
-from ... import response as response_t
+from ... import entity_t
+from ... import response_t
 
 __all__: list[str] = [
     "Arguments",

uncountable/types/async_batch.py

@@ -1,58 +1,11 @@
-# DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# 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
-from dataclasses import dataclass
-from pkgs.serialization import serial_class
-from . import base as base_t
-
-__all__: list[str] = [
-    "AsyncBatchActionReturn",
-    "AsyncBatchRequest",
-    "AsyncBatchRequestPath",
-    "QueuedAsyncBatchRequest",
-]
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-class AsyncBatchRequestPath(StrEnum):
-    CREATE_RECIPE = "recipes/create_recipe"
-    SET_RECIPE_METADATA = "recipes/set_recipe_metadata"
-    SET_RECIPE_TAGS = "recipes/set_recipe_tags"
-    EDIT_RECIPE_INPUTS = "recipes/edit_recipe_inputs"
-    ARCHIVE_RECIPES = "recipes/archive"
-    LOCK_RECIPES = "recipes/lock_recipes"
-    UNLOCK_RECIPES = "recipes/unlock_recipes"
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@serial_class(
-    unconverted_values={"data"},
-)
-@dataclass(kw_only=True)
-class AsyncBatchRequest:
-    path: AsyncBatchRequestPath
-    data: base_t.JsonValue
-    batch_reference: str
-    depends_on: typing.Optional[list[str]] = None
-
-
-# DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True)
-class AsyncBatchActionReturn:
-    modification_made: bool
-    result_id: typing.Optional[base_t.ObjectId] = None
-
-
 # DO NOT MODIFY -- This file is generated by type_spec
-@dataclass(kw_only=True, frozen=True, eq=True)
-class QueuedAsyncBatchRequest:
-    path: AsyncBatchRequestPath
-    batch_reference: str
+# Kept only for SDK backwards compatibility
+from .async_batch_t import AsyncBatchRequestPath as AsyncBatchRequestPath
+from .async_batch_t import AsyncBatchRequest as AsyncBatchRequest
+from .async_batch_t import AsyncBatchActionReturn as AsyncBatchActionReturn
+from .async_batch_t import QueuedAsyncBatchRequest as QueuedAsyncBatchRequest
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/async_batch_processor.py

@@ -1,6 +1,6 @@
 # DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# ruff: noqa: E402 Q003
 # fmt: off
 # isort: skip_file
 # ruff: noqa: PLR0904
@@ -8,14 +8,15 @@ from __future__ import annotations
 import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
-from uncountable.types import async_batch as async_batch_t
-from uncountable.types import base as base_t
+import uncountable.types.api.equipment.associate_equipment_input as associate_equipment_input_t
+from uncountable.types import async_batch_t
+from uncountable.types import base_t
 import uncountable.types.api.recipes.create_recipe as create_recipe_t
 import uncountable.types.api.recipes.edit_recipe_inputs as edit_recipe_inputs_t
-from uncountable.types import identifier as identifier_t
-from uncountable.types import recipe_identifiers as recipe_identifiers_t
-from uncountable.types import recipe_metadata as recipe_metadata_t
-from uncountable.types import recipe_workflow_steps as recipe_workflow_steps_t
+from uncountable.types import identifier_t
+from uncountable.types import recipe_identifiers_t
+from uncountable.types import recipe_metadata_t
+from uncountable.types import recipe_workflow_steps_t
 import uncountable.types.api.recipes.set_recipe_metadata as set_recipe_metadata_t
 import uuid
 from abc import ABC, abstractmethod
@@ -31,6 +32,41 @@ class AsyncBatchProcessorBase(ABC):
     def send(self) -> base_t.ObjectId:
         ...
 
+    def associate_equipment_input(
+        self,
+        *,
+        equipment_key: identifier_t.IdentifierKey,
+        material_family_ids: list[base_t.ObjectId],
+        depends_on: typing.Optional[list[str]] = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Create or return the input association for equipment
+
+        :param equipment_key: Identifier of the equipment to associate
+        :param material_family_ids: The list of material families to add the input to. This must be non-empty
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = associate_equipment_input_t.Arguments(
+            equipment_key=equipment_key,
+            material_family_ids=material_family_ids,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.ASSOCIATE_EQUIPMENT_INPUT,
+            data=json_data,
+            depends_on=depends_on,
+            batch_reference=batch_reference,
+        )
+
+        self._enqueue(req)
+
+        return async_batch_t.QueuedAsyncBatchRequest(
+            path=req.path,
+            batch_reference=req.batch_reference,
+        )
+
     def create_recipe(
         self,
         *,

uncountable/types/async_batch_t.py

@@ -0,0 +1,59 @@
+# 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
+from dataclasses import dataclass
+from pkgs.serialization import serial_class
+from . import base_t
+
+__all__: list[str] = [
+    "AsyncBatchActionReturn",
+    "AsyncBatchRequest",
+    "AsyncBatchRequestPath",
+    "QueuedAsyncBatchRequest",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class AsyncBatchRequestPath(StrEnum):
+    CREATE_RECIPE = "recipes/create_recipe"
+    SET_RECIPE_METADATA = "recipes/set_recipe_metadata"
+    SET_RECIPE_TAGS = "recipes/set_recipe_tags"
+    EDIT_RECIPE_INPUTS = "recipes/edit_recipe_inputs"
+    ARCHIVE_RECIPES = "recipes/archive"
+    LOCK_RECIPES = "recipes/lock_recipes"
+    UNLOCK_RECIPES = "recipes/unlock_recipes"
+    ASSOCIATE_EQUIPMENT_INPUT = "equipment/associate_equipment_input"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    unconverted_values={"data"},
+)
+@dataclass(kw_only=True)
+class AsyncBatchRequest:
+    path: AsyncBatchRequestPath
+    data: base_t.JsonValue
+    batch_reference: str
+    depends_on: typing.Optional[list[str]] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class AsyncBatchActionReturn:
+    modification_made: bool
+    result_id: typing.Optional[base_t.ObjectId] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True, frozen=True, eq=True)
+class QueuedAsyncBatchRequest:
+    path: AsyncBatchRequestPath
+    batch_reference: str
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/base.py

@@ -1,85 +1,11 @@
-# DO NOT MODIFY -- This file is generated by type_spec
 # flake8: noqa: F821
-# ruff: noqa: E402
+# 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
-
-__all__: list[str] = [
-    "JsonScalar",
-    "JsonValue",
-    "ObjectId",
-]
 # DO NOT MODIFY -- This file is generated by type_spec
-# === START section from base.prepart.py ===
-"""
-    Types that type_spec will use in the emitted files.
-"""
-from typing import Union, Any, TYPE_CHECKING
-from collections.abc import Mapping, Sequence
-
-# These two are part of the core output, thus don't duplicate here
-# from decimal import Decimal
-# import datetime
-
-# Decimal/datetime are included as this list includes JSON serializable objects
-# It is preferred to use ExtJsonScalar and leave JsonScalar as deprecated, as
-# a type that we intend on migrating.
-JsonScalar = Union[str, float, bool, Decimal, None, datetime.datetime, datetime.date]
-ExtJsonScalar = JsonScalar
-
-# PureJsonScalar/PureJsonValue should be used in places where the type
-# is expected to already serialized and not expecting any high-order types.
-# this type is fully representable in the JSON format, and in JavaScript
-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}"
-
-
-if TYPE_CHECKING:
-    JsonValue = Union[JsonScalar, Mapping[str, "JsonValue"], Sequence["JsonValue"]]
-    ExtJsonValue = JsonValue
-    PureJsonValue = Union[
-        PureJsonScalar,
-        Mapping[str, "PureJsonValue"],
-        Sequence["PureJsonValue"]
-    ]
-else:
-    JsonValue = Union[JsonScalar, dict[str, Any], list[Any]]
-    ExtJsonValue = JsonValue
-    PureJsonValue = Union[PureJsonScalar, dict[str, Any], list[Any]]
-
-ObjectId = int
-
-
-def is_pure_json_value(value: ExtJsonValue) -> bool:
-    """
-    Checks if the runtime type of value is a PureJsonValue.
-    """
-    if value is None:
-        return True
-    if isinstance(value, (int, float, bool, str)):
-        return True
-
-    if isinstance(value, list):
-        for item in value:
-            if not is_pure_json_value(item):
-                return False
-        return True
-
-    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
+# Kept only for SDK backwards compatibility
 
-    return False
-# === END section from base.prepart.py ===
+from .base_t import JsonScalar as JsonScalar
+from .base_t import JsonValue as JsonValue
+from .base_t import ObjectId as ObjectId
 # DO NOT MODIFY -- This file is generated by type_spec