From 48f0178405f0a04d498287eb4e828cfa1218fb84 Mon Sep 17 00:00:00 2001 From: Izalia Mae Date: Sun, 18 Apr 2021 01:49:58 -0400 Subject: [PATCH] database: add optional session arg to delete and update --- IzzyLib/database.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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())