beenull/simple-login
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

add flush option to ModelMixin.create()

Browse source
This commit is contained in:
Son Nguyen Kim 2021-09-20 18:16:52 +02:00
parent ba3074b94a
commit 0055ca976b
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
app/models.py
View file

@ -83,12 +83,17 @@ class ModelMixin(object):
def create(cls, **kw): def create(cls, **kw):
# whether should call db.session.commit # whether should call db.session.commit
commit = kw.pop("commit", False) commit = kw.pop("commit", False)
flush = kw.pop("flush", False)
r = cls(**kw) r = cls(**kw)
db.session.add(r) db.session.add(r)
if commit: if commit:
db.session.commit() db.session.commit()
if flush:
db.session.flush()
return r return r
def save(self): def save(self):