python - 使用表条目而不是子工厂

Question

我的 MyUser 类有一个 FactoryBoy 工厂，分配了默认名称和相应的公司：

class MyUserFactory(SQLAlchemyModelFactory):      

    class Meta:                                 
        model = MyUser                
        sqlalchemy_session = db.session   

    name = 'F.Bar'
    company = factory.SubFactory(CompanyFactory)

问题：我有一个指定的可能公司表，我想用它来代替来自CompanyFactory. 有没有办法从这个表中读取一个条目来使用它company而不是 SubFactory？

Answer 1

您可以使用以下代码（请参阅http://factoryboy.readthedocs.org/en/latest/reference.html#factory.lazy_attribute）：

class MyFactory(SQLAlchemyFactory):
    @factory.lazy_attribute
    def company(self):
        return get_some_company_object()