如何设置多个速率限制，每10秒，每10分钟，每1天，使用django-ratelimit或throttling？

Question

我想根据10秒、10分钟和1天的持续时间设置我的视图的速率限制。因此假设用户每天可以发送20个请求/ 10秒、100个请求/10分钟和1000个请求。

我尝试过限制，但找不到任何方法来设置多个请求。我已经尝试过django-ratelimit包，但我也找不到任何这样的选项，因为它为rate设置了一个字符串，例如rate = '5/10m‘。如果有解决这个问题的办法，请告诉我。

Answer 1

因为我使用的是下面的，而不是泛型视图。

@method_decorator(ratelimit(key='user', rate='1000/d'), name='dispatch')
class MyViewSet(ViewSet):

我找到了另一个解决方案。

from django.http import JsonResponse
from ratelimit.decorators import ratelimit

class RateLimitForSecurity(View):
    @ratelimit(key='ip', rate='30/m')
    def dispatch(self, request, *args, **kwargs):
        was_limited = getattr(request, 'limited', False)
        if was_limited:
            return JsonResponse({"code": 1, 'msg': 'try many times'},json_dumps_params={'ensure_ascii':False})

        return super().dispatch(request, *args, **kwargs)

class IndexView(RateLimitForSecurity, generic.ListView):
      # same as before

Answer 2

我在做这方面的工作，发现您可以覆盖限制类函数parse_rate，并且可以根据需要对其进行自定义

from rest_framework.throttling import UserRateThrottle
class ThirdPartyMonthlyThrottle(UserRateThrottle):
    scope = 'third_party_monthly'

    def parse_rate(self, rate):
        if rate is None:
            return None, None
        num, period = rate.split('/')
        num_requests = int(num)
        duration = int(period)*86400
        return num_requests, duration

'DEFAULT_THROTTLE_RATES': {
        'third_party_monthly': '3/30'
    }

这可以解释为每月3次请求

Answer 3

你可以这样做

from ratelimit.decorators import ratelimit
from http import HTTPStatus

@ratelimit(key='user_or_ip', rate='1/10m', method=['POST'])
def ajax_add_review(request, username):
    was_limited = getattr(request, 'limited', False)
        if was_limited:
            return JsonResponse({"error": 'limit crossed'}, status=HTTPStatus.BAD_REQUEST)