64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
|
import os
|
|||
|
|
from flask import Flask
|
|||
|
|
from flask_sqlalchemy import SQLAlchemy
|
|||
|
|
from flask_login import LoginManager
|
|||
|
|
from flask_migrate import Migrate
|
|||
|
|
from config import config
|
|||
|
|
|
|||
|
|
# 初始化扩展
|
|||
|
|
db = SQLAlchemy()
|
|||
|
|
login_manager = LoginManager()
|
|||
|
|
migrate = Migrate()
|
|||
|
|
|
|||
|
|
def create_app(config_name='default'):
|
|||
|
|
"""应用工厂函数"""
|
|||
|
|
# 设置模板和静态文件夹
|
|||
|
|
app_dir = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
template_folder = os.path.join(app_dir, 'web', 'templates')
|
|||
|
|
static_folder = os.path.join(os.path.dirname(app_dir), 'static')
|
|||
|
|
|
|||
|
|
app = Flask(__name__,
|
|||
|
|
template_folder=template_folder,
|
|||
|
|
static_folder=static_folder)
|
|||
|
|
|
|||
|
|
# 尝试加载.env文件
|
|||
|
|
try:
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
if load_dotenv():
|
|||
|
|
print("成功加载.env文件")
|
|||
|
|
else:
|
|||
|
|
print("未找到或无法加载.env文件")
|
|||
|
|
except ImportError:
|
|||
|
|
print("python-dotenv未安装，跳过.env文件加载")
|
|||
|
|
|
|||
|
|
app.config.from_object(config[config_name])
|
|||
|
|
config[config_name].init_app(app)
|
|||
|
|
|
|||
|
|
# 特别确保MAX_CONTENT_LENGTH配置正确应用
|
|||
|
|
max_content_length = app.config.get('MAX_CONTENT_LENGTH', 16 * 1024 * 1024)
|
|||
|
|
app.config['MAX_CONTENT_LENGTH'] = max_content_length
|
|||
|
|
print(f"Setting MAX_CONTENT_LENGTH to: {max_content_length} bytes ({max_content_length / (1024*1024)} MB)")
|
|||
|
|
|
|||
|
|
# 初始化扩展
|
|||
|
|
db.init_app(app)
|
|||
|
|
login_manager.init_app(app)
|
|||
|
|
migrate.init_app(app, db)
|
|||
|
|
|
|||
|
|
# 配置登录管理器
|
|||
|
|
login_manager.login_view = 'web.login'
|
|||
|
|
login_manager.login_message = '请先登录'
|
|||
|
|
login_manager.login_message_category = 'info'
|
|||
|
|
login_manager.id_attribute = 'get_id' # 使用 get_id 方法获取用户ID
|
|||
|
|
|
|||
|
|
# 注册蓝图
|
|||
|
|
from app.api import api_bp
|
|||
|
|
app.register_blueprint(api_bp, url_prefix=f'/api/{app.config["API_VERSION"]}')
|
|||
|
|
|
|||
|
|
from app.web import web_bp
|
|||
|
|
app.register_blueprint(web_bp)
|
|||
|
|
|
|||
|
|
# 注册错误处理器
|
|||
|
|
from app.web.views import register_error_handlers
|
|||
|
|
register_error_handlers(app)
|
|||
|
|
|
|||
|
|
return app
|