Question

我正在尝试创建一个REST API并且在用户注册时陷入困境：基本上我需要在注册之前拥有访问令牌。

这是观点：

class UserViewSet(viewsets.ModelViewSet):

    """
    API endpoint that allows users to be viewed or edited.
    """
    queryset = User.objects.all()
    serializer_class = UserSerializer

    def metadata(self, request):
        """
        Don't include the view description in OPTIONS responses.
        """
        data = super(UserViewSet, self).metadata(request)
        return data

    def create(self, request):
        serializer = self.get_serializer(data=request.DATA, files=request.FILES)

        if serializer.is_valid():
            self.pre_save(serializer.object)
            self.object = serializer.save(force_insert=True)
            self.post_save(self.object, created=True)
            self.object.set_password(self.object.password)
            self.object.save()
            headers = self.get_success_headers(serializer.data)
            return Response(serializer.data, status=status.HTTP_201_CREATED,
                            headers=headers)

        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

这是解决方法：

@api_view(['POST'])
@permission_classes((AllowAny,))
@csrf_exempt
def create_auth(request, format=None):
    data = JSONParser().parse(request)
    serialized = UserSerializer(data=data)

    if serialized.is_valid():
        user = User.objects.create_user(
            serialized.init_data['email'],
            serialized.init_data['username'],
            serialized.init_data['password'],
        )
        user.groups = serialized.init_data['groups']

        user.save()

        serialized_user = UserSerializer(user)
        return Response(serialized_user.data, status=status.HTTP_201_CREATED, headers={"Access-Control-Allow-Origin": "http://127.0.0.1:8000/"})
    else:
        return Response(serialized._errors, status=status.HTTP_400_BAD_REQUEST, headers={"Access-Control-Allow-Origin": "http://127.0.0.1:8000/"})

我的问题是：如何在UserViewSet中指定创建我不需要凭据？或者指定自定义身份验证方法？我不想更改整个视图集的身份验证/权限类。

谢谢，阿迪

修改澄清：应该允许未注册的用户POST注册数据，不应该被允许。经过身份验证的用户可以获取用户列表并更新自己的配置文件...这是默认行为。这就是AllowAny 不选项的原因。在我看来，正确的地方是创建功能，但我没有得到我应该覆盖的内容。

Answer 1

自定义get_queryset方法：

def get_queryset(self):
    if self.request.user.is_superuser:
        return User.objects.all()
    else:
        return User.objects.filter(id=self.request.user.id)

这样，经过身份验证的用户只能检索，修改或删除自己的对象。

指定permission_classes = (AllowAny,)，以便经过身份验证的用户可以创建新用户。

编辑：评论中的进一步说明

以这种方式自定义get_queryset方法意味着：

是的，未经过身份验证的用户可以发送GET请求来检索用户列表，但它将为空，因为返回User.objects.filter（id = self.request.user.id）确保只有信息关于经过身份验证的用户被返回。
这同样适用于其他方法，如果经过身份验证的用户尝试删除另一个用户对象，则会返回一个详细信息：未找到（因为它尝试访问的用户不在查询集中）。
经过身份验证的用户可以对用户对象执行任何操作。

Answer 2

您可以利用Django REST Framework定义自定义权限的能力。您可以在自定义类中指定has_permission和has_object_permission。除了发布到创建端点之外，这将为您提供向anon用户投掷403s的预期行为。它可能看起来像：

class IsAnonCreate(permissions.BasePermission):
    def has_permission(self, request, view):
        if request.method == "POST" and not request.user.is_authenticated():
            return True
        elif not request.user.is_authenticated() and request.method != "POST":
            return False
        elif request.method in permissions.SAFE_METHODS:
            return True

        return False

    def has_object_permission(self, request, view, obj):
        if not request.user.is_authenticated():
            return False
        if request.method in permissions.SAFE_METHODS:
            return True

        return obj.username == request.user.username

如果需要，您可以为经过身份验证的用户添加一些自定义处理。

然后，您需要做的就是将权限类添加到ModelViewSet：

class UserViewSet(viewsets.ModelViewSet):

    """
    API endpoint that allows users to be viewed or edited.
    """
    queryset = User.objects.all()
    serializer_class = UserSerializer
    permission_classes = (IsAnonCreate, )

Answer 3

这是基于@argaen的答案，它对我有用：

class UserViewSet(viewsets.ModelViewSet):
    serializer_class = UserSerializer
    permission_classes = (AllowAny,)
    authentication_classes = (NoAuthentication,)
    filter_backends = (filters.DjangoFilterBackend,)
    filter_fields = ('id', 'email', 'name')

    def get_queryset(self):
        user = TokenAuthentication().authenticate(self.request)
        if user is not None:
            user = user[0]
            if user.is_superuser:
                return get_user_model().objects.all()
            else:
                return get_user_model().objects.filter(id=user.id)

        return get_user_model().objects.none()

用于在viewset中创建的Django-rest-framework权限

3 个答案: