Annotate request and exception with usage data

django_ratelimit/decorators.py

@@ -5,7 +5,7 @@
 
 from django_ratelimit import ALL, UNSAFE
 from django_ratelimit.exceptions import Ratelimited
-from django_ratelimit.core import is_ratelimited
+from django_ratelimit.core import get_usage
 
 
 __all__ = ['ratelimit']
@@ -16,14 +16,24 @@ def decorator(fn):
         @wraps(fn)
         def _wrapped(request, *args, **kw):
             old_limited = getattr(request, 'limited', False)
-            ratelimited = is_ratelimited(request=request, group=group, fn=fn,
-                                         key=key, rate=rate, method=method,
-                                         increment=True)
+
+            usage = get_usage(request=request, group=group, fn=fn,
+                              key=key, rate=rate, method=method,
+                              increment=True)
+            if usage is None:
+                ratelimited = False
+            else:
+                ratelimited = usage['should_limit']
+
             request.limited = ratelimited or old_limited
+            request.usage = usage
+
             if ratelimited and block:
                 cls = getattr(
                     settings, 'RATELIMIT_EXCEPTION_CLASS', Ratelimited)
-                raise (import_string(cls) if isinstance(cls, str) else cls)()
+                exc = (import_string(cls) if isinstance(cls, str) else cls)()
+                exc.usage = usage
+                raise exc
             return fn(request, *args, **kw)
         return _wrapped
     return decorator