有没有办法让 FactoryBoy 将参数传递给我的 Django 模型的 save() 方法?

Question

我有一个这样的Django模型：

class NicePerson(models.Model):
last_name = models.CharField(max_length=100)
def save(self, make_irish=False, *args, **kwargs):
"""if make_irish is True, prepend last_name with O'"
if make_irish:
self.last_name = "O'" + self.last_name
super(MyModel, self).save(*args, **kwargs)

我有一个 FactoryBoy 类来构建 NicePerson 实例

class NicePersonFactory(factory.django.DjangoModelFactory):
class Meta:
model = NicePerson

我知道我可以像这样一起使用它们：

nice_person = NicePersonFactory(last_name='Shea')

但是如何将"make_irish"参数传递给我的save()方法呢？

Answer 1

factory_boy使用默认管理器进行创建，即它调用NicePerson.objects.create()或NicePerson.objects.get_or_create()。

对于您的示例，您可以覆盖这些管理器(在模型定义中(：

class NicePersonManager(models.Manager):
def create(self, *args, **kwargs):
if kwargs.pop('make_irish', False):
kwargs.update(...)
return super().create(*args, **kwargs)

另一种选择是覆盖工厂的_create和_get_or_create方法。例如：

class NicePersonFactory(factory.django.DjangoModelFactory):
@classmethod
def _create(cls, model_class, *args, **kwargs):
make_irish = kwargs.pop('make_irish', False)
instance = model_class(**kwargs)
instance.save(make_irish=make_irish)
return instance