如何在ModelSerializer的创建方法中实现update_or_create

Question

守则：

class OTP(AppModel):
    phone_regex = RegexValidator(regex=r'^[6789]\d{9}$', message="phone no. is invalid.")
    phone_number = models.CharField(validators=[phone_regex], max_length=10, unique=True)
    code = models.CharField(max_length=255)

    def __str__(self):
        return str(self.phone_number) + ": "+str(self.code)

class OTPSerializer(serializers.ModelSerializer):
    code = serializers.CharField(max_length=None, required=False)
    class Meta:
        model = OTP
        fields = ('id', 'code', 'phone_number')
        read_only_fields=('id', 'code')

    @transaction.atomic
    def create(self, validated_data):
        phone_number = validated_data.pop("phone_number")
        otp, created = OTP.objects.update_or_create(
            phone_number=phone_number, defaults={"code": generate_otp()})
        return otp

我试图在django rest框架的ModelSerializer的create方法中执行update\u或\u create。

但是，型号OTP内的phone\u number字段必须是唯一的。因此，unique=True。

我可以发布一个电话号码并创建对象。但是，再次发布相同的电话号码会抛出错误otp，因为此电话号码已经存在，而不是更新它（如果它已经存在），因为我已经覆盖了create方法。请帮忙！




Answer 1

您可以使用信号以干净的方式执行此操作。简单地说，您可以将创建的变量发送到您定义的接收器，并根据对象是否创建来处理它。在REST响应的情况下，只需重写序列化程序中的create方法，根据您所处的状态返回数据，或者重写APIView中使用的get/post/patch方法，不返回Serializer.data，而是返回您想要的任何数据。以下是信号接收器的示例：

@receiver(post_save, sender=settings.OTP_MODEL)
def update(sender, instance=None, created=False, **kwargs):
    if created:
        # Do Something
    else:
        # Do Some Other thing

Answer 2

尝试在序列化程序类中而不是在模型类中设置验证器。让序列化程序类看起来像这样：

class OTPSerializer(serializers.ModelSerializer):
    code = serializers.CharField(max_length=None, required=False)
    phone_regex = RegexValidator(regex=r'^[6789]\d{9}$', message="phone no. is invalid.")  # add this line
    phone_number = serializers.CharField(validators=[phone_regex])  # and this line
    
    class Meta:
        model = OTP
        fields = ('id', 'code', 'phone_number')
        read_only_fields=('id', 'code')

    @transaction.atomic
    def create(self, validated_data):
        phone_number = validated_data.pop("phone_number")
        otp, created = OTP.objects.update_or_create(
            phone_number=phone_number, defaults={"code": generate_otp()})
        return otp

Answer 3

您可以将电话号码设置为不需要，然后手动进行检查。您会得到错误，因为DRF在您验证之前验证了电话号码。因此，基本上，解决方案可以是以下（仅限serialiser代码）：

class OTPSerializer(serializers.ModelSerializer):
    code = serializers.CharField(max_length=None, required=False)
    class Meta:
        model = OTP
        fields = ('id', 'code', 'phone_number')
        read_only_fields=('id', 'code')
        extra_kwargs = {'phone_number': {'required': False}}

    @transaction.atomic
    def create(self, validated_data):
        phone_number = validated_data.pop("phone_number")
        otp, created = OTP.objects.update_or_create(
        phone_number=phone_number, defaults={"code": generate_otp()})
        return otp