from .job import Job
from .utils import get_backend

from . import app_settings

class task(object):
    def __init__(self, queue='default'):
        self.queue = queue

        app_settings.WORKERS.setdefault(self.queue, 1)

    def __call__(self, fn):
        return TaskWrapper(fn, self.queue)

class TaskWrapper(object):
    def __init__(self, fn, queue):
        self.fn = fn
        self.queue = queue

        self.path = '%s.%s' % (fn.__module__, fn.__name__)

    def __repr__(self):
        return "<TaskWrapper: %s>" % self.path

    def __call__(self, *args, **kwargs):
        job = Job(self.path, args, kwargs)
        job.validate()

        get_backend().enqueue(job, self.queue)

from .job import Job
from .utils import get_backend

from . import app_settings

class task(object):
    def __init__(self, queue='default', timeout=None):
        self.queue = queue
        self.timeout = timeout

        app_settings.WORKERS.setdefault(self.queue, 1)

    def __call__(self, fn):
        return TaskWrapper(fn, self.queue, self.timeout)

class TaskWrapper(object):
    def __init__(self, fn, queue, timeout):
        self.fn = fn
        self.queue = queue
        self.timeout = timeout

        self.path = '%s.%s' % (fn.__module__, fn.__name__)

    def __repr__(self):
        return "<TaskWrapper: %s>" % self.path

    def __call__(self, *args, **kwargs):
        job = Job(self.path, args, kwargs)
        job.validate()

        get_backend().enqueue(job, self.queue)

Store a timeout value on the TaskWrapper, defaulting to no timeout.