enhancing locks

1 files changed, 15 insertions, 0 deletions

diff --git a/helpers/lock.py b/helpers/lock.py
new file mode 100644
index 0000000..dff8b1f
--- /dev/null
+++ b/helpers/lock.py
@@ -0,0 +1,15 @@
+import threading
+
+class Lock:
+    def __init__(self, lock_type):
+        self.type = lock_type
+        self.lock = threading.RLock()
+
+    def acquire(self, *args, **kwargs):
+        #print("acquiring lock for {}".format(self.type))
+        self.lock.acquire(*args, **kwargs)
+
+    def release(self, *args, **kwargs):
+        #print("releasing lock for {}".format(self.type))
+        self.lock.release(*args, **kwargs)
+