djresttoolkit 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

{djresttoolkit-0.2.0.dist-info → djresttoolkit-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: djresttoolkit
-Version: 0.2.0
+Version: 0.3.0
 Summary: A collection of Django and DRF utilities to simplify API development.
 Project-URL: Homepage, https://github.com/shaileshpandit141/djresttoolkit
 Project-URL: Documentation, https://shaileshpandit141.github.io/djresttoolkit

{djresttoolkit-0.2.0.dist-info → djresttoolkit-0.3.0.dist-info}/RECORD

@@ -10,8 +10,11 @@ src/djresttoolkit/mail/_models.py,sha256=_41pH3xC0jP8SHSty2FkxvRh2_ddKk-4peT11OH
 src/djresttoolkit/mail/_types.py,sha256=est1mrN80vB_4-j-2yuAr_l_7rNzO4AJlqpq74zO5ow,682
 src/djresttoolkit/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-djresttoolkit-0.2.0.dist-info/METADATA,sha256=Xr6JLGRsN0NlwnQ_Nu0HcdBv6Sfs7RKeIdYOq4GVoL8,4374
-djresttoolkit-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-djresttoolkit-0.2.0.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
-djresttoolkit-0.2.0.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
-djresttoolkit-0.2.0.dist-info/RECORD,,
+src/djresttoolkit/views/__init__.py,sha256=XrxBrs6sH4HmUzp41omcmy_y94pSaXAVn01ttQ022-4,76
+src/djresttoolkit/views/_exceptions/__init__.py,sha256=DrCUxuPNyBR4WhzNutn5HDxLa--q51ykIxSG7_bFsOI,83
+src/djresttoolkit/views/_exceptions/_exception_handler.py,sha256=lg6kfAABex-UbAsF78uX-M-ZnJ3u1vK1eIXytb-4KSw,2375
+djresttoolkit-0.3.0.dist-info/METADATA,sha256=LKqn4ZSUVD-yo03t6klk-QHW1q_VwxpCJk6DfTSIDMo,4374
+djresttoolkit-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+djresttoolkit-0.3.0.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
+djresttoolkit-0.3.0.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
+djresttoolkit-0.3.0.dist-info/RECORD,,

src/djresttoolkit/views/__init__.py

@@ -0,0 +1,3 @@
+from ._exceptions import exception_handler
+
+__all__ = ["exception_handler"]

src/djresttoolkit/views/_exceptions/__init__.py

@@ -0,0 +1,3 @@
+from ._exception_handler import exception_handler
+
+__all__ = ["exception_handler"]

src/djresttoolkit/views/_exceptions/_exception_handler.py

@@ -0,0 +1,64 @@
+from typing import Any, cast
+
+from django.core.cache import cache
+from django.utils import timezone
+from rest_framework import status, views
+from rest_framework.request import Request
+from rest_framework.response import Response
+from rest_framework.throttling import AnonRateThrottle
+
+
+def exception_handler(exc: Exception, context: dict[str, Any]) -> Response | None:
+    """
+    Custom exception handler that preserves DRF's default functionality
+    while adding custom throttling behavior.
+    """
+
+    # Call DRF's default exception handler first
+    response: Response | None = views.exception_handler(exc, context)
+
+    request: Request | None = context.get("request")
+    view = context.get("view")
+
+    if request and view:
+        # Pick throttle classes from view or default to AnonRateThrottle
+        throttle_classes: list[type[AnonRateThrottle]] = getattr(
+            view, "throttle_classes", [AnonRateThrottle]
+        )
+
+        for throttle_class in throttle_classes:
+            throttle = throttle_class()
+            cache_key = throttle.get_cache_key(request, view)
+            if not cache_key:
+                continue
+
+            history: list[float] = cache.get(cache_key, [])
+            now = timezone.now().timestamp()
+            duration: float = cast(float, throttle.duration)  # type: ignore[attr-defined]
+
+            # Keep only non-expired timestamps
+            history = [float(ts) for ts in history if now - float(ts) < duration]
+
+            # If throttle limit exceeded
+            if len(history) >= throttle.num_requests:  # type: ignore[attr-defined]
+                retry_after: float = (
+                    duration - (now - history[0]) if history else duration
+                )
+
+                return Response(
+                    data={
+                        "detail": "Too many requests. Please try again later.",
+                        "retry_after": {
+                            "time": round(retry_after, 2),
+                            "unit": "seconds",
+                        },
+                    },
+                    status=status.HTTP_429_TOO_MANY_REQUESTS,
+                )
+
+            # Otherwise add current timestamp
+            history.append(now)
+            cache.set(key=cache_key, value=history, timeout=duration)
+
+    # If DRF handled the exception, return that response
+    return response