mpt-extension-sdk 4.0.4__py3-none-any.whl → 4.0.5__py3-none-any.whl

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]
 

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.dist-info → mpt_extension_sdk-4.0.5.dist-info}/METADATA

@@ -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.4.dist-info → mpt_extension_sdk-4.0.5.dist-info}/RECORD

@@ -10,7 +10,9 @@ mpt_extension_sdk/core/security.py,sha256=0eCed1cmaNK1rCpgnsN9zw6wzr4m_Qb9aKxscO
 mpt_extension_sdk/core/utils.py,sha256=LsvciwlVBJCBbd7L1rIWrTo_8Fi-W_SkL6IEezKUb4s,362
 mpt_extension_sdk/flows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mpt_extension_sdk/flows/context.py,sha256=J_U9nTtY89mUvrcOqdgpk-21F8f6YENSmjxgpVK6XWs,964
-mpt_extension_sdk/flows/pipeline.py,sha256=9E29UKUEvw1kvd_d60vT_9BRAf0jFo3JiOq3CQE-1eU,1374
+mpt_extension_sdk/flows/pipeline.py,sha256=3BO5X3jvuYnRTWrXRzImjVCVOCut8dSvkT2LZwfpaZk,1374
+mpt_extension_sdk/key_vault/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mpt_extension_sdk/key_vault/base.py,sha256=0aXH5c5n9-rUCwmBVY_HtIFUVB6JUC0137Duyh-basw,3321
 mpt_extension_sdk/mpt_http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mpt_extension_sdk/mpt_http/base.py,sha256=3qgo9BW-uTRAbex26urMvLZU0fwnYnBprDRIJCHvTOk,1284
 mpt_extension_sdk/mpt_http/mpt.py,sha256=FatFnbAgyV42N2hJ14n09-WE3V1ioL-lYxuid6vd4Wo,10580
@@ -38,8 +40,8 @@ mpt_extension_sdk/runtime/master.py,sha256=oBI3WHpf2YkTRsRyT1SFIyvCM7gGQpbtXvI1l
 mpt_extension_sdk/runtime/swoext.py,sha256=LS0YZXwQsHDaYjalDGfYQi5cAL3aKQhphf3chrNmNBk,1665
 mpt_extension_sdk/runtime/utils.py,sha256=ZnRWIBwYzOerx30KFmBXLUe19qzwE4JFNkNMGVqjuyk,3365
 mpt_extension_sdk/runtime/workers.py,sha256=k25jFlh-NdHVqeMV2AgoC6IHIlbRHcDoRB5xqWwMPFg,2411
-mpt_extension_sdk-4.0.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mpt_extension_sdk-4.0.4.dist-info/METADATA,sha256=K0Kbv4Hug_sXepfrKGlQW2OU4ZMAoBieZ5yCH2v3jJQ,1563
-mpt_extension_sdk-4.0.4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-mpt_extension_sdk-4.0.4.dist-info/entry_points.txt,sha256=8uZQihFseK4Kp5XWHl9EDYneg-CLS8BFO-ygp0h34nc,143
-mpt_extension_sdk-4.0.4.dist-info/RECORD,,
+mpt_extension_sdk-4.0.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mpt_extension_sdk-4.0.5.dist-info/METADATA,sha256=xsj7JQDNE8Mrf2zZvrGkGO4_iMNlnY6YQp5i2BMV1ww,1666
+mpt_extension_sdk-4.0.5.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+mpt_extension_sdk-4.0.5.dist-info/entry_points.txt,sha256=8uZQihFseK4Kp5XWHl9EDYneg-CLS8BFO-ygp0h34nc,143
+mpt_extension_sdk-4.0.5.dist-info/RECORD,,