|
|
|
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') |
|
|
|
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', |
|
'x-functions-key': api_key |
|
} |
|
|
|
try: |
|
response = requests.post(function_url, json={'operation': operation, 'data': data}, headers=headers) |
|
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: |
|
logging.error(f"Request exception: {str(e)}") |
|
raise |
