如何应用限制只是为了在 DRF 视图集中创建操作？

Question

I have a viewset and I want to apply throttling to only create action of that viewset and I don't want it to be applied to update , destroy , retrieve and etc...

class UserViewSet(viewsets.ModelViewSet):
    # should only be applied to the create action
    throttle_classes = [SomeThrottle] 
    ...

Answer 1

As described in Introspecting ViewSet actions [DRF docs] you can inspect the action attribute to set the throttle_classes based on the current action by overriding the get_throttles method:

class UserViewSet(viewsets.ModelViewSet):
    def get_throttles(self):
        if self.action == 'create':
            throttle_classes = [SomeThrottle]
        else:
            throttle_classes = []  # No throttle for other actions
        return [throttle() for throttle in throttle_classes]

Answer 2

if use ScopedRateThrottle instead of Custom throttles

# settings

    'DEFAULT_THROTTLE_CLASSES': [
        'rest_framework.throttling.ScopedRateThrottle',
    ],
    'DEFAULT_THROTTLE_RATES': {
        'submit': '1/minute',
    }

class SomeViewSet(mixins.CreateModelMixin, GenericViewSet)
    ....

    def get_throttles(self):
        if self.action == 'create':
            self.throttle_scope = 'submit'
        return super().get_throttles()