albert 1.12.0__py3-none-any.whl → 1.13.0__py3-none-any.whl

albert/__init__.py

@@ -4,4 +4,4 @@ from albert.core.auth.sso import AlbertSSOClient
 
 __all__ = ["Albert", "AlbertClientCredentials", "AlbertSSOClient"]
 
-__version__ = "1.12.0"
+__version__ = "1.13.0"

albert/resources/inventory.py

@@ -14,6 +14,7 @@ from albert.resources._mixins import HydrationMixin
 from albert.resources.acls import ACL
 from albert.resources.cas import Cas
 from albert.resources.companies import Company
+from albert.resources.lists import ListItem
 from albert.resources.locations import Location
 from albert.resources.tagged_base import BaseTaggedResource
 from albert.resources.tags import Tag
@@ -75,6 +76,8 @@ class CasAmount(BaseAlbertModel):
         The SMILES string of the CAS Number resource. Obtained from the Cas object when provided.
     number: str | None
         The CAS number. Obtained from the Cas object when provided.
+    inventory_function: list[ListItem | EntityLink | str] | None
+        Business-controlled functions associated with the CAS in this inventory context.
 
     !!! tip
     ---
@@ -87,6 +90,9 @@ class CasAmount(BaseAlbertModel):
     target: float | None = Field(default=None, alias="inventoryValue")
     id: str | None = Field(default=None)
     cas_category: str | None = Field(default=None, alias="casCategory")
+    inventory_function: list[SerializeAsEntityLink[ListItem] | str] | None = Field(
+        default=None, alias="inventoryFunction"
+    )
     type: str | None = Field(default=None)
     classification_type: str | None = Field(default=None, alias="classificationType")
 

albert/resources/lists.py

@@ -25,9 +25,11 @@ class ListItem(BaseResource):
     id : str | None
         The Albert ID of the list item. Set when the list item is retrieved from Albert.
     category : ListItemCategory | None
-        The category of the list item. Allowed values are `businessDefined`, `userDefined`, `projects`, and `extensions`.
+        The category of the list item. Allowed values are `businessDefined`, `userDefined`, `projects`, `extensions`,
+        and `inventory`.
     list_type : str | None
-        The type of the list item. Allowed values are `projectState` for `projects` and `extensions` for `extensions`.
+        The type of the list item. Allowed values are `projectState` for `projects`, `extensions` for `extensions`,
+        and `casCategory` or `inventoryFunction` for `inventory`.
     """
 
     name: str
@@ -37,14 +39,15 @@ class ListItem(BaseResource):
 
     @model_validator(mode="after")
     def validate_list_type(self) -> ListItem:
+        allowed_by_category = {
+            ListItemCategory.PROJECTS: {"projectState"},
+            ListItemCategory.EXTENSIONS: {"extensions"},
+            ListItemCategory.INVENTORY: {"casCategory", "inventoryFunction"},
+        }
         if (
-            self.category == ListItemCategory.PROJECTS
-            and self.list_type is not None
-            and self.list_type != "projectState"
-        ) or (
-            self.category == ListItemCategory.EXTENSIONS
-            and self.list_type is not None
-            and self.list_type != "extensions"
+            self.list_type is not None
+            and self.category in allowed_by_category
+            and self.list_type not in allowed_by_category[self.category]
         ):
             raise ValueError(
                 f"List type {self.list_type} is not allowed for category {self.category}"

albert/utils/inventory.py

@@ -1,6 +1,7 @@
 from collections.abc import Iterable
 from typing import Any
 
+from albert.core.shared.models.base import BaseResource, EntityLink
 from albert.resources.inventory import CasAmount
 
 
@@ -56,6 +57,61 @@ def _build_cas_delete_operation(identifier: str) -> dict[str, Any]:
     }
 
 
+def _normalize_inventory_function_ids(
+    value: list[BaseResource | EntityLink | str] | None,
+) -> list[str]:
+    if not value:
+        return []
+    ids: list[str] = []
+    for item in value:
+        if isinstance(item, str):
+            if item:
+                ids.append(item)
+            continue
+        if isinstance(item, BaseResource):
+            if item.id:
+                ids.append(item.id)
+            continue
+        if isinstance(item, EntityLink):
+            if item.id:
+                ids.append(item.id)
+            continue
+    return ids
+
+
+def _build_inventory_function_operations(
+    *,
+    entity_id: str,
+    existing: list[BaseResource | EntityLink | str] | None,
+    updated: list[BaseResource | EntityLink | str] | None,
+) -> list[dict[str, Any]]:
+    existing_ids = set(_normalize_inventory_function_ids(existing))
+    updated_ids = set(_normalize_inventory_function_ids(updated))
+    to_add = sorted(updated_ids - existing_ids)
+    to_delete = sorted(existing_ids - updated_ids)
+
+    operations: list[dict[str, Any]] = []
+    if to_add:
+        operations.append(
+            {
+                "attribute": "inventoryFunction",
+                "entityId": entity_id,
+                "operation": "add",
+                "newValue": to_add,
+            }
+        )
+    if to_delete:
+        operations.append(
+            {
+                "attribute": "inventoryFunction",
+                "entityId": entity_id,
+                "operation": "delete",
+                "oldValue": to_delete,
+            }
+        )
+    return operations
+
+
 def _build_cas_scalar_operation(
     *,
     attribute: str,
@@ -117,6 +173,14 @@ def _build_cas_update_operations(existing: CasAmount, updated: CasAmount) -> lis
         if operation is not None:
             operations.append(operation)
 
+    operations.extend(
+        _build_inventory_function_operations(
+            entity_id=identifier,
+            existing=existing.inventory_function,
+            updated=updated.inventory_function,
+        )
+    )
+
     return operations
 
 
@@ -159,6 +223,13 @@ def _build_cas_patch_operations(
             )
             if target_operation is not None:
                 operations.append(target_operation)
+        operations.extend(
+            _build_inventory_function_operations(
+                entity_id=identifier,
+                existing=None,
+                updated=cas_amount.inventory_function,
+            )
+        )
 
     removals = [existing_lookup[key] for key in existing_lookup.keys() - updated_lookup.keys()]
     for cas_amount in removals:

{albert-1.12.0.dist-info → albert-1.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: albert
-Version: 1.12.0
+Version: 1.13.0
 Summary: The official Python SDK for the Albert Invent platform.
 Project-URL: Homepage, https://www.albertinvent.com/
 Project-URL: Documentation, https://docs.developer.albertinvent.com/albert-python

{albert-1.12.0.dist-info → albert-1.13.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-albert/__init__.py,sha256=IxZK2r9lTXpOpps67inCsmcTp1mEgdAXXsoIgn-EO2c,239
+albert/__init__.py,sha256=GFvvmokCWEGy95K7pNqV4Rme5Dfut9E7_lnKxNTtL1o,239
 albert/client.py,sha256=9SUy9AJpnFEUlSfxvbP1gJI776WcOoZykgPHx0EcF8g,12038
 albert/exceptions.py,sha256=-oxOJGE0A__aPUhri3qqb5YQ5qanECcTqamS73vGajM,3172
 albert/collections/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -80,9 +80,9 @@ albert/resources/entity_types.py,sha256=7lessP8StBDfi4DZMifvRNN4lspdQ5X0olBIu7lT
 albert/resources/facet.py,sha256=tjm6CsOL7T5LSV4G9406C1c5mpzY1gpfQYAttP6Hckw,372
 albert/resources/files.py,sha256=0q5H6rJ9il0Oy1FnwxVo3otq_w8YQeQjo29DgsTY8JQ,1067
 albert/resources/hazards.py,sha256=ltlbYct3PhvpTojqic0OcfJVKd2s6m59XqszzZZAuWo,375
-albert/resources/inventory.py,sha256=f0f8WGEGbbDgXUe9FoipsVsEmJkeuIKEId0WuEV_QXc,13608
+albert/resources/inventory.py,sha256=xkPZDOR-KFB79We56K4A35Td_L3kqe2OJ8Tvzrj1vQE,13944
 albert/resources/links.py,sha256=te7KIO7vfXrA-Qjngvb4JHy--SUVTG5M7GBFz8XdKto,1032
-albert/resources/lists.py,sha256=-ViW_wHJbaQh47zXNr1eszOMmwSFX-nQWTfOA8BzOG8,1687
+albert/resources/lists.py,sha256=UinEaBEEo80Q1Vf5Ys4ViPcNTSvAeCRaZiW41CHY4f0,1861
 albert/resources/locations.py,sha256=xqbSoO-IjLSyK74W1jSdsNL9kUN5jD6Wf9XNxw8czYA,821
 albert/resources/lots.py,sha256=8YC6fHv08uYvhh09ULzNtFmdyfLeM1z6Hkm9wcTvcX4,6839
 albert/resources/notebooks.py,sha256=PebrXEbR8J9afqwwTwGjFlEkYCdqRh3AtvVwYMypex8,9581
@@ -114,10 +114,10 @@ albert/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 albert/utils/_auth.py,sha256=YjzaGIzI9qP53nwdyE2Ezs-9UokzA38kgdE7Sxnyjd8,124
 albert/utils/_patch.py,sha256=e-bD2x6W3Wt4FaKFK477h3kZeyucn3DEB9m5DR7LzaA,24273
 albert/utils/data_template.py,sha256=AUwzfQ-I2HY-osq_Tme5KLwXfMzW2pJpiud7HAMh148,27874
-albert/utils/inventory.py,sha256=ViHxb62DVxniEJqOfD7Gf8HltLeCbq21y4rS1xkVRnY,5349
+albert/utils/inventory.py,sha256=hhL1wCn2vtF2z5FGwlX-3XIla4paB26QbmbStkG3yTQ,7433
 albert/utils/property_data.py,sha256=xb0CZAh_0zFiN5yQwcw3Mjk4En8MxMrJF9DKQI19UhM,25074
 albert/utils/tasks.py,sha256=ejhXD9yQR_ReDKLJ3k8ZxWxkYl-Mta_4OPXrz_lQiBI,19878
-albert-1.12.0.dist-info/METADATA,sha256=fPc1wDNv8rwCKP4RMRMyrmEx9-tShO2r8UuRPMr7yw0,15427
-albert-1.12.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-albert-1.12.0.dist-info/licenses/LICENSE,sha256=S7_vRdIhQmG7PmTlU8-BCCveuEcFZ6_3IUVdcoaJMuA,11348
-albert-1.12.0.dist-info/RECORD,,
+albert-1.13.0.dist-info/METADATA,sha256=9QEz1ieKLpVnsyjDhJdyfZ0BfYZOpn6aJAP-3vdYCAU,15427
+albert-1.13.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+albert-1.13.0.dist-info/licenses/LICENSE,sha256=S7_vRdIhQmG7PmTlU8-BCCveuEcFZ6_3IUVdcoaJMuA,11348
+albert-1.13.0.dist-info/RECORD,,