4f80fd6540
Used django-ratelimit instead of django-ratelimit-backend to configure two different rate limit configurations for same endpoint. PROD-1708
32 lines
865 B
Python
32 lines
865 B
Python
"""
|
|
A utility class which wraps the RateLimitMixin 3rd party class to do bad request counting
|
|
which can be used for rate limiting
|
|
"""
|
|
|
|
from ratelimitbackend.backends import RateLimitMixin
|
|
|
|
|
|
class RequestRateLimiter(RateLimitMixin):
|
|
"""
|
|
Use the 3rd party RateLimitMixin to help do rate limiting.
|
|
"""
|
|
def is_rate_limit_exceeded(self, request):
|
|
"""
|
|
Returns if the client has been rated limited
|
|
"""
|
|
counts = self.get_counters(request)
|
|
return sum(counts.values()) >= self.requests
|
|
|
|
def tick_request_counter(self, request):
|
|
"""
|
|
Ticks any counters used to compute when rate limt has been reached
|
|
"""
|
|
self.cache_incr(self.get_cache_key(request))
|
|
|
|
|
|
class BadRequestRateLimiter(RequestRateLimiter):
|
|
"""
|
|
Default rate limit is 30 requests for every 5 minutes.
|
|
"""
|
|
pass
|