Django-rest-framework 接口实现 权限:(Permission)

时间：2021-04-16 00:07:26

权限:(Permission)

rest_framework 中 已经定义好了 权限

在 from rest_framework.permissions 中查看所有的权限

  • from rest_framework.permissions import BasePermission

自定义一个权限

要在 用户的 model 中添加一个字段 vip 字段

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

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

定义 一个中间件 permissions.py 定义一个类

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

配置 权限

# 注册全局 权限

'DEFAULT_PERMISSION_CLASSES': ['auth_demo.permissions.MyPermission', ]

# 注册局部权限

# 写在对应的视图中

from auth_demo.permissions import MyPermission

permission_classes = [MyPermission, ]

相关文章