From bcc39a90866bee104d389f1d2a52279387999fda Mon Sep 17 00:00:00 2001 From: Tai Lee Date: Tue, 22 Sep 2015 09:55:03 +1000 Subject: [PATCH] Raise Http404 if object does not exist. --- polymorphic/admin.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/polymorphic/admin.py b/polymorphic/admin.py index bf3e6cd..216f74f 100644 --- a/polymorphic/admin.py +++ b/polymorphic/admin.py @@ -198,7 +198,11 @@ class PolymorphicParentModelAdmin(admin.ModelAdmin): def _get_real_admin(self, object_id): - obj = self.model.objects.non_polymorphic().values('polymorphic_ctype').get(pk=object_id) + try: + obj = self.model.objects.non_polymorphic() \ + .values('polymorphic_ctype').get(pk=object_id) + except self.model.DoesNotExist: + raise Http404 return self._get_real_admin_by_ct(obj['polymorphic_ctype'])