Changed acquire_timeout to class __call__ method

labthings · Jun 24, 2020 · b7d1594 · b7d1594
1 parent e6a313d
commit b7d1594
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/labthings/core/lock.py b/src/labthings/core/lock.py
@@ -33,7 +33,7 @@ def __init__(self, timeout=None, name=None):
         self.name = name
 
     @contextmanager
-    def acquire_timeout(self, timeout, blocking=True):
+    def __call__(self, timeout=sentinel, blocking=True):
         result = self.acquire(timeout=timeout, blocking=blocking)
         yield result
         if result:
@@ -91,7 +91,7 @@ def __init__(self, locks, timeout=None):
         self.timeout = timeout
 
     @contextmanager
-    def acquire_timeout(self, timeout, blocking=True):
+    def __call__(self, timeout=sentinel, blocking=True):
         result = self.acquire(timeout=timeout, blocking=blocking)
         yield result
         if result: