Python开发：用自定义中间件在Django Rest Framework实现用户访问频率控制

Django Rest Framework（DRF）本身并没有提供用户访问频率控制的功能，但是你可以通过自定义中间件来实现这个功能。以下是一个简单的例子： ```python from django.shortcuts import HttpResponse from datetime import datetime, timedelta class UserAccessFrequencyMiddleware: def __init__(self, get_response): self.get_response = get_response def __call__(self, request): # 获取当前用户 user = request.user # 如果用户是匿名用户，直接返回响应 if user.is_anonymous: return self.get_response(request) # 记录用户最后一次访问时间 if not hasattr(user, 'last_access_time'): user.last_access_time = datetime.now() # 计算用户最后一次访问到现在的时间差 time_diff = datetime.now() - user.last_access_time # 如果用户访问频率超过了设定的阈值，返回错误响应 if time_diff < timedelta(seconds=60): return HttpResponse('Too frequent access', status=429) # 更新用户最后一次访问时间 user.last_access_time = datetime.now() # 更新用户并保存到数据库 user.save() # 返回响应 return self.get_response(request) ``` 这个中间件会检查每个用户的访问频率，如果用户在1分钟内访问了多次，就会返回一个错误响应。你可以根据需要调整这个阈值。 注意，这个中间件只能控制用户的访问频率，不能防止DDoS攻击。如果你需要防止DDoS攻击，你可能需要使用专门的DDoS防护服务或者防火墙。 ######[AI写代码神器 | 333点数解答 | 2023-11-08 00:53:37]