时间戳

首页 > 解决方案 > Djoser - 覆盖注册

django-rest-framework - Djoser - 覆盖注册

问题描述

我对如何修改 Djoser 的注册端点感到困惑。我想做的就是向端点添加范围限制，但我不明白如何覆盖它。文档上的这个页面讨论了它：https ://djoser.readthedocs.io/en/2.1.0/adjustment.html 但它似乎已经过时了？今天如何使用 UserViewSet 完成并确保 url 按预期工作？

标签： django-rest-frameworkdjoser

解决方案

您可以做的是继承 djoserUserViewSet并添加您的额外代码。像这样的东西应该工作

# your_views.py

from djoser.views import UserViewSet as DjoserUserViewSet


class UserViewSet(DjoserUserViewSet):

    def get_throttles(self):
        if self.action == "create":
            self.throttle_classes = [YourThrottleClass]
        return super().get_throttles()

然后在你的urls.py你不应该包括djoser.urls在你的urlpatterns

而不是这个（取自他们的文档，您可能有其他网址）：

urlpatterns = [
    (...),
    url(r'^auth/', include('djoser.urls')),
]

在您的urlpatterns（您可能已经定义了路由器）中执行此操作：

# I have use endpoint "auth/users" to keep it similar to the above, but it can be just simple "users"

router = DefaultRouter()
router.register("auth/users", your_views.UserViewSet)  

urlpatterns = [
    (...),
    url(r'^', include(router.urls)),
]

幕后djoser.urls是注册users端点，但它们是内部UserViewSet的，因此您可以使用自己的自定义类。

推荐阅读