Mail

venv/lib/python3.12/site-packages/django_ratelimit/middleware.py

@@ -0,0 +1,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)