von/pgbot
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

some refactoring, fixed db stuff and help section

Browse source
This commit is contained in:
Von Random 2023-11-05 11:31:42 +00:00
parent c9331151f8
commit 696bf5492b
4 changed files with 70 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pgbotlib/dbstuff.py
View file

@ -7,9 +7,19 @@ class DBConn:
self.connection = psycopg.connect(*args, **kwargs)
self.cursor = self.connection.cursor()
def update(self, query: str, values: tuple) -> list:
self.cursor.execute(query, values)
return self.connection.commit()
def update(self, query: str, values: tuple) -> None:
failure = None
try:
self.cursor.execute('SAVEPOINT sp1')
self.cursor.execute(query, values)
except Exception as e:
failure = e
self.cursor.execute('ROLLBACK TO SAVEPOINT sp1')
else:
self.cursor.execute('RELEASE SAVEPOINT sp1')
self.connection.commit()
if failure:
raise failure
def query_raw(self, query: str, values: tuple) -> list:
self.cursor.execute(query, values)