app.py

import os

from flask import Flask, jsonify, make_response
from flask_restful import Api
from werkzeug.exceptions import HTTPException
from werkzeug.exceptions import default_exceptions
from dotenv import load_dotenv

from config import APP_CONFIG
from endpoints import RESOURCES
from utils.start_up import init_redis

load_dotenv()


def create_app(config_mode):

    app = Flask(__name__)

    # pylint: disable=W0612
    @app.after_request
    def af_request(resp):
        resp = make_response(resp)
        resp.headers["Access-Control-Allow-Origin"] = "*"
        resp.headers["Access-Control-Allow-Methods"] = "GET,POST,PUT,DELETE,OPTIONS"
        resp.headers["Access-Control-Allow-Headers"] = "X-Requested-With, Content-Type, Authorization"
        return resp

    # pylint: enable=W0612

    @app.errorhandler(Exception)
    def handle_error(error):

        code = 500

        if isinstance(error, HTTPException):
            code = error.code

        # pylint: disable=E1101
        app.logger.warning(f"{code} - {error}")
        # pylint: enable=E1101

        return jsonify(error=str(error)), code

    for ex in default_exceptions:
        app.register_error_handler(ex, handle_error)

    # flask config
    app.config.from_object(APP_CONFIG[config_mode])
    # pylint: disable=E1101
    app.logger.info(f"APP Mode: {config_mode}")
    # pylint: enable=E1101

    # Route Init
    api = Api(app)
    for resource in RESOURCES:
        api.add_resource(RESOURCES[resource], f"/{resource}")

    # run when app start
    with app.app_context():
        init_redis()

    return app


def main():
    config_name = os.environ.get("APP_SETTINGS", "development")
    app = create_app(config_name)
    app.run(host=app.config["HOST"], port=app.config["PORT"], threaded=True)


if __name__ == "__main__":
    main()