Moved logging setupt to seperate module

requirements.txt

@@ -14,8 +14,9 @@ pycparser==2.23
 pydantic==2.12.0
 pydantic_core==2.41.1
 python-json-logger==4.0.0
+simple-logger-handler @ git+https://git.gansejunge.com/notifier/lib-logger-handler.git@main
 sniffio==1.3.1
 starlette==0.48.0
 typing-inspection==0.4.2
 typing_extensions==4.15.0
-uvicorn==0.37.0
+uvicorn==0.37.0

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")

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

@@ -7,7 +7,7 @@ from pydantic import BaseModel
 from validator import is_valid_platform,is_valid_token,verify_api_key
 from secret_handler import encrypt_token
 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
 import uuid
 from hashlib import sha256
 import uvicorn

src/secret_handler.py

@@ -1,5 +1,5 @@
 from cryptography.fernet import Fernet
-from logger_handler import setup_logger
+from simple_logger_handler import setup_logger
 
 logger = setup_logger(__name__)
 

src/validator.py

@@ -1,6 +1,6 @@
 from argon2 import PasswordHasher
 import re
-from logger_handler import setup_logger
+from simple_logger_handler import setup_logger
 
 logger = setup_logger(__name__)