HTalk/app/__init__.py

from flask import Flask, render_template, Response
from flask.logging import default_handler
import logging
import logging.handlers
import os
import sys
from datetime import datetime

from .db import db, Role, User
from .moment import moment
from .mail import mail
from .migrate import migrate
from .login import login
from .logger import Logger

from configure import conf


if conf["DEBUG_PROFILE"]:
    from werkzeug.middleware.profiler import ProfilerMiddleware


class HTalkFlask(Flask):
    def __init__(self, name):
        super(HTalkFlask, self).__init__(name)
        self.update_configure()
        self.profile_setting()
        self.logging_setting()
        self.blueprint()

        db.init_app(self)
        moment.init_app(self)
        mail.init_app(self)
        migrate.init_app(self, db)
        login.init_app(self)

        @self.context_processor
        def inject_base():
            """ app默认模板变量 """
            return {"conf": conf,
                    "Role": Role,
                    "User": User,
                    "datetime": datetime}

        self.error_page([400, 401, 403, 404, 405, 408, 410, 413, 414, 423, 500, 501, 502])

    def blueprint(self):
        from .index import index
        self.register_blueprint(index, url_prefix="/")

        from .auth import auth
        self.register_blueprint(auth, url_prefix="/auth")

        from .comment import comment
        self.register_blueprint(comment, url_prefix="/cm")

        from .archive import archive
        self.register_blueprint(archive, url_prefix="/ac")

    def profile_setting(self):
        if conf["DEBUG_PROFILE"]:
            self.wsgi_app = ProfilerMiddleware(self.wsgi_app, sort_by=("cumtime",))

    def logging_setting(self):
        self.logger.removeHandler(default_handler)
        self.logger.setLevel(conf["LOG_LEVEL"])
        self.logger.propagate = False  # 不传递给更高级别的处理器处理日志

        if len(conf["LOG_HOME"]) > 0:
            handle = logging.handlers.TimedRotatingFileHandler(
                os.path.join(conf["LOG_HOME"], f"flask.log"), backupCount=10)
            handle.setFormatter(logging.Formatter(conf["LOG_FORMAT"]))
            self.logger.addHandler(handle)

        if conf["LOG_STDERR"]:
            handle = logging.StreamHandler(sys.stderr)
            handle.setFormatter(logging.Formatter(conf["LOG_FORMAT"]))
            self.logger.addHandler(handle)

    def update_configure(self):
        """ 更新配置 """
        self.config.update(conf)

    def error_page(self, error_code):
        for i in error_code:
            def create_error_handle(status):  # 创建一个 status 变量给 error_handle
                def error_handle(e):
                    Logger.print_load_page_log(status)
                    data = render_template('error.html', error_code=status, error_info=e)
                    return Response(response=data, status=status)
                return error_handle

            self.errorhandler(i)(create_error_handle(i))