cubbyshare/frontend/__init__.py

import base64
import json
import os

import redis
import flask
import flask.json
from flask_bootstrap import Bootstrap

from .frontend import frontend
from .nav import nav


app = flask.Flask(__name__)
Bootstrap(app)
app.config["BOOTSTRAP_SERVE_LOCAL"] = True
app.config["SECRET_KEY"] = os.environ.get("PLATFORM_PROJECT_ENTROPY", "secretkey")
app.register_blueprint(frontend)
nav.init_app(app)

relationships = os.environ.get("PLATFORM_RELATIONSHIPS", None)
if relationships:
    app.config["RELATIONSHIPS"] = json.loads(
        base64.b64decode(relationships)
    )
    app.redis = redis.StrictRedis(
        host=app.config["RELATIONSHIPS"]["redis"][0]["host"],
        port=app.config["RELATIONSHIPS"]["redis"][0]["port"],
        db=0,
    )
else:
    app.redis = None


def get_token():
    return app.redis.get("token") if app.redis else None


app.get_token = get_token


if __name__ == "__main__":
    # token = redis_conn.get('token')
    app.run(host="0.0.0.0", port=8888)