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')......