From a3c32694806ffc212f987c99c20799526069d643 Mon Sep 17 00:00:00 2001
From: Pjotr Prins
Date: Thu, 23 Jun 2016 10:56:12 +0000
Subject: views: close db handling

---
 wqflask/wqflask/views.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

(limited to 'wqflask')

diff --git a/wqflask/wqflask/views.py b/wqflask/wqflask/views.py
index 10e53e5e..bce1f755 100644
--- a/wqflask/wqflask/views.py
+++ b/wqflask/wqflask/views.py
@@ -26,9 +26,10 @@ import base64
 import array
 import sqlalchemy
 from wqflask import app
-from flask import (render_template, request, make_response, Response,
-                   Flask, g, config, jsonify, redirect, url_for,
-                   send_from_directory)
+from flask import g, request, render_template
+# from flask import (render_template, request, make_response, Response,
+#                    Flask, g, config, jsonify, redirect, url_for,
+#                    send_from_directory)
 from wqflask import search_results
 from wqflask import gsearch
 from wqflask import update_search_results
@@ -73,8 +74,11 @@ def connect_db():
 
 @app.teardown_appcontext
 def shutdown_session(exception=None):
-    logger.debug("remove db_session")
-    db_session.remove()
+    db = getattr(g, '_database', None)
+    if db is not None:
+        logger.debug("remove db_session")
+        db_session.remove()
+        g.db = None
 
 #@app.before_request
 #def trace_it():
-- 
cgit v1.2.3