MPT-12840 Add agreements attachments

Author: albertsola

mpt_api_client/http/async_client.py

@@ -32,6 +32,7 @@ def __init__(
         base_headers = {
             "User-Agent": "swo-marketplace-client/1.0",
             "Authorization": f"Bearer {api_token}",
+            "Accept": "application/json",
         }
         super().__init__(
             base_url=base_url,

mpt_api_client/http/async_service.py

@@ -91,19 +91,6 @@ async def get(self, resource_id: str, select: list[str] | str | None = None) ->
             select = ",".join(select) if select else None
         return await self._resource_action(resource_id=resource_id, query_params={"select": select})
 
-    async def update(self, resource_id: str, resource_data: ResourceData) -> Model:
-        """Update a resource using `PUT /endpoint/{resource_id}`.
-
-        Args:
-            resource_id: Resource ID.
-            resource_data: Resource data.
-
-        Returns:
-            Resource object.
-
-        """
-        return await self._resource_action(resource_id, "PUT", json=resource_data)
-
     async def _fetch_page_as_response(self, limit: int = 100, offset: int = 0) -> httpx.Response:
         """Fetch one page of resources.
 
@@ -119,13 +106,14 @@ async def _fetch_page_as_response(self, limit: int = 100, offset: int = 0) -> ht
 
         return response
 
-    async def _resource_do_request(
+    async def _resource_do_request(  # noqa: WPS211
         self,
         resource_id: str,
         method: str = "GET",
         action: str | None = None,
         json: ResourceData | ResourceList | None = None,
         query_params: QueryParam | None = None,
+        headers: dict[str, str] | None = None,
     ) -> httpx.Response:
         """Perform an action on a specific resource using.
 
@@ -138,13 +126,16 @@ async def _resource_do_request(
             action: The action name to use.
             json: The updated resource data.
             query_params: Additional query parameters.
+            headers: Additional headers.
 
         Raises:
             HTTPError: If the action fails.
         """
-        resource_url = urljoin(f"{self._endpoint}/", resource_id)
+        resource_url = urljoin(f"{self.endpoint}/", resource_id)
         url = urljoin(f"{resource_url}/", action) if action else resource_url
-        response = await self.http_client.request(method, url, json=json, params=query_params)
+        response = await self.http_client.request(
+            method, url, json=json, params=query_params, headers=headers
+        )
         response.raise_for_status()
         return response
 

mpt_api_client/http/client.py

@@ -32,6 +32,7 @@ def __init__(
         base_headers = {
             "User-Agent": "swo-marketplace-client/1.0",
             "Authorization": f"Bearer {api_token}",
+            "content-type": "application/json",
         }
         super().__init__(
             base_url=base_url,

mpt_api_client/http/mixins.py

@@ -12,7 +12,7 @@ def create(self, resource_data: ResourceData) -> Model:
         Returns:
             New resource created.
         """
-        response = self.http_client.post(self._endpoint, json=resource_data)  # type: ignore[attr-defined]
+        response = self.http_client.post(self.endpoint, json=resource_data)  # type: ignore[attr-defined]
         response.raise_for_status()
 
         return self._model_class.from_response(response)  # type: ignore[attr-defined, no-any-return]
@@ -31,6 +31,23 @@ def delete(self, resource_id: str) -> None:
         response.raise_for_status()
 
 
+class UpdateMixin[Model]:
+    """Update resource mixin."""
+
+    def update(self, resource_id: str, resource_data: ResourceData) -> Model:
+        """Update a resource using `PUT /endpoint/{resource_id}`.
+
+        Args:
+            resource_id: Resource ID.
+            resource_data: Resource data.
+
+        Returns:
+            Resource object.
+
+        """
+        return self._resource_action(resource_id, "PUT", json=resource_data)  # type: ignore[attr-defined, no-any-return]
+
+
 class AsyncCreateMixin[Model]:
     """Create resource mixin."""
 
@@ -40,7 +57,7 @@ async def create(self, resource_data: ResourceData) -> Model:
         Returns:
             New resource created.
         """
-        response = await self.http_client.post(self._endpoint, json=resource_data)  # type: ignore[attr-defined]
+        response = await self.http_client.post(self.endpoint, json=resource_data)  # type: ignore[attr-defined]
         response.raise_for_status()
 
         return self._model_class.from_response(response)  # type: ignore[attr-defined, no-any-return]
@@ -55,6 +72,23 @@ async def delete(self, resource_id: str) -> None:
         Args:
             resource_id: Resource ID.
         """
-        url = urljoin(f"{self._endpoint}/", resource_id)  # type: ignore[attr-defined]
+        url = urljoin(f"{self.endpoint}/", resource_id)  # type: ignore[attr-defined]
         response = await self.http_client.delete(url)  # type: ignore[attr-defined]
         response.raise_for_status()
+
+
+class AsyncUpdateMixin[Model]:
+    """Update resource mixin."""
+
+    async def update(self, resource_id: str, resource_data: ResourceData) -> Model:
+        """Update a resource using `PUT /endpoint/{resource_id}`.
+
+        Args:
+            resource_id: Resource ID.
+            resource_data: Resource data.
+
+        Returns:
+            Resource object.
+
+        """
+        return await self._resource_action(resource_id, "PUT", json=resource_data)  # type: ignore[attr-defined, no-any-return]

mpt_api_client/http/service.py

@@ -91,19 +91,6 @@ def get(self, resource_id: str, select: list[str] | str | None = None) -> Model:
 
         return self._resource_action(resource_id=resource_id, query_params={"select": select})
 
-    def update(self, resource_id: str, resource_data: ResourceData) -> Model:
-        """Update a resource using `PUT /endpoint/{resource_id}`.
-
-        Args:
-            resource_id: Resource ID.
-            resource_data: Resource data.
-
-        Returns:
-            Resource object.
-
-        """
-        return self._resource_action(resource_id, "PUT", json=resource_data)
-
     def _fetch_page_as_response(self, limit: int = 100, offset: int = 0) -> httpx.Response:
         """Fetch one page of resources.
 
@@ -119,13 +106,14 @@ def _fetch_page_as_response(self, limit: int = 100, offset: int = 0) -> httpx.Re
 
         return response
 
-    def _resource_do_request(
+    def _resource_do_request(  # noqa: WPS211
         self,
         resource_id: str,
         method: str = "GET",
         action: str | None = None,
         json: ResourceData | ResourceList | None = None,
         query_params: QueryParam | None = None,
+        headers: dict[str, str] | None = None,
     ) -> httpx.Response:
         """Perform an action on a specific resource using `HTTP_METHOD /endpoint/{resource_id}`.
 
@@ -135,16 +123,19 @@ def _resource_do_request(
             action: The action name to use.
             json: The updated resource data.
             query_params: Additional query parameters.
+            headers: Additional headers.
 
         Returns:
             HTTP response object.
 
         Raises:
             HTTPError: If the action fails.
         """
-        resource_url = urljoin(f"{self._endpoint}/", resource_id)
+        resource_url = urljoin(f"{self.endpoint}/", resource_id)
         url = urljoin(f"{resource_url}/", action) if action else resource_url
-        response = self.http_client.request(method, url, json=json, params=query_params)
+        response = self.http_client.request(
+            method, url, json=json, params=query_params, headers=headers
+        )
         response.raise_for_status()
         return response
 
@@ -166,6 +157,11 @@ def _resource_action(
             query_params: Additional query parameters.
         """
         response = self._resource_do_request(
-            resource_id, method, action, json=json, query_params=query_params
+            resource_id,
+            method,
+            action,
+            json=json,
+            query_params=query_params,
+            headers={"Accept": "application/json"},
         )
         return self._model_class.from_response(response)

mpt_api_client/models/__init__.py

@@ -1,5 +1,6 @@
 from mpt_api_client.models.collection import Collection
+from mpt_api_client.models.download_file import DownloadFile
 from mpt_api_client.models.meta import Meta, Pagination
 from mpt_api_client.models.model import Model, ResourceData
 
-__all__ = ["Collection", "Meta", "Model", "Pagination", "ResourceData"]  # noqa: WPS410
+__all__ = ["Collection", "DownloadFile", "Meta", "Model", "Pagination", "ResourceData"]  # noqa: WPS410

mpt_api_client/models/download_file.py

@@ -0,0 +1,55 @@
+import re
+
+from httpx import Response
+
+
+class DownloadFile:
+    """File resource."""
+
+    def __init__(self, response: Response):
+        self.response = response
+
+    @property
+    def filename(self) -> str | None:
+        """Filename from Content-Disposition header.
+
+        Returns:
+            The filename if found in the Content-Disposition header, None otherwise.
+        """
+        content_disposition = self.response.headers.get("content-disposition")
+        if not content_disposition:
+            return None
+
+        filename_match = re.search(
+            r'filename\*=(?:UTF-8\'\')?([^;]+)|filename=(?:"([^"]+)"|([^;]+))',
+            content_disposition,
+            re.IGNORECASE,
+        )
+
+        if filename_match:
+            return filename_match.group(1) or filename_match.group(2) or filename_match.group(3)
+
+        return None
+
+    @property
+    def file_contents(self) -> bytes:
+        """Returns the content of the attachment.
+
+        Returns:
+            The content of the attachment in bytes
+
+        Raises:
+            ResponseNotRead()
+
+        """
+        return self.response.content
+
+    @property
+    def content_type(self) -> str | None:
+        """Returns the content type of the attachment.
+
+        Returns:
+            The content type of the attachment.
+        """
+        ctype = self.response.headers.get("content-type", "")
+        return str(ctype)

mpt_api_client/resources/commerce/agreements.py

@@ -1,5 +1,17 @@
 from mpt_api_client.http import AsyncService, Service
+from mpt_api_client.http.mixins import (
+    AsyncCreateMixin,
+    AsyncDeleteMixin,
+    AsyncUpdateMixin,
+    CreateMixin,
+    DeleteMixin,
+    UpdateMixin,
+)
 from mpt_api_client.models import Model
+from mpt_api_client.resources.commerce.agreements_attachments import (
+    AgreementsAttachmentService,
+    AsyncAgreementsAttachmentService,
+)
 
 
 class Agreement(Model):
@@ -14,7 +26,13 @@ class AgreementsServiceConfig:
     _collection_key = "data"
 
 
-class AgreementsService(Service[Agreement], AgreementsServiceConfig):
+class AgreementsService(  # noqa: WPS215
+    CreateMixin[Agreement],
+    UpdateMixin[Agreement],
+    DeleteMixin,
+    Service[Agreement],
+    AgreementsServiceConfig,
+):
     """Agreements service."""
 
     def template(self, agreement_id: str) -> str:
@@ -29,8 +47,28 @@ def template(self, agreement_id: str) -> str:
         response = self._resource_do_request(agreement_id, action="template")
         return response.text
 
+    def attachments(self, agreement_id: str) -> AgreementsAttachmentService:
+        """Get the attachments service for the given Agreement id.
 
-class AsyncAgreementsService(AsyncService[Agreement], AgreementsServiceConfig):
+        Args:
+            agreement_id: Agreement ID.
+
+        Returns:
+            Agreements Attachment service.
+        """
+        return AgreementsAttachmentService(
+            http_client=self.http_client,
+            endpoint_params={"agreement_id": agreement_id},
+        )
+
+
+class AsyncAgreementsService(  # noqa: WPS215
+    AsyncCreateMixin[Agreement],
+    AsyncUpdateMixin[Agreement],
+    AsyncDeleteMixin,
+    AsyncService[Agreement],
+    AgreementsServiceConfig,
+):
     """Agreements service."""
 
     async def template(self, agreement_id: str) -> str:
@@ -44,3 +82,17 @@ async def template(self, agreement_id: str) -> str:
         """
         response = await self._resource_do_request(agreement_id, action="template")
         return response.text
+
+    def attachments(self, agreement_id: str) -> AsyncAgreementsAttachmentService:
+        """Get the attachments service for the given Agreement id.
+
+        Args:
+            agreement_id: Agreement ID.
+
+        Returns:
+            Agreements Attachment service.
+        """
+        return AsyncAgreementsAttachmentService(
+            http_client=self.http_client,
+            endpoint_params={"agreement_id": agreement_id},
+        )