Fix deadlock in exception case

secretlounge · Aug 3, 2024 · 6e04e38 · 6e04e38
1 parent e20a4b0
commit 6e04e38
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/secretlounge_ng/database.py b/secretlounge_ng/database.py
@@ -130,10 +130,12 @@ def __init__(self, obj, func, lock=None):
 	def __enter__(self):
 		return self.obj
 	def __exit__(self, exc_type, *_):
-		if exc_type is None:
-			self.func(self.obj)
-		if self.lock is not None:
-			self.lock.release()
+		try:
+			if exc_type is None:
+				self.func(self.obj)
+		finally:
+			if self.lock is not None:
+				self.lock.release()
 
 class Database():
 	def __init__(self):