mirror of https://github.com/peter4431/rq.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import (absolute_import, division, print_function,
|
|
unicode_literals)
|
|
|
|
from functools import wraps
|
|
|
|
from rq.compat import string_types
|
|
|
|
from .defaults import DEFAULT_RESULT_TTL
|
|
from .queue import Queue
|
|
from .utils import backend_class
|
|
|
|
|
|
class job(object): # noqa
|
|
queue_class = Queue
|
|
|
|
def __init__(self, queue, connection=None, timeout=None,
|
|
result_ttl=DEFAULT_RESULT_TTL, ttl=None,
|
|
queue_class=None):
|
|
"""A decorator that adds a ``delay`` method to the decorated function,
|
|
which in turn creates a RQ job when called. Accepts a required
|
|
``queue`` argument that can be either a ``Queue`` instance or a string
|
|
denoting the queue name. For example:
|
|
|
|
@job(queue='default')
|
|
def simple_add(x, y):
|
|
return x + y
|
|
|
|
simple_add.delay(1, 2) # Puts simple_add function into queue
|
|
"""
|
|
self.queue = queue
|
|
self.queue_class = backend_class(self, 'queue_class', override=queue_class)
|
|
self.connection = connection
|
|
self.timeout = timeout
|
|
self.result_ttl = result_ttl
|
|
self.ttl = ttl
|
|
|
|
def __call__(self, f):
|
|
@wraps(f)
|
|
def delay(*args, **kwargs):
|
|
if isinstance(self.queue, string_types):
|
|
queue = self.queue_class(name=self.queue,
|
|
connection=self.connection)
|
|
else:
|
|
queue = self.queue
|
|
depends_on = kwargs.pop('depends_on', None)
|
|
at_front = kwargs.pop('at_front', False)
|
|
return queue.enqueue_call(f, args=args, kwargs=kwargs,
|
|
timeout=self.timeout, result_ttl=self.result_ttl,
|
|
ttl=self.ttl, depends_on=depends_on, at_front=at_front)
|
|
f.delay = delay
|
|
return f
|