1 files changed, 131 insertions, 0 deletions
diff --git a/gn2/wqflask/__init__.py b/gn2/wqflask/__init__.py
new file mode 100644
index 00000000..9b714868
--- /dev/null
+++ b/gn2/wqflask/__init__.py
@@ -0,0 +1,131 @@
+"""Entry point for flask app"""
+# pylint: disable=C0413,E0611
+import os
+import time
+import datetime
+from typing import Tuple
+from pathlib import Path
+from urllib.parse import urljoin, urlparse
+
+import redis
+import jinja2
+from flask_session import Session
+from authlib.integrations.requests_client import OAuth2Session
+from flask import g, Flask, flash, session, url_for, redirect, current_app
+
+
+from gn2.utility import formatting
+
+from gn3.authentication import DataRole, AdminRole
+
+from gn2.wqflask.group_manager import group_management
+from gn2.wqflask.resource_manager import resource_management
+from gn2.wqflask.metadata_edits import metadata_edit
+
+from gn2.wqflask.api.markdown import glossary_blueprint
+from gn2.wqflask.api.markdown import references_blueprint
+from gn2.wqflask.api.markdown import links_blueprint
+from gn2.wqflask.api.markdown import policies_blueprint
+from gn2.wqflask.api.markdown import environments_blueprint
+from gn2.wqflask.api.markdown import facilities_blueprint
+from gn2.wqflask.api.markdown import blogs_blueprint
+from gn2.wqflask.api.markdown import news_blueprint
+from gn2.wqflask.api.jobs import jobs as jobs_bp
+from gn2.wqflask.oauth2.routes import oauth2
+from gn2.wqflask.oauth2.checks import user_logged_in
+from gn2.wqflask.oauth2.collections import num_collections
+from gn2.wqflask.oauth2.request_utils import user_details, authserver_authorise_uri
+
+from gn2.wqflask.jupyter_notebooks import jupyter_notebooks
+
+from gn2.wqflask.startup import (
+    StartupError,
+    startup_errors,
+    check_mandatory_configs)
+
+app = Flask(__name__)
+
+
+# See http://flask.pocoo.org/docs/config/#configuring-from-files
+# Note no longer use the badly named WQFLASK_OVERRIDES (nyi)
+default_settings_file = Path(Path(__file__).parent.parent.parent,
+                             "etc/default_settings.py")
+app.config.from_pyfile(default_settings_file)
+app.config.from_envvar('GN2_SETTINGS')
+
+app.jinja_env.globals.update(
+    undefined=jinja2.StrictUndefined,
+    numify=formatting.numify,
+    logged_in=user_logged_in,
+    authserver_authorise_uri=authserver_authorise_uri,
+    user_details=user_details,
+    num_collections=num_collections,
+    datetime=datetime)
+
+app.config["SESSION_REDIS"] = redis.from_url(app.config["REDIS_URL"])
+
+## BEGIN: SECRETS -- Should be the last of the settings to load
+secrets_file = os.environ.get("GN2_SECRETS")
+if secrets_file and Path(secrets_file).exists():
+    app.config.from_envvar("GN2_SECRETS")
+## END: SECRETS
+
+
+# Registering blueprints
+app.register_blueprint(glossary_blueprint, url_prefix="/glossary")
+app.register_blueprint(references_blueprint, url_prefix="/references")
+app.register_blueprint(links_blueprint, url_prefix="/links")
+app.register_blueprint(policies_blueprint, url_prefix="/policies")
+app.register_blueprint(environments_blueprint, url_prefix="/environments")
+app.register_blueprint(facilities_blueprint, url_prefix="/facilities")
+app.register_blueprint(blogs_blueprint, url_prefix="/blogs")
+app.register_blueprint(news_blueprint, url_prefix="/news")
+app.register_blueprint(jupyter_notebooks, url_prefix="/jupyter_notebooks")
+
+app.register_blueprint(resource_management, url_prefix="/resource-management")
+app.register_blueprint(metadata_edit, url_prefix="/datasets/")
+app.register_blueprint(group_management, url_prefix="/group-management")
+app.register_blueprint(jobs_bp, url_prefix="/jobs")
+app.register_blueprint(oauth2, url_prefix="/oauth2")
+
+from gn2.wqflask.app_errors import register_error_handlers
+register_error_handlers(app)
+
+try:
+    check_mandatory_configs(app)
+except StartupError as serr:
+    app.startup_error = serr
+    app.register_blueprint(startup_errors, url_prefix="/")
+
+server_session = Session(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}