如何在 ModelSerializer 的 create 方法中实现 update_or_create

Question

代码：

类 OTP（AppModel）： phone_regex = RegexValidator(regex=r'^[6789]\d{9}$', message="电话号码无效。") phone_number = models.CharField(validators=[phone_regex], max_length=10, unique=True) 代码 = models.CharField(max_length=255) def __str__(self): 返回 str(self.phone_number) + ": "+str(self.code) 类 OTPSerializer(serializers.ModelSerializer): code = serializers.CharField(max_length=None, required=False) 元类： 型号 = OTP 字段 = ('id', 'code', 'phone_number') read_only_fields=('id', 'code') @transaction.atomic 定义创建（自我，验证数据）： phone_number = 已验证数据.pop("phone_number") otp, 创建 = OTP.objects.update_or_create( phone_number=phone_number，默认值={"code": generate_otp()}) 返回 otp

我正在尝试在django-rest-framework 的ModelSerializer 的create 方法中执行update_or_create。

但是，模型OTP 中的字段phone_number 必须是unique。因此unique=True。

我能够发布phone_number 并创建对象。但是，再次发布相同的phone_number 会引发错误otp with this phone number already exists，而不是在它已经存在时更新它，因为我已经覆盖了create 方法。请帮忙！

Answer 1

您可以使用Signals 以干净的方式执行此操作。简单地说，您可以将created 变量发送到您定义的receiver，并根据您的对象是否创建来处理它。在 REST 响应的情况下，只需覆盖 Serializer 中的 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

您可以使phone_number 不需要，然后手动进行检查。您会收到错误消息，因为 DRF 在您之前验证了 phone_number。因此，基本上，解决方案可能如下（仅序列化代码）：

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