diff --git a/ratelimit/decorators.py b/ratelimit/decorators.py index 7aaf19e..5010bde 100644 --- a/ratelimit/decorators.py +++ b/ratelimit/decorators.py @@ -20,7 +20,7 @@ def _wrapped(request, *args, **kw): increment=True) request.limited = ratelimited or old_limited if ratelimited and block: - raise Ratelimited() + raise Ratelimited(group=group, key=key, rate=rate) return fn(request, *args, **kw) return _wrapped return decorator diff --git a/ratelimit/exceptions.py b/ratelimit/exceptions.py index f39a0f4..31de192 100644 --- a/ratelimit/exceptions.py +++ b/ratelimit/exceptions.py @@ -2,4 +2,9 @@ class Ratelimited(PermissionDenied): - pass + + def __init__(self, group, key, rate): + super(Ratelimited, self).__init__() + self.group = group + self.key = key + self.rate = rate