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.
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from functools import wraps
|
|
from .queue import Queue
|
|
from .connections import resolve_connection
|
|
from .worker import DEFAULT_RESULT_TTL
|
|
from rq.compat import string_types
|
|
|
|
|
|
class job(object):
|
|
def __init__(self, queue, connection=None, timeout=None,
|
|
result_ttl=DEFAULT_RESULT_TTL):
|
|
"""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.connection = resolve_connection(connection)
|
|
self.timeout = timeout
|
|
self.result_ttl = result_ttl
|
|
|
|
def __call__(self, f):
|
|
@wraps(f)
|
|
def delay(*args, **kwargs):
|
|
if isinstance(self.queue, string_types):
|
|
queue = Queue(name=self.queue, connection=self.connection)
|
|
else:
|
|
queue = self.queue
|
|
return queue.enqueue_call(f, args=args, kwargs=kwargs,
|
|
timeout=self.timeout, result_ttl=self.result_ttl)
|
|
f.delay = delay
|
|
return f
|