diff --git a/apps/notification/views/common.py b/apps/notification/views/common.py index f7fbf9fe..1861f39f 100644 --- a/apps/notification/views/common.py +++ b/apps/notification/views/common.py @@ -25,16 +25,15 @@ class SubscribeInfoView(generics.RetrieveAPIView): serializer_class = serializers.SubscribeSerializer -class SubscribeInfoAuthUserView(generics.ListAPIView): +class SubscribeInfoAuthUserView(generics.RetrieveAPIView): """Subscribe info auth user view.""" permission_classes = (permissions.IsAuthenticated,) serializer_class = serializers.SubscribeSerializer + lookup_field = None - def get_queryset(self): - user = self.request.user - queryset = self.filter_queryset(models.Subscriber.objects.all()) - return queryset.filter(user=user) + def get_object(self): + return get_object_or_404(models.Subscriber, user=self.request.user) class UnsubscribeView(generics.UpdateAPIView):