From c6aa4ab03cad373c0e482333e7405dc8d45996e6 Mon Sep 17 00:00:00 2001 From: Andres GUERRERO Date: Fri, 21 Oct 2022 13:18:01 +0200 Subject: [PATCH] Don't unlock the task if lock_expiry is still valid --- celery_singleton/singleton.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/celery_singleton/singleton.py b/celery_singleton/singleton.py index 725151e..8398c1d 100644 --- a/celery_singleton/singleton.py +++ b/celery_singleton/singleton.py @@ -150,4 +150,5 @@ def on_failure(self, exc, task_id, args, kwargs, einfo): self.release_lock(task_args=args, task_kwargs=kwargs) def on_success(self, retval, task_id, args, kwargs): - self.release_lock(task_args=args, task_kwargs=kwargs) + if self.lock_expiry is None: + self.release_lock(task_args=args, task_kwargs=kwargs)