MPT-13322 Add catalog products media

albertsola · albertsola · commit 447099e327a5 · 2025-09-10T11:33:39.000+01:00
diff --git a/mpt_api_client/resources/catalog/products.py b/mpt_api_client/resources/catalog/products.py
@@ -11,6 +11,14 @@
     AsyncPublishableMixin,
     PublishableMixin,
 )
+from mpt_api_client.resources.catalog.products_item_groups import (
+    AsyncItemGroupsService,
+    ItemGroupsService,
+)
+from mpt_api_client.resources.catalog.products_media import (
+    AsyncMediaService,
+    MediaService,
+)
 from mpt_api_client.resources.catalog.products_parameter_groups import (
     AsyncParameterGroupsService,
     ParameterGroupsService,
@@ -43,12 +51,24 @@ class ProductsService(
 ):
     """Products service."""
 
+    def item_groups(self, product_id: str) -> ItemGroupsService:
+        """Return item_groups service."""
+        return ItemGroupsService(
+            http_client=self.http_client, endpoint_params={"product_id": product_id}
+        )
+
     def parameter_groups(self, product_id: str) -> ParameterGroupsService:
         """Return parameter_groups service."""
         return ParameterGroupsService(
             http_client=self.http_client, endpoint_params={"product_id": product_id}
         )
 
+    def media(self, product_id: str) -> MediaService:
+        """Return media service."""
+        return MediaService(
+            http_client=self.http_client, endpoint_params={"product_id": product_id}
+        )
+
     def product_parameters(self, product_id: str) -> ParametersService:
         """Return product_parameters service."""
         return ParametersService(
@@ -66,12 +86,24 @@ class AsyncProductsService(
 ):
     """Products service."""
 
+    def item_groups(self, product_id: str) -> AsyncItemGroupsService:
+        """Return item_groups service."""
+        return AsyncItemGroupsService(
+            http_client=self.http_client, endpoint_params={"product_id": product_id}
+        )
+
     def parameter_groups(self, product_id: str) -> AsyncParameterGroupsService:
         """Return parameter_groups service."""
         return AsyncParameterGroupsService(
             http_client=self.http_client, endpoint_params={"product_id": product_id}
         )
 
+    def media(self, product_id: str) -> AsyncMediaService:
+        """Return media service."""
+        return AsyncMediaService(
+            http_client=self.http_client, endpoint_params={"product_id": product_id}
+        )
+
     def product_parameters(self, product_id: str) -> AsyncParametersService:
         """Return product_parameters service."""
         return AsyncParametersService(
diff --git a/mpt_api_client/resources/catalog/products_item_groups.py b/mpt_api_client/resources/catalog/products_item_groups.py
@@ -0,0 +1,40 @@
+from mpt_api_client.http import AsyncService, CreateMixin, DeleteMixin, Service
+from mpt_api_client.http.mixins import (
+    AsyncCreateMixin,
+    AsyncDeleteMixin,
+    AsyncUpdateMixin,
+    UpdateMixin,
+)
+from mpt_api_client.models import Model
+
+
+class ItemGroup(Model):
+    """Item Group resource."""
+
+
+class ItemGroupsServiceConfig:
+    """Item Groups service configuration."""
+
+    _endpoint = "/public/v1/catalog/products/{product_id}/item-groups"
+    _model_class = ItemGroup
+    _collection_key = "data"
+
+
+class ItemGroupsService(
+    CreateMixin[ItemGroup],
+    DeleteMixin,
+    UpdateMixin[ItemGroup],
+    Service[ItemGroup],
+    ItemGroupsServiceConfig,
+):
+    """Item Groups service."""
+
+
+class AsyncItemGroupsService(
+    AsyncCreateMixin[ItemGroup],
+    AsyncDeleteMixin,
+    AsyncUpdateMixin[ItemGroup],
+    AsyncService[ItemGroup],
+    ItemGroupsServiceConfig,
+):
+    """Item Groups service."""
diff --git a/mpt_api_client/resources/catalog/products_media.py b/mpt_api_client/resources/catalog/products_media.py
@@ -0,0 +1,185 @@
+import json
+from typing import override
+
+from httpx import Response
+from httpx._types import FileTypes
+
+from mpt_api_client.http import AsyncService, CreateMixin, DeleteMixin, Service
+from mpt_api_client.http.mixins import (
+    AsyncCreateMixin,
+    AsyncDeleteMixin,
+    AsyncUpdateMixin,
+    UpdateMixin,
+)
+from mpt_api_client.models import FileModel, Model, ResourceData
+
+
+def _json_to_file_payload(resource_data: ResourceData) -> bytes:
+    return json.dumps(
+        resource_data, ensure_ascii=False, separators=(",", ":"), allow_nan=False
+    ).encode("utf-8")
+
+
+class Media(Model):
+    """Media resource."""
+
+
+class MediaServiceConfig:
+    """Media service configuration."""
+
+    _endpoint = "/public/v1/catalog/products/{product_id}/media"
+    _model_class = Media
+    _collection_key = "data"
+
+
+class MediaService(
+    CreateMixin[Media],
+    DeleteMixin,
+    UpdateMixin[Media],
+    Service[Media],
+    MediaServiceConfig,
+):
+    """Media service."""
+
+    @override
+    def create(
+        self,
+        resource_data: ResourceData | None = None,
+        files: dict[str, FileTypes] | None = None,  # noqa: WPS221
+    ) -> Media:
+        """Create Media resource.
+
+        Currently are two types of media resources available image and video.
+
+        Video:
+            resource_data:
+                {
+                  "name": "SomeMediaFile",
+                  "description":"Some media description",
+                  "mediaType": "Video",
+                  "url": http://www.somemedia.com/somevideo.avi,
+                  "displayOrder": 1
+                }
+            files: Add an image with the video thumbnail
+
+        Image:
+            resource_data:
+                {
+                  "name": "SomeMediaFile",
+                  "description":"Some media description",
+                  "mediaType": "Video",
+                  "displayOrder": 1
+                }
+            files: The image itself
+
+        Args:
+            resource_data: Resource data.
+            files: Files data.
+
+        Returns:
+            Media resource.
+        """
+        files = files or {}
+
+        # Note: This is a workaround to fulfill MPT API request format
+        #
+        # HTTPx does not support sending json and files in the same call
+        # currently only supports sending form-data and files in the same call.
+        # https://www.python-httpx.org/quickstart/#sending-multipart-file-uploads
+        #
+        # MPT API expects files and data to be submitted in a multipart form-data upload.
+        #
+        # Current workaround is to send the json data as an unnamed file.
+        # This ends adding the json as payload multipart data.
+        #
+        # json.dumps is setup using the same params of httpx json encoder to produce the same
+        # encodings.
+
+        if resource_data:
+            files["_media_data"] = (
+                None,
+                _json_to_file_payload(resource_data),
+                "application/json",
+            )
+
+        response = self.http_client.post(self.endpoint, files=files)
+        response.raise_for_status()
+        return Media.from_response(response)
+
+    def download(self, media_id: str) -> FileModel:
+        """Download the media file for the given media ID.
+
+        Args:
+            media_id: Media ID.
+
+        Returns:
+            Media file.
+        """
+        response: Response = self._resource_do_request(
+            media_id, method="GET", headers={"Accept": "*"}
+        )
+        return FileModel(response)
+
+
+class AsyncMediaService(
+    AsyncCreateMixin[Media],
+    AsyncDeleteMixin,
+    AsyncUpdateMixin[Media],
+    AsyncService[Media],
+    MediaServiceConfig,
+):
+    """Media service."""
+
+    @override
+    async def create(
+        self,
+        resource_data: ResourceData | None = None,
+        files: dict[str, FileTypes] | None = None,  # noqa: WPS221
+    ) -> Media:
+        """Create Media resource.
+
+        Args:
+            resource_data: Resource data.
+            files: Files data.
+
+        Returns:
+            Media resource.
+        """
+        files = files or {}
+
+        # Note: This is a workaround to fulfill MPT API request format
+        #
+        # HTTPx does not support sending json and files in the same call
+        # currently only supports sending form-data and files in the same call.
+        # https://www.python-httpx.org/quickstart/#sending-multipart-file-uploads
+        #
+        # MPT API expects files and data to be submitted in a multipart form-data upload.
+        #
+        # Current workaround is to send the json data as an unnamed file.
+        # This ends adding the json as payload multipart data.
+        #
+        # json.dumps is setup using the same params of httpx json encoder to produce the same
+        # encodings.
+
+        if resource_data:
+            files["_media_data"] = (
+                None,
+                _json_to_file_payload(resource_data),
+                "application/json",
+            )
+
+        response = await self.http_client.post(self.endpoint, files=files)
+        response.raise_for_status()
+        return Media.from_response(response)
+
+    async def download(self, media_id: str) -> FileModel:
+        """Download the media file for the given media ID.
+
+        Args:
+            media_id: Media ID.
+
+        Returns:
+            Media file.
+        """
+        response = await self._resource_do_request(media_id, method="GET", headers={"Accept": "*"})
+        return FileModel(response)
diff --git a/setup.cfg b/setup.cfg
@@ -33,10 +33,12 @@ extend-ignore =
 
 per-file-ignores =
   mpt_api_client/rql/query_builder.py: WPS110 WPS115 WPS210 WPS214
-  mpt_api_client/resources/catalog/products.py: WPS215
+  mpt_api_client/resources/catalog/products.py: WPS204 WPS215
   mpt_api_client/resources/catalog/items.py: WPS215
+  mpt_api_client/resources/catalog/products_item_groups.py: WPS215
   mpt_api_client/resources/catalog/products_parameter_groups.py: WPS215
   mpt_api_client/resources/catalog/products_parameters.py: WPS215
+  mpt_api_client/resources/catalog/products_media.py: WPS215
   tests/http/test_async_service.py: WPS204 WPS202
   tests/http/test_service.py: WPS204 WPS202
 
diff --git a/tests/resources/catalog/test_products.py b/tests/resources/catalog/test_products.py
@@ -1,6 +1,14 @@
 import pytest
 
 from mpt_api_client.resources.catalog.products import AsyncProductsService, ProductsService
+from mpt_api_client.resources.catalog.products_item_groups import (
+    AsyncItemGroupsService,
+    ItemGroupsService,
+)
+from mpt_api_client.resources.catalog.products_media import (
+    AsyncMediaService,
+    MediaService,
+)
 from mpt_api_client.resources.catalog.products_parameter_groups import (
     AsyncParameterGroupsService,
     ParameterGroupsService,
@@ -38,7 +46,9 @@ def test_async_mixins_present(async_products_service, method):
 @pytest.mark.parametrize(
     ("service_method", "expected_service_class"),
     [
+        ("item_groups", ItemGroupsService),
         ("parameter_groups", ParameterGroupsService),
+        ("media", MediaService),
         ("product_parameters", ParametersService),
     ],
 )
@@ -52,7 +62,9 @@ def test_property_services(products_service, service_method, expected_service_cl
 @pytest.mark.parametrize(
     ("service_method", "expected_service_class"),
     [
+        ("item_groups", AsyncItemGroupsService),
         ("parameter_groups", AsyncParameterGroupsService),
+        ("media", AsyncMediaService),
         ("product_parameters", AsyncParametersService),
     ],
 )
diff --git a/tests/resources/catalog/test_products_item_groups.py b/tests/resources/catalog/test_products_item_groups.py
@@ -0,0 +1,36 @@
+import pytest
+
+from mpt_api_client.resources.catalog.products_item_groups import (
+    AsyncItemGroupsService,
+    ItemGroupsService,
+)
+
+
+@pytest.fixture
+def item_groups_service(http_client):
+    return ItemGroupsService(http_client=http_client, endpoint_params={"product_id": "PRD-001"})
+
+
+@pytest.fixture
+def async_item_groups_service(async_http_client):
+    return AsyncItemGroupsService(
+        http_client=async_http_client, endpoint_params={"product_id": "PRD-001"}
+    )
+
+
+def test_endpoint(item_groups_service):
+    assert item_groups_service.endpoint == "/public/v1/catalog/products/PRD-001/item-groups"
+
+
+def test_async_endpoint(async_item_groups_service):
+    assert async_item_groups_service.endpoint == "/public/v1/catalog/products/PRD-001/item-groups"
+
+
+@pytest.mark.parametrize("method", ["get", "create", "delete", "update"])
+def test_methods_present(item_groups_service, method):
+    assert hasattr(item_groups_service, method)
+
+
+@pytest.mark.parametrize("method", ["get", "create", "delete", "update"])
+def test_async_methods_present(async_item_groups_service, method):
+    assert hasattr(async_item_groups_service, method)
diff --git a/tests/resources/catalog/test_products_media.py b/tests/resources/catalog/test_products_media.py
