import email.parser
-from django.http import Http404
+from rest_framework.generics import get_object_or_404
from rest_framework.generics import ListAPIView
from rest_framework.serializers import SerializerMethodField
lookup_url_kwarg = 'pk'
def get_queryset(self):
- if not Cover.objects.filter(pk=self.kwargs['pk']).exists():
- raise Http404
+ get_object_or_404(Cover, pk=self.kwargs['pk'])
return CoverComment.objects.filter(
cover=self.kwargs['pk']
lookup_url_kwarg = 'patch_id'
def get_queryset(self):
- if not Patch.objects.filter(id=self.kwargs['patch_id']).exists():
- raise Http404
+ get_object_or_404(Patch, id=self.kwargs['patch_id'])
return PatchComment.objects.filter(
patch=self.kwargs['patch_id']