一个简单的rest_framework demo

models.py

from django.db import models
 
class UserInfo(models.Model):
    username = models.CharField(max_length=32)
    password = models.CharField(max_length=32)
    role_choice = ((1,'管理员'),(2,'普通用户'))
    role = models.IntegerField(choices=role_choice)
 
class BWH(models.Model):
    '''三围信息'''
    bust = models.IntegerField(verbose_name='胸围')
    waist = models.IntegerField(verbose_name='腰围')
    hips = models.IntegerField(verbose_name='臀围')
 
    user = models.OneToOneField(to=UserInfo)

serializers.py

from rest_framework import serializers
from app01 import models
 
class BWHModelSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.BWH
        fields = '__all__'
 
 
class UserModelSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.UserInfo
        fields = '__all__'

permission.py

from rest_framework.permissions import BasePermission
from app01 import models
 
class AdminPermission(BasePermission):
    '''管理员权限认证'''
    message = '您没有权限访问！'
    def has_permission(self, request, view):
        user_id = request.GET.get('uid')
        user = models.UserInfo.objects.filter(pk=user_id).first()
 
        if not user or user.role == 2:
            return False
        else:
            return True

auth.py

from rest_framework.authentication import BaseAuthentication
from rest_framework.exceptions import AuthenticationFailed
from app01 import models
 
class BlackNameAuth(BaseAuthentication):
    '''黑名单认证'''
 
    def authenticate(self, request):
        BLACK_NAME_LIST = ['小花', '小翠']
 
        # 通过从url获取user_id的方式模拟用户登录
        user_id = request.GET.get('uid')
        user = models.UserInfo.objects.filter(pk=user_id).first()
 
        if not user or user.username in BLACK_NAME_LIST:
            raise AuthenticationFailed('您没有登录或者被关小黑屋啦')
        else:
            return user.username,user_id

urls.py

from django.conf.urls import url,include
from django.contrib import admin
from rest_framework import routers
from app01 import views
 
# 实例化一个routers对象
routers = routers.DefaultRouter()
# 往对象里注册（添加）url
routers.register('bwh',views.BWHViewSet)
routers.register('user',views.UserViewSet)
 
urlpatterns = [
    url(r'^admin/', admin.site.urls),
    url(r'',include(routers.urls))
]

view.py

from app01.serializers import BWHModelSerializer
from app01.serializers import UserModelSerializer
from rest_framework.viewsets import ModelViewSet
from app01 import models
from app01.auth import BlackNameAuth
from app01.permission import AdminPermission
from rest_framework.pagination import PageNumberPagination
 
class MyPagination(PageNumberPagination):
    '''分页，自定义每页显示2条'''
    page_size = 2
 
 
 
class BWHViewSet(ModelViewSet):
    '''在黑名单中的用户禁止查看三围信息'''
    authentication_classes = [BlackNameAuth]
 
    # 分页
    pagination_class = MyPagination
 
    serializer_class = BWHModelSerializer
    queryset = models.BWH.objects.all()
 
 
 
class UserViewSet(ModelViewSet):
    '''非管理员禁止查看用户信息'''
    permission_classes = [AdminPermission]
 
    # 分页
    pagination_class = MyPagination
 
    serializer_class = UserModelSerializer
    queryset = models.UserInfo.objects.all()

posted on 2019-05-14 10:12 斜阳红红阅读(205) 评论(0) 收藏举报