feat: Add LOG_LEVEL env variable

Author: ImMohammad20000

.env.example

@@ -48,9 +48,10 @@ UVICORN_PORT = 8000
 # LOG_ROTATION_INTERVAL=1
 # LOG_ROTATION_UNIT="H" # "S", "M", "H", "D", "W0"-"W6", "midnight"
 # LOG_MAX_BYTES=10485760 # 10 MB
+# LOG_LEVEL="INFO"
 # ECHO_SQL_QUERIES=False
 # VITE_BASE_API="https://example.com/"
-# JWT_ACCESS_TOKEN_EXPIRE_MINUTES = 1440
+# JWT_ACCESS_TOKEN_EXPIRE_MINUTES=1440
 
 # due to high amount of data, this job is only available for postgresql and timescaledb
 # ENABLE_RECORDING_NODES_STATS = False

app/utils/logger.py

@@ -10,6 +10,7 @@
     ECHO_SQL_QUERIES,
     LOG_BACKUP_COUNT,
     LOG_FILE_PATH,
+    LOG_LEVEL,
     LOG_MAX_BYTES,
     LOG_ROTATION_ENABLED,
     LOG_ROTATION_INTERVAL,
@@ -41,6 +42,9 @@ def formatMessage(self, record: logging.LogRecord) -> str:
     '%(levelprefix)s %(asctime)s - %(client_addr)s - "%(request_line)s" %(status_code)s'
 )
 
+LOGGING_CONFIG["loggers"]["uvicorn"]["level"] = LOG_LEVEL
+LOGGING_CONFIG["loggers"]["uvicorn.error"]["level"] = LOG_LEVEL
+LOGGING_CONFIG["loggers"]["uvicorn.access"]["level"] = LOG_LEVEL
 
 if SAVE_LOGS_TO_FILE:
     if LOG_ROTATION_ENABLED:

config.py

@@ -70,6 +70,10 @@
 LOG_ROTATION_INTERVAL = config("LOG_ROTATION_INTERVAL", cast=int, default=1)
 LOG_ROTATION_UNIT = config("LOG_ROTATION_UNIT", default="H")
 LOG_MAX_BYTES = config("LOG_MAX_BYTES", cast=int, default=10485760)  # default: 10 MB
+VALID_LOG_LEVELS = ("CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG")
+LOG_LEVEL = config("LOG_LEVEL", default="INFO").upper()
+if LOG_LEVEL not in VALID_LOG_LEVELS:
+    LOG_LEVEL = "INFO"
 
 # USERNAME: PASSWORD
 SUDOERS = (

main.py

@@ -1,5 +1,4 @@
 import ipaddress
-import logging
 import os
 import socket
 import ssl
@@ -14,6 +13,7 @@
 from app.utils.logger import LOGGING_CONFIG
 from config import (
     DEBUG,
+    LOG_LEVEL,
     UVICORN_HOST,
     UVICORN_LOOP,
     UVICORN_PORT,
@@ -134,10 +134,9 @@ def validate_cert_and_key(cert_file_path, key_file_path):
         bind_args["uds"] = None
         bind_args["host"] = "0.0.0.0"
 
-    log_level = logging.DEBUG if DEBUG else logging.INFO
-    LOGGING_CONFIG["loggers"]["uvicorn"]["level"] = log_level
-    LOGGING_CONFIG["loggers"]["uvicorn.error"]["level"] = log_level
-    LOGGING_CONFIG["loggers"]["uvicorn.access"]["level"] = log_level
+    effective_log_level = LOG_LEVEL
+    for logger_name in ("uvicorn", "uvicorn.error", "uvicorn.access"):
+        LOGGING_CONFIG["loggers"][logger_name]["level"] = effective_log_level
 
     try:
         uvicorn.run(
@@ -146,6 +145,7 @@ def validate_cert_and_key(cert_file_path, key_file_path):
             workers=1,
             reload=DEBUG,
             log_config=LOGGING_CONFIG,
+            log_level=effective_log_level.lower(),
             loop=UVICORN_LOOP,
         )
     except FileNotFoundError:  # to prevent error on removing unix sock