v1

sharepoint_client.py

@@ -0,0 +1,125 @@
+import os
+from dotenv import load_dotenv
+import requests
+from datetime import datetime
+import logging
+import json
+
+# Load environment variables
+load_dotenv()
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+def update_excel_file(file_id, data):
+    """
+    Update Excel Online file with new data using OneDrive for Business API
+    """
+    client_id = os.getenv('AZURE_CLIENT_ID')
+    client_secret = os.getenv('AZURE_CLIENT_SECRET')
+    tenant_id = os.getenv('AZURE_TENANT_ID')
+    user_email = os.getenv('USER_EMAIL')
+
+    # Get access token
+    token_url = f"https://login.microsoftonline.com/{tenant_id}/oauth2/v2.0/token"
+    token_data = {
+        'grant_type': 'client_credentials',
+        'client_id': client_id,
+        'client_secret': client_secret,
+        'scope': 'https://graph.microsoft.com/.default'
+    }
+
+    token_response = requests.post(token_url, data=token_data)
+    access_token = token_response.json()['access_token']
+
+    headers = {
+        'Authorization': f'Bearer {access_token}',
+        'Content-Type': 'application/json'
+    }
+
+    try:
+        # Get the drive ID first
+        drives_url = f"https://graph.microsoft.com/v1.0/users/{user_email}/drive"
+        drive_response = requests.get(drives_url, headers=headers)
+        drive_response.raise_for_status()
+        drive_id = drive_response.json().get('id')
+        logging.info(f"Found drive ID: {drive_id}")
+
+        # Get the file details to verify it exists
+        file_url = f"https://graph.microsoft.com/v1.0/drives/{drive_id}/items/{file_id}"
+        file_response = requests.get(file_url, headers=headers)
+        file_response.raise_for_status()
+        logging.info(f"Found file: {file_response.json().get('name')}")
+
+        # Clear existing data (except header)
+        clear_url = f"https://graph.microsoft.com/v1.0/drives/{drive_id}/items/{file_id}/workbook/worksheets/Sheet1/range(address='A2:D1000')"
+        clear_data = {
+            "values": [[""]*4]*999
+        }
+        response = requests.patch(clear_url, headers=headers, json=clear_data)
+        response.raise_for_status()
+        logging.info("Cleared existing data from Excel file")
+
+        # Write new data
+        update_url = f"https://graph.microsoft.com/v1.0/drives/{drive_id}/items/{file_id}/workbook/worksheets/Sheet1/range(address='A2:D{len(data)+1}')"
+        update_data = {
+            "values": data
+        }
+        response = requests.patch(update_url, headers=headers, json=update_data)
+        response.raise_for_status()
+        logging.info(f"Successfully wrote {len(data)} rows to Excel file")
+
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Error updating Excel file: {e}")
+        if hasattr(e.response, 'text'):
+            logging.error(f"Response content: {e.response.text}")
+        raise
+
+def store_app_registrations(app_registrations):
+    # Get current date
+    current_date = datetime.utcnow()
+
+    # Prepare data for Excel
+    excel_data = []
+    for app in app_registrations:
+        password_credentials = app.get('passwordCredentials', [])
+        if not password_credentials:
+            logging.warning(f"No password credentials found for {app['displayName']}")
+            continue
+
+        expiry_date = password_credentials[0].get('endDateTime')
+        if expiry_date:
+            if (expiry_date.endswith('ZZ')):
+                expiry_date = expiry_date[:-1]
+            elif expiry_date.endswith('Z'):
+                expiry_date = expiry_date[:-1]
+            
+            try:
+                expiry_date_obj = datetime.strptime(expiry_date.split('.')[0], '%Y-%m-%dT%H:%M:%S')
+                days_to_expiry = (expiry_date_obj - current_date).days
+
+                # Get owner information
+                owners = app.get('owners', [])
+                owner_upns = [owner.get('userPrincipalName') for owner in owners if owner.get('userPrincipalName')]
+                owner_list = ', '.join(owner_upns) if owner_upns else 'No owners'
+
+                excel_data.append([
+                    app['displayName'],
+                    expiry_date_obj.strftime('%Y-%m-%d'),
+                    str(days_to_expiry),
+                    owner_list
+                ])
+
+            except ValueError as e:
+                logging.error(f"Error parsing expiry date for {app['displayName']}: {e}")
+                continue
+
+    # Update Excel file
+    try:
+        excel_file_id = os.getenv('EXCEL_FILE_ID')
+        update_excel_file(excel_file_id, excel_data)
+        logging.info("Successfully updated Excel file")
+    except Exception as e:
+        logging.error(f"Failed to update Excel file: {e}")
+
+    logging.info("Finished processing app registrations")