#!/usr/bin/env python # -*- coding:utf-8 -*- # project: 1月 # author: NinEveN # date: 2022/1/6 from rest_framework.pagination import PageNumberPagination from rest_framework.viewsets import ModelViewSet from common.core.response import ApiResponse class AppsPageNumber(PageNumberPagination): page_size = 20 # 每页显示多少条 page_size_query_param = 'limit' # URL中每页显示条数的参数 page_query_param = 'page' # URL中页码的参数 max_page_size = 100 # 最大页码数限制 class BaseModelSet(ModelViewSet): def retrieve(self, request, *args, **kwargs): data = super().retrieve(request, *args, **kwargs).data return ApiResponse(data=data) def list(self, request, *args, **kwargs): data = super().list(request, *args, **kwargs).data return ApiResponse(data=data) def destroy(self, request, *args, **kwargs): instance = self.get_object() self.perform_destroy(instance) return ApiResponse() def update(self, request, *args, **kwargs): data = super().update(request, *args, **kwargs).data return ApiResponse(data=data)