UncountablePythonSDK 0.0.75__py3-none-any.whl → 0.0.77__py3-none-any.whl

pkgs/type_spec/emit_typescript_util.py

@@ -153,7 +153,7 @@ def emit_type_ts(ctx: EmitTypescriptContext, stype: builder.SpecType) -> None:
         ctx.out.write(f"export type {stype.name} = {base_type}{{}}\n")
     else:
         if isinstance(stype, builder.SpecTypeDefnObject) and len(stype.parameters) > 0:
-            full_type_name = f'{stype.name}<{", ".join(stype.parameters)}>'
+            full_type_name = f"{stype.name}<{', '.join(stype.parameters)}>"
         else:
             full_type_name = stype.name
         ctx.out.write(f"export type {full_type_name} = {base_type}{{")
@@ -201,7 +201,7 @@ def refer_to_impl(
             return f"readonly ({spec})[]" if multi else f"readonly {spec}[]", False
         if stype.defn_type.name == builder.BaseTypeName.s_union:
             return (
-                f'({" | ".join([refer_to(ctx, p) for p in stype.parameters])})',
+                f"({' | '.join([refer_to(ctx, p) for p in stype.parameters])})",
                 False,
             )
         if stype.defn_type.name == builder.BaseTypeName.s_literal:
@@ -209,11 +209,11 @@ def refer_to_impl(
             for parameter in stype.parameters:
                 assert isinstance(parameter, builder.SpecTypeLiteralWrapper)
                 parts.append(refer_to(ctx, parameter))
-            return f'({" | ".join(parts)})', False
+            return f"({' | '.join(parts)})", False
         if stype.defn_type.name == builder.BaseTypeName.s_optional:
             return f"{refer_to(ctx, stype.parameters[0])} | null", True
         if stype.defn_type.name == builder.BaseTypeName.s_tuple:
-            return f"[{", ".join([refer_to(ctx, p) for p in stype.parameters])}]", False
+            return f"[{', '.join([refer_to(ctx, p) for p in stype.parameters])}]", False
         params = ", ".join([refer_to(ctx, p) for p in stype.parameters])
         return f"{refer_to(ctx, stype.defn_type)}<{params}>", False
 

pkgs/type_spec/type_info/emit_type_info.py

@@ -234,9 +234,9 @@ def _extract_and_validate_layout(
 
         local_ref_name = None
         if group.ref_name is not None:
-            assert (
-                base_layout is None or base_layout.get(group.ref_name) is None
-            ), f"group-name-duplicate-in-base:{group.ref_name}"
+            assert base_layout is None or base_layout.get(group.ref_name) is None, (
+                f"group-name-duplicate-in-base:{group.ref_name}"
+            )
             local_ref_name = group.ref_name
 
         if group.extends:
@@ -255,9 +255,9 @@ def _extract_and_validate_layout(
         assert group_ref_name in layout, f"missing-base-group:{group_ref_name}"
 
     for prop_ref_name in stype.properties:
-        assert (
-            prop_ref_name in all_fields_group
-        ), f"layout-missing-field:{prop_ref_name}"
+        assert prop_ref_name in all_fields_group, (
+            f"layout-missing-field:{prop_ref_name}"
+        )
 
     return layout
 

uncountable/core/environment.py

@@ -2,6 +2,8 @@ import functools
 import os
 from importlib.metadata import PackageNotFoundError, version
 
+from uncountable.types import integration_server_t
+
 
 @functools.cache
 def get_version() -> str:
@@ -12,8 +14,8 @@ def get_version() -> str:
     return version_str
 
 
-def get_integration_env() -> str | None:
-    return os.environ.get("UNC_INTEGRATION_ENV")
+def get_server_env() -> str | None:
+    return os.environ.get("UNC_SERVER_ENV")
 
 
 def get_webhook_server_port() -> int:
@@ -26,3 +28,14 @@ def get_local_admin_server_port() -> int:
 
 def get_otel_enabled() -> bool:
     return os.environ.get("UNC_OTEL_ENABLED") == "true"
+
+
+def get_profiles_module() -> str:
+    return os.environ["UNC_PROFILES_MODULE"]
+
+
+def get_integration_envs() -> list[integration_server_t.IntegrationEnvironment]:
+    return [
+        integration_server_t.IntegrationEnvironment(env)
+        for env in os.environ.get("UNC_INTEGRATION_ENVS", "prod").split(",")
+    ]

uncountable/core/file_upload.py

@@ -99,9 +99,7 @@ class FileUploader:
                     file_bytes.bytes_data,
                     {"filename": file_bytes.name.encode()},
                     client_session=session,
-                    config=aiotus.RetryConfiguration(
-                        ssl=False if self._allow_insecure_tls else None
-                    ),
+                    config=aiotus.RetryConfiguration(ssl=not self._allow_insecure_tls),
                     chunksize=_CHUNK_SIZE,
                 )
                 if location is None:

uncountable/integration/construct_client.py

@@ -3,22 +3,21 @@ from uncountable.core.client import ClientConfig
 from uncountable.core.types import AuthDetailsAll, AuthDetailsOAuth
 from uncountable.integration.secret_retrieval.retrieve_secret import retrieve_secret
 from uncountable.integration.telemetry import JobLogger
+from uncountable.types import auth_retrieval_t
 from uncountable.types.job_definition_t import (
-    AuthRetrievalBasic,
-    AuthRetrievalOAuth,
     ProfileMetadata,
 )
 
 
 def _construct_auth_details(profile_meta: ProfileMetadata) -> AuthDetailsAll:
     match profile_meta.auth_retrieval:
-        case AuthRetrievalOAuth():
+        case auth_retrieval_t.AuthRetrievalOAuth():
             refresh_token = retrieve_secret(
                 profile_meta.auth_retrieval.refresh_token_secret,
                 profile_metadata=profile_meta,
             )
             return AuthDetailsOAuth(refresh_token=refresh_token)
-        case AuthRetrievalBasic():
+        case auth_retrieval_t.AuthRetrievalBasic():
             api_id = retrieve_secret(
                 profile_meta.auth_retrieval.api_id_secret, profile_metadata=profile_meta
             )

uncountable/integration/executors/generic_upload_executor.py

@@ -103,7 +103,7 @@ def _pull_remote_directory_data(
     files_to_pull = _filter_by_max_files(remote_directory, files_to_pull)
 
     logger.log_info(
-        f"Accessing SFTP directory: {remote_directory.src_path} and pulling files: {", ".join([f.filename for f in files_to_pull if f.filename is not None])}",
+        f"Accessing SFTP directory: {remote_directory.src_path} and pulling files: {', '.join([f.filename for f in files_to_pull if f.filename is not None])}",
     )
     return filesystem_session.download_files(files_to_pull)
 
@@ -205,16 +205,16 @@ class GenericUploadJob(Job[None]):
                     secret_access_key = None
 
                 if self.data_source.endpoint_url is None:
-                    assert (
-                        self.data_source.cloud_provider is not None
-                    ), "either cloud_provider or endpoint_url must be specified"
+                    assert self.data_source.cloud_provider is not None, (
+                        "either cloud_provider or endpoint_url must be specified"
+                    )
                     match self.data_source.cloud_provider:
                         case S3CloudProvider.AWS:
                             endpoint_url = "https://s3.amazonaws.com"
                         case S3CloudProvider.OVH:
-                            assert (
-                                self.data_source.region_name is not None
-                            ), "region_name must be specified for cloud_provider OVH"
+                            assert self.data_source.region_name is not None, (
+                                "region_name must be specified for cloud_provider OVH"
+                            )
                             endpoint_url = f"https://s3.{self.data_source.region_name}.cloud.ovh.net"
                 else:
                     endpoint_url = self.data_source.endpoint_url

uncountable/integration/executors/script_executor.py

@@ -19,7 +19,7 @@ def resolve_script_executor(
     for _, job_class in inspect.getmembers(job_module, inspect.isclass):
         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}, found {len(found_jobs)}"
+    assert len(found_jobs) == 1, (
+        f"expected exactly one job class in {executor.import_path}, found {len(found_jobs)}"
+    )
     return found_jobs[0]

uncountable/integration/queue_runner/job_scheduler.py

@@ -55,7 +55,7 @@ def _start_workers(
     job_worker_lookup: dict[str, Worker] = {}
     job_definition_lookup: dict[str, job_definition_t.JobDefinition] = {}
     for profile in profiles:
-        for job_definition in profile.definition.jobs:
+        for job_definition in profile.jobs:
             job_definition_lookup[job_definition.id] = job_definition
             job_worker_key = _get_job_worker_key(job_definition, profile.name)
             if job_worker_key not in job_queue_worker_lookup:

uncountable/integration/queue_runner/worker.py

@@ -51,16 +51,11 @@ class RegisteredJobDetails:
 
 def get_registered_job_details(job_ref_name: str) -> RegisteredJobDetails:
     profiles = load_profiles()
-    for profile in profiles:
-        for job_definition in profile.definition.jobs:
+    for profile_metadata in profiles:
+        for job_definition in profile_metadata.jobs:
             if job_definition.id == job_ref_name:
                 return RegisteredJobDetails(
-                    profile_metadata=job_definition_t.ProfileMetadata(
-                        name=profile.name,
-                        base_url=profile.definition.base_url,
-                        auth_retrieval=profile.definition.auth_retrieval,
-                        client_options=profile.definition.client_options,
-                    ),
+                    profile_metadata=profile_metadata,
                     job_definition=job_definition,
                 )
     raise Exception(f"profile not found for job {job_ref_name}")

uncountable/integration/scan_profiles.py

@@ -1,27 +1,26 @@
 import functools
-import os
-from dataclasses import dataclass
 from importlib import resources
 
 from pkgs.argument_parser import CachedParser
-from uncountable.types import job_definition_t
+from uncountable.core import environment
+from uncountable.types import integration_server_t, job_definition_t
 
 profile_parser = CachedParser(job_definition_t.ProfileDefinition)
 
-
-@dataclass(kw_only=True)
-class ProfileDetails:
-    name: str
-    definition: job_definition_t.ProfileDefinition
+_DEFAULT_PROFILE_ENV = integration_server_t.IntegrationEnvironment.PROD
+_IGNORED_PROFILE_FOLDERS = ["__pycache__"]
 
 
 @functools.cache
-def load_profiles() -> list[ProfileDetails]:
-    profiles_module = os.environ["UNC_PROFILES_MODULE"]
+def load_profiles() -> list[job_definition_t.ProfileMetadata]:
+    profiles_module = environment.get_profiles_module()
+    integration_envs = environment.get_integration_envs()
     profiles = [
-        entry for entry in resources.files(profiles_module).iterdir() if entry.is_dir()
+        entry
+        for entry in resources.files(profiles_module).iterdir()
+        if entry.is_dir() and entry.name not in _IGNORED_PROFILE_FOLDERS
     ]
-    profile_details: list[ProfileDetails] = []
+    profile_details: list[job_definition_t.ProfileMetadata] = []
     seen_job_ids: set[str] = set()
     for profile_file in profiles:
         profile_name = profile_file.name
@@ -34,9 +33,34 @@ def load_profiles() -> list[ProfileDetails]:
                 if job.id in seen_job_ids:
                     raise Exception(f"multiple jobs with id {job.id}")
                 seen_job_ids.add(job.id)
-            profile_details.append(
-                ProfileDetails(name=profile_name, definition=definition)
-            )
+
+            if definition.environments is not None:
+                for integration_env in integration_envs:
+                    environment_config = definition.environments.get(integration_env)
+                    if environment_config is not None:
+                        profile_details.append(
+                            job_definition_t.ProfileMetadata(
+                                name=profile_name,
+                                jobs=definition.jobs,
+                                base_url=environment_config.base_url,
+                                auth_retrieval=environment_config.auth_retrieval,
+                                client_options=environment_config.client_options,
+                            )
+                        )
+            elif _DEFAULT_PROFILE_ENV in integration_envs:
+                assert (
+                    definition.base_url is not None
+                    and definition.auth_retrieval is not None
+                ), f"define environments in profile.yaml for {profile_name}"
+                profile_details.append(
+                    job_definition_t.ProfileMetadata(
+                        name=profile_name,
+                        jobs=definition.jobs,
+                        base_url=definition.base_url,
+                        auth_retrieval=definition.auth_retrieval,
+                        client_options=definition.client_options,
+                    )
+                )
         except FileNotFoundError as e:
             print(f"WARN: profile.yaml not found for {profile_name}", e)
             continue

uncountable/integration/secret_retrieval/retrieve_secret.py

@@ -15,7 +15,7 @@ from uncountable.types.secret_retrieval_t import (
 )
 
 
-class SecretRetrievalError(BaseException):
+class SecretRetrievalError(Exception):
     def __init__(
         self, secret_retrieval: SecretRetrieval, message: str | None = None
     ) -> None:

uncountable/integration/server.py

@@ -12,12 +12,10 @@ from opentelemetry.trace import get_current_span
 from sqlalchemy.engine.base import Engine
 
 from uncountable.integration.cron import CronJobArgs, cron_job_executor
-from uncountable.integration.scan_profiles import ProfileDetails
 from uncountable.integration.telemetry import Logger
-from uncountable.types import base_t
+from uncountable.types import base_t, job_definition_t
 from uncountable.types.job_definition_t import (
     CronJobDefinition,
-    ProfileMetadata,
     WebhookJobDefinition,
 )
 
@@ -38,17 +36,11 @@ class IntegrationServer:
         )
         self._server_logger = Logger(get_current_span())
 
-    def register_jobs(self, profiles: list[ProfileDetails]) -> None:
+    def register_jobs(self, profiles: list[job_definition_t.ProfileMetadata]) -> None:
         valid_job_ids = []
-        for profile in profiles:
-            for job_defn in profile.definition.jobs:
+        for profile_metadata in profiles:
+            for job_defn in profile_metadata.jobs:
                 valid_job_ids.append(job_defn.id)
-                profile_metadata = ProfileMetadata(
-                    name=profile.name,
-                    auth_retrieval=profile.definition.auth_retrieval,
-                    base_url=profile.definition.base_url,
-                    client_options=profile.definition.client_options,
-                )
                 match job_defn:
                     case CronJobDefinition():
                         # Add to ap scheduler

uncountable/integration/telemetry.py

@@ -21,8 +21,8 @@ from opentelemetry.sdk.trace.export import (
 from opentelemetry.trace import DEFAULT_TRACE_OPTIONS, Span, Tracer
 
 from uncountable.core.environment import (
-    get_integration_env,
     get_otel_enabled,
+    get_server_env,
     get_version,
 )
 from uncountable.types import base_t, job_definition_t
@@ -41,7 +41,7 @@ def get_otel_resource() -> Resource:
     unc_version = os.environ.get("UNC_VERSION")
     if unc_version is not None:
         attributes["service.version"] = unc_version
-    unc_env = get_integration_env()
+    unc_env = get_server_env()
     if unc_env is not None:
         attributes["deployment.environment"] = unc_env
     resource = Resource.create(attributes=_cast_attributes(attributes))

uncountable/integration/webhook_server/entrypoint.py

@@ -8,8 +8,8 @@ from flask.typing import ResponseReturnValue
 from flask.wrappers import Response
 from opentelemetry.trace import get_current_span
 from uncountable.core.environment import (
-    get_integration_env,
     get_local_admin_server_port,
+    get_server_env,
     get_webhook_server_port,
 )
 from uncountable.integration.queue_runner.command_server.command_client import (
@@ -143,15 +143,9 @@ def register_route(
 
 def main() -> None:
     profiles = load_profiles()
-    for profile in profiles:
+    for profile_metadata in profiles:
         server_logger = Logger(get_current_span())
-        profile_metadata = job_definition_t.ProfileMetadata(
-            name=profile.name,
-            auth_retrieval=profile.definition.auth_retrieval,
-            base_url=profile.definition.base_url,
-            client_options=profile.definition.client_options,
-        )
-        for job in profile.definition.jobs:
+        for job in profile_metadata.jobs:
             if isinstance(job, job_definition_t.WebhookJobDefinition):
                 register_route(
                     server_logger=server_logger, profile_meta=profile_metadata, job=job
@@ -165,6 +159,6 @@ if __name__ == "__main__":
     app.run(
         host="0.0.0.0",
         port=get_webhook_server_port(),
-        debug=get_integration_env() == "playground",
+        debug=get_server_env() == "playground",
         exclude_patterns=[],
     )

uncountable/types/__init__.py

@@ -9,6 +9,7 @@ from .api.equipment import associate_equipment_input as associate_equipment_inpu
 from .api.recipes import associate_recipe_as_input as associate_recipe_as_input_t
 from .api.recipes import associate_recipe_as_lot as associate_recipe_as_lot_t
 from . import async_batch_t as async_batch_t
+from . import auth_retrieval_t as auth_retrieval_t
 from . import base_t as base_t
 from . import calculations_t as calculations_t
 from . import chemical_structure_t as chemical_structure_t
@@ -31,6 +32,7 @@ from . import experiment_groups_t as experiment_groups_t
 from . import field_values_t as field_values_t
 from . import fields_t as fields_t
 from . import generic_upload_t as generic_upload_t
+from .api.recipes import get_column_calculation_values as get_column_calculation_values_t
 from .api.recipes import get_curve as get_curve_t
 from .api.entity import get_entities_data as get_entities_data_t
 from .api.inputs import get_input_data as get_input_data_t
@@ -47,10 +49,12 @@ from .api.recipe_metadata import get_recipe_metadata_data as get_recipe_metadata
 from .api.recipes import get_recipe_names as get_recipe_names_t
 from .api.recipes import get_recipe_output_metadata as get_recipe_output_metadata_t
 from .api.recipes import get_recipes_data as get_recipes_data_t
+from .api.entity import grant_entity_permissions as grant_entity_permissions_t
 from . import id_source_t as id_source_t
 from . import identifier_t as identifier_t
 from . import input_attributes_t as input_attributes_t
 from . import inputs_t as inputs_t
+from . import integration_server_t as integration_server_t
 from .api.uploader import invoke_uploader as invoke_uploader_t
 from . import job_definition_t as job_definition_t
 from .api.entity import list_entities as list_entities_t
@@ -110,6 +114,7 @@ __all__: list[str] = [
     "associate_recipe_as_input_t",
     "associate_recipe_as_lot_t",
     "async_batch_t",
+    "auth_retrieval_t",
     "base_t",
     "calculations_t",
     "chemical_structure_t",
@@ -132,6 +137,7 @@ __all__: list[str] = [
     "field_values_t",
     "fields_t",
     "generic_upload_t",
+    "get_column_calculation_values_t",
     "get_curve_t",
     "get_entities_data_t",
     "get_input_data_t",
@@ -148,10 +154,12 @@ __all__: list[str] = [
     "get_recipe_names_t",
     "get_recipe_output_metadata_t",
     "get_recipes_data_t",
+    "grant_entity_permissions_t",
     "id_source_t",
     "identifier_t",
     "input_attributes_t",
     "inputs_t",
+    "integration_server_t",
     "invoke_uploader_t",
     "job_definition_t",
     "list_entities_t",

uncountable/types/api/entity/create_entities.py

@@ -12,6 +12,7 @@ from pkgs.serialization import serial_class
 from ... import base_t
 from ... import entity_t
 from ... import field_values_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",
@@ -40,9 +41,10 @@ class EntityToCreate:
 )
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
-    definition_id: base_t.ObjectId
-    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG], typing.Literal[entity_t.EntityType.CONDITION_PARAMETER], typing.Literal[entity_t.EntityType.OUTPUT]]
+    entity_type: entity_t.LimitedEntityType
     entities_to_create: list[EntityToCreate]
+    definition_id: typing.Optional[base_t.ObjectId] = None
+    definition_key: typing.Optional[identifier_t.IdentifierKey] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/create_entity.py

@@ -12,6 +12,7 @@ from pkgs.serialization import serial_class
 from ... import base_t
 from ... import entity_t
 from ... import field_values_t
+from ... import identifier_t
 
 __all__: list[str] = [
     "Arguments",
@@ -43,8 +44,9 @@ class EntityFieldInitialValue:
 )
 @dataclasses.dataclass(kw_only=True)
 class Arguments:
-    definition_id: base_t.ObjectId
-    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.INVENTORY_AMOUNT], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS], typing.Literal[entity_t.EntityType.FIELD_OPTION_SET], typing.Literal[entity_t.EntityType.WEBHOOK], typing.Literal[entity_t.EntityType.SPECS], typing.Literal[entity_t.EntityType.GOAL], typing.Literal[entity_t.EntityType.INGREDIENT_TAG_MAP], typing.Literal[entity_t.EntityType.INGREDIENT_TAG], typing.Literal[entity_t.EntityType.CONDITION_PARAMETER], typing.Literal[entity_t.EntityType.OUTPUT]]
+    entity_type: entity_t.LimitedEntityType
+    definition_id: typing.Optional[base_t.ObjectId] = None
+    definition_key: typing.Optional[identifier_t.IdentifierKey] = None
     field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None
 
 

uncountable/types/api/entity/grant_entity_permissions.py

@@ -0,0 +1,48 @@
+# 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
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import async_batch_t
+from ... import entity_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/entity/grant_entity_permissions"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.grant_entity_permissions.Arguments",
+)
+@dataclasses.dataclass(kw_only=True)
+class Arguments:
+    entity_type: entity_t.LimitedEntityType
+    entity_key: identifier_t.IdentifierKey
+    permission_types: list[entity_t.GrantableEntityPermissionType]
+    user_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None
+    user_group_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None
+    all_users: typing.Optional[bool] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.entity.grant_entity_permissions.Data",
+)
+@dataclasses.dataclass(kw_only=True)
+class Data(async_batch_t.AsyncBatchActionReturn):
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/recipes/get_column_calculation_values.py

@@ -0,0 +1,58 @@
+# 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
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import base_t
+from ... import identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "RecipeColumnCalculationValue",
+]
+
+ENDPOINT_METHOD = "GET"
+ENDPOINT_PATH = "api/external/recipes/get_column_calculation_values"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.get_column_calculation_values.Arguments",
+)
+@dataclasses.dataclass(kw_only=True)
+class Arguments:
+    recipe_keys: list[identifier_t.IdentifierKey]
+    calculation_key: identifier_t.IdentifierKey
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.get_column_calculation_values.RecipeColumnCalculationValue",
+    to_string_values={"calculation_quantity"},
+)
+@dataclasses.dataclass(kw_only=True)
+class RecipeColumnCalculationValue:
+    recipe_id: base_t.ObjectId
+    ingredient_id: base_t.ObjectId
+    recipe_step_id: base_t.ObjectId
+    recipe_input_id: base_t.ObjectId
+    calculation_quantity: typing.Optional[Decimal] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.get_column_calculation_values.Data",
+)
+@dataclasses.dataclass(kw_only=True)
+class Data:
+    values: list[RecipeColumnCalculationValue]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/async_batch_processor.py

@@ -15,7 +15,9 @@ from uncountable.types import base_t
 import uncountable.types.api.recipes.clear_recipe_outputs as clear_recipe_outputs_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 entity_t
 from uncountable.types import generic_upload_t
+import uncountable.types.api.entity.grant_entity_permissions as grant_entity_permissions_t
 from uncountable.types import identifier_t
 import uncountable.types.api.uploader.invoke_uploader as invoke_uploader_t
 from uncountable.types import recipe_identifiers_t
@@ -230,6 +232,47 @@ class AsyncBatchProcessorBase(ABC):
             batch_reference=req.batch_reference,
         )
 
+    def grant_entity_permissions(
+        self,
+        *,
+        entity_type: entity_t.LimitedEntityType,
+        entity_key: identifier_t.IdentifierKey,
+        permission_types: list[entity_t.GrantableEntityPermissionType],
+        user_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None,
+        user_group_keys: typing.Optional[list[identifier_t.IdentifierKey]] = None,
+        all_users: typing.Optional[bool] = None,
+        depends_on: typing.Optional[list[str]] = None,
+    ) -> async_batch_t.QueuedAsyncBatchRequest:
+        """Grant entity permissions to a list of users or user groups or to all users.
+
+        :param depends_on: A list of batch reference keys to process before processing this request
+        """
+        args = grant_entity_permissions_t.Arguments(
+            entity_type=entity_type,
+            entity_key=entity_key,
+            permission_types=permission_types,
+            user_keys=user_keys,
+            user_group_keys=user_group_keys,
+            all_users=all_users,
+        )
+        json_data = serialize_for_api(args)
+
+        batch_reference = str(uuid.uuid4())
+
+        req = async_batch_t.AsyncBatchRequest(
+            path=async_batch_t.AsyncBatchRequestPath.GRANT_ENTITY_PERMISSIONS,
+            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 invoke_uploader(
         self,
         *,

uncountable/types/async_batch_t.py

@@ -34,6 +34,7 @@ class AsyncBatchRequestPath(StrEnum):
     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"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec