1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
from django.conf import settings
from django.utils.module_loading import import_string
from django_ratelimit.exceptions import Ratelimited
class RatelimitMiddleware:
def __init__(self, get_response):
self.get_response = get_response
def __call__(self, request):
return self.get_response(request)
def process_exception(self, request, exception):
if not isinstance(exception, Ratelimited):
return None
view = import_string(settings.RATELIMIT_VIEW)
return view(request, exception)
|
