目录html
1、认证python
2、权限django
3、限制访问频率api
4、总结缓存
1、认证(补充的一个点)app
认证请求头ide
1 #!/usr/bin/env python 2 # -*- coding:utf-8 -*- 3 from rest_framework.views import APIView 4 from rest_framework.response import Response 5 from rest_framework.authentication import BaseAuthentication 6 from rest_framework.permissions import BasePermission 7 8 from rest_framework.request import Request 9 from rest_framework import exceptions 10 11 token_list = [ 12 'sfsfss123kuf3j123', 13 'asijnfowerkkf9812', 14 ] 15 16 17 class TestAuthentication(BaseAuthentication): 18 def authenticate(self, request): 19 """ 20 用户认证,若是验证成功后返回元组: (用户,用户Token) 21 :param request: 22 :return: 23 None,表示跳过该验证; 24 若是跳过了全部认证,默认用户和Token和使用配置文件进行设置 25 self._authenticator = None 26 if api_settings.UNAUTHENTICATED_USER: 27 self.user = api_settings.UNAUTHENTICATED_USER() # 默认值为:匿名用户 28 else: 29 self.user = None 30 31 if api_settings.UNAUTHENTICATED_TOKEN: 32 self.auth = api_settings.UNAUTHENTICATED_TOKEN()# 默认值为:None 33 else: 34 self.auth = None 35 (user,token)表示验证经过并设置用户名和Token; 36 AuthenticationFailed异常 37 """ 38 val = request.query_params.get('token') 39 if val not in token_list: 40 raise exceptions.AuthenticationFailed("用户认证失败") 41 42 return ('登陆用户', '用户token') 43 44 def authenticate_header(self, request): 45 """ 46 Return a string to be used as the value of the `WWW-Authenticate` 47 header in a `401 Unauthenticated` response, or `None` if the 48 authentication scheme should return `403 Permission Denied` responses. 49 """ 50 pass 51 52 53 class TestPermission(BasePermission): 54 message = "权限验证失败" 55 56 def has_permission(self, request, view): 57 """ 58 判断是否有权限访问当前请求 59 Return `True` if permission is granted, `False` otherwise. 60 :param request: 61 :param view: 62 :return: True有权限;False无权限 63 """ 64 if request.user == "管理员": 65 return True 66 67 # GenericAPIView中get_object时调用 68 def has_object_permission(self, request, view, obj): 69 """ 70 视图继承GenericAPIView,并在其中使用get_object时获取对象时,触发单独对象权限验证 71 Return `True` if permission is granted, `False` otherwise. 72 :param request: 73 :param view: 74 :param obj: 75 :return: True有权限;False无权限 76 """ 77 if request.user == "管理员": 78 return True 79 80 81 class TestView(APIView): 82 # 认证的动做是由request.user触发 83 authentication_classes = [TestAuthentication, ] 84 85 # 权限 86 # 循环执行全部的权限 87 permission_classes = [TestPermission, ] 88 89 def get(self, request, *args, **kwargs): 90 # self.dispatch 91 print(request.user) 92 print(request.auth) 93 return Response('GET请求,响应内容') 94 95 def post(self, request, *args, **kwargs): 96 return Response('POST请求,响应内容') 97 98 def put(self, request, *args, **kwargs): 99 return Response('PUT请求,响应内容')
1 # 2 class MyAuthtication(BasicAuthentication): 3 def authenticate(self, request): 4 token = request.query_params.get('token') #注意是没有GET的,用query_params表示 5 if token == 'zxxzzxzc': 6 return ('uuuuuu','afsdsgdf') #返回user,auth 7 # raise AuthenticationFailed('认证错误') #只要抛出认证错误这样的异常就会去执行下面的函数 8 raise APIException('认证错误') 9 def authenticate_header(self, request): #认证不成功的时候执行 10 return 'Basic reala="api"' 11 12 class UserView(APIView): 13 authentication_classes = [MyAuthtication,] 14 def get(self,request,*args,**kwargs): 15 print(request.user) 16 print(request.auth) 17 return Response('用户列表')
2、权限函数
一、需求:Host是匿名用户和用户都能访问 #匿名用户的request.user = none;User只有注册用户能访问源码分析
1 from app03 import views 2 from django.conf.urls import url 3 urlpatterns = [ 4 # django rest framework 5 url('^auth/', views.AuthView.as_view()), 6 url(r'^hosts/', views.HostView.as_view()), 7 url(r'^users/', views.UsersView.as_view()), 8 url(r'^salary/', views.SalaryView.as_view()), 9 ]
1 class SalaryView(APIView): 2 '''用户能访问''' 3 message ='无权访问' 4 authentication_classes = [MyAuthentication,] #验证是否是用户 5 permission_classes = [MyPermission,AdminPermission,] #再看用户有没有权限,若是有权限在判断有没有管理员的权限 6 def get(self,request): 7 return Response('薪资列表') 8 9 def permission_denied(self, request, message=None): 10 """ 11 If request is not permitted, determine what kind of exception to raise. 12 """ 13 if request.authenticators and not request.successful_authenticator: 14 '''若是没有经过认证,而且权限中return False了,就会报下面的这个异常了''' 15 raise exceptions.NotAuthenticated(detail='无权访问') 16 raise exceptions.PermissionDenied(detail=message)
1 from django.shortcuts import render 2 from rest_framework.views import APIView #继承的view 3 from rest_framework.response import Response #友好的返回 4 from rest_framework.authentication import BaseAuthentication #认证的类 5 from rest_framework.authentication import BasicAuthentication 6 from rest_framework.permissions import BasePermission 7 from app01 import models 8 from rest_framework import exceptions 9 from rest_framework.permissions import AllowAny #权限在这个类里面 10 from rest_framework.throttling import BaseThrottle,SimpleRateThrottle 11 # Create your views here. 12 # +++++++++++++++认证类和权限类======================== 13 class MyAuthentication(BaseAuthentication): 14 def authenticate(self, request): 15 token = request.query_params.get('token') 16 obj = models.UserInfo.objects.filter(token=token).first() 17 if obj : #若是认证成功,返回用户名和auth 18 return (obj.username,obj) 19 return None #若是没有认证成功就不处理,进行下一步 20 21 def authenticate_header(self, request): 22 pass 23 24 class MyPermission(BasePermission): 25 message = '无权访问' 26 def has_permission(self,request,view): #has_permission里面的self是view视图对象 27 if request.user: 28 return True #若是不是匿名用户就说明有权限 29 return False #不然无权限 30 31 class AdminPermission(BasePermission): 32 message = '无权访问' 33 def has_permission(self, request, view): # has_permission里面的self是view视图对象 34 if request.user=='haiyun': 35 return True # 返回True表示有权限 36 return False #返回False表示无权限 37 38 # +++++++++++++++++++++++++++ 39 class AuthView(APIView): 40 authentication_classes = [] #认证页面不须要认证 41 42 def get(self,request): 43 self.dispatch 44 return '认证列表' 45 46 class HostView(APIView): 47 '''需求: 48 Host是匿名用户和用户都能访问 #匿名用户的request.user = none 49 User只有注册用户能访问 50 ''' 51 authentication_classes = [MyAuthentication,] 52 permission_classes = [] #都能访问就不必设置权限了 53 def get(self,request): 54 print(request.user) 55 print(request.auth) 56 print(111111) 57 return Response('主机列表') 58 59 class UsersView(APIView): 60 '''用户能访问,request.user里面有值''' 61 authentication_classes = [MyAuthentication,] 62 permission_classes = [MyPermission,AdminPermission] 63 def get(self,request): 64 print(request.user,'111111111') 65 return Response('用户列表') 66 67 def permission_denied(self, request, message=None): 68 """ 69 If request is not permitted, determine what kind of exception to raise. 70 """ 71 if request.authenticators and not request.successful_authenticator: 72 '''若是没有经过认证,而且权限中return False了,就会报下面的这个异常了''' 73 raise exceptions.NotAuthenticated(detail='无权访问22222') 74 raise exceptions.PermissionDenied(detail=message) 75 76 77 class SalaryView(APIView): 78 '''用户能访问''' 79 message ='无权访问' 80 authentication_classes = [MyAuthentication,] #验证是否是用户 81 permission_classes = [MyPermission,AdminPermission,] #再看用户有没有权限,若是有权限在判断有没有管理员的权限 82 def get(self,request): 83 return Response('薪资列表') 84 85 def permission_denied(self, request, message=None): 86 """ 87 If request is not permitted, determine what kind of exception to raise. 88 """ 89 if request.authenticators and not request.successful_authenticator: 90 '''若是没有经过认证,而且权限中return False了,就会报下面的这个异常了''' 91 raise exceptions.NotAuthenticated(detail='无权访问') 92 raise exceptions.PermissionDenied(detail=message)
若是赶上这样的,还能够自定制,参考源码post
def check_permissions(self, request): """ Check if the request should be permitted. Raises an appropriate exception if the request is not permitted. """ for permission in self.get_permissions(): #循环每个permission对象,调用has_permission #若是False,则抛出异常 #True 说明有权访问 if not permission.has_permission(request, self): self.permission_denied( request, message=getattr(permission, 'message', None) )
def permission_denied(self, request, message=None): """ If request is not permitted, determine what kind of exception to raise. """ if request.authenticators and not request.successful_authenticator: '''若是没有经过认证,而且权限中return False了,就会报下面的这个异常了''' raise exceptions.NotAuthenticated() raise exceptions.PermissionDenied(detail=message)
那么咱们能够重写permission_denied这个方法,以下:
1 class UsersView(APIView): 2 '''用户能访问,request.user里面有值''' 3 authentication_classes = [MyAuthentication,] 4 permission_classes = [MyPermission,] 5 def get(self,request): 6 return Response('用户列表') 7 8 def permission_denied(self, request, message=None): 9 """ 10 If request is not permitted, determine what kind of exception to raise. 11 """ 12 if request.authenticators and not request.successful_authenticator: 13 '''若是没有经过认证,而且权限中return False了,就会报下面的这个异常了''' 14 raise exceptions.NotAuthenticated(detail='无权访问') 15 raise exceptions.PermissionDenied(detail=message)
2. 全局使用
上述操做中均是对单独视图进行特殊配置,若是想要对全局进行配置,则须要再配置文件中写入便可。
1 REST_FRAMEWORK = { 2 'UNAUTHENTICATED_USER': None, 3 'UNAUTHENTICATED_TOKEN': None, #将匿名用户设置为None 4 "DEFAULT_AUTHENTICATION_CLASSES": [ 5 "app01.utils.MyAuthentication", 6 ], 7 'DEFAULT_PERMISSION_CLASSES':[ 8 "app03.utils.MyPermission",#设置路径, 9 ] 10 }
1 class AuthView(APIView): 2 authentication_classes = [] #认证页面不须要认证 3 4 def get(self,request): 5 self.dispatch 6 return '认证列表' 7 8 class HostView(APIView): 9 '''需求: 10 Host是匿名用户和用户都能访问 #匿名用户的request.user = none 11 User只有注册用户能访问 12 ''' 13 authentication_classes = [MyAuthentication,] 14 permission_classes = [] #都能访问就不必设置权限了 15 def get(self,request): 16 print(request.user) 17 print(request.auth) 18 return Response('主机列表') 19 20 class UsersView(APIView): 21 '''用户能访问,request.user里面有值''' 22 authentication_classes = [MyAuthentication,] 23 permission_classes = [MyPermission,] 24 def get(self,request): 25 print(request.user,'111111111') 26 return Response('用户列表') 27 28 def permission_denied(self, request, message=None): 29 """ 30 If request is not permitted, determine what kind of exception to raise. 31 """ 32 if request.authenticators and not request.successful_authenticator: 33 '''若是没有经过认证,而且权限中return False了,就会报下面的这个异常了''' 34 raise exceptions.NotAuthenticated(detail='无权访问') 35 raise exceptions.PermissionDenied(detail=message) 36 37 38 class SalaryView(APIView): 39 '''用户能访问''' 40 message ='无权访问' 41 authentication_classes = [MyAuthentication,] #验证是否是用户 42 permission_classes = [MyPermission,AdminPermission,] #再看用户有没有权限,若是有权限在判断有没有管理员的权限 43 def get(self,request): 44 return Response('薪资列表') 45 46 def permission_denied(self, request, message=None): 47 """ 48 If request is not permitted, determine what kind of exception to raise. 49 """ 50 if request.authenticators and not request.successful_authenticator: 51 '''若是没有经过认证,而且权限中return False了,就会报下面的这个异常了''' 52 raise exceptions.NotAuthenticated(detail='无权访问') 53 raise exceptions.PermissionDenied(detail=message)
3、限流
一、为何要限流呢?
答:
只要你想爬,迟早有一天能够爬。
二、限制访问频率源码分析
1 self.check_throttles(request)
1 def check_throttles(self, request): 2 """ 3 Check if request should be throttled. 4 Raises an appropriate exception if the request is throttled. 5 """ 6 for throttle in self.get_throttles(): 7 #循环每个throttle对象,执行allow_request方法 8 # allow_request: 9 #返回False,说明限制访问频率 10 #返回True,说明不限制,通行 11 if not throttle.allow_request(request, self): 12 self.throttled(request, throttle.wait()) 13 #throttle.wait()表示还要等多少秒就能访问了
1 def get_throttles(self): 2 """ 3 Instantiates and returns the list of throttles that this view uses. 4 """ 5 #返回对象 6 return [throttle() for throttle in self.throttle_classes]
1 throttle_classes = api_settings.DEFAULT_THROTTLE_CLASSES
1 class BaseThrottle(object): 2 """ 3 Rate throttling of requests. 4 """ 5 6 def allow_request(self, request, view): 7 """ 8 Return `True` if the request should be allowed, `False` otherwise. 9 """ 10 raise NotImplementedError('.allow_request() must be overridden') 11 12 def get_ident(self, request): 13 """ 14 Identify the machine making the request by parsing HTTP_X_FORWARDED_FOR 15 if present and number of proxies is > 0. If not use all of 16 HTTP_X_FORWARDED_FOR if it is available, if not use REMOTE_ADDR. 17 """ 18 xff = request.META.get('HTTP_X_FORWARDED_FOR') 19 remote_addr = request.META.get('REMOTE_ADDR') 20 num_proxies = api_settings.NUM_PROXIES 21 22 if num_proxies is not None: 23 if num_proxies == 0 or xff is None: 24 return remote_addr 25 addrs = xff.split(',') 26 client_addr = addrs[-min(num_proxies, len(addrs))] 27 return client_addr.strip() 28 29 return ''.join(xff.split()) if xff else remote_addr 30 31 def wait(self): 32 """ 33 Optionally, return a recommended number of seconds to wait before 34 the next request. 35 """ 36 return None
1 zz
1 def throttled(self, request, wait): 2 """ 3 If request is throttled, determine what kind of exception to raise. 4 """ 5 raise exceptions.Throttled(wait)
1 class Throttled(APIException): 2 status_code = status.HTTP_429_TOO_MANY_REQUESTS 3 default_detail = _('Request was throttled.') 4 extra_detail_singular = 'Expected available in {wait} second.' 5 extra_detail_plural = 'Expected available in {wait} seconds.' 6 default_code = 'throttled' 7 8 def __init__(self, wait=None, detail=None, code=None): 9 if detail is None: 10 detail = force_text(self.default_detail) 11 if wait is not None: 12 wait = math.ceil(wait) 13 detail = ' '.join(( 14 detail, 15 force_text(ungettext(self.extra_detail_singular.format(wait=wait), 16 self.extra_detail_plural.format(wait=wait), 17 wait)))) 18 self.wait = wait 19 super(Throttled, self).__init__(detail, code)
下面来看看最简单的从源码中分析的示例,这只是举例说明了一下
1 from django.conf.urls import url 2 from app04 import views 3 urlpatterns = [ 4 url('limit/',views.LimitView.as_view()), 5 6 ]
1 from django.shortcuts import render 2 from rest_framework.views import APIView 3 from rest_framework.response import Response 4 from rest_framework import exceptions 5 # from rest_framewor import 6 # Create your views here. 7 class MyThrottle(object): 8 def allow_request(self,request,view): 9 #返回False,限制 10 #返回True,不限制 11 pass 12 def wait(self): 13 return 1000 14 15 16 class LimitView(APIView): 17 authentication_classes = [] #不让认证用户 18 permission_classes = [] #不让验证权限 19 throttle_classes = [MyThrottle, ] 20 def get(self,request): 21 # self.dispatch 22 return Response('控制访问频率示例') 23 24 def throttled(self, request, wait): 25 '''可定制方法设置中文错误''' 26 # raise exceptions.Throttled(wait) 27 class MyThrottle(exceptions.Throttled): 28 default_detail = '请求被限制' 29 extra_detail_singular = 'Expected available in {wait} second.' 30 extra_detail_plural = 'Expected available in {wait} seconds.' 31 default_code = '还须要再等{wait}秒' 32 raise MyThrottle(wait)
三、需求:对匿名用户进行限制,每一个用户一分钟容许访问10次(只针对用户来讲)
a、基于用户IP限制访问频率
流程分析:
具体实现:
1 from django.shortcuts import render 2 from rest_framework.views import APIView 3 from rest_framework.response import Response 4 from rest_framework import exceptions 5 from rest_framework.throttling import BaseThrottle,SimpleRateThrottle #限制访问频率 6 import time 7 # Create your views here. 8 RECORD = {} 9 class MyThrottle(BaseThrottle): 10 11 def allow_request(self,request,view): 12 '''对匿名用户进行限制,每一个用户一分钟访问10次 ''' 13 ctime = time.time() 14 ip = '1.1.1.1' 15 if ip not in RECORD: 16 RECORD[ip] = [ctime] 17 else: 18 #[152042123,15204212,3152042,123152042123] 19 time_list = RECORD[ip] #获取ip里面的值 20 while True: 21 val = time_list[-1]#取出最后一个时间,也就是访问最先的时间 22 if (ctime-60)>val: #吧时间大于60秒的给剔除了 23 time_list.pop() 24 #剔除了以后timelist里面就是有效的时间了,在进行判断他的访问次数是否是超过10次 25 else: 26 break 27 if len(time_list) >10: 28 return False # 返回False,限制 29 time_list.insert(0, ctime) 30 return True #返回True,不限制 31 32 def wait(self): 33 ctime = time.time() 34 first_in_time = RECORD['1.1.1.1'][-1] 35 wt = 60-(ctime-first_in_time) 36 return wt 37 38 39 class LimitView(APIView): 40 authentication_classes = [] #不让认证用户 41 permission_classes = [] #不让验证权限 42 throttle_classes = [MyThrottle, ] 43 def get(self,request): 44 # self.dispatch 45 return Response('控制访问频率示例') 46 47 def throttled(self, request, wait): 48 '''可定制方法设置中文错误''' 49 # raise exceptions.Throttled(wait) 50 class MyThrottle(exceptions.Throttled): 51 default_detail = '请求被限制' 52 extra_detail_singular = 'Expected available in {wait} second.' 53 extra_detail_plural = 'Expected available in {wait} seconds.' 54 default_code = '还须要再等{wait}秒' 55 raise MyThrottle(wait)
1 # from django.shortcuts import render 2 # from rest_framework.views import APIView 3 # from rest_framework.response import Response 4 # from rest_framework import exceptions 5 # from rest_framework.throttling import BaseThrottle,SimpleRateThrottle #限制访问频率 6 # import time 7 # # Create your views here. 8 # RECORD = {} 9 # class MyThrottle(BaseThrottle): 10 # 11 # def allow_request(self,request,view): 12 # '''对匿名用户进行限制,每一个用户一分钟访问10次 ''' 13 # ctime = time.time() 14 # ip = '1.1.1.1' 15 # if ip not in RECORD: 16 # RECORD[ip] = [ctime] 17 # else: 18 # #[152042123,15204212,3152042,123152042123] 19 # time_list = RECORD[ip] #获取ip里面的值 20 # while True: 21 # val = time_list[-1]#取出最后一个时间,也就是访问最先的时间 22 # if (ctime-60)>val: #吧时间大于60秒的给剔除了 23 # time_list.pop() 24 # #剔除了以后timelist里面就是有效的时间了,在进行判断他的访问次数是否是超过10次 25 # else: 26 # break 27 # if len(time_list) >10: 28 # return False # 返回False,限制 29 # time_list.insert(0, ctime) 30 # return True #返回True,不限制 31 # 32 # def wait(self): 33 # ctime = time.time() 34 # first_in_time = RECORD['1.1.1.1'][-1] 35 # wt = 60-(ctime-first_in_time) 36 # return wt 37 # 38 # 39 # class LimitView(APIView): 40 # authentication_classes = [] #不让认证用户 41 # permission_classes = [] #不让验证权限 42 # throttle_classes = [MyThrottle, ] 43 # def get(self,request): 44 # # self.dispatch 45 # return Response('控制访问频率示例') 46 # 47 # def throttled(self, request, wait): 48 # '''可定制方法设置中文错误''' 49 # # raise exceptions.Throttled(wait) 50 # class MyThrottle(exceptions.Throttled): 51 # default_detail = '请求被限制' 52 # extra_detail_singular = 'Expected available in {wait} second.' 53 # extra_detail_plural = 'Expected available in {wait} seconds.' 54 # default_code = '还须要再等{wait}秒' 55 # raise MyThrottle(wait) 56 57 58 59 from django.shortcuts import render 60 from rest_framework.views import APIView 61 from rest_framework.response import Response 62 from rest_framework import exceptions 63 from rest_framework.throttling import BaseThrottle,SimpleRateThrottle #限制访问频率 64 import time 65 # Create your views here. 66 RECORD = {} 67 class MyThrottle(BaseThrottle): 68 69 def allow_request(self,request,view): 70 '''对匿名用户进行限制,每一个用户一分钟访问10次 ''' 71 ctime = time.time() 72 self.ip =self.get_ident(request) 73 if self.ip not in RECORD: 74 RECORD[self.ip] = [ctime] 75 else: 76 #[152042123,15204212,3152042,123152042123] 77 time_list = RECORD[self.ip] #获取ip里面的值 78 while True: 79 val = time_list[-1]#取出最后一个时间,也就是访问最先的时间 80 if (ctime-60)>val: #吧时间大于60秒的给剔除了 81 time_list.pop() 82 #剔除了以后timelist里面就是有效的时间了,在进行判断他的访问次数是否是超过10次 83 else: 84 break 85 if len(time_list) >10: 86 return False # 返回False,限制 87 time_list.insert(0, ctime) 88 return True #返回True,不限制 89 90 def wait(self): 91 ctime = time.time() 92 first_in_time = RECORD[self.ip][-1] 93 wt = 60-(ctime-first_in_time) 94 return wt 95 96 97 class LimitView(APIView): 98 authentication_classes = [] #不让认证用户 99 permission_classes = [] #不让验证权限 100 throttle_classes = [MyThrottle, ] 101 def get(self,request): 102 # self.dispatch 103 return Response('控制访问频率示例') 104 105 def throttled(self, request, wait): 106 '''可定制方法设置中文错误''' 107 # raise exceptions.Throttled(wait) 108 class MyThrottle(exceptions.Throttled): 109 default_detail = '请求被限制' 110 extra_detail_singular = 'Expected available in {wait} second.' 111 extra_detail_plural = 'Expected available in {wait} seconds.' 112 default_code = '还须要再等{wait}秒' 113 raise MyThrottle(wait)
b、用resetframework内部的限制访问频率(利于Django缓存)
源码分析:
from rest_framework.throttling import BaseThrottle,SimpleRateThrottle #限制访问频率
1 class BaseThrottle(object): 2 """ 3 Rate throttling of requests. 4 """ 5 6 def allow_request(self, request, view): 7 """ 8 Return `True` if the request should be allowed, `False` otherwise. 9 """ 10 raise NotImplementedError('.allow_request() must be overridden') 11 12 def get_ident(self, request): #惟一标识 13 """ 14 Identify the machine making the request by parsing HTTP_X_FORWARDED_FOR 15 if present and number of proxies is > 0. If not use all of 16 HTTP_X_FORWARDED_FOR if it is available, if not use REMOTE_ADDR. 17 """ 18 xff = request.META.get('HTTP_X_FORWARDED_FOR') 19 remote_addr = request.META.get('REMOTE_ADDR') #获取IP等 20 num_proxies = api_settings.NUM_PROXIES 21 22 if num_proxies is not None: 23 if num_proxies == 0 or xff is None: 24 return remote_addr 25 addrs = xff.split(',') 26 client_addr = addrs[-min(num_proxies, len(addrs))] 27 return client_addr.strip() 28 29 return ''.join(xff.split()) if xff else remote_addr 30 31 def wait(self): 32 """ 33 Optionally, return a recommended number of seconds to wait before 34 the next request. 35 """ 36 return None
1 class SimpleRateThrottle(BaseThrottle): 2 """ 3 一个简单的缓存实现,只须要` get_cache_key() `。被覆盖。 4 速率(请求/秒)是由视图上的“速率”属性设置的。类。该属性是一个字符串的形式number_of_requests /期。 5 周期应该是:(的),“秒”,“M”,“min”,“h”,“小时”,“D”,“一天”。 6 之前用于节流的请求信息存储在高速缓存中。 7 A simple cache implementation, that only requires `.get_cache_key()` 8 to be overridden. 9 10 The rate (requests / seconds) is set by a `rate` attribute on the View 11 class. The attribute is a string of the form 'number_of_requests/period'. 12 13 Period should be one of: ('s', 'sec', 'm', 'min', 'h', 'hour', 'd', 'day') 14 15 Previous request information used for throttling is stored in the cache. 16 """ 17 cache = default_cache 18 timer = time.time 19 cache_format = 'throttle_%(scope)s_%(ident)s' 20 scope = None 21 THROTTLE_RATES = api_settings.DEFAULT_THROTTLE_RATES 22 23 def __init__(self): 24 if not getattr(self, 'rate', None): 25 self.rate = self.get_rate() 26 self.num_requests, self.duration = self.parse_rate(self.rate) 27 28 def get_cache_key(self, request, view):#这个至关因而一个半成品,咱们能够来补充它 29 """ 30 Should return a unique cache-key which can be used for throttling. 31 Must be overridden. 32 33 May return `None` if the request should not be throttled. 34 """ 35 raise NotImplementedError('.get_cache_key() must be overridden') 36 37 def get_rate(self): 38 """ 39 Determine the string representation of the allowed request rate. 40 """ 41 if not getattr(self, 'scope', None): 42 msg = ("You must set either `.scope` or `.rate` for '%s' throttle" % 43 self.__class__.__name__) 44 raise ImproperlyConfigured(msg) 45 46 try: 47 return self.THROTTLE_RATES[self.scope] 48 except KeyError: 49 msg = "No default throttle rate set for '%s' scope" % self.scope 50 raise ImproperlyConfigured(msg) 51 52 def parse_rate(self, rate): 53 """ 54 Given the request rate string, return a two tuple of: 55 <allowed number of requests>, <period of time in seconds> 56 """ 57 if rate is None: 58 return (None, None) 59 num, period = rate.split('/') 60 num_requests = int(num) 61 duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]] 62 return (num_requests, duration) 63 64 #一、一进来会先执行他, 65 def allow_request(self, request, view): 66 """ 67 Implement the check to see if the request should be throttled. 68 69 On success calls `throttle_success`. 70 On failure calls `throttle_failure`. 71 """ 72 if self.rate is None: 73 return True 74 75 self.key = self.get_cache_key(request, view) #二、执行get_cache_key,这里的self.key就至关于咱们举例ip 76 if self.key is None: 77 return True 78 79 self.history = self.cache.get(self.key, []) #三、获得的key,默认是一个列表,赋值给了self.history, 80 # 这时候self.history就是每个ip对应的访问记录 81 self.now = self.timer() 82 83 # Drop any requests from the history which have now passed the 84 # throttle duration 85 while self.history and self.history[-1] <= self.now - self.duration: 86 self.history.pop() 87 if len(self.history) >= self.num_requests: 88 return self.throttle_failure() 89 return self.throttle_success() 90 91 def throttle_success(self): 92 """ 93 Inserts the current request's timestamp along with the key 94 into the cache. 95 """ 96 self.history.insert(0, self.now) 97 self.cache.set(self.key, self.history, self.duration) 98 return True 99 100 def throttle_failure(self): 101 """ 102 Called when a request to the API has failed due to throttling. 103 """ 104 return False 105 106 def wait(self): 107 """ 108 Returns the recommended next request time in seconds. 109 """ 110 if self.history: 111 remaining_duration = self.duration - (self.now - self.history[-1]) 112 else: 113 remaining_duration = self.duration 114 115 available_requests = self.num_requests - len(self.history) + 1 116 if available_requests <= 0: 117 return None 118 119 return remaining_duration / float(available_requests)
请求一进来会先执行SimpleRateThrottle这个类的构造方法
1 def __init__(self): 2 if not getattr(self, 'rate', None): 3 self.rate = self.get_rate() #点进去看到须要些一个scope ,2/m 4 self.num_requests, self.duration = self.parse_rate(self.rate)
1 def get_rate(self): 2 """ 3 Determine the string representation of the allowed request rate. 4 """ 5 if not getattr(self, 'scope', None): #检测必须有scope,没有就报错了 6 msg = ("You must set either `.scope` or `.rate` for '%s' throttle" % 7 self.__class__.__name__) 8 raise ImproperlyConfigured(msg) 9 10 try: 11 return self.THROTTLE_RATES[self.scope] 12 except KeyError: 13 msg = "No default throttle rate set for '%s' scope" % self.scope 14 raise ImproperlyConfigured(msg)
1 def parse_rate(self, rate): 2 """ 3 Given the request rate string, return a two tuple of: 4 <allowed number of requests>, <period of time in seconds> 5 """ 6 if rate is None: 7 return (None, None) 8 num, period = rate.split('/') 9 num_requests = int(num) 10 duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]] 11 return (num_requests, duration)
1 #二、接下来会先执行他, 2 def allow_request(self, request, view): 3 """ 4 Implement the check to see if the request should be throttled. 5 6 On success calls `throttle_success`. 7 On failure calls `throttle_failure`. 8 """ 9 if self.rate is None: 10 return True 11 12 self.key = self.get_cache_key(request, view) #二、执行get_cache_key,这里的self.key就至关于咱们举例ip 13 if self.key is None: 14 return True #不限制 15 # [114521212,11452121211,45212121145,21212114,521212] 16 self.history = self.cache.get(self.key, []) #三、获得的key,默认是一个列表,赋值给了self.history, 17 # 这时候self.history就是每个ip对应的访问记录 18 self.now = self.timer() 19 20 # Drop any requests from the history which have now passed the 21 # throttle duration 22 while self.history and self.history[-1] <= self.now - self.duration: 23 self.history.pop() 24 if len(self.history) >= self.num_requests: 25 return self.throttle_failure() 26 return self.throttle_success()
1 def wait(self): 2 """ 3 Returns the recommended next request time in seconds. 4 """ 5 if self.history: 6 remaining_duration = self.duration - (self.now - self.history[-1]) 7 else: 8 remaining_duration = self.duration 9 10 available_requests = self.num_requests - len(self.history) + 1 11 if available_requests <= 0: 12 return None 13 14 return remaining_duration / float(available_requests)
代码实现:
1 ###########用resetframework内部的限制访问频率############## 2 class MySimpleRateThrottle(SimpleRateThrottle): 3 scope = 'xxx' 4 def get_cache_key(self, request, view): 5 return self.get_ident(request) #返回惟一标识IP 6 7 class LimitView(APIView): 8 authentication_classes = [] #不让认证用户 9 permission_classes = [] #不让验证权限 10 throttle_classes = [MySimpleRateThrottle, ] 11 def get(self,request): 12 # self.dispatch 13 return Response('控制访问频率示例') 14 15 def throttled(self, request, wait): 16 '''可定制方法设置中文错误''' 17 # raise exceptions.Throttled(wait) 18 class MyThrottle(exceptions.Throttled): 19 default_detail = '请求被限制' 20 extra_detail_singular = 'Expected available in {wait} second.' 21 extra_detail_plural = 'Expected available in {wait} seconds.' 22 default_code = '还须要再等{wait}秒' 23 raise MyThrottle(wait)
记得在settings里面配置
1 REST_FRAMEWORK = { 2 'UNAUTHENTICATED_USER': None, 3 'UNAUTHENTICATED_TOKEN': None, #将匿名用户设置为None 4 "DEFAULT_AUTHENTICATION_CLASSES": [ 5 "app01.utils.MyAuthentication", 6 ], 7 'DEFAULT_PERMISSION_CLASSES':[ 8 # "app03.utils.MyPermission",#设置路径, 9 ], 10 'DEFAULT_THROTTLE_RATES':{ 11 'xxx':'2/minute' #2分钟 12 } 13 } 14 15 #缓存:放在文件 16 CACHES = { 17 'default': { 18 'BACKEND': 'django.core.cache.backends.filebased.FileBasedCache', 19 'LOCATION': 'cache', #文件路径 20 } 21 }
四、对匿名用户进行限制,每一个用户1分钟容许访问5次,对于登陆的普通用户1分钟访问10次,VIP用户一分钟访问20次
1 from django.contrib import admin 2 3 from django.conf.urls import url, include 4 from app05 import views 5 6 urlpatterns = [ 7 url('index/',views.IndexView.as_view()), 8 url('manage/',views.ManageView.as_view()), 9 ]
1 from django.shortcuts import render 2 from rest_framework.views import APIView 3 from rest_framework.response import Response 4 from rest_framework.authentication import BaseAuthentication #认证须要 5 from rest_framework.throttling import BaseThrottle,SimpleRateThrottle #限流处理 6 from rest_framework.permissions import BasePermission 7 from rest_framework import exceptions 8 from app01 import models 9 # Create your views here. 10 ###############3##认证##################### 11 class MyAuthentcate(BaseAuthentication): 12 '''检查用户是否存在,若是存在就返回user和auth,若是没有就返回''' 13 def authenticate(self, request): 14 token = request.query_params.get('token') 15 obj = models.UserInfo.objects.filter(token=token).first() 16 if obj: 17 return (obj.username,obj.token) 18 return None #表示我不处理 19 20 ##################权限##################### 21 class MyPermission(BasePermission): 22 message='无权访问' 23 def has_permission(self, request, view): 24 if request.user: 25 return True #true表示有权限 26 return False #false表示无权限 27 28 class AdminPermission(BasePermission): 29 message = '无权访问' 30 31 def has_permission(self, request, view): 32 if request.user=='haiyan': 33 return True # true表示有权限 34 return False # false表示无权限 35 36 ############3#####限流##################3## 37 class AnonThrottle(SimpleRateThrottle): 38 scope = 'wdp_anon' #至关于设置了最大的访问次数和时间 39 def get_cache_key(self, request, view): 40 if request.user: 41 return None #返回None表示我不限制,登陆用户我无论 42 #匿名用户 43 return self.get_ident(request) #返回一个惟一标识IP 44 45 class UserThrottle(SimpleRateThrottle): 46 scope = 'wdp_user' 47 def get_cache_key(self, request, view): 48 #登陆用户 49 if request.user: 50 return request.user 51 return None #返回NOne表示匿名用户我无论 52 53 54 ##################视图##################### 55 #首页支持匿名访问, 56 #无须要登陆就能够访问 57 class IndexView(APIView): 58 authentication_classes = [MyAuthentcate,] #认证判断他是否是匿名用户 59 permission_classes = [] #通常主页就不须要权限验证了 60 throttle_classes = [AnonThrottle,UserThrottle,] #对匿名用户和普通用户的访问限制 61 62 def get(self,request): 63 # self.dispatch 64 return Response('访问首页') 65 66 def throttled(self, request, wait): 67 '''可定制方法设置中文错误''' 68 69 # raise exceptions.Throttled(wait) 70 class MyThrottle(exceptions.Throttled): 71 default_detail = '请求被限制' 72 extra_detail_singular = 'Expected available in {wait} second.' 73 extra_detail_plural = 'Expected available in {wait} seconds.' 74 default_code = '还须要再等{wait}秒' 75 76 raise MyThrottle(wait) 77 78 #需登陆就能够访问 79 class ManageView(APIView): 80 authentication_classes = [MyAuthentcate, ] # 认证判断他是否是匿名用户 81 permission_classes = [MyPermission,] # 通常主页就不须要权限验证了 82 throttle_classes = [AnonThrottle, UserThrottle, ] # 对匿名用户和普通用户的访问限制 83 84 def get(self, request): 85 # self.dispatch 86 return Response('管理人员访问页面') 87 88 def throttled(self, request, wait): 89 '''可定制方法设置中文错误''' 90 91 # raise exceptions.Throttled(wait) 92 class MyThrottle(exceptions.Throttled): 93 default_detail = '请求被限制' 94 extra_detail_singular = 'Expected available in {wait} second.' 95 extra_detail_plural = 'Expected available in {wait} seconds.' 96 default_code = '还须要再等{wait}秒' 97 98 raise MyThrottle(wait)
4、总结
一、认证:就是检查用户是否存在;若是存在返回(request.user,request.auth);不存在request.user/request.auth=None
二、权限:进行职责的划分
三、限制访问频率
认证 - 类:authenticate/authenticate_header ##验证不成功的时候执行的 - 返回值: - return None, - return (user,auth), - raise 异常 - 配置: - 视图: class IndexView(APIView): authentication_classes = [MyAuthentication,] - 全局: REST_FRAMEWORK = { 'UNAUTHENTICATED_USER': None, 'UNAUTHENTICATED_TOKEN': None, "DEFAULT_AUTHENTICATION_CLASSES": [ # "app02.utils.MyAuthentication", ], } 权限 - 类:has_permission/has_object_permission - 返回值: - True、#有权限 - False、#无权限 - exceptions.PermissionDenied(detail="错误信息") #异常本身随意,想抛就抛,错误信息本身指定 - 配置: - 视图: class IndexView(APIView): permission_classes = [MyPermission,] - 全局: REST_FRAMEWORK = { "DEFAULT_PERMISSION_CLASSES": [ # "app02.utils.MyAuthentication", ], } 限流 - 类:allow_request/wait PS: scope = "wdp_user" - 返回值:
return True、#不限制
return False #限制 - 配置: - 视图: class IndexView(APIView): throttle_classes=[AnonThrottle,UserThrottle,] def get(self,request,*args,**kwargs): self.dispatch return Response('访问首页') - 全局 REST_FRAMEWORK = { "DEFAULT_THROTTLE_CLASSES":[ ], 'DEFAULT_THROTTLE_RATES':{ 'wdp_anon':'5/minute', 'wdp_user':'10/minute', } }