Django REST framework 实现基于 IP 和用户名的访问频率控制

橘子海
橘子海
发布时间:2026-10-13 | 5人浏览过
原创

本文介绍如何在 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 中实现内置访问频率控制的完整内容,希望本文能对大家的学习和工作提供帮助。

本站声明

本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。

Python 相关产品和应用

全部下载

专题列表