Added acquire_timeout context manager

Author: jtc42

src/labthings/core/lock.py

@@ -1,6 +1,7 @@
 from gevent.hub import getcurrent
 from gevent.lock import RLock as _RLock
 
+from contextlib import contextmanager
 import logging
 
 from .exceptions import LockError
@@ -31,6 +32,13 @@ def __init__(self, timeout=None, name=None):
         self.timeout = timeout
         self.name = name
 
+    @contextmanager
+    def acquire_timeout(self, timeout, blocking=True):
+        result = self.acquire(timeout=timeout, blocking=blocking)
+        yield result
+        if result:
+            self.release()
+        
     def locked(self):
         return self._lock.locked()
 
@@ -82,6 +90,13 @@ def __init__(self, locks, timeout=None):
         self.locks = locks
         self.timeout = timeout
 
+    @contextmanager
+    def acquire_timeout(self, timeout, blocking=True):
+        result = self.acquire(timeout=timeout, blocking=blocking)
+        yield result
+        if result:
+            self.release()
+        
     def acquire(self, blocking=True, timeout=sentinel):
         if timeout is sentinel:
             timeout = self.timeout