如何使用 django rest 框架为不同的用户设置不同的节流范围？

Question

例如：为客户和员工设置不同的节流率。 这可以使用UserRateThrottle来完成吗？ 例如：

REST_FRAMEWORK['DEFAULT_THROTTLE_CLASSES'] = ('backend.throttles.EmployeeThrottle')
REST_FRAMEWORK['DEFAULT_THROTTLE_RATES'] = {
    'user': config.THROTTLE_RATE,
    'employee': config.EMPLOYEE_THROTTLE_RATE
}

from rest_framework.throttling import UserRateThrottle

class EmployeeThrottle(UserRateThrottle):
    *WHAT TO DO HERE?*

我可以在这个EmployeeThrottle类中以某种方式获取请求并根据该请求内容设置范围。

编辑：例如： UserRateThrottle这里BurstRateThrottle ，如果不知何故，我能得到我的请求可以设置基于该范围。

Answer 1

如果这是每个用户类型，那么您应该使用BaseThrottle作为基类。 我认为这样的事情应该有效吗？

class UserRateThrottle(throttling.BaseThrottle):
    scope = 'user'
    def allow_request(self, request, view):
        return request.user.type == 'user'


class EmployeeRateThrottle(throttling.BaseThrottle):
    scope = 'employee'
    def allow_request(self, request, view):
        return request.user.type == 'employee'

Answer 2

from rest_framework.throttling import UserRateThrottle


class CustomThrottle(UserRateThrottle):
    def __init__(self):
        pass
    def allow_request(self, request, view):
        employee_scope = getattr(view, 'employee', None)
        user_scope = getattr(view, 'user', None)
        if request.user.profile.is_employee:
            if not employee_scope:
                return True
            self.scope = employee_scope
        else:
            if not user_scope:
                return True
            self.scope = user_scope 
            
        self.rate = self.get_rate()
        self.num_requests, self.duration = self.parse_rate(self.rate)
        
        return super().allow_request(request, view)