python - 如何让 FactoryBoy 的 ImageField 在调用 save() 之前生成图像？

Question

主题。

现在(Factory Boy 版本 2.4.1.)使用此代码:

class ImageFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Image

    image = factory.django.ImageField(width=1024, height=768)

image 在保存时将为 None，因此如果 Image 模型覆盖了 save 并且其操作对于 image，它将失败。这正是我的情况。

那么 - 如何在 save 调用之前生成图像？

Answer 1

我找到了解决方法:

from django.core.files.base import ContentFile

class ImageFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Image

    image = factory.LazyAttribute(
            lambda _: ContentFile(
                factory.django.ImageField()._make_data(
                    {'width': 1024, 'height': 768}
                ), 'example.jpg'
            )
        )