权限类(BasePermission)
与认证一样,同样创建一个py文件,负责放校验权限的类
from rest_framework.permissions import BasePermission
导入模块后,写一个类来继承。与认证一样,总得来说就是继承后重写,然后到视图层用。
permission_classes = [UserPermission]
这是重写的权限类,首先当然是继承BasePermission,然后request.user.user_type取出对应的user_type值,因为我想让我写了一些接口只让超级管理员使用,所以我设定了超级管理员的user_type为3
这样,只有user_type为3的用户才能进入,返回True,不为超级管理员返回False.
我在视图层随便写了个接口,然后校验权限,注意⚠️:因为在上面认证里面我设置了全局配置,所以需要先登录才能进入,当然,也必须要登录,因为不登录它获取不到你的user_type的值,还有,记得加token。
当然,还得再改改,不然会报错就很难受,加一个判断是否登录:request.user.is_authenticated
想要自定义返回的数据:
那么就需要在重写权限的地方改改
只要使用的choice,想要拿到对应的中文,就可以用:
get_字段名_display()
以上是局部使用,想要全局使用如下:
局部禁用如下
就是设置它为空
permission_classes = []