web-gpt/project/__init__.py

from flask import Flask, request, send_from_directory
from flask_sqlalchemy import SQLAlchemy
from flask_login import LoginManager
import yaml
import logging
import time
import traceback
import os

# init SQLAlchemy so we can use it later in our models
db = SQLAlchemy()


def create_app():
    app = Flask(__name__)

    print("project instance dir: ", app.instance_path)
    config_file = app.instance_path+"/config.yaml"
    print("config file path: ", config_file)
    with open(config_file) as config_file:
        conf = yaml.safe_load(config_file)
        app.config['SECRET_KEY'] = conf['app']['SECRET_KEY']
        app.config['LOGGING_LEVEL'] = conf['app']['LOGGING_LEVEL']
        app.config['SQLALCHEMY_DATABASE_URI'] = conf['app']['SQLALCHEMY_DATABASE_URI']
        app.config['HOMEPAGE_NOTICE'] = conf['app']['HOMEPAGE_NOTICE']
        app.config['NETWORK_PROXY'] = conf['network']['PROXY']
        app.config['OPENAI_API_KEY'] = conf['openai']['API_KEY']
        app.config['OPENAI_MODEL_NAME'] = conf['openai']['MODEL_NAME']
        app.config['OPENAI_PROMPT'] = conf['openai']['PROMPT']

    db.init_app(app)
    login_manager = LoginManager()
    login_manager.login_view = 'auth.login'
    login_manager.init_app(app)
    login_manager.login_message = "请先登录"

    from .models import User

    @login_manager.user_loader
    def load_user(user_id):
        # since the user_id is just the primary key of our user table, use it in the query for the user
        return User.query.get(int(user_id))

    with app.app_context():
        db.create_all()

    # blueprint for auth routes in our app
    from .auth import auth as auth_blueprint
    app.register_blueprint(auth_blueprint)

    # blueprint for non-auth parts of app
    from .main import main as main_blueprint
    app.register_blueprint(main_blueprint)

    return app


app = create_app()
logger = logging.getLogger('waitress')
logger.setLevel(app.config['LOGGING_LEVEL'])


def get_actual_addr(request):
    ip = request.remote_addr
    if request.headers.getlist("X-Forwarded-For"):
        ip = request.headers.getlist("X-Forwarded-For")[0]
    return ip


@app.route('/favicon.ico')
def favicon():
    return send_from_directory(os.path.join(app.root_path, 'static'),
                               'favicon.ico', mimetype='image/vnd.microsoft.icon')


@app.before_request
def before_request():
    timestamp = time.strftime('[%Y-%b-%d %H:%M]')
    logger.info('%s > %s %s %s %s', timestamp, get_actual_addr(request),
                request.method, request.scheme, request.full_path)


@app.after_request
def after_request(response):
    timestamp = time.strftime('[%Y-%b-%d %H:%M]')
    logger.info('%s < %s %s %s %s %s', timestamp, get_actual_addr(request),
                request.method, request.scheme, request.full_path, response.status)
    return response


@app.errorhandler(Exception)
def exceptions(e):
    tb = traceback.format_exc()
    timestamp = time.strftime('[%Y-%b-%d %H:%M]')
    logger.error('%s %s %s %s %s 5xx INTERNAL SERVER ERROR\n%s', timestamp,
                 get_actual_addr(request), request.method, request.scheme, request.full_path, tb)
    return e.status_code