celery/celery/task/__init__.py

# -*- coding: utf-8 -*-
"""
    celery.task
    ~~~~~~~~~~~

    Creating tasks, subtasks, sets and chords.

    :copyright: (c) 2009 - 2012 by Ask Solem.
    :license: BSD, see LICENSE for more details.

"""
from __future__ import absolute_import

from celery.app.state import current_app, current_task as current
from celery.__compat__ import MagicModule, recreate_module
from celery.local import Proxy


class module(MagicModule):

    def __call__(self, *args, **kwargs):
        return self.task(*args, **kwargs)


old_module, new_module = recreate_module(__name__,
    by_module={
        "celery.task.base":   ["BaseTask", "Task", "PeriodicTask",
                               "task", "periodic_task"],
        "celery.canvas":      ["chain", "group", "chord", "subtask"],
        "celery.task.sets":   ["TaskSet"],
    },
    base=module,
    __package__="celery.task",
    __file__=__file__,
    __path__=__path__,
    __doc__=__doc__,
    current=current,
    discard_all=Proxy(lambda: current_app.control.discard_all),
    backend_cleanup=Proxy(
        lambda: current_app.tasks["celery.backend_cleanup"]
    ),
)