mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-22 17:47:04 +03:00
Tweak throttles and improve docs
This commit is contained in:
parent
3005079824
commit
886f8b4751
|
@ -14,10 +14,12 @@ from djangorestframework.response import Response
|
|||
|
||||
class User3SecRateThrottle(UserRateThrottle):
|
||||
rate = '3/sec'
|
||||
scope = 'seconds'
|
||||
|
||||
|
||||
class User3MinRateThrottle(UserRateThrottle):
|
||||
rate = '3/min'
|
||||
scope = 'minutes'
|
||||
|
||||
|
||||
class MockView(APIView):
|
||||
|
|
|
@ -44,7 +44,7 @@ class SimpleRateThottle(BaseThrottle):
|
|||
|
||||
timer = time.time
|
||||
settings = api_settings
|
||||
cache_format = '%(class)s_%(scope)s_%(ident)s'
|
||||
cache_format = 'throtte_%(scope)s_%(ident)s'
|
||||
scope = None
|
||||
|
||||
def __init__(self, view):
|
||||
|
@ -144,7 +144,6 @@ class AnonRateThrottle(SimpleRateThottle):
|
|||
ident = request.META.get('REMOTE_ADDR', None)
|
||||
|
||||
return self.cache_format % {
|
||||
'class': self.__class__.__name__,
|
||||
'scope': self.scope,
|
||||
'ident': ident
|
||||
}
|
||||
|
@ -167,7 +166,6 @@ class UserRateThrottle(SimpleRateThottle):
|
|||
ident = request.META.get('REMOTE_ADDR', None)
|
||||
|
||||
return self.cache_format % {
|
||||
'class': self.__class__.__name__,
|
||||
'scope': self.scope,
|
||||
'ident': ident
|
||||
}
|
||||
|
@ -181,11 +179,13 @@ class ScopedRateThrottle(SimpleRateThottle):
|
|||
user id of the request, and the scope of the view being accessed.
|
||||
"""
|
||||
|
||||
scope_attr = 'throttle_scope'
|
||||
|
||||
def __init__(self, view):
|
||||
"""
|
||||
Scope is determined from the view being accessed.
|
||||
"""
|
||||
self.scope = getattr(self.view, 'throttle_scope', None)
|
||||
self.scope = getattr(self.view, self.scope_attr, None)
|
||||
super(ScopedRateThrottle, self).__init__(view)
|
||||
|
||||
def parse_rate_description(self, rate):
|
||||
|
@ -204,7 +204,7 @@ class ScopedRateThrottle(SimpleRateThottle):
|
|||
with the '.throttle_scope` property of the view.
|
||||
"""
|
||||
if not self.scope:
|
||||
return None # Only throttle views with `.throttle_scope` set.
|
||||
return None # Only throttle views if `.throttle_scope` is set.
|
||||
|
||||
if request.user.is_authenticated():
|
||||
ident = request.user.id
|
||||
|
@ -212,7 +212,6 @@ class ScopedRateThrottle(SimpleRateThottle):
|
|||
ident = request.META.get('REMOTE_ADDR', None)
|
||||
|
||||
return self.cache_format % {
|
||||
'class': self.__class__.__name__,
|
||||
'scope': self.scope,
|
||||
'ident': ident
|
||||
}
|
||||
|
|
|
@ -83,6 +83,29 @@ The allowed request rate is determined from one of the following (in order of pr
|
|||
* The `rate` property on the class, which may be provided by overriding `UserThrottle` and setting the property.
|
||||
* The `DEFAULT_THROTTLE_RATES['user']` setting.
|
||||
|
||||
An API may have multiple `UserRateThrottles` in place at the same time. To do so, override `UserRateThrottle` and set a unique "scope" for each class.
|
||||
|
||||
For example, multiple user throttle rates could be implemented by using the following classes...
|
||||
|
||||
class BurstRateThrottle(UserRateThrottle):
|
||||
scope = 'burst'
|
||||
|
||||
class SustainedRateThrottle(UserRateThrottle):
|
||||
scope = 'sustained'
|
||||
|
||||
...and the following settings.
|
||||
|
||||
API_SETTINGS = {
|
||||
'DEFAULT_THROTTLES': (
|
||||
'example.throttles.BurstRateThrottle',
|
||||
'example.throttles.SustainedRateThrottle',
|
||||
)
|
||||
'DEFAULT_THROTTLE_RATES': {
|
||||
'burst': '60/min',
|
||||
'sustained': '1000/day'
|
||||
}
|
||||
}
|
||||
|
||||
`UserThrottle` is suitable if you want a simple global rate restriction per-user.
|
||||
|
||||
## ScopedRateThrottle
|
||||
|
|
Loading…
Reference in New Issue
Block a user