Django之会员权限管理vip

vip_choices = ((0,'普通用户'),(1,'vip'),(2,'svip'),(3,'sss'))

vip=models.IntegerField(choices=vip_choices，defaurt=0)

自定义权限:

from rest_framework.permissions import BasePermission


class MyPermission(BasePermission):
    message = '只有VIP才能访问'  # 错误提示 系统会自动调用  通过反射

    def has_permission(self, request, view):
        # 判断是否已经通过认证  通过 request.auth 判断 认证成功后保存的token 参数
        #  如果没有 认证  request.user  也是空的  下面的操作就会报错
        if not request.auth:
            return False
        # 如果你是VIP才有权限访问
        # request.user:当前经过认证的用户对象
        if request.user.vip:
            return True
        else:
            # 如果不是VIP就拒绝范围跟
            return False

配置权限

注册全局 权限

在settings文件中

# 如果未指定，则此设置默认允许不受限制的访问
 REST_FRAMEWORK = {
     'DEFAULT_PERMISSION_CLASSES': (
         	'auth_demo.permissions.MyPermission',
     )
 }

注册局部权限

写在对应的视图中

from auth_demo.permissions import MyPermission
from rest_framework.views import APIView

class myview(APIView):
	permission_classes = [MyPermission, ]
	def get(self,request):
		pass

局部禁用

permission_classes = []