import requests from requests.exceptions import RequestException, Timeout, ConnectionError, HTTPError from fastapi import HTTPException from secret_handler import return_credentials import os import time from logger_handler import setup_logger backend_api_url=os.getenv("BACKEND_API_URL","localhost:8101/internal/receive-notifications") api_key= return_credentials("/etc/secrets/api_key") logger = setup_logger(__name__) def send_notification(notification:str,max_retries: int = 5,timeout: int = 5)->str: """ Sends a notification to the internal backend service when a new docker image is released. Parameters: notification: Name of the docker repository. """ headers = { "X-API-Key-Internal": api_key, "Content-Type": "application/json" } data = { "receipent_user_id": 1, "message": { "title": notification, "body": "A new version is available.", "link": None, "category":"utility", "timestamp": int(time.time()) }} logger.debug(f"[Notify] Preparing to send notification: title='{notification}'") with requests.Session() as session: for attempt in range(1, max_retries + 1): try: logger.debug(f"[Notify] Sending request to backend (attempt {attempt}/{max_retries})") response = session.post(backend_api_url, headers=headers, json=data, timeout=timeout) response.raise_for_status() logger.info(f"[Notify] Notification sent successfully for '{notification}'") return response.text except (Timeout, ConnectionError) as e: logger.warning(f"[Notify] Attempt {attempt}/{max_retries} failed: {type(e).__name__}") if attempt == max_retries: logger.error(f"[Notify] All retry attempts failed for '{notification}'") raise HTTPException(status_code=503, detail=f"Notification service unavailable: {type(e).__name__}") sleep_time = 2 ** (attempt - 1) logger.debug(f"[Notify] Retrying in {sleep_time} seconds...") time.sleep(sleep_time) except HTTPError as e: logger.error(f"[Notify] HTTP {e.response.status_code}: {e.response.text}") raise HTTPException(status_code=e.response.status_code, detail=e.response.text) except RequestException as e: logger.error(f"[Notify] Unexpected request failure: {e}") raise HTTPException(status_code=500, detail=f"Request failed: {str(e)}")