import os
from celery import Celery
os.environ.setdefault('DJANGO_SETTINGS_MODULE',
'stackdio.server.settings.production')
# pylint: disable=wrong-import-position
from django.conf import settings # nopep8
app = Celery('stackdio')
# Using a string here means the worker will not have to
# pickle the object when using Windows.
app.config_from_object('django.conf:settings')
app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)
# Add opbeat for celery if it's enabled
if settings.OPBEAT_ENABLED:
from opbeat.contrib.django.models import client, logger, register_handlers
from opbeat.contrib.celery import register_signal
try:
register_signal(client)
except Exception as e:
logger.exception('Failed installing celery hook: %s', e)
if 'opbeat.contrib.django' in settings.INSTALLED_APPS:
register_handlers()
# set the default Django settings module for the 'celery' program.
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'perma.settings')
app = Celery('perma')
# Using a string here means the worker will not have to
# pickle the object when using Windows.
app.config_from_object('django.conf:settings')
app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)
# configure opbeat
# Imports have to come after DJANGO_SETTINGS_MODULE
if settings.USE_OPBEAT:
from celery.signals import task_prerun, task_postrun
from opbeat.contrib.django.models import client, logger, register_handlers
from opbeat.contrib.celery import register_signal
try:
register_signal(client)
except Exception as e:
logger.exception('Failed installing celery hook: %s' % e)
register_handlers()
@task_prerun.connect
def task_prerun_handler(task_id, task, *args, **kwargs):
client.begin_transaction("task.celery")
@task_postrun.connect
def task_postrun_handler(task_id, task, *args, **kwargs):
client.end_transaction('celery:'+task.__name__, 200 if kwargs.get('state') == 'SUCCESS' else 500)