This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/root/celery.py

56 lines
1.8 KiB
Python
Raw Normal View History

"""passbook core celery"""
import os
2019-10-01 08:24:10 +00:00
from logging.config import dictConfig
from celery import Celery
2019-12-31 11:51:16 +00:00
from celery.signals import after_task_publish, setup_logging, task_postrun, task_prerun
from django.conf import settings
2019-10-01 08:24:10 +00:00
from structlog import get_logger
2019-04-04 19:48:50 +00:00
# set the default Django settings module for the 'celery' program.
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "passbook.root.settings")
LOGGER = get_logger()
2019-12-31 11:51:16 +00:00
CELERY_APP = Celery("passbook")
# pylint: disable=unused-argument
@setup_logging.connect
def config_loggers(*args, **kwags):
"""Apply logging settings from settings.py to celery"""
2019-10-01 08:24:10 +00:00
dictConfig(settings.LOGGING)
# pylint: disable=unused-argument
@after_task_publish.connect
def after_task_publish(sender=None, headers=None, body=None, **kwargs):
"""Log task_id after it was published"""
2019-12-31 11:51:16 +00:00
info = headers if "task" in headers else body
LOGGER.debug(
"Task published", task_id=info.get("id", ""), task_name=info.get("task", "")
)
# pylint: disable=unused-argument
@task_prerun.connect
def task_prerun(task_id, task, *args, **kwargs):
"""Log task_id on worker"""
2019-12-31 11:51:16 +00:00
LOGGER.debug("Task started", task_id=task_id, task_name=task.__name__)
# pylint: disable=unused-argument
@task_postrun.connect
def task_postrun(task_id, task, *args, retval=None, state=None, **kwargs):
"""Log task_id on worker"""
2019-12-31 11:51:16 +00:00
LOGGER.debug("Task finished", task_id=task_id, task_name=task.__name__, state=state)
# Using a string here means the worker doesn't have to serialize
# the configuration object to child processes.
# - namespace='CELERY' means all celery-related configuration keys
# should have a `CELERY_` prefix.
2019-12-31 11:51:16 +00:00
CELERY_APP.config_from_object(settings, namespace="CELERY")
# Load task modules from all registered Django app configs.
CELERY_APP.autodiscover_tasks()