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

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

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

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

@@ -20,24 +20,28 @@ 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 = self.interfaces.credentials.get(system='azure-entra-o-auth-2', 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}"
+        if credentials.get("type") == 'custom':
+            tenant_id = credentials['tenant_id']
+            client_id = credentials['client_id']
+            client_secret = credentials['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.get('access_token')
 
-        # 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'
         }
 

{brynq_sdk_azure-3.0.0 → brynq_sdk_azure-3.0.1}/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.1
 Summary: Azure wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_azure-3.0.0 → brynq_sdk_azure-3.0.1}/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.1',
     description='Azure wrapper from BrynQ',
     long_description='Azure wrapper from BrynQ',
     author='BrynQ',