REST framework:过滤

一、数据准备

模型类

class UserInfo(models.Model):
    name = models.CharField(max_length=20, verbose_name='用户名')
    pwd = models.CharField(max_length=18, verbose_name='密码')
    email = models.EmailField(max_length=40, verbose_name='邮箱')
    age = models.IntegerField(verbose_name='年龄', default=18)

    def __str__(self):
        return self.name

    class Meta:
        db_table = 't_user'


# 模型Addr关联userinfo，
class Addr(models.Model):
    user = models.ForeignKey('UserInfo', verbose_name='所属用户', on_delete=models.CASCADE)
    mobile = models.CharField(verbose_name='手机号', max_length=18)
    city = models.CharField(verbose_name='城市', max_length=10)
    info = models.CharField(verbose_name='详细地址', max_length=200)

    def __str__(self):
        return self.info

    class Meta:
        db_table = 't_address'

序列化器类

class UserInfoSerializer(serializers.ModelSerializer):

    class Meta:
        model = UserInfo
        fields = '__all__'


class AddrSerializer(serializers.ModelSerializer):

    class Meta:
        model = Addr
        fields = '__all__'

视图类

class UserView(ModelViewSet):
    # 指定模型类查询集
    queryset = UserInfo.objects.all()
    # 指定序列化器类
    serializer_class = UserInfoSerializerclass AddrView(ModelViewSet):
    # 指定模型类查询集
    queryset = Addr.objects.all()
    # 指定序列化器类
    serializer_class = AddrSerializer

二、相关配置

对于列表数据可能需要根据字段进行过滤，我们可以通过添加django-fitlter扩展来增强支持

pip install django-filter -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

在配置文件中增加过滤后端的设置

 # 注册应用，
INSTALLED_APPS = [
    ...
    'django_filters', 
]
# 指定过滤器
REST_FRAMEWORK = {
    'DEFAULT_FILTER_BACKENDS': ('django_filters.rest_framework.DjangoFilterBackend',)
}

 三、过滤器的基本使用

在视图中添加filterset_fields属性，指定可以过滤的字段

class UserView(ModelViewSet):
    # 指定模型类查询集
    queryset = UserInfo.objects.all()
    # 指定序列化器类
    serializer_class = UserInfoSerializer

    filterset_fields = ('name', 'email',)

指定过滤字段之前

指定过滤字段之后