Merge pull request #159 from ixc/object-does-not-exist

Raise Http404 if object does not exist.
jazzband · Oct 1, 2015 · f4898c3 · f4898c3
2 parents 670b2be + bcc39a9
commit f4898c3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/polymorphic/admin.py b/polymorphic/admin.py
@@ -198,7 +198,11 @@ def get_child_type_choices(self, request, action):
 
 
     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'])