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)