Add @throttle_scope function based view decorator

Author: d-ryzhykau

docs/api-guide/views.md

@@ -182,6 +182,7 @@ The available decorators are:
 * `@parser_classes(...)`
 * `@authentication_classes(...)`
 * `@throttle_classes(...)`
+* `@throttle_scope(...)`
 * `@permission_classes(...)`
 * `@content_negotiation_class(...)`
 * `@metadata_class(...)`

rest_framework/decorators.py

@@ -67,6 +67,9 @@ def handler(self, *args, **kwargs):
         WrappedAPIView.throttle_classes = getattr(func, 'throttle_classes',
                                                   APIView.throttle_classes)
 
+        WrappedAPIView.throttle_scope = getattr(func, 'throttle_scope',
+                                                None)
+
         WrappedAPIView.permission_classes = getattr(func, 'permission_classes',
                                                     APIView.permission_classes)
 
@@ -136,6 +139,14 @@ def decorator(func):
     return decorator
 
 
+def throttle_scope(throttle_scope):
+    def decorator(func):
+        _check_decorator_order(func, 'throttle_scope')
+        func.throttle_scope = throttle_scope
+        return func
+    return decorator
+
+
 def permission_classes(permission_classes):
     def decorator(func):
         _check_decorator_order(func, 'permission_classes')

tests/test_decorators.py

@@ -1,14 +1,14 @@
 import sys
 
 import pytest
-from django.test import TestCase
+from django.test import TestCase, override_settings
 
 from rest_framework import status
 from rest_framework.authentication import BasicAuthentication
 from rest_framework.decorators import (
     action, api_view, authentication_classes, content_negotiation_class,
     metadata_class, parser_classes, permission_classes, renderer_classes,
-    schema, throttle_classes, versioning_class
+    schema, throttle_classes, throttle_scope, versioning_class
 )
 from rest_framework.negotiation import BaseContentNegotiation
 from rest_framework.parsers import JSONParser
@@ -153,6 +153,25 @@ def view(request):
         response = view(request)
         assert response.status_code == status.HTTP_429_TOO_MANY_REQUESTS
 
+    @override_settings(
+        REST_FRAMEWORK={
+            "DEFAULT_THROTTLE_CLASSES": ['rest_framework.throttling.ScopedRateThrottle'],
+            "DEFAULT_THROTTLE_RATES": {"x": "1/day"},
+        }
+    )
+    def test_throttle_scope(self):
+        @api_view(['GET'])
+        @throttle_scope("x")
+        def view(request):
+            return Response({})
+
+        request = self.factory.get('/')
+        response = view(request)
+        assert response.status_code == status.HTTP_200_OK
+
+        response = view(request)
+        assert response.status_code == status.HTTP_429_TOO_MANY_REQUESTS
+
     def test_versioning_class(self):
         @api_view(["GET"])
         @versioning_class(QueryParameterVersioning)