Corvus/server/corvus/__init__.py

"""Corvus Flask Application."""
import os
from logging.config import dictConfig

from flask import Flask
from flask_migrate import Migrate

from corvus.db import db
from corvus.errors import BaseError, handle_corvus_base_error, \
    handle_corvus_404_error
from corvus.utility import json_utility, session_utility

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


def create_app(test_config: dict = None) -> Flask:
    """
    Create an instance of Corvus.

    :param test_config:
    :return:
    """
    app = Flask(__name__, instance_relative_config=True)
    app.logger.debug('Creating Corvus Server')

    data_directory = os.getenv('CORVUS_DATA_DIRECTORY', '/tmp')
    app.logger.debug('Corvus Data Directory: %s', data_directory)

    default_database_uri = 'sqlite:///{}/corvus.db'.format(data_directory)
    app.config.from_mapping(
        SECRET_KEY='dev',
        SQLALCHEMY_DATABASE_URI=default_database_uri,
        SQLALCHEMY_TRACK_MODIFICATIONS=False,
        TRAP_HTTP_EXCEPTIONS=True,
    )

    if test_config is None:
        app.logger.debug('Loading configurations')
        app.config.from_object('corvus.default_settings')
        app.config.from_pyfile('config.py', silent=True)
        if os.getenv('CORVUS_SETTINGS', None):
            app.config.from_envvar('CORVUS_SETTINGS')
    else:
        app.logger.debug('Loading test configuration')
        app.config.from_mapping(test_config)

    try:
        os.makedirs(app.instance_path)
    except OSError:
        pass

    app.json_encoder = json_utility.CustomJSONEncoder
    app.session_interface = session_utility.DisableSessionInterface()

    app.logger.debug('Initializing Application')
    db.init_app(app)
    app.logger.debug('Registering Database Models')
    Migrate(app, db)

    return app


def register_blueprints(app: Flask) -> None:
    """
    Register blueprints for the application.

    :param app:
    :return:
    """
    from corvus.api import AUTH_BLUEPRINT, USER_BLUEPRINT, HEALTH_BLUEPRINT
    app.register_blueprint(AUTH_BLUEPRINT)
    app.register_blueprint(USER_BLUEPRINT)
    app.register_blueprint(HEALTH_BLUEPRINT)


def register_error_handlers(app: Flask) -> None:
    """
    Register error handlers for the application.

    :param app:
    :return:
    """
    app.register_error_handler(404, handle_corvus_404_error)
    app.register_error_handler(BaseError, handle_corvus_base_error)


corvus = create_app()  # pylint: disable=C0103
register_blueprints(corvus)
register_error_handlers(corvus)

logger = corvus.logger  # pylint: disable=C0103

if __name__ == "__main__":
    corvus.run()