Briq
/
django-polymorphic
6
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Raise ParentAdminNotRegistered instead of KeyError

fix_request_path_info
Diederik van der Boor 2016-07-08 12:06:04 +02:00
parent 8fb1fa924e
commit ac98a8123a
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
polymorphic/admin/childadmin.py
View File

@ -7,6 +7,11 @@ from django.utils import six
from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ugettext_lazy as _
class ParentAdminNotRegistered(RuntimeError):
"The admin site for the model is not registered."
pass
class PolymorphicChildModelAdmin(admin.ModelAdmin): class PolymorphicChildModelAdmin(admin.ModelAdmin):
""" """
The *optional* base class for the admin interface of derived models. The *optional* base class for the admin interface of derived models.
@ -112,7 +117,11 @@ class PolymorphicChildModelAdmin(admin.ModelAdmin):
if parent_model == self.model: if parent_model == self.model:
# when parent_model is in among child_models, just return super instance # when parent_model is in among child_models, just return super instance
return super(PolymorphicChildModelAdmin, self) return super(PolymorphicChildModelAdmin, self)
return self.admin_site._registry.get(parent_model)
try:
return self.admin_site._registry[parent_model]
except KeyError:
raise ParentAdminNotRegistered("No parent admin was registered for a '{0}' model.".format(parent_model))
def response_post_save_add(self, request, obj): def response_post_save_add(self, request, obj):
return self._get_parent_admin().response_post_save_add(request, obj) return self._get_parent_admin().response_post_save_add(request, obj)