KaiZhaoKZ 2020-02-14
在REST后台开发中,我们需要通过业务逻辑来选择数据集或者序列化器。
# 重写get_queryset实现通过业务逻辑选择指定数据集 def get_queryset(self): ‘‘‘ 通过前段传递过来的keyword选择指定数据集 :return: ‘‘‘ # 获取keyword keyword = self.request.query_params.get(‘keyword‘) # 通过前段传递过来的keyword选择指定数据集 if keyword: users = User.objects.filter(username__contains=keyword) else: users = User.objects.all() # 响应数据 return users
# 重写get_serializer_class方法,实现通过业务逻辑选择指定序列化器 def get_serializer_class(self): if self.request.method == ‘GET‘: return UserMSGSerializers elif self.request.method == ‘POST‘: return UserAddSerializers
我们只需要在API视图中重写以上两个方法即可达到目的。