wizata-dsapi 2.2.0.dev2__py3-none-any.whl → 2.2.0.dev3__py3-none-any.whl

wizata_dsapi/__init__.py

@@ -65,6 +65,7 @@ from .connection import Connection, ConnectionMapping, ConnectionDirection, Tran
     EdgeDeviceKey, EdgeDeviceKeyStatus
 from .connection_types import ConnectionType, CONNECTION_TYPES, TRANSFORM_SCHEMAS, get_connection_type, \
     list_connection_types
+from .product import ProductKind, Product, ProductComponent, BUILTIN_PRODUCT_KINDS, is_valid_product_key
 from .dashboard import Dashboard, DashboardType
 from .mobile_asset import MobileAssetTile, MobileAssetTileType, MobileAssetConfig
 from .graylog_log import GraylogLog

wizata_dsapi/product.py

@@ -0,0 +1,210 @@
+"""
+Production Planner product definitions (DEV-6070, epic DEV-6044).
+
+ProductKind — what role a product plays: raw material, container, finished product,
+  or a kind the customer adds. The three built-in kinds (``BUILTIN_PRODUCT_KINDS``)
+  are seeded on every tenant and cannot be deleted.
+
+Product — something the plant consumes or makes, identified by a Wizata key
+  (e.g. ``can_33_coke``). ``category`` is a backend category id (Volume, Count, ...):
+  it says how a quantity of that product is expressed.
+
+ProductComponent — one line of a product's bill of materials: ``product`` is made of
+  ``quantity`` ``unit`` of ``composed_of``. The unit is a backend unit id and must
+  belong to the category of ``composed_of`` (33 ml of coke, 1 pcs of can).
+  ``unit_symbol`` is filled on read only.
+
+All of it is served under ``/dsapi/planner`` and gated by the Production Planner
+add-on license (DEV-6069).
+"""
+
+import re
+from datetime import datetime, timezone
+from typing import Optional
+
+#: key -> (name, description). Seeded by alembic 0019; editable, never deletable.
+BUILTIN_PRODUCT_KINDS = {
+    "raw": ("Raw Material",
+            "A material bought or produced upstream and consumed to make other "
+            "products (e.g. syrup, water, aluminium)."),
+    "container": ("Containers",
+                  "Packaging that holds or carries a product (e.g. can, bottle, box, pallet)."),
+    "finished": ("Finished Product",
+                 "A product ready to ship or sell, made from raw materials and containers."),
+}
+
+#: A Wizata key: starts with a letter or digit, then letters, digits, ``_ - .``.
+PRODUCT_KEY_PATTERN = re.compile(r"^[A-Za-z0-9][A-Za-z0-9_\-.]{0,63}$")
+
+
+def is_valid_product_key(key) -> bool:
+    return isinstance(key, str) and PRODUCT_KEY_PATTERN.match(key) is not None
+
+
+def _ts_out(value):
+    if value is None:
+        return None
+    if value.tzinfo is None:
+        value = value.replace(tzinfo=timezone.utc)
+    return value.isoformat()
+
+
+def _ts_in(value):
+    if value is None or isinstance(value, datetime):
+        return value
+    return datetime.fromisoformat(str(value).replace("Z", "+00:00"))
+
+
+def _uuid_str(value):
+    """Backend ids travel as strings, lower-cased so equal ids compare equal."""
+    if value is None or value == "":
+        return None
+    return str(value).lower()
+
+
+class ProductKind:
+    """ProductKind entity stored in PostgreSQL ``"ProductKind"``. PK ``key``."""
+
+    @classmethod
+    def route(cls):
+        return "planner/productkinds"
+
+    def __init__(self, key: str = None, name: str = None, description: str = None):
+        self.key: Optional[str] = key
+        self.name: Optional[str] = name
+        self.description: Optional[str] = description
+        self.created_at: Optional[datetime] = None
+        self.updated_at: Optional[datetime] = None
+
+    @property
+    def builtin(self) -> bool:
+        return self.key in BUILTIN_PRODUCT_KINDS
+
+    def to_dict(self) -> dict:
+        obj = {}
+        if self.key is not None:
+            obj["key"] = self.key
+        if self.name is not None:
+            obj["name"] = self.name
+        if self.description is not None:
+            obj["description"] = self.description
+        obj["builtin"] = self.builtin
+        if self.created_at is not None:
+            obj["createdAt"] = _ts_out(self.created_at)
+        if self.updated_at is not None:
+            obj["updatedAt"] = _ts_out(self.updated_at)
+        return obj
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "ProductKind":
+        obj = cls()
+        obj.key = data.get("key")
+        obj.name = data.get("name")
+        obj.description = data.get("description")
+        obj.created_at = _ts_in(data.get("createdAt") or data.get("created_at"))
+        obj.updated_at = _ts_in(data.get("updatedAt") or data.get("updated_at"))
+        return obj
+
+    def api_id(self) -> str:
+        return self.key
+
+    def __repr__(self) -> str:
+        return f"ProductKind(key={self.key!r}, name={self.name!r})"
+
+
+class Product:
+    """Product entity stored in PostgreSQL ``"Product"``. PK ``key``."""
+
+    @classmethod
+    def route(cls):
+        return "planner/products"
+
+    def __init__(self, key: str = None, name: str = None, category=None,
+                 kind: str = None, description: str = None):
+        self.key: Optional[str] = key
+        self.name: Optional[str] = name
+        self.category: Optional[str] = _uuid_str(category)
+        self.kind: Optional[str] = kind
+        self.description: Optional[str] = description
+        self.created_at: Optional[datetime] = None
+        self.updated_at: Optional[datetime] = None
+
+    def to_dict(self) -> dict:
+        obj = {}
+        if self.key is not None:
+            obj["key"] = self.key
+        if self.name is not None:
+            obj["name"] = self.name
+        if self.category is not None:
+            obj["category"] = self.category
+        if self.kind is not None:
+            obj["kind"] = self.kind
+        if self.description is not None:
+            obj["description"] = self.description
+        if self.created_at is not None:
+            obj["createdAt"] = _ts_out(self.created_at)
+        if self.updated_at is not None:
+            obj["updatedAt"] = _ts_out(self.updated_at)
+        return obj
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "Product":
+        obj = cls()
+        obj.key = data.get("key")
+        obj.name = data.get("name")
+        obj.category = _uuid_str(data.get("category"))
+        obj.kind = data.get("kind")
+        obj.description = data.get("description")
+        obj.created_at = _ts_in(data.get("createdAt") or data.get("created_at"))
+        obj.updated_at = _ts_in(data.get("updatedAt") or data.get("updated_at"))
+        return obj
+
+    def api_id(self) -> str:
+        return self.key
+
+    def __repr__(self) -> str:
+        return f"Product(key={self.key!r}, kind={self.kind!r}, category={self.category!r})"
+
+
+class ProductComponent:
+    """
+    One bill-of-materials line, stored in PostgreSQL ``"ProductComponent"``.
+    PK (``product``, ``composed_of``).
+    """
+
+    def __init__(self, product: str = None, composed_of: str = None,
+                 quantity: float = None, unit=None):
+        self.product: Optional[str] = product
+        self.composed_of: Optional[str] = composed_of
+        self.quantity: Optional[float] = quantity
+        self.unit: Optional[str] = _uuid_str(unit)
+        #: read-only, filled from the backend unit on read
+        self.unit_symbol: Optional[str] = None
+
+    def to_dict(self) -> dict:
+        obj = {}
+        if self.product is not None:
+            obj["product"] = self.product
+        if self.composed_of is not None:
+            obj["composedOf"] = self.composed_of
+        if self.quantity is not None:
+            obj["quantity"] = self.quantity
+        if self.unit is not None:
+            obj["unit"] = self.unit
+        if self.unit_symbol is not None:
+            obj["unitSymbol"] = self.unit_symbol
+        return obj
+
+    @classmethod
+    def from_dict(cls, data: dict) -> "ProductComponent":
+        obj = cls()
+        obj.product = data.get("product")
+        obj.composed_of = data.get("composedOf") or data.get("composed_of")
+        obj.quantity = data.get("quantity")
+        obj.unit = _uuid_str(data.get("unit"))
+        obj.unit_symbol = data.get("unitSymbol") or data.get("unit_symbol")
+        return obj
+
+    def __repr__(self) -> str:
+        return (f"ProductComponent({self.product!r} <- {self.quantity!r} {self.unit_symbol or self.unit!r} "
+                f"of {self.composed_of!r})")

wizata_dsapi/version.py

@@ -1 +1 @@
-__version__ = "2.2.0.dev2"
+__version__ = "2.2.0.dev3"

wizata_dsapi/wizata_dsapi_client.py

@@ -54,6 +54,7 @@ from .mobile_asset import MobileAssetTile, MobileAssetConfig
 from .version import __version__
 import ast
 from .deployment import Deployment, DeploymentStatus
+from .product import ProductKind, Product, ProductComponent
 from .api_config import _registry
 
 from .streamlit_utils import get_streamlit_token, get_streamlit_domain
@@ -2731,6 +2732,142 @@ class WizataDSAPIClient(ApiInterface, ApiDtoInterface):
             raise self.__raise_error(response)
 
 
+    # ------------------------------------------------------------------
+    # Production Planner - product definitions (DEV-6070)
+    #
+    # Everything below answers 403 PLANNER_LICENSE_DISABLED while the
+    # Production Planner add-on is not active on the environment (DEV-6069).
+    # ------------------------------------------------------------------
+
+    def _planner_request(self, method: str, route: str, body=None, params: dict = None,
+                         expected=(200,)):
+        response = requests.request(
+            method,
+            self.__url() + route,
+            headers=self.__header(),
+            data=json.dumps(body) if body is not None else None,
+            params=params,
+        )
+        if response.status_code not in expected:
+            raise self.__raise_error(response)
+        return response
+
+    @staticmethod
+    def _planner_key(value) -> str:
+        key = value.key if hasattr(value, "key") else value
+        if not key:
+            raise ValueError("please provide a key")
+        return urllib.parse.quote(str(key), safe="")
+
+    def search_product_kinds(self, search_query=None) -> PagedQueryResult:
+        """
+        The only way to list product kinds: the built-in raw / container / finished and
+        the tenant's own.
+
+        :param search_query: SearchQuery or dict; filters on ``name``, ``_search`` over
+            key and name.
+        :return: PagedQueryResult with ProductKind results.
+        """
+        if search_query is None:
+            search_query = SearchQuery()
+        elif isinstance(search_query, dict):
+            search_query = SearchQuery.from_dict(search_query)
+        response = self._planner_request("POST", "planner/productkinds/search",
+                                         body=search_query.to_dict())
+        return PagedQueryResult.from_dict(response.json(), dto_class=ProductKind)
+
+    def get_product_kind(self, key: str):
+        """
+        :param key: product kind key (e.g. ``raw``).
+        :return: ProductKind, or None when it does not exist.
+        """
+        response = self._planner_request("GET", f"planner/productkinds/{self._planner_key(key)}/",
+                                         expected=(200, 404))
+        return ProductKind.from_dict(response.json()) if response.status_code == 200 else None
+
+    def create_product_kind(self, kind: ProductKind) -> ProductKind:
+        response = self._planner_request("POST", "planner/productkinds/", body=kind.to_dict(),
+                                         expected=(201,))
+        return ProductKind.from_dict(response.json())
+
+    def update_product_kind(self, kind: ProductKind) -> ProductKind:
+        response = self._planner_request("PUT", f"planner/productkinds/{self._planner_key(kind)}/",
+                                         body=kind.to_dict())
+        return ProductKind.from_dict(response.json())
+
+    def delete_product_kind(self, kind):
+        """
+        :param kind: ProductKind or its key. A built-in kind, or a kind still used by a
+            product, is refused (409).
+        """
+        self._planner_request("DELETE", f"planner/productkinds/{self._planner_key(kind)}/",
+                              expected=(204,))
+
+    def search_products(self, search_query=None) -> PagedQueryResult:
+        """
+        The only way to list products.
+
+        :param search_query: SearchQuery or dict; filters on ``kind``, ``category``
+            (backend category id), ``name``, and ``_search`` over key and name.
+        :return: PagedQueryResult with Product results.
+        """
+        if search_query is None:
+            search_query = SearchQuery()
+        elif isinstance(search_query, dict):
+            search_query = SearchQuery.from_dict(search_query)
+        response = self._planner_request("POST", "planner/products/search",
+                                         body=search_query.to_dict())
+        return PagedQueryResult.from_dict(response.json(), dto_class=Product)
+
+    def get_product(self, key: str):
+        """
+        :return: Product, or None when it does not exist.
+        """
+        response = self._planner_request("GET", f"planner/products/{self._planner_key(key)}/",
+                                         expected=(200, 404))
+        return Product.from_dict(response.json()) if response.status_code == 200 else None
+
+    def create_product(self, product: Product) -> Product:
+        response = self._planner_request("POST", "planner/products/", body=product.to_dict(),
+                                         expected=(201,))
+        return Product.from_dict(response.json())
+
+    def update_product(self, product: Product) -> Product:
+        response = self._planner_request("PUT", f"planner/products/{self._planner_key(product)}/",
+                                         body=product.to_dict())
+        return Product.from_dict(response.json())
+
+    def delete_product(self, product):
+        """
+        :param product: Product or its key. Refused (409) while another product is made
+            of it; its own bill of materials is deleted with it.
+        """
+        self._planner_request("DELETE", f"planner/products/{self._planner_key(product)}/",
+                              expected=(204,))
+
+    def get_product_components(self, product) -> list:
+        """
+        :param product: Product or its key.
+        :return: its bill of materials, a list of ProductComponent with unit_symbol set.
+        """
+        response = self._planner_request(
+            "GET", f"planner/products/{self._planner_key(product)}/components/")
+        return [ProductComponent.from_dict(c) for c in response.json()]
+
+    def set_product_components(self, product, components: list) -> list:
+        """
+        Replace a product's bill of materials (an empty list clears it).
+
+        :param product: Product or its key.
+        :param components: list of ProductComponent (``product`` may be left unset).
+        :return: the stored components, unit_symbol set.
+        """
+        body = [c.to_dict() if isinstance(c, ProductComponent) else c for c in components]
+        response = self._planner_request(
+            "PUT", f"planner/products/{self._planner_key(product)}/components/", body=body)
+        return [ProductComponent.from_dict(c) for c in response.json()]
+
+
 def api() -> WizataDSAPIClient:
     """
     Create a WizataDSAPIClient from environment variables.

{wizata_dsapi-2.2.0.dev2.dist-info → wizata_dsapi-2.2.0.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 2.2.0.dev2
+Version: 2.2.0.dev3
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-2.2.0.dev2.dist-info → wizata_dsapi-2.2.0.dev3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-wizata_dsapi/__init__.py,sha256=8zazLs_NYKRhveJt3vhy1SxwXHvBv5fEl6yfX-UilrI,3374
+wizata_dsapi/__init__.py,sha256=E5OJj1DhWW9I96DmE1fzMPsgvkPCqeEtnNAK05sEQgQ,3479
 wizata_dsapi/api_config.py,sha256=zOgvpkB3FE2ny_2vAui7B1hcAMFCLvfb2ep3m-b7dIw,8270
 wizata_dsapi/api_dto.py,sha256=F9HjNa_vqYhdEMZwwk8E3SmCxuMBcxFnMp1eE9dxLvg,2910
 wizata_dsapi/api_interface.py,sha256=mCq0-rlwWL4GKB2QnxSPuyTecYOXXknmPOlQe_SEf2I,14670
@@ -33,6 +33,7 @@ wizata_dsapi/paged_query_result.py,sha256=QOpJMAS6rxJfiz3a5oSfc0gOvwQeZTEhOtBoSs
 wizata_dsapi/pipeline.py,sha256=mMkNzFQsC9s1ZuC0HrfLG1BsZCnYYlrrhPlo6Out1Zs,40872
 wizata_dsapi/pipeline_image.py,sha256=WtMY0_ncba9WzZ96KQINiYshJegQdArzhspKTGWBgVA,7758
 wizata_dsapi/plot.py,sha256=KLeYj6xpG_XFSDuR4A8Iqp2fmAVLeDsHaUOVT2sRUYM,4033
+wizata_dsapi/product.py,sha256=DEuqi-1KDih3fyLEmVlsKao57tCpPJQ2sTwhEqe-r0A,7577
 wizata_dsapi/request.py,sha256=t5eykjLnhREYrcBcZcFyxtk89nBCP44jC0SjEChmHRg,27785
 wizata_dsapi/script.py,sha256=1Mjw3BNxN_R5dx0f55T0WuebqqRr_69rsK5yuQZOjmQ,19467
 wizata_dsapi/search.py,sha256=pHX-xGIB0E6gJkTxXmLjFe_b8-0gei7hfkW0JnxzCl0,4691
@@ -44,10 +45,10 @@ wizata_dsapi/trigger.py,sha256=waOX2AhrLvFzNFRQ535j6iemgg_QJP2QsWGL2ipljFg,8848
 wizata_dsapi/twin.py,sha256=8AnrxSaL7dMrILXJ3-GGmmUz078j3qqwzDTe6i86xqA,12672
 wizata_dsapi/twinregistration.py,sha256=iXt55AVGAQNl0jmxa3ZVynPqlSsIunFf4sLfp-MWTw0,13244
 wizata_dsapi/user.py,sha256=-hchKAMZULu9P00RUHDqu_5AT4NI5BYOJd8XCBrt3eQ,1950
-wizata_dsapi/version.py,sha256=bKxMxf4FLo3j7fCZ_ayiGhLIAkPquwpEeY8g9TOU0P8,27
+wizata_dsapi/version.py,sha256=8rZEHkrNQw-J8A6mFRKebsH-vIimD3thK_hO18YaEmk,27
 wizata_dsapi/wizard_function.py,sha256=RbM7W7Gf-6Rhp_1dU9DBYkHaciknGAGvuAndhAS_vyo,942
 wizata_dsapi/wizard_request.py,sha256=v6BaqKLKvTWmUSo0_gda9FabAQz5x_-GOH1Av50GzFo,3762
-wizata_dsapi/wizata_dsapi_client.py,sha256=DX8WXKT5gi_nfHHX2-jz4dvUe-Bo_HszQmr94R3JvV0,112418
+wizata_dsapi/wizata_dsapi_client.py,sha256=bhAaInsbPBVc6EPe56_-U2_ggoEqHBfVH-1FhmobjqI,118779
 wizata_dsapi/words.py,sha256=tV8CqzCqODZCV7PgBxBF5exBxeF_ya9t5DiUy-cg6Sg,1535
 wizata_dsapi/models/__init__.py,sha256=O3Tusyhl8hFb-5QaEBY7DJISRB8Gf_Xbxxr-aFDR1_U,380
 wizata_dsapi/models/common.py,sha256=5BC3Dc6kMvqPGxHgoCuGVOynu_F1MWh7EJSzXIICSSg,27196
@@ -56,8 +57,8 @@ wizata_dsapi/plots/common.py,sha256=7F4WMBaddubnCpdSafgaI9U9Sz0Sngg0VREP_vnIMno,
 wizata_dsapi/plots/theme.py,sha256=D4sSgEwabXVxF3z1GV8DivXJbbY61KgxXAWfjUbNDAc,3594
 wizata_dsapi/scripts/__init__.py,sha256=9fRdTMuuXcqMkIsUPf5dwFsTWwBzKteQdAMVkyy2acQ,212
 wizata_dsapi/scripts/common.py,sha256=T2cP8aGPg8OgIvBWuQBNonVtph_dHlNFubByuYrLb0o,15675
-wizata_dsapi-2.2.0.dev2.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-wizata_dsapi-2.2.0.dev2.dist-info/METADATA,sha256=dVvL2OBGr42dKhKzMnfULhMExHhmUkJkAW1X8oR7_uU,14037
-wizata_dsapi-2.2.0.dev2.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-wizata_dsapi-2.2.0.dev2.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
-wizata_dsapi-2.2.0.dev2.dist-info/RECORD,,
+wizata_dsapi-2.2.0.dev3.dist-info/licenses/LICENSE.txt,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+wizata_dsapi-2.2.0.dev3.dist-info/METADATA,sha256=ZxA42rrbx5Sv6s_IHW7bl-wnqUG7h-EeVdFyIoPMFy0,14037
+wizata_dsapi-2.2.0.dev3.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+wizata_dsapi-2.2.0.dev3.dist-info/top_level.txt,sha256=-OeTJbEnh5DuWyTOHtvw0Dw3LRg3G27TNS6W4ZtfwPs,13
+wizata_dsapi-2.2.0.dev3.dist-info/RECORD,,