MPT-12329 Order collection client (#14)

Author: albertsola

mpt_api_client/http/collection.py

@@ -23,8 +23,8 @@ class CollectionBaseClient[ResourceType: Resource](ABC):  # noqa: WPS214
     """
 
     _endpoint: str
-    _resource_class: type[Resource]
-    _collection_class: type[Collection[Resource]]
+    _resource_class: type[ResourceType]
+    _collection_class: type[Collection[ResourceType]]
 
     def __init__(
         self,
@@ -181,7 +181,7 @@ def create(self, resource_data: dict[str, Any]) -> ResourceType:
         response = self.mpt_client.post(self._endpoint, json=resource_data)
         response.raise_for_status()
 
-        return self._resource_class.from_response(response)  # type: ignore[return-value]
+        return self._resource_class.from_response(response)
 
     def _fetch_page_as_response(self, limit: int = 100, offset: int = 0) -> httpx.Response:
         """Fetch one page of resources.

mpt_api_client/modules/__init__.py

@@ -0,0 +1,3 @@
+from mpt_api_client.modules.order import Order, OrderCollectionClient
+
+__all__ = ["Order", "OrderCollectionClient"]  # noqa: WPS410

mpt_api_client/modules/order.py

@@ -0,0 +1,16 @@
+from mpt_api_client.http.collection import CollectionBaseClient
+from mpt_api_client.models import Collection, Resource
+from mpt_api_client.registry import commerce
+
+
+class Order(Resource):
+    """Order resource."""
+
+
+@commerce("orders")
+class OrderCollectionClient(CollectionBaseClient[Order]):
+    """Orders client."""
+
+    _endpoint = "/public/v1/commerce/orders"
+    _resource_class = Order
+    _collection_class = Collection[Order]

mpt_api_client/mpt.py

@@ -0,0 +1,54 @@
+from mpt_api_client.http.client import MPTClient
+from mpt_api_client.modules import OrderCollectionClient
+from mpt_api_client.registry import Registry, commerce
+
+
+class MPT:
+    """MPT API Client."""
+
+    def __init__(
+        self,
+        base_url: str | None = None,
+        api_key: str | None = None,
+        registry: Registry | None = None,
+        mpt_client: MPTClient | None = None,
+    ):
+
+        self.mpt_client = mpt_client or MPTClient(base_url=base_url, api_token=api_key)
+        self.registry: Registry = registry or Registry()
+
+    def __getattr__(self, name):  # type: ignore[no-untyped-def]
+        return self.registry.get(name)(client=self.mpt_client)
+
+    @property
+    def commerce(self) -> "CommerceMpt":
+        """Commerce MPT API Client.
+
+        The Commerce API provides a comprehensive set of endpoints
+        for managing agreements, requests, subscriptions, and orders
+        within a vendor-client-ops ecosystem.
+        """
+        return CommerceMpt(mpt_client=self.mpt_client, registry=commerce)
+
+
+class CommerceMpt(MPT):
+    """Commerce MPT API Client."""
+
+    @property
+    def orders(self) -> OrderCollectionClient:
+        """Orders MPT API collection.
+
+        The Orders API provides a comprehensive set of endpoints
+        for creating, updating, and retrieving orders.
+
+
+
+        Returns: Order collection
+
+        Examples:
+            active=RQLQuery("status=active")
+            for order in mpt.orders.filter(active).iterate():
+               [...]
+
+        """
+        return self.registry.get("orders")(client=self.mpt_client)  # type: ignore[return-value]

mpt_api_client/registry.py

@@ -0,0 +1,74 @@
+from collections.abc import Callable
+from typing import Any
+
+from mpt_api_client.http.collection import CollectionBaseClient
+
+ItemType = type[CollectionBaseClient[Any]]
+
+
+class Registry:
+    """Registry for MPT collection clients."""
+
+    def __init__(self) -> None:
+        self.items: dict[str, ItemType] = {}  # noqa: WPS110
+
+    def __call__(self, keyname: str) -> Callable[[ItemType], ItemType]:
+        """Decorator to register a CollectionBaseClient class.
+
+        Args:
+            keyname: The key to register the class under
+
+        Returns:
+            The decorator function
+
+        Examples:
+            registry = Registry()
+            @registry("orders")
+            class OrderCollectionClient(CollectionBaseClient):
+                _endpoint = "/api/v1/orders"
+                _resource_class = Order
+
+            registry.get("orders") == OrderCollectionClient
+        """
+
+        def decorator(cls: ItemType) -> ItemType:
+            self.register(keyname, cls)
+            return cls
+
+        return decorator
+
+    def register(self, keyname: str, item: ItemType) -> None:  # noqa: WPS110
+        """Register a collection client class with a keyname.
+
+        Args:
+            keyname: The key to register the client under
+            item: The collection client class to register
+        """
+        self.items[keyname] = item
+
+    def get(self, keyname: str) -> ItemType:
+        """Get a registered collection client class by keyname.
+
+        Args:
+            keyname: The key to look up
+
+        Returns:
+            The registered collection client class
+
+        Raises:
+            KeyError: If keyname is not registered
+        """
+        if keyname not in self.items:
+            raise KeyError(f"No collection client registered with keyname: {keyname}")
+        return self.items[keyname]
+
+    def list_keys(self) -> list[str]:
+        """Get all registered keynames.
+
+        Returns:
+            List of all registered keynames
+        """
+        return list(self.items.keys())
+
+
+commerce = Registry()

tests/http/conftest.py renamed to tests/conftest.py

@@ -2,7 +2,7 @@
 
 from mpt_api_client.http.collection import CollectionBaseClient
 from mpt_api_client.models import Collection
-from tests.http.conftest import DummyResource
+from tests.conftest import DummyResource
 
 
 class DummyCollectionClient(CollectionBaseClient[DummyResource]):

tests/http/collection/conftest.py

@@ -2,7 +2,7 @@
 
 from mpt_api_client.http.client import MPTClient
 from mpt_api_client.http.resource import ResourceBaseClient
-from tests.http.conftest import DummyResource
+from tests.conftest import DummyResource
 
 
 class DummyResourceClient(ResourceBaseClient[DummyResource]):

tests/http/resource/conftest.py

@@ -3,7 +3,7 @@
 from httpx import ConnectTimeout, Response, codes
 
 from mpt_api_client.http.client import MPTClient
-from tests.http.conftest import API_TOKEN, API_URL
+from tests.conftest import API_TOKEN, API_URL
 
 
 def test_mpt_client_initialization():

tests/http/test_client.py

@@ -0,0 +1,12 @@
+
+from mpt_api_client.modules.order import Order, OrderCollectionClient
+
+
+def test_order_collection_client(mpt_client):
+    order_cc = OrderCollectionClient(client=mpt_client)
+    assert order_cc.query_rql is None
+
+
+def test_order():
+    order = Order()
+    assert order is not None