diff options
Diffstat (limited to 'redis/lock.py')
| -rw-r--r-- | redis/lock.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/redis/lock.py b/redis/lock.py index fb73ce2..05d4153 100644 --- a/redis/lock.py +++ b/redis/lock.py @@ -125,8 +125,9 @@ class Lock(object): def __enter__(self): # force blocking, as otherwise the user would have to check whether # the lock was actually acquired or not. - self.acquire(blocking=True) - return self + if self.acquire(blocking=True): + return self + raise LockError("Unable to acquire lock within the time specified") def __exit__(self, exc_type, exc_value, traceback): self.release() |