mpt-extension-sdk 4.0.4__tar.gz → 4.0.5__tar.gz

{mpt_extension_sdk-4.0.4 → mpt_extension_sdk-4.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mpt-extension-sdk
-Version: 4.0.4
+Version: 4.0.5
 Summary: Extensions SDK for SoftwareONE Marketplace Platform
 License: Apache-2.0
 Author: SoftwareOne AG
@@ -9,6 +9,8 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+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)
 Requires-Dist: boto3 (==1.34.*)
 Requires-Dist: click (==8.1.*)

mpt_extension_sdk-4.0.5/mpt_extension_sdk/key_vault/base.py

@@ -0,0 +1,110 @@
+import logging
+
+from azure.core.exceptions import (
+    HttpResponseError,
+    ResourceNotFoundError,
+)
+from azure.identity import DefaultAzureCredential
+from azure.keyvault.secrets import SecretClient
+from requests import Session
+
+logger = logging.getLogger(__name__)
+
+
+class KeyVault(Session):
+    def __init__(self, key_vault_name: str):
+        """
+        Initialize the KeyVault client with the provided Key Vault name.
+
+        Args:
+            key_vault_name (str): The name of the Azure Key Vault.
+        """
+        super().__init__()
+        self.key_vault_name = key_vault_name
+
+    def get_secret(self, secret_name: str):
+        """
+        Retrieve a secret from the Azure Key Vault.
+
+        Args:
+            key_vault_url (str): The URL of the Azure Key Vault.
+            secret_name (str): The name of the secret to retrieve.
+
+        Returns:
+            str: The value of the secret.
+        """
+        try:
+            client = self._get_key_vault_client(self.key_vault_name)
+
+            # Retrieve the secret from the Key Vault
+            secret = client.get_secret(secret_name)
+
+            return secret.value
+        except ResourceNotFoundError as e:
+            logger.error(
+                f"Secret '{secret_name}' not found in Key Vault '{self.key_vault_name}': {e}"
+            )
+            return None
+
+    def set_secret(self, secret_name: str, secret_value: str):
+        """
+        Set a secret in the Azure Key Vault.
+
+        Args:
+            key_vault_url (str): The URL of the Azure Key Vault.
+            secret_name (str): The name of the secret to set.
+            secret_value (str): The value of the secret to set.
+
+        Returns:
+            None
+        """
+
+        # Set the secret in the Key Vault
+        try:
+            client = self._get_key_vault_client(self.key_vault_name)
+
+            client.set_secret(secret_name, secret_value)
+
+            updated_secret = client.get_secret(secret_name)
+
+            return updated_secret.value
+        except HttpResponseError as e:
+            logger.error(
+                f"Failed to set secret '{secret_name}' in Key Vault '{self.key_vault_name}': {e}"
+            )
+            return None
+
+    def _get_key_vault_url(self, key_vault_name: str):
+        """
+        Construct the Key Vault URL using the provided Key Vault name.
+
+        Args:
+            key_vault_name (str): The name of the Azure Key Vault.
+
+        Returns:
+            str: The URL of the Azure Key Vault.
+        """
+
+        # Construct the Key Vault URL
+        return f"https://{key_vault_name}.vault.azure.net/"
+
+    def _get_key_vault_client(self, key_vault_name: str):
+        """
+        Create a Key Vault client using the provided Key Vault URL and secret name.
+
+        Args:
+            key_vault_url (str): The URL of the Azure Key Vault.
+
+        Returns:
+            SecretClient: An instance of the SecretClient for the specified Key Vault.
+        """
+
+        # Get the Key Vault URL
+        key_vault_url = self._get_key_vault_url(key_vault_name)
+        # Create a credential object using DefaultAzureCredential
+        credential = DefaultAzureCredential()
+
+        # Create a Key Vault client using the credential
+        client = SecretClient(vault_url=key_vault_url, credential=credential)
+
+        return client

{mpt_extension_sdk-4.0.4 → mpt_extension_sdk-4.0.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mpt-extension-sdk"
-version = "4.0.4"
+version = "4.0.5"
 description = "Extensions SDK for SoftwareONE Marketplace Platform"
 authors = ["SoftwareOne AG"]
 readme = "README.md"
@@ -8,31 +8,33 @@ license = "Apache-2.0"
 
 [tool.poetry.dependencies]
 python = ">=3.12,<4"
+azure-identity = "^1.21.0"
+azure-keyvault-secrets = "^4.9.0"
+azure-monitor-opentelemetry-exporter = "1.0.0b25"
 boto3 = "1.34.*"
+click = "8.1.*"
+debugpy = "1.8.*"
 django = "4.2.*" # should it be a dependency of the extension ? it is already a dependency of the sdk.
+django-ninja = "1.1.*"
+gunicorn = "23.0.*"
 jinja2 = "3.1.*"
 markdown-it-py = "3.0.*"
 openpyxl = "3.1.*"
-phonenumbers = "8.13.*"
-pyairtable = "2.3.*"
-pymsteams = "0.2.*"
-regex = "^2024.7.24"
-requests = "2.32.*"
-types-openpyxl = "3.1.*"
-django-ninja = "1.1.*"
-pyjwt = "2.8.*"
-typing-extensions = "4.9.*"
-click = "8.1.*"
-debugpy = "1.8.*"
-gunicorn = "23.0.*"
-azure-monitor-opentelemetry-exporter = "1.0.0b25"
 opentelemetry-api = "1.30.*"
 opentelemetry-instrumentation-django = "0.51b0"
 opentelemetry-instrumentation-logging = "0.51b0"
 opentelemetry-instrumentation-requests = "0.51b0"
 opentelemetry-sdk = "1.30.*"
+phonenumbers = "8.13.*"
+pyairtable = "2.3.*"
 pyfiglet = "1.0.*"
+pyjwt = "2.8.*"
+pymsteams = "0.2.*"
+regex = "^2024.7.24"
+requests = "2.32.*"
 rich = "13.7.*"
+types-openpyxl = "3.1.*"
+typing-extensions = "4.9.*"
 watchfiles = "0.21.*"
 
 [tool.poetry.group.dev.dependencies]

{mpt_extension_sdk-4.0.4 → mpt_extension_sdk-4.0.5}/mpt_extension_sdk/flows/pipeline.py

@@ -1,8 +1,8 @@
 from abc import ABC, abstractmethod
 from typing import Callable
 
-from mpt_extension_sdk.mpt_http.base import MPTClient
 from mpt_extension_sdk.flows.context import Context
+from mpt_extension_sdk.mpt_http.base import MPTClient
 
 NextStep = Callable[[MPTClient, Context], None]