move with_commit into class in db.py

page_analyzer/db.py

@@ -14,15 +14,6 @@
 )
 
 
-def with_commit(func):
-    def inner(self, *args, **kwargs):
-        result = func(self, *args, **kwargs)
-        self.conn.commit()
-        return result
-
-    return inner
-
-
 class DbConnectionProcessor:
     def __init__(self):
         self.conn = psycopg2.connect(DATABASE_URL)
@@ -37,6 +28,19 @@ def _execute_query(self, request, params=None, fetch_type=None):
                 case FETCH_TYPES.ALL:
                     return curs.fetchall()
 
+    @staticmethod
+    def with_commit(func):
+        def inner(self, *args, **kwargs):
+            try:
+                result = func(self, *args, **kwargs)
+                self.conn.commit()
+                return result
+            except psycopg2.Error as e:
+                self.conn.rollback()
+                raise e
+
+        return inner
+
     def get_all_urls(self):
         return self._execute_query(
             'SELECT * FROM urls ORDER BY id DESC', fetch_type=FETCH_TYPES.ALL