UncountablePythonSDK 0.0.159__tar.gz → 0.0.160__tar.gz

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.159
+Version: 0.0.160
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/UncountablePythonSDK.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.159
+Version: 0.0.160
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -60,6 +60,7 @@ pkgs/argument_parser/argument_parser.py
 pkgs/argument_parser/case_convert.py
 pkgs/filesystem_utils/__init__.py
 pkgs/filesystem_utils/_blob_session.py
+pkgs/filesystem_utils/_file_share_session.py
 pkgs/filesystem_utils/_gdrive_session.py
 pkgs/filesystem_utils/_local_session.py
 pkgs/filesystem_utils/_s3_session.py

uncountablepythonsdk-0.0.160/docs/static/favicons/manifest.json

@@ -0,0 +1,18 @@
+{
+  "name": "",
+  "icons": [
+    {
+      "src": "/static/img/icons/android-chrome-192x192.png",
+      "sizes": "192x192",
+      "type": "image/png"
+    },
+    {
+      "src": "/static/img/icons/android-chrome-512x512.png",
+      "sizes": "512x512",
+      "type": "image/png"
+    }
+  ],
+  "theme_color": "#ffffff",
+  "background_color": "#ffffff",
+  "display": "standalone"
+}

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/examples/integration-server/pyproject.toml

@@ -12,7 +12,7 @@ dependencies = [
   "types-paramiko ==4.0.0.20250822",
   "types-openpyxl == 3.*",
   "types-pysftp == 0.*",
-  "types-pytz ==2025.*",
+  "types-pytz ==2026.*",
   "types-requests == 2.*",
   "types-simplejson == 3.*",
   "pandas-stubs",

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/pkgs/filesystem_utils/__init__.py

@@ -1,4 +1,5 @@
 from ._blob_session import BlobSession as BlobSession
+from ._file_share_session import FileShareSession as FileShareSession
 from ._gdrive_session import GDriveSession as GDriveSession
 from ._gdrive_session import delete_gdrive_file as delete_gdrive_file
 from ._gdrive_session import download_gdrive_file as download_gdrive_file
@@ -13,6 +14,7 @@ from ._sftp_session import move_sftp_files as move_sftp_files
 from .file_type_utils import FileObjectData as FileObjectData
 from .file_type_utils import FileSystemBlobConfig as FileSystemBlobConfig
 from .file_type_utils import FileSystemFileReference as FileSystemFileReference
+from .file_type_utils import FileSystemFileShareConfig as FileSystemFileShareConfig
 from .file_type_utils import FileSystemObject as FileSystemObject
 from .file_type_utils import FileSystemS3Config as FileSystemS3Config
 from .file_type_utils import FileSystemSFTPConfig as FileSystemSFTPConfig

uncountablepythonsdk-0.0.160/pkgs/filesystem_utils/_file_share_session.py

@@ -0,0 +1,131 @@
+from io import BytesIO
+from typing import Self
+
+from azure.core.credentials import AzureSasCredential
+from azure.storage.fileshare import ShareClient, ShareDirectoryClient
+
+from pkgs.filesystem_utils.file_type_utils import (
+    FileObjectData,
+    FileSystemFileReference,
+    FileSystemFileShareConfig,
+    FileSystemObject,
+    FileTransfer,
+    IncompatibleFileReference,
+)
+from pkgs.filesystem_utils.filesystem_session import FileSystemSession
+
+
+class FileShareSession(FileSystemSession):
+    def __init__(self, config: FileSystemFileShareConfig) -> None:
+        super().__init__()
+        self.config = config
+        self._share_client: ShareClient | None = None
+
+    def start(self) -> None:
+        self._share_client = ShareClient(
+            account_url=self.config.account_url,
+            share_name=self.config.share_name,
+            credential=self.config.credential,
+        )
+
+    def __enter__(self) -> Self:
+        self.start()
+        return self
+
+    def __exit__(self, exc_type: object, exc_val: object, exc_tb: object) -> None:
+        self._share_client = None
+
+    def _list_dir(
+        self,
+        dir_client: ShareDirectoryClient,
+        *,
+        recursive: bool,
+        valid_extensions: list[str] | None,
+    ) -> list[FileSystemObject]:
+        results: list[FileSystemObject] = []
+        for item in dir_client.list_directories_and_files():
+            item_path = f"{dir_client.directory_path}/{item['name']}".lstrip("/")
+            if item["is_directory"]:
+                if recursive:
+                    assert self._share_client is not None
+                    results.extend(
+                        self._list_dir(
+                            self._share_client.get_directory_client(item_path),
+                            recursive=recursive,
+                            valid_extensions=valid_extensions,
+                        )
+                    )
+            elif valid_extensions is None or any(
+                item_path.endswith(ext) for ext in valid_extensions
+            ):
+                results.append(FileSystemFileReference(filepath=item_path))
+        return results
+
+    def list_files(
+        self,
+        dir_path: FileSystemObject,
+        *,
+        recursive: bool = False,
+        valid_extensions: list[str] | None = None,
+    ) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReference()
+        assert self._share_client is not None, (
+            "call to list_files on uninitialized file share session"
+        )
+        dir_client = self._share_client.get_directory_client(dir_path.filepath)
+        return self._list_dir(
+            dir_client, recursive=recursive, valid_extensions=valid_extensions
+        )
+
+    def download_files(self, filepaths: list[FileSystemObject]) -> list[FileObjectData]:
+        assert self._share_client is not None, (
+            "call to download_files on uninitialized file share session"
+        )
+        downloaded: list[FileObjectData] = []
+        for file_object in filepaths:
+            if (
+                not isinstance(file_object, FileSystemFileReference)
+                or not file_object.filename
+            ):
+                raise IncompatibleFileReference()
+            file_client = self._share_client.get_file_client(file_object.filepath)
+            stream = file_client.download_file()
+            file_data = stream.readall()
+            downloaded.append(
+                FileObjectData(
+                    file_data=file_data,
+                    file_IO=BytesIO(file_data),
+                    filename=file_object.filename,
+                    filepath=file_object.filepath,
+                )
+            )
+        return downloaded
+
+    def move_files(self, file_mappings: list[FileTransfer]) -> None:
+        assert self._share_client is not None, (
+            "call to move_files on uninitialized file share session"
+        )
+        for src, dest in file_mappings:
+            if not isinstance(src, FileSystemFileReference) or not isinstance(
+                dest, FileSystemFileReference
+            ):
+                raise IncompatibleFileReference()
+            src_client = self._share_client.get_file_client(src.filepath)
+            dest_client = self._share_client.get_file_client(dest.filepath)
+            source_url = (
+                f"{src_client.url}?{self.config.credential.signature}"
+                if isinstance(self.config.credential, AzureSasCredential)
+                else src_client.url
+            )
+            dest_client.start_copy_from_url(source_url)
+            src_client.delete_file()
+
+    def delete_files(self, filepaths: list[FileSystemObject]) -> None:
+        assert self._share_client is not None, (
+            "call to delete_files on uninitialized file share session"
+        )
+        for file_object in filepaths:
+            if not isinstance(file_object, FileSystemFileReference):
+                raise IncompatibleFileReference()
+            self._share_client.get_file_client(file_object.filepath).delete_file()

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/pkgs/filesystem_utils/file_type_utils.py

@@ -89,3 +89,12 @@ class FileSystemBlobConfig:
         | None
     )
     container: ContainerProperties | str
+
+
+@dataclass(frozen=True, kw_only=True)
+class FileSystemFileShareConfig:
+    account_url: str
+    credential: (
+        AzureNamedKeyCredential | AzureSasCredential | TokenCredential | str | None
+    )
+    share_name: str

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/pkgs/type_spec/emit_open_api.py

@@ -93,18 +93,29 @@ def _write_guide_as_html(guide: EmitOpenAPIGuide, *, is_open: bool) -> str:
         </details>"""
 
 
+_FRONTMATTER_RE = re.compile(r"^---\n.*?\n---\n", re.DOTALL)
+
+
+def _strip_guide_frontmatter(content: str) -> str:
+    return _FRONTMATTER_RE.sub("", content).lstrip("\n")
+
+
 def _open_api_info(
-    config: OpenAPIConfig, guides: list[EmitOpenAPIGuide]
+    *, config: OpenAPIConfig, root_guides: list[builder.SpecGuide]
 ) -> GlobalContextInfo:
-    full_guides = "<br/>".join([
-        _write_guide_as_html(guide, is_open=True)
-        for guide in sorted(guides, key=lambda g: g.ref_name)
-    ])
-    full_description = f"{config.description}<br/>{full_guides}"
     info: GlobalContextInfo = dict()
     info["version"] = "1.0.0"
     info["title"] = "Uncountable API Documentation"
-    info["description"] = full_description
+    sorted_guides = sorted(root_guides, key=lambda g: g.ref_name)
+    description_parts = [
+        config.description,
+        *[
+            f'<a id="{guide.ref_name}" />\n\n'
+            + _strip_guide_frontmatter(guide.markdown_content)
+            for guide in sorted_guides
+        ],
+    ]
+    info["description"] = "\n\n".join(description_parts)
     info["x-logo"] = {
         "url": "../static/images/logo_blue.png",
         "altText": "Logo",
@@ -119,16 +130,11 @@ def _open_api_servers(config: OpenAPIConfig) -> list[EmitOpenAPIServer]:
 
 
 def emit_open_api(builder: builder.SpecBuilder, *, config: OpenAPIConfig) -> None:
-    root_guides = builder.guides.get(RootGuideKey(), [])
-    openapi_guides = [
-        EmitOpenAPIGuide(
-            ref_name=guide.ref_name, title=guide.title, html_content=guide.html_content
-        )
-        for guide in root_guides
-    ]
     gctx = EmitOpenAPIGlobalContext(
         version="3.0.0",
-        info=_open_api_info(config, openapi_guides),
+        info=_open_api_info(
+            config=config, root_guides=builder.guides.get(RootGuideKey(), [])
+        ),
         servers=_open_api_servers(config),
     )
 

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/uncountable/core/client.py

@@ -330,6 +330,11 @@ class Client(ClientMethods):
         headers = self._build_auth_headers()
         headers[UNC_REQUEST_ID_HEADER] = request_id
         headers[UNC_SDK_VERSION_HEADER] = get_version()
+        if self._cfg.request_context is not None:
+            if self._cfg.request_context.user_locale is not None:
+                headers["X-User-Locale"] = self._cfg.request_context.user_locale
+            if self._cfg.request_context.user_timezone is not None:
+                headers["X-User-Timezone"] = self._cfg.request_context.user_timezone
         method = api_request.method.lower()
         data = {"data": json.dumps(serialize_for_api(api_request.args))}
         match method:

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/uncountable/integration/queue_runner/worker.py

@@ -23,7 +23,7 @@ from uncountable.integration.queue_runner.datastore.interface import Datastore
 from uncountable.integration.queue_runner.types import ListenQueue, ResultQueue
 from uncountable.integration.scan_profiles import load_profiles
 from uncountable.integration.telemetry import JobLogger, Logger, get_otel_tracer
-from uncountable.types import base_t, job_definition_t, queued_job_t
+from uncountable.types import base_t, client_config_t, job_definition_t, queued_job_t
 
 
 class Worker:
@@ -89,6 +89,16 @@ def _resolve_queued_job_payload(queued_job: queued_job_t.QueuedJob) -> base_t.Js
             return queued_job.payload.invocation_context.webhook_payload
 
 
+def _resolve_queued_job_request_context(
+    queued_job: queued_job_t.QueuedJob,
+) -> client_config_t.RequestContext | None:
+    match queued_job.payload.invocation_context:
+        case queued_job_t.InvocationContextWebhook():
+            return queued_job.payload.invocation_context.request_context
+        case _:
+            return None
+
+
 def run_queued_job(
     queued_job: queued_job_t.QueuedJob,
 ) -> job_definition_t.JobResult:
@@ -108,8 +118,11 @@ def run_queued_job(
         )
         with job_logger.resource_tracking():
             try:
+                request_context = _resolve_queued_job_request_context(queued_job)
                 client = construct_uncountable_client(
-                    profile_meta=job_details.profile_metadata, logger=job_logger
+                    profile_meta=job_details.profile_metadata,
+                    logger=job_logger,
+                    request_context=request_context,
                 )
                 batch_processor = AsyncBatchProcessor(client=client)
 

{uncountablepythonsdk-0.0.159 → uncountablepythonsdk-0.0.160}/uncountable/types/api/recipes/edit_recipe_inputs.py

@@ -23,12 +23,15 @@ __all__: list[str] = [
     "ENDPOINT_METHOD",
     "ENDPOINT_PATH",
     "MixStepMatchStrategy",
+    "PlaceholderAssignmentPlaceholderType",
     "PlaceholderBase",
     "PlaceholderDisplayMode",
     "PlaceholderType",
+    "ProductRecipePlaceholderAssignment",
     "RecipeInputEdit",
     "RecipeInputEditAddInput",
     "RecipeInputEditAddInstructions",
+    "RecipeInputEditAssignPlaceholder",
     "RecipeInputEditBase",
     "RecipeInputEditChangeBasisViewed",
     "RecipeInputEditClearInputs",
@@ -41,6 +44,8 @@ __all__: list[str] = [
     "RecipeInputEditUpdateAnnotations",
     "RecipeInputEditUpsertInput",
     "RecipeInputPlaceholder",
+    "RecipeInputPlaceholderAssignment",
+    "RecipeInputPlaceholderAssignmentBase",
     "SubcategoryPlaceholder",
 ]
 
@@ -60,6 +65,7 @@ class RecipeInputEditType(StrEnum):
     ADD_INSTRUCTIONS = "add_instructions"
     SET_ROLE = "set_role"
     SET_PLACEHOLDER = "set_placeholder"
+    ASSIGN_PLACEHOLDER = "assign_placeholder"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -269,9 +275,60 @@ class RecipeInputEditPlaceholder(RecipeInputEditBase):
     mix_step_match_strategy: MixStepMatchStrategy = MixStepMatchStrategy.FIRST
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+class PlaceholderAssignmentPlaceholderType(StrEnum):
+    PRODUCT = "product"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.edit_recipe_inputs.RecipeInputPlaceholderAssignmentBase",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class RecipeInputPlaceholderAssignmentBase:
+    type: PlaceholderAssignmentPlaceholderType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.edit_recipe_inputs.ProductRecipePlaceholderAssignment",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class ProductRecipePlaceholderAssignment(RecipeInputPlaceholderAssignmentBase):
+    type: typing.Literal[PlaceholderAssignmentPlaceholderType.PRODUCT] = PlaceholderAssignmentPlaceholderType.PRODUCT
+    product_key: identifier_t.IdentifierKey
+    recipe_key: identifier_t.IdentifierKey
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+RecipeInputPlaceholderAssignment = typing.Annotated[
+    typing.Union[ProductRecipePlaceholderAssignment],
+    serial_union_annotation(
+    named_type_path="sdk.api.recipes.edit_recipe_inputs.RecipeInputPlaceholderAssignment",
+        discriminator="type",
+        discriminator_map={
+            "product": ProductRecipePlaceholderAssignment,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.edit_recipe_inputs.RecipeInputEditAssignPlaceholder",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class RecipeInputEditAssignPlaceholder(RecipeInputEditBase):
+    type: typing.Literal[RecipeInputEditType.ASSIGN_PLACEHOLDER] = RecipeInputEditType.ASSIGN_PLACEHOLDER
+    assignment: RecipeInputPlaceholderAssignment
+    mix_step_match_strategy: MixStepMatchStrategy = MixStepMatchStrategy.FIRST
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 RecipeInputEdit = typing.Annotated[
-    RecipeInputEditClearInputs | RecipeInputEditUpsertInput | RecipeInputEditAddInput | RecipeInputEditUpdateAnnotations | RecipeInputEditSetLot | RecipeInputEditSetPropagatedPercentage | RecipeInputEditChangeBasisViewed | RecipeInputEditAddInstructions | RecipeInputEditSetRole | RecipeInputEditPlaceholder,
+    RecipeInputEditClearInputs | RecipeInputEditUpsertInput | RecipeInputEditAddInput | RecipeInputEditUpdateAnnotations | RecipeInputEditSetLot | RecipeInputEditSetPropagatedPercentage | RecipeInputEditChangeBasisViewed | RecipeInputEditAddInstructions | RecipeInputEditSetRole | RecipeInputEditPlaceholder | RecipeInputEditAssignPlaceholder,
     serial_union_annotation(
     named_type_path="sdk.api.recipes.edit_recipe_inputs.RecipeInputEdit",
         discriminator="type",
@@ -286,6 +343,7 @@ RecipeInputEdit = typing.Annotated[
             "add_instructions": RecipeInputEditAddInstructions,
             "set_role": RecipeInputEditSetRole,
             "set_placeholder": RecipeInputEditPlaceholder,
+            "assign_placeholder": RecipeInputEditAssignPlaceholder,
         },
     ),
 ]

uncountablepythonsdk-0.0.159/docs/static/favicons/manifest.json

@@ -1,18 +0,0 @@
-{
-    "name": "",
-    "icons": [
-        {
-            "src": "/static/img/icons/android-chrome-192x192.png",
-            "sizes": "192x192",
-            "type": "image/png"
-        },
-        {
-            "src": "/static/img/icons/android-chrome-512x512.png",
-            "sizes": "512x512",
-            "type": "image/png"
-        }
-    ],
-    "theme_color": "#ffffff",
-    "background_color": "#ffffff",
-    "display": "standalone"
-}