Django 中的自定义权限

Question

在 Django Rest 框架中，我们可以验证诸如 (isAuthenticated, isAdminUser...) 之类的权限，但是我们如何添加自定义权限并确定 django 可以使用哪些权限？

我真的很想了解背后发生了什么（我没有找到解释这一点的文档）：

@permission_classes([IsAdminUser])

谢谢

Answer 1

如果您使用VSCode ， hover 而不是@permission_classes([IsAdminUser]) ，请单击命令（在您的键盘上）。

您可以查看幕后发生的事情，播放并创建您的自定义 Django 版本（不推荐），或者您可以覆盖 function。

Answer 2

编写您自己的权限，如下所示：

def permission_valid_token(func):
    # first arg is the viewset, second is the request
    def wrapper(*args, **kwargs):
        valid_token, user_token = test_token_authorization(args[1].headers)
        if not valid_token:
            return Response(status=status.HTTP_401_UNAUTHORIZED)
        return func(*args, **kwargs)
    return wrapper

这是我在应用程序中使用的权限，可能您将拥有 valid_token 部分

你在视图中导入它们

from file_that_you_use.permissions import permission_valid_token

而你用作装饰者

class WeatherViewSet(viewsets.ViewSet):
    @permission_valid_token
    def list(self, request):

例如，这只是一个视图集，您可以使用通用视图集或任何您想要的。