Added missing billing statement attachments endpoints

Author: Robert Segal

mpt_api_client/resources/billing/credit_memo_attachments.py

@@ -1,13 +1,10 @@
 from mpt_api_client.http import AsyncService, Service
 from mpt_api_client.http.mixins import (
     AsyncCollectionMixin,
-    AsyncFilesOperationsMixin,
-    AsyncModifiableResourceMixin,
     CollectionMixin,
-    FilesOperationsMixin,
-    ModifiableResourceMixin,
 )
 from mpt_api_client.models import Model
+from mpt_api_client.resources.billing.mixins import AsyncAttachmentMixin, AttachmentMixin
 
 
 class CreditMemoAttachment(Model):
@@ -23,8 +20,7 @@ class CreditMemoAttachmentsServiceConfig:
 
 
 class CreditMemoAttachmentsService(
-    FilesOperationsMixin[CreditMemoAttachment],
-    ModifiableResourceMixin[CreditMemoAttachment],
+    AttachmentMixin[CreditMemoAttachment],
     CollectionMixin[CreditMemoAttachment],
     Service[CreditMemoAttachment],
     CreditMemoAttachmentsServiceConfig,
@@ -33,8 +29,7 @@ class CreditMemoAttachmentsService(
 
 
 class AsyncCreditMemoAttachmentsService(
-    AsyncFilesOperationsMixin[CreditMemoAttachment],
-    AsyncModifiableResourceMixin[CreditMemoAttachment],
+    AsyncAttachmentMixin[CreditMemoAttachment],
     AsyncCollectionMixin[CreditMemoAttachment],
     AsyncService[CreditMemoAttachment],
     CreditMemoAttachmentsServiceConfig,

mpt_api_client/resources/billing/custom_ledger_attachments.py

@@ -1,13 +1,10 @@
 from mpt_api_client.http import AsyncService, Service
 from mpt_api_client.http.mixins import (
     AsyncCollectionMixin,
-    AsyncFilesOperationsMixin,
-    AsyncModifiableResourceMixin,
     CollectionMixin,
-    FilesOperationsMixin,
-    ModifiableResourceMixin,
 )
 from mpt_api_client.models import Model
+from mpt_api_client.resources.billing.mixins import AsyncAttachmentMixin, AttachmentMixin
 
 
 class CustomLedgerAttachment(Model):
@@ -23,8 +20,7 @@ class CustomLedgerAttachmentsServiceConfig:
 
 
 class CustomLedgerAttachmentsService(
-    FilesOperationsMixin[CustomLedgerAttachment],
-    ModifiableResourceMixin[CustomLedgerAttachment],
+    AttachmentMixin[CustomLedgerAttachment],
     CollectionMixin[CustomLedgerAttachment],
     Service[CustomLedgerAttachment],
     CustomLedgerAttachmentsServiceConfig,
@@ -33,8 +29,7 @@ class CustomLedgerAttachmentsService(
 
 
 class AsyncCustomLedgerAttachmentsService(
-    AsyncFilesOperationsMixin[CustomLedgerAttachment],
-    AsyncModifiableResourceMixin[CustomLedgerAttachment],
+    AsyncAttachmentMixin[CustomLedgerAttachment],
     AsyncCollectionMixin[CustomLedgerAttachment],
     AsyncService[CustomLedgerAttachment],
     CustomLedgerAttachmentsServiceConfig,

mpt_api_client/resources/billing/invoice_attachments.py

@@ -1,13 +1,10 @@
 from mpt_api_client.http import AsyncService, Service
 from mpt_api_client.http.mixins import (
     AsyncCollectionMixin,
-    AsyncFilesOperationsMixin,
-    AsyncModifiableResourceMixin,
     CollectionMixin,
-    FilesOperationsMixin,
-    ModifiableResourceMixin,
 )
 from mpt_api_client.models import Model
+from mpt_api_client.resources.billing.mixins import AsyncAttachmentMixin, AttachmentMixin
 
 
 class InvoiceAttachment(Model):
@@ -23,8 +20,7 @@ class InvoiceAttachmentsServiceConfig:
 
 
 class InvoiceAttachmentsService(
-    FilesOperationsMixin[InvoiceAttachment],
-    ModifiableResourceMixin[InvoiceAttachment],
+    AttachmentMixin[InvoiceAttachment],
     CollectionMixin[InvoiceAttachment],
     Service[InvoiceAttachment],
     InvoiceAttachmentsServiceConfig,
@@ -33,8 +29,7 @@ class InvoiceAttachmentsService(
 
 
 class AsyncInvoiceAttachmentsService(
-    AsyncFilesOperationsMixin[InvoiceAttachment],
-    AsyncModifiableResourceMixin[InvoiceAttachment],
+    AsyncAttachmentMixin[InvoiceAttachment],
     AsyncCollectionMixin[InvoiceAttachment],
     AsyncService[InvoiceAttachment],
     InvoiceAttachmentsServiceConfig,

mpt_api_client/resources/billing/journal_attachments.py

@@ -1,19 +1,10 @@
 from mpt_api_client.http import AsyncService, Service
 from mpt_api_client.http.mixins import (
     AsyncCollectionMixin,
-    AsyncCreateFileMixin,
-    AsyncDeleteMixin,
-    AsyncDownloadFileMixin,
-    AsyncGetMixin,
-    AsyncUpdateMixin,
     CollectionMixin,
-    CreateFileMixin,
-    DeleteMixin,
-    DownloadFileMixin,
-    GetMixin,
-    UpdateMixin,
 )
 from mpt_api_client.models import Model
+from mpt_api_client.resources.billing.mixins import AsyncAttachmentMixin, AttachmentMixin
 
 
 class JournalAttachment(Model):
@@ -31,11 +22,7 @@ class JournalAttachmentsServiceConfig:
 
 
 class JournalAttachmentsService(
-    CreateFileMixin[JournalAttachment],
-    UpdateMixin[JournalAttachment],
-    DownloadFileMixin[JournalAttachment],
-    DeleteMixin,
-    GetMixin[JournalAttachment],
+    AttachmentMixin[JournalAttachment],
     CollectionMixin[JournalAttachment],
     Service[JournalAttachment],
     JournalAttachmentsServiceConfig,
@@ -44,11 +31,7 @@ class JournalAttachmentsService(
 
 
 class AsyncJournalAttachmentsService(
-    AsyncCreateFileMixin[JournalAttachment],
-    AsyncUpdateMixin[JournalAttachment],
-    AsyncDownloadFileMixin[JournalAttachment],
-    AsyncDeleteMixin,
-    AsyncGetMixin[JournalAttachment],
+    AsyncAttachmentMixin[JournalAttachment],
     AsyncCollectionMixin[JournalAttachment],
     AsyncService[JournalAttachment],
     JournalAttachmentsServiceConfig,

mpt_api_client/resources/billing/ledger_attachments.py

@@ -1,13 +1,10 @@
 from mpt_api_client.http import AsyncService, Service
 from mpt_api_client.http.mixins import (
     AsyncCollectionMixin,
-    AsyncFilesOperationsMixin,
-    AsyncModifiableResourceMixin,
     CollectionMixin,
-    FilesOperationsMixin,
-    ModifiableResourceMixin,
 )
 from mpt_api_client.models import Model
+from mpt_api_client.resources.billing.mixins import AsyncAttachmentMixin, AttachmentMixin
 
 
 class LedgerAttachment(Model):
@@ -23,8 +20,7 @@ class LedgerAttachmentsServiceConfig:
 
 
 class LedgerAttachmentsService(
-    FilesOperationsMixin[LedgerAttachment],
-    ModifiableResourceMixin[LedgerAttachment],
+    AttachmentMixin[LedgerAttachment],
     CollectionMixin[LedgerAttachment],
     Service[LedgerAttachment],
     LedgerAttachmentsServiceConfig,
@@ -33,8 +29,7 @@ class LedgerAttachmentsService(
 
 
 class AsyncLedgerAttachmentsService(
-    AsyncFilesOperationsMixin[LedgerAttachment],
-    AsyncModifiableResourceMixin[LedgerAttachment],
+    AsyncAttachmentMixin[LedgerAttachment],
     AsyncCollectionMixin[LedgerAttachment],
     AsyncService[LedgerAttachment],
     LedgerAttachmentsServiceConfig,

mpt_api_client/resources/billing/mixins.py

@@ -1,3 +1,15 @@
+from mpt_api_client.http.mixins import (
+    AsyncCreateFileMixin,
+    AsyncDeleteMixin,
+    AsyncDownloadFileMixin,
+    AsyncGetMixin,
+    AsyncUpdateMixin,
+    CreateFileMixin,
+    DeleteMixin,
+    DownloadFileMixin,
+    GetMixin,
+    UpdateMixin,
+)
 from mpt_api_client.models import ResourceData
 
 
@@ -392,3 +404,23 @@ async def queue(self, resource_id: str, resource_data: ResourceData | None = Non
         return await self._resource_action(  # type: ignore[attr-defined, no-any-return]
             resource_id, "POST", "queue", json=resource_data
         )
+
+
+class AttachmentMixin[Model](
+    CreateFileMixin[Model],
+    UpdateMixin[Model],
+    DeleteMixin,
+    DownloadFileMixin[Model],
+    GetMixin[Model],
+):
+    """Attachment mixin."""
+
+
+class AsyncAttachmentMixin[Model](
+    AsyncCreateFileMixin[Model],
+    AsyncUpdateMixin[Model],
+    AsyncDeleteMixin,
+    AsyncDownloadFileMixin[Model],
+    AsyncGetMixin[Model],
+):
+    """Async Attachment mixin."""

mpt_api_client/resources/billing/statement_attachments.py

@@ -0,0 +1,39 @@
+from mpt_api_client.http import AsyncService, Service
+from mpt_api_client.http.mixins import (
+    AsyncCollectionMixin,
+    CollectionMixin,
+)
+from mpt_api_client.models import Model
+from mpt_api_client.resources.billing.mixins import AsyncAttachmentMixin, AttachmentMixin
+
+
+class StatementAttachment(Model):
+    """Statement Attachment resource."""
+
+
+class StatementAttachmentsServiceConfig:
+    """Statement Attachments service configuration."""
+
+    _endpoint = "/public/v1/billing/statements/{statement_id}/attachments"
+    _model_class = StatementAttachment
+    _collection_key = "data"
+    _upload_file_key = "file"
+    _upload_data_key = "attachment"
+
+
+class StatementAttachmentsService(
+    AttachmentMixin[StatementAttachment],
+    CollectionMixin[StatementAttachment],
+    Service[StatementAttachment],
+    StatementAttachmentsServiceConfig,
+):
+    """Statement Attachments service."""
+
+
+class AsyncStatementAttachmentsService(
+    AsyncAttachmentMixin[StatementAttachment],
+    AsyncCollectionMixin[StatementAttachment],
+    AsyncService[StatementAttachment],
+    StatementAttachmentsServiceConfig,
+):
+    """Statement Attachments service."""

mpt_api_client/resources/billing/statements.py

@@ -9,6 +9,10 @@
 )
 from mpt_api_client.models import Model
 from mpt_api_client.resources.billing.mixins import AsyncIssuableMixin, IssuableMixin
+from mpt_api_client.resources.billing.statement_attachments import (
+    AsyncStatementAttachmentsService,
+    StatementAttachmentsService,
+)
 from mpt_api_client.resources.billing.statement_charges import (
     AsyncStatementChargesService,
     StatementChargesService,
@@ -44,6 +48,13 @@ def charges(self, statement_id: str) -> StatementChargesService:
             endpoint_params={"statement_id": statement_id},
         )
 
+    def attachments(self, statement_id: str) -> StatementAttachmentsService:
+        """Return statement attachments service."""
+        return StatementAttachmentsService(
+            http_client=self.http_client,
+            endpoint_params={"statement_id": statement_id},
+        )
+
 
 class AsyncStatementsService(
     AsyncUpdateMixin[Statement],
@@ -61,3 +72,10 @@ def charges(self, statement_id: str) -> AsyncStatementChargesService:
             http_client=self.http_client,
             endpoint_params={"statement_id": statement_id},
         )
+
+    def attachments(self, statement_id: str) -> AsyncStatementAttachmentsService:
+        """Return statement attachments service."""
+        return AsyncStatementAttachmentsService(
+            http_client=self.http_client,
+            endpoint_params={"statement_id": statement_id},
+        )

tests/unit/resources/billing/test_credit_memo_attachments.py

@@ -38,14 +38,14 @@ def test_async_endpoint(async_credit_memo_attachments_service):
     assert result is True
 
 
-@pytest.mark.parametrize("method", ["get", "create", "update", "delete"])
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "iterate", "download"])
 def test_methods_present(credit_memo_attachments_service, method: str):
     result = hasattr(credit_memo_attachments_service, method)
 
     assert result is True
 
 
-@pytest.mark.parametrize("method", ["get", "create", "update", "delete"])
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "iterate", "download"])
 def test_async_methods_present(async_credit_memo_attachments_service, method: str):
     result = hasattr(async_credit_memo_attachments_service, method)
 

tests/unit/resources/billing/test_custom_ledger_attachments.py

@@ -36,14 +36,14 @@ def test_async_endpoint(async_custom_ledger_attachments_service):
     assert result is True
 
 
-@pytest.mark.parametrize("method", ["get", "create", "update", "delete"])
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "iterate", "download"])
 def test_methods_present(custom_ledger_attachments_service, method: str):
     result = hasattr(custom_ledger_attachments_service, method)
 
     assert result is True
 
 
-@pytest.mark.parametrize("method", ["get", "create", "update", "delete"])
+@pytest.mark.parametrize("method", ["get", "create", "update", "delete", "iterate", "download"])
 def test_async_methods_present(async_custom_ledger_attachments_service, method: str):
     result = hasattr(async_custom_ledger_attachments_service, method)