MRE/app/__init__.py

from flask import Flask
import os
from app.MRE import blueprint as mre_blueprint
from .api import blueprint as api_blueprint
from logging import Logger, Formatter, getLogger, DEBUG, INFO, FileHandler


def load_config_from_environ(app: Flask) -> Flask:
    for env_var in filter(lambda x: x.startswith(app.name.upper()), os.environ.keys()):
        config_name = env_var[len(app.name):]
        app.config[config_name] = os.environ[env_var]

    return app


def init_logger(app: Flask, log_level=INFO) -> Flask:
    logger = getLogger(app.name)
    formatter = Formatter(f'[%(asctime)s] {app.name.upper()} %(levelname)s: %(message)s (%(filename)s.%(funcName)s)')
    logger.setLevel(log_level)
    to_file = FileHandler(os.environ.get('MRE_LOG_PATH', f'./{app.name}.log'))
    to_file.setFormatter(formatter)
    logger.addHandler(to_file)

    app.logger = logger
    return app


def create_app(app_name: str='MRE') -> Flask:
    app = Flask(app_name)
    app = load_config_from_environ(app)
    app = init_logger(app)

    app.register_blueprint(mre_blueprint)
    app.register_blueprint(api_blueprint)

    return app