shds/bootstrap/__init__.py

import os

from dotenv import dotenv_values
from flask import Flask
from flask_cors import CORS
import algorithm


def create_app():
    app = Flask(import_name=__name__, root_path=os.path.abspath(os.path.join(os.path.dirname(__file__), ".")),
                static_folder='output',
                static_url_path='/static'
                )
    CORS(app, supports_credentials=True)

    config = dotenv_values(".env")
    app.config['APP_NAME'] = config.get('APP_NAME') if config.get('APP_NAME') is not None else 'FLASK APP'
    app.config['APP_DEBUG'] = eval(config.get('APP_DEBUG'))
    # app.config['DB_HOST'] = config.get('DB_HOST')
    # app.config['DB_PORT'] = config.get('DB_PORT')
    # app.config['DB_USERNAME'] = config.get('DB_USERNAME')
    # app.config['DB_PASSWORD'] = config.get('DB_PASSWORD')
    # app.config['DB_DATABASE'] = config.get('DB_DATABASE')

    # 启用FLASK DEBUG
    app.debug = app.config['APP_DEBUG']

    # # 初始化模型算法变量
    # algorithm.init()

    print('''
    -----------------------------
    {}
    -----------------------------
    '''.format(app.config['APP_NAME']))

    return app