32 lines
830 B
Python
32 lines
830 B
Python
from flask import Flask
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_login import LoginManager
|
|
from config import config
|
|
|
|
# 初始化扩展
|
|
db = SQLAlchemy()
|
|
login_manager = LoginManager()
|
|
|
|
def create_app(config_name=None):
|
|
app = Flask(__name__)
|
|
|
|
# 加载配置
|
|
config_name = config_name or 'default'
|
|
app.config.from_object(config[config_name])
|
|
|
|
# 初始化扩展
|
|
db.init_app(app)
|
|
login_manager.init_app(app)
|
|
login_manager.login_view = 'auth.login'
|
|
login_manager.login_message = '请先登录访问此页面'
|
|
login_manager.login_message_category = 'info'
|
|
|
|
# 注册蓝图
|
|
from app.routes.auth import auth_bp
|
|
from app.routes.main import main_bp
|
|
|
|
app.register_blueprint(main_bp)
|
|
app.register_blueprint(auth_bp, url_prefix='/auth')
|
|
|
|
return app
|