|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235 |
- """
- The child admin displays the change/delete view of the subclass model.
- """
- import inspect
-
- from django.contrib import admin
- from django.urls import resolve
- from django.utils.translation import gettext_lazy as _
-
- from polymorphic.utils import get_base_polymorphic_model
-
- from ..admin import PolymorphicParentModelAdmin
-
-
- class ParentAdminNotRegistered(RuntimeError):
- "The admin site for the model is not registered."
-
-
- class PolymorphicChildModelAdmin(admin.ModelAdmin):
- """
- The *optional* base class for the admin interface of derived models.
-
- This base class defines some convenience behavior for the admin interface:
-
- * It corrects the breadcrumbs in the admin pages.
- * It adds the base model to the template lookup paths.
- * It allows to set ``base_form`` so the derived class will automatically include other fields in the form.
- * It allows to set ``base_fieldsets`` so the derived class will automatically display any extra fields.
- """
-
- #: The base model that the class uses (auto-detected if not set explicitly)
- base_model = None
-
- #: By setting ``base_form`` instead of ``form``, any subclass fields are automatically added to the form.
- #: This is useful when your model admin class is inherited by others.
- base_form = None
-
- #: By setting ``base_fieldsets`` instead of ``fieldsets``,
- #: any subclass fields can be automatically added.
- #: This is useful when your model admin class is inherited by others.
- base_fieldsets = None
-
- #: Default title for extra fieldset
- extra_fieldset_title = _("Contents")
-
- #: Whether the child admin model should be visible in the admin index page.
- show_in_index = False
-
- def __init__(self, model, admin_site, *args, **kwargs):
- super().__init__(model, admin_site, *args, **kwargs)
-
- if self.base_model is None:
- self.base_model = get_base_polymorphic_model(model)
-
- def get_form(self, request, obj=None, **kwargs):
- # The django admin validation requires the form to have a 'class Meta: model = ..'
- # attribute, or it will complain that the fields are missing.
- # However, this enforces all derived ModelAdmin classes to redefine the model as well,
- # because they need to explicitly set the model again - it will stick with the base model.
- #
- # Instead, pass the form unchecked here, because the standard ModelForm will just work.
- # If the derived class sets the model explicitly, respect that setting.
- kwargs.setdefault("form", self.base_form or self.form)
-
- # prevent infinite recursion when this is called from get_subclass_fields
- if not self.fieldsets and not self.fields:
- kwargs.setdefault("fields", "__all__")
-
- return super().get_form(request, obj, **kwargs)
-
- def get_model_perms(self, request):
- match = resolve(request.path_info)
-
- if (
- not self.show_in_index
- and match.app_name == "admin"
- and match.url_name in ("index", "app_list")
- ):
- return {"add": False, "change": False, "delete": False}
- return super().get_model_perms(request)
-
- @property
- def change_form_template(self):
- opts = self.model._meta
- app_label = opts.app_label
-
- # Pass the base options
- base_opts = self.base_model._meta
- base_app_label = base_opts.app_label
-
- return [
- f"admin/{app_label}/{opts.object_name.lower()}/change_form.html",
- "admin/%s/change_form.html" % app_label,
- # Added:
- "admin/%s/%s/change_form.html" % (base_app_label, base_opts.object_name.lower()),
- "admin/%s/change_form.html" % base_app_label,
- "admin/polymorphic/change_form.html",
- "admin/change_form.html",
- ]
-
- @property
- def delete_confirmation_template(self):
- opts = self.model._meta
- app_label = opts.app_label
-
- # Pass the base options
- base_opts = self.base_model._meta
- base_app_label = base_opts.app_label
-
- return [
- "admin/%s/%s/delete_confirmation.html" % (app_label, opts.object_name.lower()),
- "admin/%s/delete_confirmation.html" % app_label,
- # Added:
- "admin/%s/%s/delete_confirmation.html"
- % (base_app_label, base_opts.object_name.lower()),
- "admin/%s/delete_confirmation.html" % base_app_label,
- "admin/polymorphic/delete_confirmation.html",
- "admin/delete_confirmation.html",
- ]
-
- @property
- def object_history_template(self):
- opts = self.model._meta
- app_label = opts.app_label
-
- # Pass the base options
- base_opts = self.base_model._meta
- base_app_label = base_opts.app_label
-
- return [
- f"admin/{app_label}/{opts.object_name.lower()}/object_history.html",
- "admin/%s/object_history.html" % app_label,
- # Added:
- "admin/%s/%s/object_history.html" % (base_app_label, base_opts.object_name.lower()),
- "admin/%s/object_history.html" % base_app_label,
- "admin/polymorphic/object_history.html",
- "admin/object_history.html",
- ]
-
- def _get_parent_admin(self):
- # this returns parent admin instance on which to call response_post_save methods
- parent_model = self.model._meta.get_field("polymorphic_ctype").model
- if parent_model == self.model:
- # when parent_model is in among child_models, just return super instance
- return super()
-
- try:
- return self.admin_site._registry[parent_model]
- except KeyError:
- # Admin is not registered for polymorphic_ctype model, but perhaps it's registered
- # for a intermediate proxy model, between the parent_model and this model.
- for klass in inspect.getmro(self.model):
- if not issubclass(klass, parent_model):
- continue # e.g. found a mixin.
-
- # Fetch admin instance for model class, see if it's a possible candidate.
- model_admin = self.admin_site._registry.get(klass)
- if model_admin is not None and isinstance(
- model_admin, PolymorphicParentModelAdmin
- ):
- return model_admin # Success!
-
- # If we get this far without returning there is no admin available
- raise ParentAdminNotRegistered(
- f"No parent admin was registered for a '{parent_model}' model."
- )
-
- def response_post_save_add(self, request, obj):
- return self._get_parent_admin().response_post_save_add(request, obj)
-
- def response_post_save_change(self, request, obj):
- return self._get_parent_admin().response_post_save_change(request, obj)
-
- def render_change_form(self, request, context, add=False, change=False, form_url="", obj=None):
- context.update({"base_opts": self.base_model._meta})
- return super().render_change_form(
- request, context, add=add, change=change, form_url=form_url, obj=obj
- )
-
- def delete_view(self, request, object_id, context=None):
- extra_context = {"base_opts": self.base_model._meta}
- return super().delete_view(request, object_id, extra_context)
-
- def history_view(self, request, object_id, extra_context=None):
- # Make sure the history view can also display polymorphic breadcrumbs
- context = {"base_opts": self.base_model._meta}
- if extra_context:
- context.update(extra_context)
- return super().history_view(request, object_id, extra_context=context)
-
- # ---- Extra: improving the form/fieldset default display ----
-
- def get_base_fieldsets(self, request, obj=None):
- return self.base_fieldsets
-
- def get_fieldsets(self, request, obj=None):
- base_fieldsets = self.get_base_fieldsets(request, obj)
-
- # If subclass declares fieldsets or fields, this is respected
- if self.fieldsets or self.fields or not self.base_fieldsets:
- return super().get_fieldsets(request, obj)
-
- # Have a reasonable default fieldsets,
- # where the subclass fields are automatically included.
- other_fields = self.get_subclass_fields(request, obj)
-
- if other_fields:
- return (
- base_fieldsets[0],
- (self.extra_fieldset_title, {"fields": other_fields}),
- ) + base_fieldsets[1:]
- else:
- return base_fieldsets
-
- def get_subclass_fields(self, request, obj=None):
- # Find out how many fields would really be on the form,
- # if it weren't restricted by declared fields.
- exclude = list(self.exclude or [])
- exclude.extend(self.get_readonly_fields(request, obj))
-
- # By not declaring the fields/form in the base class,
- # get_form() will populate the form with all available fields.
- form = self.get_form(request, obj, exclude=exclude)
- subclass_fields = list(form.base_fields.keys()) + list(
- self.get_readonly_fields(request, obj)
- )
-
- # Find which fields are not part of the common fields.
- for fieldset in self.get_base_fieldsets(request, obj):
- for field in fieldset[1]["fields"]:
- try:
- subclass_fields.remove(field)
- except ValueError:
- pass # field not found in form, Django will raise exception later.
- return subclass_fields
|