diff options
| author | Yang Yang <y4n9squared@gmail.com> | 2022-08-02 19:09:24 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-08-03 06:09:24 +0700 |
| commit | 3f59a7e6e3d1e1aafb55a027f19221c954a2c8b4 (patch) | |
| tree | 316e8654e6960282cf84ec575e921228a5b9460a | |
| parent | 98c200d07ab23e31843284507b7004e45b5b5b46 (diff) | |
| download | rq-3f59a7e6e3d1e1aafb55a027f19221c954a2c8b4.tar.gz | |
Support callbacks in enqueue_many (#1686)
The on_success and on_failure callbacks are currently supported in
Queue.enqueue but not Queue.enqueue_many, despite documentary indicating
the contrary.
This commit brings the bulk enqueuing API back into parity.
Fixes #1680
| -rw-r--r-- | rq/queue.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/rq/queue.py b/rq/queue.py index 3472d69..f7d2214 100644 --- a/rq/queue.py +++ b/rq/queue.py @@ -23,7 +23,7 @@ def compact(lst): class EnqueueData(namedtuple('EnqueueData', ["func", "args", "kwargs", "timeout", "result_ttl", "ttl", "failure_ttl", "description", "job_id", - "at_front", "meta", "retry"])): + "at_front", "meta", "retry", "on_success", "on_failure"])): """Helper type to use when calling enqueue_many NOTE: Does not support `depends_on` yet. """ @@ -417,14 +417,14 @@ class Queue: def prepare_data(func, args=None, kwargs=None, timeout=None, result_ttl=None, ttl=None, failure_ttl=None, description=None, job_id=None, - at_front=False, meta=None, retry=None): + at_front=False, meta=None, retry=None, on_success=None, on_failure=None): # Need this till support dropped for python_version < 3.7, where defaults can be specified for named tuples # And can keep this logic within EnqueueData return EnqueueData( func, args, kwargs, timeout, result_ttl, ttl, failure_ttl, description, job_id, - at_front, meta, retry + at_front, meta, retry, on_success, on_failure ) def enqueue_many( @@ -446,7 +446,9 @@ class Queue: depends_on=None, job_id=job_data.job_id, meta=job_data.meta, status=JobStatus.QUEUED, timeout=job_data.timeout, - retry=job_data.retry + retry=job_data.retry, + on_success=job_data.on_success, + on_failure=job_data.on_failure ), pipeline=pipe, at_front=job_data.at_front |