return 404 in DestroyModelMixin

In case self.get_object() returns None we should return a 404.
This commit is contained in:
Rui Lima 2018-07-08 18:26:31 +01:00 committed by GitHub
parent 38b3d0109b
commit f9de3913ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -90,6 +90,8 @@ class DestroyModelMixin(object):
""" """
def destroy(self, request, *args, **kwargs): def destroy(self, request, *args, **kwargs):
instance = self.get_object() instance = self.get_object()
if not instance:
return Response(status=status.HTTP_404_NOT_FOUND)
self.perform_destroy(instance) self.perform_destroy(instance)
return Response(status=status.HTTP_204_NO_CONTENT) return Response(status=status.HTTP_204_NO_CONTENT)