Make AcquirerProxy.acquire() support timeout argument

41eb85b1 · Richard Oudkerk · b78174c0 · 41eb85b1
Commit 41eb85b1 authored May 06, 2012 by Richard Oudkerk
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

Lib/multiprocessing/managers.py Lib/multiprocessing/managers.py +3 -2

No files found.
--- a/Lib/multiprocessing/managers.py
+++ b/Lib/multiprocessing/managers.py
@@ -957,8 +957,9 @@ class IteratorProxy(BaseProxy):
 class AcquirerProxy(BaseProxy):
    _exposed_ = ('acquire', 'release')
-    def acquire(self, blocking=True):
+    def acquire(self, blocking=True, timeout=None):
-        return self._callmethod('acquire', (blocking,))
+        args = (blocking,) if timeout is None else (blocking, timeout)
+        return self._callmethod('acquire', args)
    def release(self):
        return self._callmethod('release')
    def __enter__(self):