Clean up database connector after session

1 files changed, 12 insertions, 2 deletions

diff --git a/wqflask/wqflask/views.py b/wqflask/wqflask/views.py
index ac45815b..10e53e5e 100644
--- a/wqflask/wqflask/views.py
+++ b/wqflask/wqflask/views.py
@@ -48,7 +48,7 @@ from wqflask.wgcna import wgcna_analysis
 from wqflask.ctl import ctl_analysis
 
 from utility import temp_data
-from utility.tools import TEMPDIR,USE_REDIS,USE_GN_SERVER,GN_SERVER_URL
+from utility.tools import SQL_URI,TEMPDIR,USE_REDIS,USE_GN_SERVER,GN_SERVER_URL
 
 from base import webqtlFormData
 from base.webqtlConfig import GENERATED_IMAGE_DIR
@@ -58,13 +58,23 @@ from pprint import pformat as pf
 
 from wqflask import user_manager
 from wqflask import collect
+from wqflask.database import db_session
 
 import utility.logger
 logger = utility.logger.getLogger(__name__ )
 
 @app.before_request
 def connect_db():
-    g.db = sqlalchemy.create_engine(app.config['SQL_URI'])
+    db = getattr(g, '_database', None)
+    if db is None:
+        logger.debug("Get new database connector")
+        g.db = g._database = sqlalchemy.create_engine(SQL_URI)
+        logger.debug(g.db)
+
+@app.teardown_appcontext
+def shutdown_session(exception=None):
+    logger.debug("remove db_session")
+    db_session.remove()
 
 #@app.before_request
 #def trace_it():