python - 如何使用 FactoryBoy 指定列表大小

Question

假设我有这个模型：

class FooContainerModel(object):
    def __init__(self, foos):
        self.foos = foos

我希望能够决定在创建时将创建多少个 foo，例如。

model = FooContainerFactory(count=15)

尝试过的工厂如

class FooContainerFactory(factory.Factory):
    class Meta:
        model = FooContainerModel

    foos = factory.List([Foo() for _ in xrange(20)]) # fixed amount
    foos = factory.lazy_attribute(lambda o: [Foo() for _ in xrange(20)]) # same

当然，我可以手动创建具有所需长度的 Foo() 列表并实例化 FooContainerModel，但这不是我想要的。有什么解决办法吗？

Answer 1

我不知道 FactoryBoy 是如何工作的，但是当我需要 Django 模型中的字段的动态属性时，这种策略通常对我有用：

def FooContainerFactory(factory.Factory):

    def __init__(self, count=20, *args, **kwargs):
        self.foos = factory.List([Foo() for _ in range(20)])
        super(FooContainerFactory, self).__init__(*args, **kwargs)

    class Meta:
        model = FooContainerModel

Answer 2

您可以使用create_batch方法来做到这一点。

models = FooContainerFactory.create_batch(size=15)

如果你想创建 15 个完全不同的模型，你可能需要使用Sequences和Faker（另外）