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

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

本文介绍如何在 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 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。