使用表条目而不是子工厂

Question

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

class MyUserFactory(SQLAlchemyModelFactory):      

    class Meta:                                 
        model = MyUser                
        sqlalchemy_session = db.session   

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

问题：我有一个指定的可能公司表，我想用它来代替CompanyFactory中的虚拟公司。有没有办法从这个表中读取一个条目来将它用于company而不是子工厂？你知道吗

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