brynq-sdk-azure 3.0.1__tar.gz → 3.0.3__tar.gz

{brynq_sdk_azure-3.0.1 → brynq_sdk_azure-3.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_azure
-Version: 3.0.1
+Version: 3.0.3
 Summary: Azure wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_azure-3.0.1 → brynq_sdk_azure-3.0.3}/brynq_sdk_azure/entra.py

@@ -11,6 +11,7 @@ from msal import ConfidentialClientApplication
 from typing import Union, List, Literal, Optional
 import os
 
+
 class Entra(BrynQ):
 
     def __init__(self, system_type: Optional[Literal['source', 'target']] = None, debug: bool = False):
@@ -21,11 +22,11 @@ class Entra(BrynQ):
 
     def __get_headers(self, system_type):
         credentials = self.interfaces.credentials.get(system='azure-entra-o-auth-2', system_type=system_type)
-        credentials = credentials.get('data')
+        credentials_data = credentials.get('data')
         if credentials.get("type") == 'custom':
-            tenant_id = credentials['tenant_id']
-            client_id = credentials['client_id']
-            client_secret = credentials['client_secret']
+            tenant_id = credentials_data.get('tenant_id')
+            client_id = credentials_data.get('client_id')
+            client_secret = credentials_data.get('client_secret')
             authority = f"https://login.microsoftonline.com/{tenant_id}"
             # Create a ConfidentialClientApplication for authentication
             app = ConfidentialClientApplication(
@@ -38,7 +39,9 @@ class Entra(BrynQ):
             result = app.acquire_token_for_client(scopes=["https://graph.microsoft.com/.default"])
             access_token = result.get('access_token')
         elif credentials.get("type") == 'oauth2':
-            access_token = credentials.get('access_token')
+            access_token = credentials_data.get('access_token')
+        else:
+            raise ValueError("The retrieved credentials are not part of the supported system types, please extend the SDK to support the specified type")
 
         headers = {
             'Authorization': f"Bearer {access_token}",
@@ -329,7 +332,7 @@ class Entra(BrynQ):
         return: response
         """
         url = f"https://graph.microsoft.com/v1.0/users/{user_id}/manager/$ref"
-        content ={f"@odata.id": f"https://graph.microsoft.com/v1.0/users/{manager_id}"}
+        content = {f"@odata.id": f"https://graph.microsoft.com/v1.0/users/{manager_id}"}
         response = requests.put(url, headers=self.headers, data=json.dumps(content), timeout=self.timeout)
         return response
 

{brynq_sdk_azure-3.0.1 → brynq_sdk_azure-3.0.3}/brynq_sdk_azure.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-azure
-Version: 3.0.1
+Version: 3.0.3
 Summary: Azure wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_azure-3.0.1 → brynq_sdk_azure-3.0.3}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 
 setup(
     name='brynq_sdk_azure',
-    version='3.0.1',
+    version='3.0.3',
     description='Azure wrapper from BrynQ',
     long_description='Azure wrapper from BrynQ',
     author='BrynQ',