120 lines
4.4 KiB
Python
120 lines
4.4 KiB
Python
# Flask 应用工厂
|
|
|
|
from flask import Flask
|
|
from app.models import db
|
|
import os
|
|
import pathlib
|
|
|
|
|
|
def create_app():
|
|
app = Flask(__name__)
|
|
|
|
# 使用 run.py 的目录作为项目根目录（绝对路径，不依赖工作目录）
|
|
import sys
|
|
if hasattr(sys, '_MEIPASS'):
|
|
BASE_DIR = pathlib.Path(sys._MEIPASS)
|
|
else:
|
|
BASE_DIR = pathlib.Path(__file__).resolve().parent.parent
|
|
|
|
# 存储到 app.config，各处引用，不再各自计算
|
|
app.config["BASE_DIR"] = BASE_DIR
|
|
app.config["SECRET_KEY"] = "piano-practice-plan-secret-key-2026"
|
|
app.config["DEBUG"] = True
|
|
app.config["SQLALCHEMY_DATABASE_URI"] = (
|
|
f"sqlite:///{BASE_DIR / 'data' / 'piano_plans.db'}"
|
|
)
|
|
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
|
|
|
|
# 问题文件目录
|
|
is_docker = os.environ.get("FLASK_ENV") == "production"
|
|
if is_docker:
|
|
app.config["PROBLEMS_DIR"] = BASE_DIR / "个性化方案"
|
|
else:
|
|
# 本地开发：问题文件在父目录的 个性化方案/针对性练习（拆分为单独文件）
|
|
app.config["PROBLEMS_DIR"] = BASE_DIR.parent / "个性化方案" / "针对性练习（拆分为单独文件）"
|
|
|
|
app.config["PDF_OUTPUT_DIR"] = BASE_DIR / "output"
|
|
app.config["API_CONFIG_FILE"] = BASE_DIR / "config" / "api_config.json"
|
|
|
|
# 初始化数据库
|
|
db.init_app(app)
|
|
|
|
# 注册蓝图
|
|
from app.routes import main_bp
|
|
from app.routes.templates import templates_bp
|
|
|
|
app.register_blueprint(main_bp)
|
|
app.register_blueprint(templates_bp)
|
|
|
|
# 创建数据库和目录
|
|
with app.app_context():
|
|
os.makedirs(os.path.join(BASE_DIR, "data"), exist_ok=True)
|
|
os.makedirs(app.config["PDF_OUTPUT_DIR"], exist_ok=True)
|
|
db.create_all()
|
|
|
|
# 简单迁移：为已存在的数据库添加新字段（必须在init_default_templates之前）
|
|
try:
|
|
from sqlalchemy import text
|
|
|
|
# 检查practice_time字段是否存在
|
|
result = db.session.execute(text("PRAGMA table_info(students)"))
|
|
columns = [row[1] for row in result]
|
|
if "practice_time" not in columns:
|
|
db.session.execute(
|
|
text(
|
|
"ALTER TABLE students ADD COLUMN practice_time VARCHAR(20) DEFAULT '30-60分钟'"
|
|
)
|
|
)
|
|
db.session.commit()
|
|
if "wechat_nickname" not in columns:
|
|
db.session.execute(
|
|
text("ALTER TABLE students ADD COLUMN wechat_nickname VARCHAR(100)")
|
|
)
|
|
db.session.commit()
|
|
|
|
# 检查student_problems表的level字段
|
|
result2 = db.session.execute(text("PRAGMA table_info(student_problems)"))
|
|
columns2 = [row[1] for row in result2]
|
|
if "level" not in columns2:
|
|
db.session.execute(
|
|
text("ALTER TABLE student_problems ADD COLUMN level VARCHAR(20)")
|
|
)
|
|
db.session.commit()
|
|
|
|
# 检查users表是否存在
|
|
result3 = db.session.execute(
|
|
text(
|
|
"SELECT name FROM sqlite_master WHERE type='table' AND name='users'"
|
|
)
|
|
)
|
|
if not result3.fetchone():
|
|
db.session.execute(
|
|
text(
|
|
"""
|
|
CREATE TABLE users (
|
|
id INTEGER PRIMARY KEY,
|
|
username VARCHAR(50) UNIQUE NOT NULL,
|
|
password_hash VARCHAR(200) NOT NULL,
|
|
role VARCHAR(20) DEFAULT 'user',
|
|
created_at TIMESTAMP
|
|
)
|
|
"""
|
|
)
|
|
)
|
|
db.session.commit()
|
|
|
|
# 检查templates表是否有sort_order字段
|
|
result4 = db.session.execute(text("PRAGMA table_info(templates)"))
|
|
template_columns = [row[1] for row in result4]
|
|
if "sort_order" not in template_columns:
|
|
db.session.execute(text("ALTER TABLE templates ADD COLUMN sort_order INTEGER DEFAULT 0"))
|
|
db.session.commit()
|
|
except Exception as e:
|
|
print(f"数据库迁移: {e}")
|
|
|
|
# 初始化默认模板（必须在迁移之后）
|
|
from app.routes.templates import init_default_templates
|
|
init_default_templates()
|
|
|
|
return app
|