使用表条目而不是subfactory

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()