【django framework】ModelSerializer+GenericAPIView，如何在提交前修改某些字段值

【django framework】ModelSerializer+GenericAPIView，如何在提交前修改某些字段值

我们经常会遇到下面这种情况：
序列化器用的是ModelSerializer，写视图的时候继承的是generics.CreateAPIView。现在我想在正式提交到数据库(perform_create)之前，修改几个字段的值，应该怎么去做呢？

---

---

---

1 在Serializer序列化器中重写create

我们知道序列化器承担了部分视图的作用，但是它主要的作用有两个：
① 对序列化进来的数据进行校验，比如validate一下手机号格式是不是正确
② 写入一些默认或者不需要前端传递的字段
// 如果说执意要在Serializer中写业务逻辑，也可以。只不过这种写法会和rest_framework的设计理念相违背，增加了序列化器和视图的耦合性。

from rest_framework import serializers
from .models import *
import re

class YourModelSerializer(serializers.ModelSerializer):
    """
    YourModel 创建 序列化器
    """

    class Meta:
        model = YourModel
        fields = '__all__'

    def validate(self, kwargs):
    """
    校验
    校验前端传过来的用户手机号格式对不对
    """

    MOBILE = kwargs.get('mobile').mobile
    REGEX_MOBILE = '1[3475698]\d{9}$|^147\d{8}$|^176\d{8}|^198\d{8}$'

    # 客户手机号格式
    if not re.match(REGEX_MOBILE, MOBILE):
        raise serializers.ValidationError("客户手机号格式有误")
    
    return kwargs


    def create(self, validated_data):
        """
        重写create方法
        // 添加唯一序列号
        序列号不需要前端传递
        """

        # 唯一序列号
        validated_data['serial'] = 'serial102356478'

		# 可以在此写业务逻辑，但是并不推荐

        return super().create(validated_data)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

---

2 继承generics.CreateAPIView后，重写perform_create()

秉承着谁的活由谁来干的原则，第一种方法虽然可行，但是增加了序列化器与视图之间的耦合性，所以并不是一个好方法。
所以我们通过重写perform_create()方法来修改我们想要修改的字段。
而视图本就是写业务逻辑的，这也符合rest_framework设计初衷。

from rest_framework import generics
from .models import YourModel
from .serializers import YourModelSerializer

class YourModelCreateAPIView(generics.CreateAPIView):
    queryset = YourModel.objects.all()
    serializer_class = YourModelSerializer

    def perform_create(self, serializer):
        # 获取需要修改的字段的值
        some_field_value = serializer.validated_data.get('field_name')
        
        # 修改字段的值
        modified_field_value = some_field_value + '_modified'
        
        # 更新字段的值
        serializer.validated_data['some_field'] = modified_field_value
        
        # 继续调用serializer的save方法保存对象
        serializer.save()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

如此，即完成了功能，又不破坏框架功能模块间的耦合性，而代码的质量也由此而提升，皆大欢喜。