diff --git a/IzzyLib/database.py b/IzzyLib/database.py index 9715724..4d13891 100644 --- a/IzzyLib/database.py +++ b/IzzyLib/database.py @@ -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())