20 lines
509 B
Python
20 lines
509 B
Python
#!/usr/bin/env python
|
|
import logging
|
|
|
|
from atat.app import celery, make_app, make_config
|
|
from celery.signals import after_setup_task_logger
|
|
|
|
from atat.utils.logging import JsonFormatter
|
|
|
|
config = make_config()
|
|
app = make_app(config)
|
|
app.app_context().push()
|
|
|
|
|
|
@after_setup_task_logger.connect
|
|
def setup_task_logger(*args, **kwargs):
|
|
if app.config.get("LOG_JSON"):
|
|
logger = logging.getLogger()
|
|
for handler in logger.handlers:
|
|
handler.setFormatter(JsonFormatter(source="queue"))
|