Move from pygame to sounddevice for sound handling

[perso/Immae/Projets/Python/MusicSampler.git] / helpers / lock.py

diff --git a/helpers/lock.py b/helpers/lock.py
index dff8b1fb87ad373c58370113339a6114ce255b4e..85d281ac8e576731e0362e67a36ad19d15c508da 100644 (file)
--- a/helpers/lock.py
+++ b/helpers/lock.py
@@ -5,6 +5,12 @@ class Lock:
         self.type = lock_type
         self.lock = threading.RLock()
 
+    def __enter__(self, *args, **kwargs):
+        self.acquire(*args, **kwargs)
+
+    def __exit__(self, type, value, traceback, *args, **kwargs):
+        self.release(*args, **kwargs)
+
     def acquire(self, *args, **kwargs):
         #print("acquiring lock for {}".format(self.type))
         self.lock.acquire(*args, **kwargs)