fix(db): execute not committing if fetch flag was set

This commit is contained in:
2026-03-03 13:10:50 +00:00
parent 6d60820800
commit a4c527ce5b

View File

@@ -27,19 +27,21 @@ class PostgresConnector:
self.connection.autocommit = False self.connection.autocommit = False
def execute(self, query, params=None, fetch=False) -> list: def execute(self, query, params=None, fetch=False) -> list:
with self.connection.cursor(cursor_factory=RealDictCursor) as cursor: try:
cursor.execute(query, params) with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
if fetch: cursor.execute(query, params)
return cursor.fetchall() result = cursor.fetchall() if fetch else None
self.connection.commit() self.connection.commit()
return result
except Exception:
self.connection.rollback()
raise
def execute_batch(self, query, values): def execute_batch(self, query, values):
with self.connection.cursor(cursor_factory=RealDictCursor) as cursor: with self.connection.cursor(cursor_factory=RealDictCursor) as cursor:
execute_batch(cursor, query, values) execute_batch(cursor, query, values)
self.connection.commit() self.connection.commit()
## User Management Methods
def close(self): def close(self):
if self.connection: if self.connection:
self.connection.close() self.connection.close()