def get_object(self): try: user = User.objects(id=self.kwargs['id'])[0] except IndexError: raise Http404 self.check_object_permissions(self.request, user) return user
def get(self, request, pk=None, *args, **kwargs): if not pk: users = User.objects() serializer = UserSerializer(users, many=True) return Response(serializer.data) try: user = User.objects.get(pk=pk) except: return Response({}, status=status.HTTP_204_NO_CONTENT) else: serializer = UserSerializer(user) return Response(serializer.data)