Skip休息节流计数器

Question

当用户请求无效或服务器无法完成进程时，如何防止Django rest节流计数请求？

例如，我需要来自用户的params，但是当用户不给params时，Django rest节流仍然计算它。

当请求不成功时，是否有跳过节流计数器的解决方案？

示例

class OncePerHourAnonThrottle(AnonRateThrottle):
    rate = "1/hour"


class Autoliker(APIView):
    throttle_classes = [OncePerHourAnonThrottle]

    def get(self, request):
        content = {"status": "get"}
        return Response(content)

    def post(self, request):
        post_url = request.POST.get("url", None)
        print(post_url)

        content = {"status": "post"}
        return Response(content)

    def throttled(self, request, wait):
        raise Throttled(
            detail={
                "message": "request limit exceeded",
                "availableIn": f"{wait} seconds",
                "throttleType": "type",
            }
        )

Answer 1

您可以为此创建一个装饰器。

class OncePerHourAnonThrottle(AnonRateThrottle):
    rate = "1/hour"

    def allow_request(self, request, view):
        """
        This function is copy of SimpleRateThrottle.allow_request
        The only difference is, instead of executing self.throttle_success
        it directly returns True and doesn't mark this request as success yet.
        """
        if self.rate is None:
            return True

        self.key = self.get_cache_key(request, view)
        if self.key is None:
            return True

        self.history = self.cache.get(self.key, [])
        self.now = self.timer()

        # Drop any requests from the history which have now passed the
        # throttle duration
        while self.history and self.history[-1] <= self.now - self.duration:
            self.history.pop()
        if len(self.history) >= self.num_requests:
            return False
        return True

def rate_limiter(view_function):
    @wraps(view_function)
    def inner(view_obj, request, *args, **kwargs):
        throttle = OncePerHourAnonThrottle()
        allowed = throttle.allow_request(request, None)
        if not allowed:
            raise exceptions.Throttled(throttle.wait())
        try:
            response = view_function(view_obj, request, *args, **kwargs)
        except Exception as exc:
            response = view_obj.handle_exception(exc)
        if response.status_code == 200:
            # now if everything goes OK, count this request as success
            throttle.throttle_success()
        return response

    return inner

class Autoliker(APIView):
   
    @rate_limiter
    def post(requests):
        # view logic
        pass

这是你怎么做的基本想法，现在你可以把它变成一个通用装饰器，甚至是基于类的装饰器。