ListAPIView
'''
重构list方法，
def list(self, request, *args, **kwargs):
response = super().list(request, *args, **kwargs)
response 父类list方法的返回值Response(serializer.data),即序列化后的数据serializer.data
response.data = context 给response添加键值对
return response 再将新的response返回给前端
'''
父类list方法
class ListModelMixin:
"""
List a queryset.
"""
def list(self, request, *args, **kwargs):
queryset = self.filter_queryset(self.get_queryset())
page = self.paginate_queryset(queryset)
if page is not None:
serializer = self.get_serializer(page, many=True)
return self.get_paginated_response(serializer.data)
serializer = self.get_serializer(queryset, many=True)
return Response(serializer.data)
重构list方法
class BidView(ListAPIView, CreateAPIView):
authentication_classes = [UserAuthentication, ]
permission_classes = [BidPermission, ]
queryset = models.BidRecord.objects.all().order_by('-id')
serializer_class = BidModelSerializer
filter_backends = [BidItemFilter, ]
def list(self, request, *args, **kwargs):
response = super().list(request, *args, **kwargs)
if response.status_code != status.HTTP_200_OK:
return response
lot_id = request.query_params.get('lot_id')
lot_object = models.Lot.objects.filter(id=lot_id).first()
context = {
'unit': lot_object.unit,
'price': response.data[0]['price'] if response.data else lot_object.start_price,
'bid_list': response.data
}
response.data = context
return response
CreateAPIView
'''
重构perform_create方法
前端数据校验成功后，执行perform_create(self, serializer)方法保存数据到数据库
'''
父类list方法
class CreateModelMixin:
"""
Create a model instance.
"""
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
self.perform_create(serializer)
headers = self.get_success_headers(serializer.data)
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
def perform_create(self, serializer):
serializer.save()
def get_success_headers(self, data):
try:
return {'Location': str(data[api_settings.URL_FIELD_NAME])}
except (TypeError, KeyError):
return {}
class BidView(ListAPIView, CreateAPIView):
def perform_create(self, serializer):
serializer.save(user_id=self.request.user.id, status=1)
RetrieveAPIView
'''
重构get方法/retrieve方法
def get(self, request, *args, **kwargs):
# 1. 获取详细信息
response = self.retrieve(request, *args, **kwargs)
同ListAPIView相似
'''
父类方法
class RetrieveAPIView(mixins.RetrieveModelMixin,
GenericAPIView):
"""
Concrete view for retrieving a model instance.
"""
def get(self, request, *args, **kwargs):
return self.retrieve(request, *args, **kwargs)
class RetrieveModelMixin:
"""
Retrieve a model instance.
"""
def retrieve(self, request, *args, **kwargs):
instance = self.get_object()
serializer = self.get_serializer(instance)
return Response(serializer.data)
重构get方法/retrieve方法
class NewsDetailView(RetrieveAPIView):
"""
获取动态详细接口
"""
queryset = models.News.objects
serializer_class = RetrieveNewsDetailModelSerializerSerializer
def get(self, request, *args, **kwargs):
# 1. 获取详细信息
response = self.retrieve(request, *args, **kwargs)
# 2. 处理用户浏览记录，当前用户添加到记录中
# 2.1 用户未登录，不记录
# 2.2 用户登录，已记录则不再记录记录，未记录则添加到浏览记录中。
if not request.user:
return response
news_object = self.get_object()
if not news_object:
return response
viewer_queryset = models.ViewerRecord.objects.filter(news=news_object, user_id=request.user.id)
if not viewer_queryset.exists():
models.ViewerRecord.objects.create(news=news_object, user_id=request.user.id)
models.News.objects.filter(id=news_object.id).update(favor_count=F("viewer_count") + 1)
return response
DestroyAPIView
'''
啦啦啦，没用过。。。，用过补充
'''
父类方法
class DestroyModelMixin:
"""
Destroy a model instance.
"""
def destroy(self, request, *args, **kwargs):
instance = self.get_object()
self.perform_destroy(instance)
return Response(status=status.HTTP_204_NO_CONTENT)
def perform_destroy(self, instance):
instance.delete()
UpdateAPIView
'''
啦啦啦，没用过。。。，用过补充
'''
父类方法
class UpdateAPIView(mixins.UpdateModelMixin,
GenericAPIView):
"""
Concrete view for updating a model instance.
"""
def put(self, request, *args, **kwargs):
return self.update(request, *args, **kwargs)
def patch(self, request, *args, **kwargs):
return self.partial_update(request, *args, **kwargs)
class UpdateModelMixin:
"""
Update a model instance.
"""
def update(self, request, *args, **kwargs):
partial = kwargs.pop('partial', False)
instance = self.get_object()
serializer = self.get_serializer(instance, data=request.data, partial=partial)
serializer.is_valid(raise_exception=True)
self.perform_update(serializer)
if getattr(instance, '_prefetched_objects_cache', None):
# If 'prefetch_related' has been applied to a queryset, we need to
# forcibly invalidate the prefetch cache on the instance.
instance._prefetched_objects_cache = {}
return Response(serializer.data)
def perform_update(self, serializer):
serializer.save()
def partial_update(self, request, *args, **kwargs):
kwargs['partial'] = True
return self.update(request, *args, **kwargs)