python - 重写_create类方法

Question

我正在使用 django，并且有一个自定义方法来创建特定模型的新对象:

class DiseaseFactory(factory.django.DjangoModelFactory):

    class Meta:
        model = models.Disease

    def _create(cls, model_class, *args, **kwargs):
        obj = model_class.create_new(first_name=f"Disease{uuid.uuid4()}")
        return obj

当我尝试使用DiseaseFactory.create()创建DiseaseFactory的实例时，它会抛出错误:

TypeError: _create() missing 1 required positional argument: 'model_class'

我做错了什么？

Answer 1

为了使其工作，您需要添加 @classmethod 装饰器，如下所示:

class DiseaseFactory(factory.django.DjangoModelFactory):

    class Meta:
        model = models.Disease

    @classmethod
    def _create(cls, model_class, *args, **kwargs):
        obj = model_class.create_new(first_name=f"Disease{uuid.uuid4()}")
        return obj