Flask-Migrate与Flask-Fixtures配合时数据加载失败问题求助

问题 #

我是Flask和Postgres新手，现在需要在测试用Postgres Docker容器里，用一个函数同时运行Flask-Migrate和Flask-Fixtures：前者负责升级数据库表结构、关系，后者从YAML文件加载测试数据。

遇到的问题：用已预建表的Postgres数据库时，Flask-Fixtures能正常加载数据；但用空容器加上Flask-Migrate的升级代码后，会触发SQLAlchemy完整性错误：

sqlalchemy.exc.IntegrityError: (psycopg2.errors.NotNullViolation) null value in column "id" of relation "accounts" violates not-null

奇怪的是，第一次运行报错后，注释掉Migrate相关代码再跑，Flask-Fixtures又能正常加载数据。我怀疑是Migrate(app, db)这块导致数据库没法正常加载数据，需要排查问题并优化代码。

正常加载数据的代码 #

def _load_db_fixtures(self, input_db_fixture_path, db_migrations_directory):
    app = Flask(__name__)
    app.config['SQLALCHEMY_DATABASE_URI'] = "postgresql://test:test@postgres:5432/test"
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
    db = SQLAlchemy(app)
    db.metadata.reflect(bind=db.engine)

    def _reload_fixture(fixture_file):
        """Reload fixtures."""
        import flask_fixtures
        import yaml
        from yaml import SafeLoader
        meta = db.metadata
        session = db.session
        with session.begin(nested=True), open(fixture_file, mode='r') as fp:
            flask_fixtures.load_fixtures(db, yaml.load(fp, SafeLoader))
        return "Reloaded"
    _reload_fixture(input_db_fixture_path) 

添加Migrate后报错的代码 #

def _load_db_fixtures(self, input_db_fixture_path, db_migrations_directory):
    app = Flask(__name__)
    app.config['SQLALCHEMY_DATABASE_URI'] = "postgresql://test:test@postgres:5432/test"
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
    db = SQLAlchemy(app)
    db.metadata.reflect(bind=db.engine)
    migrate = Migrate(app, db)
    with app.app_context():
        flask_migrate.upgrade(directory = db_migrations_directory)
    
    def _reload_fixture(fixture_file):
        """Reload fixtures."""
        import flask_fixtures
        import yaml
        from yaml import SafeLoader
        meta = db.metadata
        session = db.session
        with session.begin(nested=True), open(fixture_file, mode='r') as fp:
            flask_fixtures.load_fixtures(db, yaml.load(fp, SafeLoader))
        return "Reloaded"
    _reload_fixture(input_db_fixture_path) 

解决方案 #

问题根源 #

1. 元数据加载时机错误：你在执行db.metadata.reflect(bind=db.engine)之后才运行migrate升级，此时数据库还是空的，根本反射不到任何表结构。Flask-Fixtures依赖正确的元数据识别表的主键、字段约束，导致加载数据时无法自动生成id（如果你的fixture里没写id的话），触发非空约束错误。
2. 元数据未同步：Migrate升级后，数据库表结构已更新，但之前反射的元数据还是空状态，没有同步新结构，Flask-Fixtures用旧元数据处理自然出问题。

优化后的代码 #

调整步骤顺序，先执行migrate升级，再反射最新元数据：

def _load_db_fixtures(self, input_db_fixture_path, db_migrations_directory):
    app = Flask(__name__)
    app.config['SQLALCHEMY_DATABASE_URI'] = "postgresql://test:test@postgres:5432/test"
    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
    
    # 先初始化db和migrate
    db = SQLAlchemy(app)
    migrate = Migrate(app, db)
    
    # 在上下文里执行migrate升级，创建表结构
    with app.app_context():
        flask_migrate.upgrade(directory=db_migrations_directory)
        # 升级完成后，立即反射最新的数据库元数据
        db.metadata.reflect(bind=db.engine)
    
    def _reload_fixture(fixture_file):
        """Reload fixtures."""
        import flask_fixtures
        import yaml
        from yaml import SafeLoader
        meta = db.metadata
        session = db.session
        with session.begin(nested=True), open(fixture_file, mode='r') as fp:
            flask_fixtures.load_fixtures(db, yaml.load(fp, SafeLoader))
        return "Reloaded"
    
    _reload_fixture(input_db_fixture_path) 

额外注意点 #

• 如果YAML fixture里没显式指定id值，要确保表的id字段是自增主键（比如模型里定义为db.Column(db.Integer, primary_key=True, autoincrement=True)），这样Flask-Fixtures才能触发自增逻辑生成id。
• 每次migrate升级后必须重新反射元数据，否则后续操作会基于旧的空元数据执行，导致各种约束错误。

内容的提问来源于stack exchange，提问作者Aidos Kenessov