权限
问题:不同视图不同权限可以访问
1.models.py
from django.db import models class UserInfo(models.Model): user_type_choices = ( (1,‘普通用户‘), (2,‘VIP‘), (3,‘SVIP‘), ) user_type = models.IntegerField(choices=user_type_choices) # username = models.CharField(max_length=32) username = models.CharField(max_length=32,unique=True) password = models.CharField(max_length=64) class UserToken(models.Model): user = models.OneToOneField(to=‘UserInfo‘) token = models.CharField(max_length=64)
2.用户类型:
3.views.py:
假设订单相关业务(只有SVIP用户有权限)
假设用户信息相关业务(只有普通用户、VIP有权限)
4.运行结果:
基本使用
以上的权限代码封装到一个类中,以后各个视图的get请求直接调用即可。不用每个类都写一遍权限。
自定义权限类: 需要找到has_permission方法
0.看源码:
1.
2.
3.复制源码has_permission方法
4.继续定义权限类:has_permission方法返回True表示有权访问。
5.定义好了权限类,局部使用。
局部权限
6.运行结果:
7.再定义一个权限,出SVIP以外,其他用户都能访问。
8.将全权都放到另一个单独的文件中。
然后视图文件需要则引入。
views.py应用即可
代码:
基本使用: class MyPermission(object): def has_permission(self,request,view): if request.user.user_type != 3: return False return True class OrderView(APIView): """ 订单相关业务(只有SVIP用户有权限) """ permission_classes = [MyPermission,] def get(self,request,*args,**kwargs): # request.user # request.auth self.dispatch ret = {‘code‘:1000,‘msg‘:None,‘data‘:None} try: ret[‘data‘] = ORDER_DICT except Exception as e: pass return JsonResponse(ret)
permission.py from rest_framework.permissions import BasePermission # 权限 class SvipPermission(BasePermission): message = "必须是SVIP才能访问" def has_permission(self,request,view): if request.user.user_type != 3: # SVIP return False # 无权访问 return True # 有权访问 class MyPermission1(BasePermission): def has_permission(self, request, view): if request.user.user_type == 3: # 普通用户、VIP return False # 无权访问 return True # 有权访问 views.py from api.utils.permission import MyPermission1 from api.utils.throttle import VisitThrottle class OrderView(APIView): """ 订单相关业务(只有SVIP用户有权限) """ # 权限 # permission_classes = [SvipPermission,] def get(self, request, *args, **kwargs): # request.user # request.auth self.dispatch # if request.user.user_type != 3: # return HttpResponse(‘无权访问‘) ret = {‘code‘:1000,‘msg‘:None,‘data‘:None} try: ret[‘data‘] = ORDER_DICT except Exception as e: pass return JsonResponse(ret) class UserInfoView(APIView): """ 用户信息相关业务(只有普通用户、VIP有权限) """ # 权限 写了就不用全局的svip权限了。 permission_classes = [MyPermission1, ] def get(self, request, *args, **kwargs): # if request.user.user_type == 3: # return HttpResponse(‘无权访问‘) return HttpResponse(‘用户信息‘)
以上的单视图应用。全局使用怎么办呢?
全局权限
1.权限源码流程:
2.
3.
4.
5.
6.
7.局部配置
8.permission.py
9.全局配置 settings.py
"DEFAULT_PERMISSION_CLASSES":[‘api.utils.permission.SvipPermission‘], # 权限
10.继续看源码:
11.每一个权限都有该方法 has_permission
12.如果返回Fasle,则执行报错,抛出异常。
permission_denied
13.抛出异常
14.默认英文报错
运行结果:
15.如何改成中文报错信息呢?
运行结果:
内置权限
1.内置权限源码流程
2.为了让我们的权限更正规,必须继承该内置权限。
代码:
from rest_framework.permissions import BasePermission # 权限 class SvipPermission(BasePermission): message = "必须是SVIP才能访问" def has_permission(self,request,view): if request.user.user_type != 3: # SVIP return False # 无权访问 return True # 有权访问 class MyPermission1(BasePermission): def has_permission(self, request, view): if request.user.user_type == 3: # 普通用户、VIP return False # 无权访问 return True # 有权访问
梳理: 1. 使用 - 类,必须继承:BasePermission,必须实现:has_permission方法 from rest_framework.permissions import BasePermission class SVIPPermission(BasePermission): message = "必须是SVIP才能访问" def has_permission(self,request,view): if request.user.user_type != 3: return False return True - 返回值: - True, 有权访问 - False,无权访问 - 局部 class UserInfoView(APIView): """ 订单相关业务(普通用户、VIP) """ permission_classes = [MyPermission1, ] def get(self,request,*args,**kwargs): return HttpResponse(‘用户信息‘) - 全局 REST_FRAMEWORK = { "DEFAULT_PERMISSION_CLASSES":[‘api.utils.permission.SVIPPermission‘] }