代码之家 › 专栏 › 技术社区 › Sreekanth Reddy Balne

如何在ModelSerializer的create方法中实现update_或_create

django-rest-framework django-models django python

Sreekanth Reddy Balne · 技术社区 · 6 年前

代码:

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

我正试着去做 update_or_create 里面 create 方法 django-rest-framework 的 ModelSerializer 。

但是,这个领域 phone_number 模型内部 OTP 必须是 unique . 因此 unique=True .

我能发布一个 电话号码 然后创建对象。但是,张贴同样的 电话号码 再次抛出错误 otp with this phone number already exists ,而不是更新它,如果它已经存在,因为我已经重写了 创造 方法。请帮助!

1 回复 | 直到 6 年前

pomo_mondreganto 6 年前

你可以 phone_number 不需要,然后手动进行检查。你得到了错误,因为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

推荐文章

Florent · 如何在Django中聚合多个字段?

1 年前

KWunsch · Django没有将pk引入模板

1 年前

Akshay mek · Vscode扩展建议不适用于某些文件夹

1 年前

Farid · 限制django每个客户的访问

2 年前

Shanmukh Adari · 是否可以更改Django模型中的内置验证器消息

2 年前

Hoshmand Qadir · 将数据库中的特定项提取到Django模板中(如果它是特定类型的)

2 年前

Maruf Khan · GitHub操作无法找到Django的环境变量

2 年前

Blue · 从Django中的输入字段获取用户输入

2 年前

4SadMemeXD · 无法在Django中选择相关类别

2 年前

JasonTS · 如何在Django中应用与m2m关系的对象相关的select_?[副本]

2 年前