new @locked decorator

1 files changed, 18 insertions, 0 deletions

diff --git a/decorators.py b/decorators.py
index 72b88ff..b952a56 100644
--- a/decorators.py
+++ b/decorators.py
@@ -112,7 +112,9 @@ class classproperty(object):
     def __get__(self, inst, cls):
         return self.get(cls)
 
+
 from time import clock
+
 def timed(f):
     def wrap(*args, **kwargs):
         t = clock()
@@ -122,3 +124,19 @@ def timed(f):
         return res
     return wrap
 
+
+def locked(acquire, release):
+    """decorator taking two methods to acquire/release a lock as argument,
+    returning a decorator function which will call the inner method after
+    having called acquire(self) et will call release(self) afterwards.
+    """
+    def decorator(f):
+        def wrapper(self, *args, **kwargs):
+            acquire(self)
+            try:
+                return f(self, *args, **kwargs)
+            finally:
+                release(self)
+        return wrapper
+    return decorator
+