1 files changed, 9 insertions, 58 deletions
diff --git a/wqflask/gn2_main.py b/wqflask/gn2_main.py
index 923a35d3..3fd18b6d 100644
--- a/wqflask/gn2_main.py
+++ b/wqflask/gn2_main.py
@@ -1,60 +1,11 @@
-"""Main app creation module"""
-import time
+"""Application entry-point module"""
 
-from flask import g, session, request
+from wqflask import app
 
-from wqflask import create_app
-from wqflask.user_session import UserSession
-from gn3.authentication import DataRole, AdminRole
-
-app = create_app()
-
-@app.before_request
-def before_request():
-    g.request_start_time = time.time()
-    g.request_time = lambda: "%.5fs" % (time.time() - g.request_start_time)
-
-    token = session.get("oauth2_token", False)
-    if token and not bool(session.get("user_details", False)):
-        config = current_app.config
-        client = OAuth2Session(
-            config["OAUTH2_CLIENT_ID"], config["OAUTH2_CLIENT_SECRET"],
-            token=token)
-        resp = client.get(
-            urljoin(config["GN_SERVER_URL"], "oauth2/user"))
-        user_details = resp.json()
-        session["user_details"] = user_details
-
-        if user_details.get("error") == "invalid_token":
-            flash(user_details["error_description"], "alert-danger")
-            flash("You are now logged out.", "alert-info")
-            session.pop("user_details", None)
-            session.pop("oauth2_token", None)
-
-@app.context_processor
-def include_admin_role_class():
-    return {'AdminRole': AdminRole}
-
-
-@app.context_processor
-def include_data_role_class():
-    return {'DataRole': DataRole}
-
-@app.before_request
-def get_user_session():
-    g.user_session = UserSession()
-    # I think this should solve the issue of deleting the cookie and redirecting to the home page when a user's session has expired
-    if not g.user_session:
-        response = make_response(redirect(url_for('login')))
-        response.set_cookie('session_id_v2', '', expires=0)
-        return response
-
-@app.after_request
-def set_user_session(response):
-    if hasattr(g, 'user_session'):
-        if not request.cookies.get(g.user_session.cookie_name):
-            response.set_cookie(g.user_session.cookie_name,
-                                g.user_session.cookie)
-    else:
-        response.set_cookie('session_id_v2', '', expires=0)
-    return response
+from base import trait
+from wqflask import views
+from wqflask import collect
+from wqflask.api import router
+from wqflask import user_login
+from wqflask import user_session
+from wqflask import partial_correlations_views