1cfe1aff13
* root: initial rename * web: rename custom element prefix * root: rename external functions with pb_ prefix * root: fix formatting * root: replace domain with goauthentik.io * proxy: update path * root: rename remaining prefixes * flows: rename file extension * root: pbadmin -> akadmin * docs: fix image filenames * lifecycle: ignore migration files * ci: copy default config from current source before loading last tagged * *: new sentry dsn * tests: fix missing python3.9-dev package * root: add additional migrations for service accounts created by outposts * core: mark system-created service accounts with attribute * policies/expression: fix pb_ replacement not working * web: fix last linting errors, add lit-analyse * policies/expressions: fix lint errors * web: fix sidebar display on screens where not all items fit * proxy: attempt to fix proxy pipeline * proxy: use go env GOPATH to get gopath * lib: fix user_default naming inconsistency * docs: add upgrade docs * docs: update screenshots to use authentik * admin: fix create button on empty-state of outpost * web: fix modal submit not refreshing SiteShell and Table * web: fix height of app-card and height of generic icon * web: fix rendering of subtext * admin: fix version check error not being caught * web: fix worker count not being shown * docs: update screenshots * root: new icon * web: fix lint error * admin: fix linting error * root: migrate coverage config to pyproject
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
"""authentik core celery"""
|
|
import os
|
|
from logging.config import dictConfig
|
|
|
|
from celery import Celery
|
|
from celery.signals import after_task_publish, setup_logging, task_postrun, task_prerun
|
|
from django.conf import settings
|
|
from structlog import get_logger
|
|
|
|
# set the default Django settings module for the 'celery' program.
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "authentik.root.settings")
|
|
|
|
LOGGER = get_logger()
|
|
CELERY_APP = Celery("authentik")
|
|
|
|
|
|
# pylint: disable=unused-argument
|
|
@setup_logging.connect
|
|
def config_loggers(*args, **kwags):
|
|
"""Apply logging settings from settings.py to celery"""
|
|
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"""
|
|
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"""
|
|
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"""
|
|
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.
|
|
CELERY_APP.config_from_object(settings, namespace="CELERY")
|
|
|
|
# Load task modules from all registered Django app configs.
|
|
CELERY_APP.autodiscover_tasks()
|