Django REST framework 实现基于 IP 和用户名的访问频率控制
本文介绍如何在 Django REST framework 中配置内置的访问频率限制。通过继承 SimpleRateThrottle 类,分别针对匿名用户基于 IP 地址和登录用户基于用户名进行频率控制。文章详细说明了如何配置 Redis 缓存后端,在全站设置匿名用户限制,并在具体视图中为登录用户应用特定频率策略,同时结合 Token 身份认证实现完整的访问控制流程。
Django REST framework 访问频率控制策略
在 Django REST framework 中,可以通过内置的频率控制机制来实现对不同用户的访问限制。具体策略为:对未登录的匿名用户,采用其 IP 地址作为识别依据来控制访问频率;对已登录的用户,则采用其用户名作为识别依据进行频率控制。
自定义频率限制类
from rest_framework.throttling import SimpleRateThrottle
class 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 作为缓存后端来存储访问记录。
配置 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
用户身份认证的具体实现代码如下:
Token 身份认证实现
from rest_framework import authentication
from rest_framework import exceptionsfrom appxx import models
class 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)
以上即为 Django REST framework 中实现内置访问频率控制的完整内容,希望本文能对大家的学习和工作提供帮助。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













