Include lock name when task fails to execute due to TaskLocked.

Fixes #757
coleifer · Aug 1, 2023 · fdbd988 · fdbd988
1 parent 9cbf17e
commit fdbd988
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/huey/api.py b/huey/api.py
@@ -384,7 +384,7 @@ def _execute(self, task, timestamp):
                 self._tasks_in_flight.remove(task)
                 duration = time_clock() - start
         except TaskLockedException as exc:
-            logger.warning('Task %s not run, unable to acquire lock.', task.id)
+            logger.warning('Task %s not run, %s.', task.id, exc)
             exception = exc
             self._emit(S.SIGNAL_LOCKED, task)
         except RetryTask as exc:
@@ -907,7 +907,7 @@ def inner(*args, **kwargs):
 
     def __enter__(self):
         if not self._huey.put_if_empty(self._key, '1'):
-            raise TaskLockedException('unable to set lock: %s' % self._name)
+            raise TaskLockedException('unable to acquire lock %s' % self._name)
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         self._huey.delete(self._key)

diff --git a/huey/contrib/helpers.py b/huey/contrib/helpers.py
@@ -59,7 +59,7 @@ def decorator(fn):
         def inner(*args, **kwargs):
             tid = sem.acquire()
             if tid is None:
-                raise TaskLockedException('unable to acquire lock: %s' %
+                raise TaskLockedException('unable to acquire lock %s' %
                                           lock_name)
             try:
                 return fn(*args, **kwargs)