From 0055ca976bb685c670f375f63400c1deab5507f9 Mon Sep 17 00:00:00 2001
From: Son Nguyen Kim <nguyenkims@users.noreply.github.com>
Date: Mon, 20 Sep 2021 18:16:52 +0200
Subject: [PATCH] add flush option to ModelMixin.create()

---
 app/models.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/app/models.py b/app/models.py
index 099fb09f..42ed70e4 100644
--- a/app/models.py
+++ b/app/models.py
@@ -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):