2019独角兽企业重金招聘Python工程师标准>>>
记录在第一次使用django-rest-framework框架使用时遇到的问题,为了便于理解在这里创建了Person和Grade这两个model
from django.db import models
class Person(models.Model):SHIRT_SIZES = (('S', 'Small'),('M', 'Medium'),('L', 'Large'),)name = models.CharField(max_length=60)shirt_size = models.CharField(max_length=1, choices=SHIRT_SIZES)class Grade(models.Model):score=models.CharField(max_length=60)person = models.ForeignKey(Person, on_delete=models.CASCADE)
接口返回下拉选项对应的值
在Person这个model中的shirt_size字段 存储到数据库中的值未S/M/L,但是在前端用户需要看到他们对应value,那么我们接口如何返回呢
需要在模块对应的serializer.py文件中小小的修改一下
class PersonSerializer(serializers.ModelSerializer):shirt_size_name = serializers.CharField(source='get_shirt_size_display', required=False)class Meta:model = Personfields = ('id','shirt_size', 'shirt_size_name')
接口返回外键对应的值
在Grade查询接口中返回对应外键人员的姓名,这有两种方法
- Grade在序列化的时候,就创建一个新的字段叫 person_name,指定为 serializers.CharField,而且字段使用 source 这个属性,具体而言格式为
CharField(source='<本model中的外键>.<外键指向的model的相应属性>')
```
class GradeSerializer(serializers.ModelSerializer):person_name = serializers.CharField(source='person.name'')class Meta:model = Gradefields = ('id','person_name')
```
- 在 models 里面 @property 装饰符先创建一个 person_name 函数,这个函数返回的是外键对应 model 的相应字段,比如 person.name。 然后,序列化的时候,指定为 serializers.ReadOnlyField() 类型
这种方法也可以增加给接口增加额外的字段
```
class Grade(models.Model):score=models.CharField(max_length=60)person = models.ForeignKey(Person, on_delete=models.CASCADE)......@propertydef person_name(self):return self.person.name``````
class GradeSerializer(serializers.ModelSerializer):person_name = serializers.ReadOnlyField()class Meta:model = Gradefields = ('id','person_name')
```
模糊查询
首先增加一个过滤器filters.py
class PersonFilter(django_filters.rest_framework.FilterSet):"""人员的过滤类"""name = django_filters.CharFilter(name='name', lookup_expr='icontains')class Meta:model = Personfields = ('id', 'name')
有时自带的过滤方法无法满足,就需要我们自定义过滤方法,例:
class PersonFilter(django_filters.rest_framework.FilterSet):# 自定义了过滤方法person_filterperson_name = django_filters.CharFilter(method='person_filter')# 通过传入的中文名模糊查询到对应90分数人的IDdef person_filter(self, queryset, name, value):personList = []qs = Grade.objects.filter(name__contains=value).filter(score='90')if qs.exists():for q in qs:personList.append(q.id)# 返回对应人员的查询集合return queryset.filter(id__in=userList)
然后在对应的view中配置一下ViewSet
class PersonViewSet(viewsets.ModelViewSet):"""LIST用户列表页, 分页, 搜索, 过滤, 排序""".......filter_backends = (DjangoFilterBackend, SearchFilter, OrderingFilter)filter_class = PersonFiltersearch_fields = ('id', 'name')......