Django REST не может написать пользовательский Throttle и получает ImportError:

структура папки api/api_views.py

api_views.py

class BurstRateThrottle(UserRateThrottle):
    scope = 'burst'


class SustainedRateThrottle(UserRateThrottle):
    scope = 'sustained'

settings.py

REST_FRAMEWORK = {
    'DEFAULT_THROTTLE_CLASSES': [
        'api.api_views.BurstRateThrottle',
        'api.api_views.SustainedRateThrottle'
    ],
    'DEFAULT_THROTTLE_RATES': {
        'burst': '60/min',
        'sustained': '1000/day'
    }
}

Получение этой ошибки : Could not import 'api.api_views.BurstRateThrottle' for API setting 'DEFAULT_THROTTLE_CLASSES'. ImportError: Module "api.api_views" does not define a "BurstRateThrottle" attribute/class.

Я бы посоветовал ознакомиться с официальной документацией, которая очень понятна. https://www.django-rest-framework.org/api-guide/throttling/#custom-throttles. Более того, есть определенные модели, которые должны быть переопределены, чтобы иметь возможность реализовать полностью рабочее пользовательское решение дросселирования. Однако вы можете использовать встроенную функциональность, которую предлагают модули дросселирования, и настроить ее в соответствии с вашими потребностями. Простой пример:

from rest_framework.throttling import UserRateThrottle

class ThrottleCheck(APIView):
    throttle_class = UserRateThrottle
    throttle_scope = "THIS SHOULD BE THE NAME YOU DEFINE IN YOUR SETTING FILE"
    def get(self,request:Request)-> Response:
        """
        SOME CODE GOES HERE.  
        The throttle would be triggered here
        """
The last words, Ideally if you can handle throttling from your Nginx side would make more sense Since on production this might be challenging using the Throttle class of Django-rest-framework.
Hope this helps. Happy Hacking
Вернуться на верх