如何使用Flask-Migrate迁移其他文件夹多文件中的Flask-SQLAlchemy模型

问题根源 #

1. 多SQLAlchemy实例冲突：每个模型文件自行创建db = SQLAlchemy()，导致这些模型不属于app.py中初始化的db实例，Flask-Migrate无法识别。
2. 模型未被加载：app.py未导入任何外部模型，SQLAlchemy无法自动发现这些模型类。

修复步骤 #

1. 统一模型中的db实例 #

修改所有模型文件（model1.py、model2.py、model3.py），删除自行创建的db对象，改为从app.py导入统一的db实例：

以model1.py为例：

from sqlalchemy.sql import func
# 替换原有的db创建代码，导入app中的统一db实例
from app import db

class Student(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    firstname = db.Column(db.String(100), nullable=False)
    lastname = db.Column(db.String(100), nullable=False)
    email = db.Column(db.String(80), unique=True, nullable=False)
    age = db.Column(db.Integer)
    created_at = db.Column(db.DateTime(timezone=True),
                           server_default=func.now())
    bio = db.Column(db.Text)

    def __repr__(self):
        return f'<Student {self.firstname}>'

2. 在app.py中显式导入所有模型 #

在app.py中migrate初始化完成后，添加所有模型的导入语句，确保模型被加载到内存中：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate


app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = "My_Database_Url"
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False


db = SQLAlchemy(app)
migrate = Migrate(app, db)

# 导入所有模型，让SQLAlchemy注册这些模型类
from API.models.model1 import Student
from API.models.model2 import 你的Model2类名
from API.models.model3 import 你的Model3类名

@app.route('/')
def hello():
    return 'Hello, World!'

3. 重新执行迁移命令 #

如果之前的migrations目录为测试用空白目录，可重置迁移环境后执行：

# 删除旧迁移目录（如有重要迁移记录请跳过此步）
rm -rf migrations

# 初始化迁移环境
flask db init

# 生成迁移脚本
flask db migrate -m "初始化模型迁移"

# 将迁移应用到数据库
flask db upgrade

若保留旧migrations目录，直接执行flask db migrate即可，此时应能检测到模型变化。

内容的提问来源于stack exchange，提问作者Ankit prajapat