33 lines
964 B
Python
33 lines
964 B
Python
import logging
|
|
import socket
|
|
|
|
from celery import Celery
|
|
from celery.signals import task_failure
|
|
|
|
app = Celery("dmoj")
|
|
|
|
from django.conf import settings # noqa: E402, I202, django must be imported here
|
|
|
|
app.config_from_object(settings, namespace="CELERY")
|
|
|
|
if hasattr(settings, "CELERY_BROKER_URL_SECRET"):
|
|
app.conf.broker_url = settings.CELERY_BROKER_URL_SECRET
|
|
if hasattr(settings, "CELERY_RESULT_BACKEND_SECRET"):
|
|
app.conf.result_backend = settings.CELERY_RESULT_BACKEND_SECRET
|
|
|
|
# Load task modules from all registered Django app configs.
|
|
app.autodiscover_tasks()
|
|
|
|
# Logger to enable errors be reported.
|
|
logger = logging.getLogger("judge.celery")
|
|
|
|
|
|
@task_failure.connect()
|
|
def celery_failure_log(sender, task_id, exception, traceback, *args, **kwargs):
|
|
logger.error(
|
|
"Celery Task %s: %s on %s",
|
|
sender.name,
|
|
task_id,
|
|
socket.gethostname(), # noqa: G201
|
|
exc_info=(type(exception), exception, traceback),
|
|
)
|