wip refactoring

apis/microsoft_graph/client.py

@@ -0,0 +1,85 @@
+"""
+HTTP client for Microsoft Graph API.
+"""
+import aiohttp
+import asyncio
+import orjson
+
+# Define a global semaphore for throttling
+semaphore = asyncio.Semaphore(4)
+
+async def fetch_with_aiohttp(url, headers):
+    """
+    Fetch data from Microsoft Graph API.
+
+    Args:
+        url (str): The URL to fetch data from.
+        headers (dict): Headers including authentication.
+
+    Returns:
+        dict: JSON response data.
+
+    Raises:
+        Exception: If the request fails.
+    """
+    async with semaphore:
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url, headers=headers) as response:
+                if response.status != 200:
+                    raise Exception(f"Failed to fetch {url}: {response.status} {await response.text()}")
+                raw_bytes = await response.read()
+                content_length = response.headers.get('Content-Length')
+                if content_length and len(raw_bytes) != int(content_length):
+                    print("Warning: Incomplete response received!")
+                    return None
+                return orjson.loads(raw_bytes)
+
+async def post_with_aiohttp(url, headers, json_data):
+    """
+    Post data to Microsoft Graph API.
+
+    Args:
+        url (str): The URL to post data to.
+        headers (dict): Headers including authentication.
+        json_data (dict): JSON data to post.
+
+    Returns:
+        int: HTTP status code.
+    """
+    async with semaphore:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(url, headers=headers, json=json_data) as response:
+                return response.status
+
+async def patch_with_aiohttp(url, headers, json_data):
+    """
+    Patch data to Microsoft Graph API.
+
+    Args:
+        url (str): The URL to patch data to.
+        headers (dict): Headers including authentication.
+        json_data (dict): JSON data to patch.
+
+    Returns:
+        int: HTTP status code.
+    """
+    async with semaphore:
+        async with aiohttp.ClientSession() as session:
+            async with session.patch(url, headers=headers, json=json_data) as response:
+                return response.status
+
+async def delete_with_aiohttp(url, headers):
+    """
+    Delete data from Microsoft Graph API.
+
+    Args:
+        url (str): The URL to delete data from.
+        headers (dict): Headers including authentication.
+
+    Returns:
+        int: HTTP status code.
+    """
+    async with semaphore:
+        async with aiohttp.ClientSession() as session:
+            async with session.delete(url, headers=headers) as response:
+                return response.status