Update __init__.py

Teahouse-Studios · Sep 16, 2024 · 66c8843 · 66c8843
1 parent b3a5418
commit 66c8843
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/database/__init__.py b/database/__init__.py
@@ -303,11 +303,17 @@ def check(target_id):
                 return True
             return False
 
+        @staticmethod
+        @retry(stop=stop_after_attempt(3))
+        @auto_rollback_error
         def add(target_id):
             session.add(GroupBlockList(targetId=target_id))
             session.commit()
             return True
 
+        @staticmethod
+        @retry(stop=stop_after_attempt(3))
+        @auto_rollback_error
         def remove(target_id):
             entry = session.query(GroupBlockList).filter_by(targetId=target_id).first()
             if entry: