brynq-sdk-azure 2.0.2__tar.gz → 2.0.3__tar.gz

{brynq_sdk_azure-2.0.2 → brynq_sdk_azure-2.0.3}/PKG-INFO

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

{brynq_sdk_azure-2.0.2 → brynq_sdk_azure-2.0.3}/brynq_sdk_azure/entra.py

@@ -17,6 +17,7 @@ class Entra(BrynQ):
         super().__init__()
         self.headers = self.__get_headers(label=label)
         self.endpoint = "https://graph.microsoft.com/v1.0"
+        self.timeout = 3600
 
     def __get_headers(self, label):
         credentials = self.get_system_credential(system='azure-entra-token', label=label)
@@ -44,7 +45,7 @@ class Entra(BrynQ):
     def __add_attribute_information(self, payload, custom_attributes):
         # First get the official name of the custom attribute and all the other information
         payload.update({"customSecurityAttributes": {}})
-        metadata = requests.get('https://graph.microsoft.com/v1.0/directory/customSecurityAttributeDefinitions', headers=self.headers).json()
+        metadata = requests.get('https://graph.microsoft.com/v1.0/directory/customSecurityAttributeDefinitions', headers=self.headers, timeout=self.timeout).json()
         # Now loop through the given metadata and add the corresponding metadata and the values itself to the payload
         for attr, value in custom_attributes.items():
             for meta in metadata["value"]:
@@ -106,7 +107,7 @@ class Entra(BrynQ):
         loop = True
         url = f"{endpoint}/groups"
         while loop:
-            response = requests.get(url, headers=self.headers)
+            response = requests.get(url, headers=self.headers, timeout=self.timeout)
             groups = response.json()['value']
             df_temp = pd.json_normalize(groups)
             df = pd.concat([df, df_temp], ignore_index=True)
@@ -126,7 +127,7 @@ class Entra(BrynQ):
         group_url = "https://graph.microsoft.com/v1.0/groups/"
         df = pd.DataFrame()
         while group_url:
-            graph_r = requests.get(group_url, headers=self.headers)
+            graph_r = requests.get(group_url, headers=self.headers, timeout=self.timeout)
             graph_json = graph_r.json()
             groups = graph_json.get('value')
             for group in groups:
@@ -134,7 +135,7 @@ class Entra(BrynQ):
                 # Get users in each group
                 next_url_members = f"https://graph.microsoft.com/v1.0/groups/{group['id']}/members"
                 while next_url_members:
-                    members_r = requests.get(next_url_members, headers=self.headers)
+                    members_r = requests.get(next_url_members, headers=self.headers, timeout=self.timeout)
                     members_json = members_r.json()
                     members = members_json.get('value')
                     df_temp = pd.json_normalize(members)
@@ -167,7 +168,7 @@ class Entra(BrynQ):
             "mailNickname": f"{mail_nickname}",
             "securityEnabled": security_enabled
         }
-        response = requests.post(endpoint, headers=self.headers, json=payload)
+        response = requests.post(endpoint, headers=self.headers, json=payload, timeout=self.timeout)
         return response
 
     def update_group(self, id: int, name: str = '', description: str = '', mail_enabled: bool = False, mail_nickname: str = '', security_enabled: bool = True):
@@ -189,7 +190,7 @@ class Entra(BrynQ):
             "mailNickname": f"{mail_nickname}",
             "securityEnabled": security_enabled
         }
-        response = requests.patch(endpoint, headers=self.headers, json=payload)
+        response = requests.patch(endpoint, headers=self.headers, json=payload, timeout=self.timeout)
         return response
 
     def delete_group(self, group_id):
@@ -199,7 +200,7 @@ class Entra(BrynQ):
         :return: Response of the request
         """
         endpoint = f"https://graph.microsoft.com/v1.0/groups/{group_id}"
-        response = requests.delete(endpoint, headers=self.headers)
+        response = requests.delete(endpoint, headers=self.headers, timeout=self.timeout)
         return response
 
     def get_users(self, extra_fields: list = [], custom_attributes: bool = False, expand: str = '', expand_select: str = '') -> pd.DataFrame:
@@ -224,7 +225,7 @@ class Entra(BrynQ):
 
         df = pd.DataFrame()
         while endpoint:
-            response = requests.get(endpoint, headers=self.headers)
+            response = requests.get(endpoint, headers=self.headers, timeout=self.timeout)
             endpoint = response.json().get('@odata.nextLink')
             data = response.json().get('value')
             df_temp = json_normalize(data, sep='.')
@@ -265,7 +266,7 @@ class Entra(BrynQ):
         # If there are any custom attributes, add them to the payload. But since the endpoint needs extra metadata, we need to do some extra work
         if len(custom_attributes) > 0:
             payload = self.__add_attribute_information(payload, custom_attributes)
-        response = requests.post(endpoint, headers=self.headers, json=payload)
+        response = requests.post(endpoint, headers=self.headers, json=payload, timeout=self.timeout)
         return response
 
     def update_user(self, user_id, fields_to_update: dict = {}, custom_attributes: dict = {}, update_password: bool = False):
@@ -286,7 +287,7 @@ class Entra(BrynQ):
             }})
         if len(custom_attributes) > 0:
             payload = self.__add_attribute_information(payload, custom_attributes)
-        response = requests.patch(endpoint, headers=self.headers, json=payload)
+        response = requests.patch(endpoint, headers=self.headers, json=payload, timeout=self.timeout)
         return response
 
     def delete_user(self, user_id, delete=False):
@@ -297,10 +298,10 @@ class Entra(BrynQ):
         """
         endpoint = f"https://graph.microsoft.com/v1.0/users/{user_id}"
         if delete:
-            response = requests.delete(endpoint, headers=self.headers)
+            response = requests.delete(endpoint, headers=self.headers, timeout=self.timeout)
         else:
             payload = {"accountEnabled": False}
-            response = requests.patch(endpoint, headers=self.headers, data=json.dumps(payload))
+            response = requests.patch(endpoint, headers=self.headers, data=json.dumps(payload), timeout=self.timeout)
         return response
 
     def assign_user_to_group(self, user_id, group_id):
@@ -312,7 +313,7 @@ class Entra(BrynQ):
         """
         url = f"https://graph.microsoft.com/v1.0/groups/{group_id}/members/$ref"
         data = {"@odata.id": f"https://graph.microsoft.com/v1.0/directoryObjects/{user_id}"}
-        response = requests.post(url, headers=self.headers, data=json.dumps(data))
+        response = requests.post(url, headers=self.headers, data=json.dumps(data), timeout=self.timeout)
         return response
 
     def update_manager(self, user_id, manager_id):
@@ -324,7 +325,7 @@ class Entra(BrynQ):
         """
         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}"}
-        response = requests.put(url, headers=self.headers, data=json.dumps(content))
+        response = requests.put(url, headers=self.headers, data=json.dumps(content), timeout=self.timeout)
         return response
 
     def remove_user_from_group(self, user_id, group_id):
@@ -335,5 +336,5 @@ class Entra(BrynQ):
         return: response
         """
         url = f"https://graph.microsoft.com/v1.0/groups/{group_id}/members/{user_id}/$ref"
-        response = requests.delete(url, headers=self.headers)
+        response = requests.delete(url, headers=self.headers, timeout=self.timeout)
         return response

{brynq_sdk_azure-2.0.2 → brynq_sdk_azure-2.0.3}/brynq_sdk_azure.egg-info/PKG-INFO

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

{brynq_sdk_azure-2.0.2 → brynq_sdk_azure-2.0.3}/setup.py

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