brynq-sdk-azure 3.0.0__tar.gz → 3.0.2__tar.gz

{brynq_sdk_azure-3.0.0 → brynq_sdk_azure-3.0.2}/PKG-INFO

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

{brynq_sdk_azure-3.0.0 → brynq_sdk_azure-3.0.2}/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):
@@ -20,24 +21,30 @@ class Entra(BrynQ):
         self.timeout = 3600
 
     def __get_headers(self, system_type):
-        credentials = self.interfaces.credentials.get(system='azure-entra-token', system_type=system_type)
-        credentials = credentials.get('data')
-        tenant_id = credentials['tenant_id']
-        client_id = credentials['client_id']
-        client_secret = credentials['client_secret']
-        authority = f"https://login.microsoftonline.com/{tenant_id}"
+        credentials = self.interfaces.credentials.get(system='azure-entra-o-auth-2', system_type=system_type)
+        credentials_data = credentials.get('data')
+        if credentials.get("type") == 'custom':
+            tenant_id = credentials_data.get('tenant_id')
+            client_id = credentials.get('client_id')
+            client_secret = credentials.get('client_secret')
+            authority = f"https://login.microsoftonline.com/{tenant_id}"
+            # Create a ConfidentialClientApplication for authentication
+            app = ConfidentialClientApplication(
+                client_id,
+                authority=authority,
+                client_credential=client_secret,
+            )
 
-        # Create a ConfidentialClientApplication for authentication
-        app = ConfidentialClientApplication(
-            client_id,
-            authority=authority,
-            client_credential=client_secret,
-        )
+            # Get an access token for the Graph API
+            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_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")
 
-        # Get an access token for the Graph API
-        result = app.acquire_token_for_client(scopes=["https://graph.microsoft.com/.default"])
         headers = {
-            'Authorization': f"Bearer {result['access_token']}",
+            'Authorization': f"Bearer {access_token}",
             'Content-Type': 'application/json'
         }
 
@@ -325,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.0 → brynq_sdk_azure-3.0.2}/brynq_sdk_azure.egg-info/PKG-INFO

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

{brynq_sdk_azure-3.0.0 → brynq_sdk_azure-3.0.2}/setup.py

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