Flask框架下使用Factory Boy如何从已填充表中选取关联对象

问题根因 #

你遇到的报错本质是 User.query.all() 在模块导入阶段就被直接求值执行，此时Flask应用上下文尚未推送，自然触发上下文缺失错误。Factory Boy 的 Iterator 默认接收已实例化的可迭代对象，你直接传入查询语句的执行结果，就会导致查询时机提前到导入阶段，和预期的首次调用工厂时再执行的逻辑不符。

解决方案 #

通过 lazy_attribute 装饰器或 LazyAttribute 字段将查询逻辑延迟到首次生成工厂实例时再执行，此时你已经完成了应用初始化、上下文推送、前置数据填充的操作。

方案1：内置迭代遍历所有已有数据 #

如果需要循环遍历已填充的User表所有数据作为外键关联值，可以用延迟初始化迭代器的方式实现：

import factory
from factory.alchemy import SQLAlchemyModelFactory
from your_app import db
from your_app.models import Order, User

class OrderFactory(SQLAlchemyModelFactory):
    class Meta:
        model = Order
        sqlalchemy_session = db.session

    _user_iterator = None

    @classmethod
    def _init_user_iterator(cls):
        if not cls._user_iterator:
            # 仅首次调用工厂时执行一次查询，此时上下文已就绪
            cls._user_iterator = iter(User.query.all())
        return cls._user_iterator

    @factory.lazy_attribute
    def user(self):
        return next(OrderFactory._init_user_iterator())

方案2：测试场景动态传入前置数据 #

如果是测试场景下，你已经在测试用例前置逻辑中生成了指定的测试用户列表，可以直接动态给工厂赋值迭代源，不需要额外改工厂定义：

# 测试用例内的示例逻辑
def test_order_creation(app):
    # 已完成应用上下文推送
    # 提前生成测试用用户列表
    test_users = [UserFactory.create(name=f"用户{i}") for i in range(5)]
    # 动态给OrderFactory绑定迭代源
    OrderFactory.user = factory.Iterator(test_users)

    # 生成的订单会自动按顺序关联test_users里的用户
    order1 = OrderFactory.create()
    order2 = OrderFactory.create()

注意事项 #

首次调用工厂生成实例前，必须确保已完成两个前置操作：

• 已推送Flask应用上下文
• 关联表的前置数据已填充完成

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