Moved logging setup to seperate module

src/db.py

@@ -3,7 +3,7 @@ import threading
 from secret_handler import return_credentials
 import os
 import time
-from logger_handler import setup_logger
+from simple_logger_handler import setup_logger
 
 db_username = return_credentials("/etc/secrets/db_username")
 db_password = return_credentials("/etc/secrets/db_password")

src/dockerhub_api.py

@@ -1,6 +1,6 @@
 import requests
 from secret_handler import return_credentials
-from logger_handler import setup_logger
+from simple_logger_handler import setup_logger
 
 dockerhub_token = return_credentials("/etc/secrets/dockerhub_token")
 dockerhub_username = return_credentials("/etc/secrets/dockerhub_username")

src/github_api.py

@@ -1,6 +1,6 @@
 import requests
 from secret_handler import return_credentials
-from logger_handler import setup_logger
+from simple_logger_handler import setup_logger
 
 github_token = return_credentials("/etc/secrets/github_token")
 logger = setup_logger(__name__)

src/logger_handler.py

@@ -1,30 +0,0 @@
-import os
-import logging
-
-try:
-	from pythonjsonlogger import jsonlogger
-	JSON_LOGGING = True
-except ImportError:
-	JSON_LOGGING = False
-
-LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()
-if LOG_LEVEL not in {"ERROR", "DEBUG", "INFO", "WARNING", "CRITICAL"}:
-	LOG_LEVEL = "INFO"
-
-def setup_logger(name: str) -> logging.Logger:
-	logger = logging.getLogger(name)
-	if not logger.handlers:
-		handler = logging.StreamHandler()
-		if JSON_LOGGING:
-			formatter = jsonlogger.JsonFormatter(
-				"%(asctime)s %(name)s %(levelname)s %(message)s"
-			)
-		else:
-			formatter = logging.Formatter(
-				"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
-			)
-		handler.setFormatter(formatter)
-		logger.addHandler(handler)
-	logger.setLevel(getattr(logging, LOG_LEVEL))
-	logger.debug(f"Logger {name} initialized with level {LOG_LEVEL}")
-	return logger

src/main.py

@@ -5,7 +5,7 @@ from dockerhub_api import find_package_version_with_tag as find_package_version_
 import uvicorn
 from contextlib import asynccontextmanager
 from db import get_db, create_connection_pool, close_connection_pool, start_healthcheck_thread
-from logger_handler import setup_logger, LOG_LEVEL
+from simple_logger_handler import setup_logger, LOG_LEVEL
 from send_notification import send_notification
 from metrics_server import REQUEST_COUNTER
 import asyncio

src/secret_handler.py

@@ -1,4 +1,4 @@
-from logger_handler import setup_logger
+from simple_logger_handler import setup_logger
 
 logger = setup_logger(__name__)
 

src/send_notification.py

@@ -4,7 +4,7 @@ from fastapi import HTTPException
 from secret_handler import return_credentials
 import os
 import time
-from logger_handler import setup_logger
+from simple_logger_handler import setup_logger
 
 
 backend_api_url=os.getenv("BACKEND_API_URL","localhost:8101/internal/receive-notifications")