Update modules/db_connection.py

modules/db_connection.py

@@ -1,13 +1,18 @@
 import requests
 import os
+import logging
 
 def call_azure_function(operation, data):
     function_url = os.getenv('AZURE_FUNCTION_URL')
     api_key = os.getenv('AZURE_FUNCTION_API_KEY')
 
-    print(f"Calling Azure Function: {function_url}")
-    print(f"Operation: {operation}")
-    print(f"Data: {data}")
+    if not function_url or not api_key:
+        logging.error("AZURE_FUNCTION_URL or AZURE_FUNCTION_API_KEY is not set")
+        raise ValueError("Azure Function URL or API Key is not configured")
+
+    logging.info(f"Calling Azure Function: {function_url}")
+    logging.info(f"Operation: {operation}")
+    logging.info(f"Data: {data}")
 
     headers = {
         'Content-Type': 'application/json',
@@ -16,13 +21,17 @@ def call_azure_function(operation, data):
 
     try:
         response = requests.post(function_url, json={'operation': operation, 'data': data}, headers=headers)
-        print(f"Response status code: {response.status_code}")
-        print(f"Response content: {response.text}")
+        logging.info(f"Response status code: {response.status_code}")
+        logging.info(f"Response content: {response.text}")
 
         if response.status_code == 200:
             return response.json()
+        elif response.status_code == 401:
+            logging.error("Authentication failed. Check if the API key is correct.")
+            raise Exception("Authentication failed. Check if the API key is correct.")
         else:
+            logging.error(f"Error calling Azure Function: {response.text}")
             raise Exception(f"Error calling Azure Function: {response.text}")
     except requests.exceptions.RequestException as e:
-        print(f"Request exception: {str(e)}")
+        logging.error(f"Request exception: {str(e)}")
         raise