PythonProject/util/database.py

@@ -6,4 +6,11 @@ engine = create_engine(db_url,pool_size=50)
 Base = declarative_base()
 
 Session = sessionmaker(bind=engine,autoflush=False,autocommit=False)
-db = Session()
+
+def get_db():
+    db = Session()
+    try:
+        yield db
+    finally:
+        db.close()
+

tmpl/work/util/database.py

@@ -6,3 +6,10 @@ engine = create_engine(db_url,pool_size=50)
 Base = declarative_base()
 
 Session = sessionmaker(bind=engine,autoflush=False,autocommit=False)
+
+def get_db():
+    db = Session()
+    try:
+        yield db
+    finally:
+        db.close()