mpt-extension-sdk 4.4.0__tar.gz → 4.6.0__tar.gz

{mpt_extension_sdk-4.4.0 → mpt_extension_sdk-4.6.0}/PKG-INFO

@@ -1,14 +1,16 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: mpt-extension-sdk
-Version: 4.4.0
+Version: 4.6.0
 Summary: Extensions SDK for SoftwareONE Marketplace Platform
 License: Apache-2.0
+License-File: LICENSE
 Author: SoftwareOne AG
 Requires-Python: >=3.12,<4
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: azure-identity (>=1.21.0,<2.0.0)
 Requires-Dist: azure-keyvault-secrets (>=4.9.0,<5.0.0)
 Requires-Dist: azure-monitor-opentelemetry-exporter (==1.0.0b25)

{mpt_extension_sdk-4.4.0 → mpt_extension_sdk-4.6.0}/mpt_extension_sdk/mpt_http/base.py

@@ -21,6 +21,13 @@ class MPTClient(Session):
                 pool_maxsize=36,
             ),
         )
+        self.mount(
+            "https://",
+            HTTPAdapter(
+                max_retries=retries,
+                pool_maxsize=36,
+            ),
+        )
         self.headers.update(
             {
                 "User-Agent": USER_AGENT,

{mpt_extension_sdk-4.4.0 → mpt_extension_sdk-4.6.0}/mpt_extension_sdk/mpt_http/mpt.py

@@ -102,7 +102,15 @@ def set_processing_template(mpt_client, order_id, template):
 
 
 @wrap_mpt_http_error
-def create_asset(mpt_client, order_id, asset):
+def create_asset(mpt_client, asset):
+    """Create a new asset."""
+    response = mpt_client.post("/commerce/assets", json=asset)
+    response.raise_for_status()
+    return response.json()
+
+
+@wrap_mpt_http_error
+def create_order_asset(mpt_client, order_id, asset):
     """Create a new asset for an order."""
     response = mpt_client.post(f"/commerce/orders/{order_id}/assets", json=asset)
     response.raise_for_status()
@@ -110,13 +118,57 @@ def create_asset(mpt_client, order_id, asset):
 
 
 @wrap_mpt_http_error
-def update_asset(mpt_client, order_id, asset_id, **kwargs):
+def update_asset(mpt_client, asset_id, **kwargs):
+    """Update an asset."""
+    response = mpt_client.put(f"/commerce/assets/{asset_id}", json=kwargs)
+    response.raise_for_status()
+    return response.json()
+
+
+@wrap_mpt_http_error
+def update_order_asset(mpt_client, order_id, asset_id, **kwargs):
     """Update an order asset."""
     response = mpt_client.put(f"/commerce/orders/{order_id}/assets/{asset_id}", json=kwargs)
     response.raise_for_status()
     return response.json()
 
 
+@wrap_mpt_http_error
+def get_agreement_asset_by_external_id(mpt_client, agreement_id, asset_external_id):
+    """Retrieve an agreement asset by external ID."""
+    response = mpt_client.get(
+        f"/commerce/assets?eq(externalIds.vendor,{asset_external_id})"
+        f"&eq(agreement.id,{agreement_id})"
+        f"&eq(status,Active)"
+        f"&select=agreement.id&limit=1"
+    )
+    response.raise_for_status()
+    assets = response.json()
+    return assets["data"][0] if assets["data"] else None
+
+
+@wrap_mpt_http_error
+def get_asset_by_id(mpt_client, asset_id):
+    """Get an asset by ID."""
+    response = mpt_client.get(f"/commerce/assets/{asset_id}")
+    response.raise_for_status()
+    return response.json()
+
+
+@wrap_mpt_http_error
+def get_order_asset_by_external_id(mpt_client, order_id, asset_external_id):
+    """Retrieve an order asset by its external ID."""
+    response = mpt_client.get(
+        f"/commerce/orders/{order_id}/assets?eq(externalIds.vendor,{asset_external_id})&limit=1",
+    )
+    response.raise_for_status()
+    assets = response.json()
+    if assets["$meta"]["pagination"]["total"] == 1:
+        return assets["data"][0]
+
+    return None
+
+
 @wrap_mpt_http_error
 def create_subscription(mpt_client, order_id, subscription):
     response = mpt_client.post(
@@ -188,6 +240,18 @@ def get_template_by_name(mpt_client, product_id, template_name):
     return templates["data"][0]
 
 
+@wrap_mpt_http_error
+def get_asset_template_by_name(mpt_client, product_id, template_name):
+    """Retrieve an asset template by name."""
+    rql_filter = f"and(eq(type,Asset),eq(name,{template_name}))"
+    url = f"/catalog/products/{product_id}/templates?{rql_filter}&limit=1"
+    response = mpt_client.get(url)
+    response.raise_for_status()
+
+    templates = response.json()
+    return templates["data"][0] if templates["data"] else None
+
+
 @wrap_mpt_http_error
 def update_agreement(mpt_client, agreement_id, **kwargs):
     response = mpt_client.put(
@@ -277,21 +341,23 @@ def get_product_items_by_period(
 
 
 def get_agreements_by_ids(mpt_client, ids):
+    """Retrieve agreements by their IDs."""
+    ids_str = ",".join(ids)
     rql_query = (
-        f"and(in(id,({','.join(ids)})),eq(status,Active))"
-        "&select=lines,parameters,subscriptions,product,listing"
+        f"and(in(id,({ids_str})),eq(status,Active))"
+        "&select=assets,lines,parameters,subscriptions,product,listing"
     )
     return get_agreements_by_query(mpt_client, rql_query)
 
 
-def get_all_agreements(
-    mpt_client,
-):
-    product_condition = f"in(product.id,({','.join(settings.MPT_PRODUCTS_IDS)}))"
+def get_all_agreements(mpt_client):
+    """Retrieve all active agreements for specific products."""
+    product_ids_str = ",".join(settings.MPT_PRODUCTS_IDS)
+    product_condition = f"in(product.id,({product_ids_str}))"
 
     return get_agreements_by_query(
         mpt_client,
-        f"and(eq(status,Active),{product_condition})&select=lines,parameters,subscriptions,product,listing",
+        f"and(eq(status,Active),{product_condition})&select=assets,lines,parameters,subscriptions,product,listing",
     )
 
 

{mpt_extension_sdk-4.4.0 → mpt_extension_sdk-4.6.0}/mpt_extension_sdk/mpt_http/wrap_http_error.py

@@ -2,12 +2,17 @@ import json
 from functools import wraps
 
 from requests import HTTPError, JSONDecodeError
+from requests.exceptions import RetryError
 
 
 class MPTError(Exception):
     pass
 
 
+class MPTMaxRetryError(MPTError):
+    pass
+
+
 class MPTHttpError(MPTError):
     def __init__(self, status_code: int, content: str):
         self.status_code = status_code
@@ -46,6 +51,8 @@ def wrap_mpt_http_error(func):
                 raise MPTAPIError(e.response.status_code, e.response.json())
             except JSONDecodeError:
                 raise MPTHttpError(e.response.status_code, e.response.content.decode())
+        except RetryError as e:
+            raise MPTMaxRetryError(str(e))
 
     return _wrapper
 

{mpt_extension_sdk-4.4.0 → mpt_extension_sdk-4.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mpt-extension-sdk"
-version = "4.4.0"
+version = "4.6.0"
 description = "Extensions SDK for SoftwareONE Marketplace Platform"
 authors = ["SoftwareOne AG"]
 readme = "README.md"