add flush option to ModelMixin.create()

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

View File

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