Django REST framework 如何实现内置访问频率控制

2019-11-25 12:15:18

字体：大中小

来源：转载

供稿：网友

对匿名用户采用 IP 控制访问频率，对登录用户采用用户名控制访问频率。

from rest_framework.throttling import SimpleRateThrottleclass VisitThrottle(SimpleRateThrottle):  """匿名用户访问频率限制"""  scope = "AnonymousUser" # 随便写的，可以作为key保存在缓存中  def get_cache_key(self, request, view):    return self.get_ident(request)class UserThrottle(SimpleRateThrottle):  """登录用户访问频率限制"""  scope = "LoginUser"  def get_cache_key(self, request, view):return request.user

可以配置redis

CACHES = {  "default": {    "BACKEND": "django_redis.cache.RedisCache",    "LOCATION": "redis://127.0.0.1:6379",    "OPTIONS": {      "CLIENT_CLASS": "django_redis.client.DefaultClient",      "CONNECTION_POOL_KWARGS": {"max_connections": 100}      # "PASSWORD": "密码",    }  }}

匿名用户的访问频率限制，这里设置在全站下，如下：

REST_FRAMEWORK = {  "DEFAULT_THROTTLE_CLASSES": ["appxx.utils.VisitThrottle"],  "DEFAULT_THROTTLE_RATES":{      "AnonymousUser": "3/m", # 匿名用户一分钟可以访问3次，秒(s)、分(m)、时(h)、天(d)      "LoginUser": "10/m", # 登录用户一分钟可以访问10次    }}

登录用户的访问频率设置在单独的视图中，而视图依赖身份认证才能辨别用户是否登陆了，所以设置如下：

class BookViewSet(viewsets.ModelViewSet):  authentication_classes = [TokenAuthentication]  throttle_classes = [UserThrottle]  queryset = models.Book.objects.all()  serializer_class = serializers.BookSerializer

用户身份认证如下：

from rest_framework import authenticationfrom rest_framework import exceptionsfrom appxx import modelsclass TokenAuthentication(authentication.BaseAuthentication):  """身份认证"""  def authenticate(self, request):    token = request.GET.get("token")    obj = models.UserAuthToken.objects.filter(token=token).first()    if not obj:      raise exceptions.AuthenticationFailed("验证失败！")    else:      return (obj.user.username, obj.token)

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持武林网。

上一篇：python实现DEM数据的阴影生成的方法

下一篇：详解pandas中MultiIndex和对象实际索引不一致问题