ContextTask

Complexity

Total Complexity

2

Size/Duplication

Total Lines	6
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	__call__()	0	3	2

"""
Starting our celery instance
"""
from __future__ import absolute_import

import logging
import os

from celery import Celery
from flask import current_app
from raven import Client
from raven.contrib.celery import register_signal, register_logger_signal
from raven.handlers.logging import SentryHandler
from raven.conf import setup_logging

from config import Config, config

celery = Celery(__name__, broker=Config.CELERY_BROKER_URL)

env = os.environ.get('FLASK_CONFIG', 'default')
client = Client(config[env].SENTRY_DSN)
handler = SentryHandler(client)
setup_logging(handler)
register_logger_signal(client, loglevel=logging.INFO)
register_signal(client)

TaskBase = celery.Task


class ContextTask(TaskBase):
    abstract = True

    def __call__(self, *args, **kwargs):
        with current_app.app_context():
            return TaskBase.__call__(self, *args, **kwargs)

celery.Task = ContextTask