refactor: update logger setup for improved clarity and consistency in worker logging

This commit is contained in:
zurdi
2025-05-09 13:35:34 +00:00
parent b5ca8b437b
commit c4b9ecd3a6
2 changed files with 20 additions and 6 deletions

View File

@@ -5,7 +5,7 @@ from config import LOGLEVEL
from logger.formatter import Formatter
# Set up logger
log = logging.getLogger("romm")
log = logging.getLogger()
log.setLevel(LOGLEVEL)
# Define stdout handler

View File

@@ -1,9 +1,28 @@
import logging
import sentry_sdk
from config import SENTRY_DSN
from handler.redis_handler import redis_client
from logger.logger import log
from rq import Queue, Worker
from utils import get_version
# Get the rq.worker logger
rq_logger = logging.getLogger("rq.worker")
# Set its level (optional; you can match your app's LOGLEVEL if you want)
rq_logger.setLevel(log.level)
# Apply the same formatter to rq.worker handlers
if not rq_logger.hasHandlers():
# You can reuse the same handler as your app logger OR create a new one
for handler in log.handlers:
rq_logger.addHandler(handler)
else:
# If rq.worker already has handlers, just update their formatter
for handler in rq_logger.handlers:
handler.setFormatter(log.handlers[0].formatter)
listen = ("high", "default", "low")
sentry_sdk.init(
@@ -11,11 +30,6 @@ sentry_sdk.init(
release=f"romm@{get_version()}",
)
# TODO: setup custom logger for background workers
# Set up custom logging for Worker logging
# logging.basicConfig(format=common_log_format, datefmt=common_date_format)
if __name__ == "__main__":
# Start the worker
worker = Worker([Queue(name, connection=redis_client) for name in listen])
worker.work()