from rest_framework.pagination import PageNumberPagination
class MyPageNumberPagination(PageNumberPagination):
page_size = 3 # 每页显示3条
max_page_size = 5 # 每页最大显示5条
新建serializer.py分页
python复制代码
from rest_framework import serializers
from app01.models import User, CarModel, CarFactory, Distributor
class UserSerializer(serializers.ModelSerializer):
class Meta:
model = User
fields = ['username', 'password', 'mobile']
extra_kwargs = {
'password': {'write_only': True}
}
def create(self, validated_data):
user = User.objects.create_user(**validated_data)
return user
class CarModelSerializer(serializers.ModelSerializer):
class Meta:
model = CarModel
fields = ['id', 'name', 'init_price', 'factory', 'distributors', 'factory_info', 'distributor_info']
extra_kwargs = {
'factory': {'write_only': True},
'distributors': {'write_only': True},
'factory_info': {'read_only': True},
'distributor_info': {'read_only': True},
}
class CarFactorySerializer(serializers.ModelSerializer):
class Meta:
model = CarFactory
fields = '__all__'
class DistributorSerializer(serializers.ModelSerializer):
class Meta:
model = Distributor
fields = '__all__'
新建permission.py权限
python复制代码
from rest_framework.permissions import BasePermission
from rest_framework.exceptions import AuthenticationFailed
class MyPermission(BasePermission):
def has_permission(self, request, view):
print(view.action)
if not request.user.is_superuser and request.method != 'GET':
raise AuthenticationFailed('普通用户，权限不足')
return True
views.py视图中
python复制代码
from rest_framework.viewsets import ViewSet
from rest_framework.generics import GenericAPIView
from rest_framework.mixins import CreateModelMixin
from .models import User
from .serializer import UserSerializer
from rest_framework.response import Response
class UserView(ViewSet, GenericAPIView, CreateModelMixin):
queryset = User.objects.all()
serializer_class = UserSerializer
def create(self, request, *args, **kwargs):
ser = self.get_serializer(data=request.data)
if ser.is_valid():
ser.save()
return Response({'code': 100, 'msg': '注册成功', 'result': ser.data})
return Response({'code': 101, 'msg': '注册失败', 'result': ser.errors})
from rest_framework.mixins import DestroyModelMixin
from rest_framework_jwt.authentication import JSONWebTokenAuthentication
from rest_framework.permissions import IsAuthenticated, IsAdminUser
from rest_framework.decorators import action
class AdminView(ViewSet, GenericAPIView, DestroyModelMixin):
queryset = User.objects.all()
authentication_classes = [JSONWebTokenAuthentication]
permission_classes = [IsAuthenticated, IsAdminUser]
# detail=True 表示可以删除单个资源
@action(methods=['DELETE'], detail=True)
def delete_user(self, request, *args, **kwargs):
return self.destroy(request, *args, **kwargs)
@action(methods=['GET'], detail=True)
def lock(self, request, *args, **kwargs):
user = self.get_object()
if user is None:
return Response({'code': 101, 'msg': '用户不存在'})
if user.is_active:
user.is_active = False
user.save()
return Response({'code': 100, 'msg': '用户锁定成功'})
return Response({'code': 102, 'msg': '用户已经被锁定'})
@action(methods=['GET'], detail=True)
def unlock(self, request, *args, **kwargs):
user = self.get_object()
if user is None:
return Response({'code': 101, 'msg': '用户不存在'})
if user.is_active is False:
user.is_active = True
user.save()
return Response({'code': 100, 'msg': '用户解锁成功'})
return Response({'code': 102, 'msg': '用户已经解锁过了'})
from rest_framework.viewsets import ModelViewSet
from .models import CarModel
from .serializer import CarModelSerializer
from .permissions import MyPermission
from .page import MyPageNumberPagination
from django_filters.rest_framework import DjangoFilterBackend
class CarModelView(ModelViewSet):
queryset = CarModel.objects.all()
serializer_class = CarModelSerializer
# 认证
authentication_classes = [JSONWebTokenAuthentication]
# 权限
permission_classes = [IsAuthenticated, MyPermission]
# 分页
pagination_class = MyPageNumberPagination
# 过滤
filter_backends = [DjangoFilterBackend]
filterset_fields = ['name']
from .models import CarFactory
from .serializer import CarFactorySerializer
class CarFactoryView(ModelViewSet):
queryset = CarFactory.objects.all()
serializer_class = CarFactorySerializer
pagination_class = MyPageNumberPagination
authentication_classes = [JSONWebTokenAuthentication]
permission_classes = [IsAuthenticated, MyPermission]
from .models import Distributor
from .serializer import DistributorSerializer
class DistributorView(ModelViewSet):
queryset = Distributor.objects.all()
serializer_class = DistributorSerializer
pagination_class = MyPageNumberPagination
authentication_classes = [JSONWebTokenAuthentication]
permission_classes = [IsAuthenticated, MyPermission]
urls.py路由中
python复制代码
from django.contrib import admin
from django.urls import path, include
from rest_framework_jwt.views import obtain_jwt_token
from rest_framework.routers import SimpleRouter
from app01.views import UserView, AdminView, CarModelView, CarFactoryView, DistributorView
router = SimpleRouter()
router.register('user', UserView, 'user')
router.register('admin', AdminView, 'admin')
router.register('carMode', CarModelView, 'carMode')
router.register('carFactory', CarFactoryView, 'carFactory')
router.register('distributor', DistributorView, 'distributor')
urlpatterns = [
path('login/', obtain_jwt_token),
path('', include(router.urls))
]