izaliamae/izzylib
1
0
Fork 0
Code Issues Pull requests Releases 7 Wiki Activity

database: add optional session arg to delete and update

Browse source
This commit is contained in:
Izalia Mae 2021-04-18 01:49:58 -04:00
parent 7f23bd9ece
commit 48f0178405
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
IzzyLib/database.py
View file

@ -336,7 +336,10 @@ class CustomRows(object):
self[k] = v
def delete(self):
def delete(self, s=None):
if s:
return self.delete_session(s)
with self._db.session() as s:
return self.delete_session(s)
@ -345,10 +348,13 @@ class CustomRows(object):
return s.remove(row=self)
def update(self, dict_data={}, **data):
def update(self, dict_data={}, s=None, **data):
dict_data.update(data)
self._update(dict_data)
if s:
return self.update_session(s, **self._filter_data())
with self._db.session() as s:
s.update(row=self, **self._filter_data())