from flask import Flask,session,g
import config
from exts import db,mail
# from blueprints.qa import bp as qa_bp
# 导入蓝图
from blueprints import qa_bp
from blueprints import user_bp
from flask_migrate import Migrate
from models import UserModel



app = Flask(__name__)
app.config.from_object(config)  # 绑定数据库
db.init_app(app)                 # 绑定类方法
mail.init_app(app)

migrate = Migrate(app,db)

# 注册蓝图
app.register_blueprint(qa_bp)
app.register_blueprint(user_bp)


@app.before_request
def before_request():
    user_id = session.get("user_id")
    if user_id:
        try:
            user = UserModel.query.get(user_id)
            # 给g绑定一个叫做user的变量，它的值是user这个变量
            # setattr(g,"user",user)
            g.user = user
        except:
            g.user = None

# 上下文处理器
@app.context_processor
def context_processor():
    if hasattr(g,"user"):
        return {"user":g.user}
    else:
        return {}

if __name__ == '__main__':
    app.run(debug=True)