app.py

from logging.config import dictConfig

import flask
from flask import Flask, render_template, redirect, Blueprint, url_for

from apis import api_blueprint

from model.models import db

import os


def get_env_variable(name):
    try:
        return os.environ[name]
    except KeyError:
        message = "Expected environment variable '{}' not set.".format(name)
        raise Exception(message)

def get_db_uri():
    #postgres_url = get_env_variable("POSTGRES_URL")
    #postgres_user = get_env_variable("POSTGRES_USER")
    #postgres_pw = get_env_variable("POSTGRES_PW")
    #postgres_db = get_env_variable("POSTGRES_DB")
    postgres_url = "localhost"
    postgres_user = "geco_ro"
    postgres_pw = "geco78"
    postgres_db = "gmql_meta_new13"
    return 'postgresql+psycopg2://{user}:{pw}@{url}/{db}'.format(user=postgres_user,
                                                                 pw=postgres_pw,
                                                                 url=postgres_url,
                                                                 db=postgres_db)


dictConfig({
    'version': 1,
    'formatters': {'default': {
        'format': '[%(asctime)s] %(levelname)s in %(module)s:%(lineno)d: %(message)s',
    }},
    'handlers': {'wsgi': {
        'class': 'logging.StreamHandler',
        'stream': 'ext://flask.logging.wsgi_errors_stream',
        'formatter': 'default'
    }},
    'root': {
        'level': 'DEBUG',
        'handlers': ['wsgi']
    }
})

base_url = '/genosurf/'
api_url = base_url + 'api'
repo_static_url = base_url + 'repo_static'

my_app = Flask(__name__)


my_app.config['SQLALCHEMY_DATABASE_URI'] = get_db_uri()
my_app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
my_app.config['SQLALCHEMY_POOL_SIZE'] = 1
my_app.config['SQLALCHEMY_MAX_OVERFLOW'] = 30

db.init_app(my_app)

simple_page = Blueprint('root_pages', __name__,
                        static_folder='../vue-metadata/dist/static',
                        template_folder='../vue-metadata/dist')

graph_pages = Blueprint('static', __name__,
                        # static_url_path='/',
                        static_folder='./repo_static/',
                        # template_folder='../vue-metadata/dist'
                        )


# base url defined in apis init
@simple_page.route('/')
def index():
    flask.current_app.logger.info("serve index")
    return render_template('index.html')


# Make a "catch all route" so all requests match our index.html file.
# This lets us use the new history APIs in the browser.
@simple_page.route('/', defaults={'path': ''})
@simple_page.route('/<path:path>')
def redirect_all(path):
    return redirect(url_for('.index'))


# register blueprints
my_app.register_blueprint(api_blueprint, url_prefix=api_url)
my_app.register_blueprint(graph_pages, url_prefix=repo_static_url)
my_app.register_blueprint(simple_page, url_prefix=base_url)
my_app.app_context().push()


# redirect all to base url
@my_app.route('/', defaults={'path': ''})
@my_app.route('/<path:path>')
def index_all(path):
    return redirect(base_url)


# if __name__ == '__main__':
#     my_app.run()


# prevent cached responses
@my_app.after_request
def add_header(r):
    """
    Add headers to both force latest IE rendering engine or Chrome Frame,
    and also to cache the rendered page for 10 minutes.
    """
    if "Cache-Control" not in r.headers:
        r.cache_control.max_age = 300  # 5 min
        # r.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
        # r.headers["Pragma"] = "no-cache"
        # r.headers["Expires"] = "0"
        # r.headers['Cache-Control'] = 'public, max-age=0'
    return r