make every request a database transaction.

This commit is contained in:
fiatjaf
2020-09-11 14:17:09 -03:00
committed by fiatjaf
parent 487f082eb2
commit 4855e2cd3d
5 changed files with 174 additions and 159 deletions
+13 -2
View File
@@ -15,9 +15,20 @@ class Database:
return self
def __exit__(self, exc_type, exc_val, exc_tb):
if exc_val:
self.connection.rollback()
self.cursor.close()
self.cursor.close()
else:
self.connection.commit()
self.cursor.close()
self.connection.close()
def commit(self):
self.connection.commit()
self.cursor.close()
self.connection.close()
def rollback(self):
self.connection.rollback()
def fetchall(self, query: str, values: tuple = ()) -> list:
"""Given a query, return cursor.fetchall() rows."""