uhm

RAIILock.py

@@ -0,0 +1,35 @@
+
+
+import threading
+
+
+class RAIILock:
+	
+	def __init__(self, lock: threading.RLock, defer=False):
+		
+		self.__lock = lock
+		self.__defer = defer
+		
+		self.__acquired = False
+	
+	def __enter__(self):
+		
+		if self.__defer is False:
+			self.acquire()
+			
+		return self
+	
+	def __exit__(self, exc_type, exc_val, exc_tb):
+		
+		self.release()
+	
+	def acquire(self):
+		
+		self.__lock.acquire()
+		self.__acquired = True
+	
+	def release(self):
+		
+		if self.__acquired:
+			self.__lock.release()
+			self.__acquired = False