wsgi.py

from api import app
from flask_sqlalchemy import SQLAlchemy
import logging, os
import yaml, sys
import uuid

# Port
port = int(os.environ.get("AS_PORT", 8080))

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=port)
else:
    # Running inside gunicorn, set logger
    gunicorn_logger = logging.getLogger('gunicorn.error')
    app.logger.handlers = gunicorn_logger.handlers
    app.logger.setLevel(gunicorn_logger.level)
    app.logger.info("Setting gunicorn logger...")

# Load config
app.logger.info("Loading config from " + "config/as.yml")
try:
    with open("config/as.yml", "r") as stream:
        conf = yaml.safe_load(stream)
        app.logger.debug("... config loaded")
        app.config['as'] = conf
except yaml.YAMLError as exc:
    app.logger.error('Error loading YAML: {}'.format(exc))
    sys.exit(4)
except FileNotFoundError as fnfe:
    app.logger.error('Could not load config file: {}'.format(fnfe))
    sys.exit(4)
conf = app.config['as']


# Init API-Key config
app.logger.info("Init API-Key config...")
if 'apikeys' in conf:
    apikey_conf = conf['apikeys']

    # iSHARE flow
    if apikey_conf['ishare'] and (apikey_conf['ishare']['enabledToken'] or apikey_conf['ishare']['enabledCreatePolicy']):
        app.logger.info("... enabled for iSHARE flows ...")
        if 'apiKey' not in apikey_conf['ishare'] or len(apikey_conf['ishare']['apiKey']) < 1:
            # Check for ENV
            if os.environ.get('AS_APIKEY_ISHARE'):
                app.logger.info("... ... no API-Key provided in config, using API-Key from ENV ...")
                apikey_conf['ishare']['apiKey'] = os.environ.get('AS_APIKEY_ISHARE')
            else:
                app.logger.info("... ... no API-Key provided, generating random API-Key ...")
                apikey_conf['ishare']['apiKey'] = str(uuid.uuid4())
        if apikey_conf['ishare']['enabledToken']:
            app.logger.info("... ... requiring API-Key in Header '{}' for /token endpoint: {}".format(apikey_conf['ishare']['headerName'], apikey_conf['ishare']['apiKey']))
        if apikey_conf['ishare']['enabledCreatePolicy']:
            app.logger.info("... ... requiring API-Key in Header '{}' for /createpolicy endpoint: {}".format(apikey_conf['ishare']['headerName'], apikey_conf['ishare']['apiKey']))

    # TIL flow
    if apikey_conf['issuer'] and apikey_conf['issuer']['enabledIssuer']:
        app.logger.info("... enabled for Trusted-Issuers-List flow ...")
        if 'apiKey' not in apikey_conf['issuer'] or len(apikey_conf['issuer']['apiKey']) < 1:
            if os.environ.get('AS_APIKEY_ISSUER'):
                app.logger.info("... ... no API-Key provided in config, using API-Key from ENV ...")
                apikey_conf['issuer']['apiKey'] = os.environ.get('AS_APIKEY_ISSUER')
            else:
                app.logger.info("... ... no API-Key provided, generating random API-Key ...")
                apikey_conf['issuer']['apiKey'] = str(uuid.uuid4())
        app.logger.info("... ... requiring API-Key in Header '{}' for /issuer endpoint: {}".format(apikey_conf['issuer']['headerName'], apikey_conf['issuer']['apiKey']))
            
else:
    app.logger.info("... no 'apikeys' config found. Not requiring any API Keys!")


# Create database
app.logger.info("Creating database...")
if 'db' not in conf:
    app.logger.error('No database configuration in config file')
    sys.exit(4)
db_conf = conf['db']

if os.environ.get('AS_DATABASE_URI'):
    app.logger.info("... taking URI from ENV...")
    app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('AS_DATABASE_URI')
if 'useMemory' in db_conf and db_conf['useMemory']:
    app.logger.info("... using in-memory SQLite ...")
    app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
elif 'useFile' in db_conf and db_conf['useFile'] and 'filename' in db_conf['useFile']:
    basedir = os.path.abspath(os.path.dirname(__file__))
    if 'filepath' in db_conf['useFile'] and len(db_conf['useFile']['filepath']) > 0:
        basedir = db_conf['useFile']['filepath']
    dbpath = os.path.join(basedir, db_conf['useFile']['filename'])
    app.logger.info("... using file-based SQLite (" +  dbpath + ") ...")
    app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + dbpath
elif 'useURI' in db_conf and db_conf['useURI'] and len(db_conf['useURI']) > 0:
    app.logger.info("... using specified URI '" + db_conf['useURI'] + "' ...")
    app.config['SQLALCHEMY_DATABASE_URI'] = db_conf['useURI']

app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
if db_conf['modTracking']:
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True

app.config['SQLALCHEMY_ECHO'] = False
if db_conf['echo']:
    app.config['SQLALCHEMY_ECHO'] = True

db = SQLAlchemy(app)
app.config['db'] = db
app.logger.info("... database created!")

with app.app_context():
    app.logger.info("Creating database tables...")
    from api.models import token
    #db.drop_all() # TODO: Make configurable
    db.create_all()
    app.logger.info("... database tables created")