|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142 |
- """
- Rendering utils for admin forms;
-
- This makes sure that admin fieldsets/layout settings are exported to the template.
- """
- import json
-
- from django.contrib.admin.helpers import AdminField, InlineAdminForm, InlineAdminFormSet
- from django.utils.encoding import force_str
- from django.utils.text import capfirst
- from django.utils.translation import gettext
-
- from polymorphic.formsets import BasePolymorphicModelFormSet
-
-
- class PolymorphicInlineAdminForm(InlineAdminForm):
- """
- Expose the admin configuration for a form
- """
-
- def polymorphic_ctype_field(self):
- return AdminField(self.form, "polymorphic_ctype", False)
-
- @property
- def is_empty(self):
- return "__prefix__" in self.form.prefix
-
-
- class PolymorphicInlineAdminFormSet(InlineAdminFormSet):
- """
- Internally used class to expose the formset in the template.
- """
-
- def __init__(self, *args, **kwargs):
- # Assigned later via PolymorphicInlineSupportMixin later.
- self.request = kwargs.pop("request", None)
- self.obj = kwargs.pop("obj", None)
- super().__init__(*args, **kwargs)
-
- def __iter__(self):
- """
- Output all forms using the proper subtype settings.
- """
- for form, original in zip(self.formset.initial_forms, self.formset.get_queryset()):
- # Output the form
- model = original.get_real_instance_class()
- child_inline = self.opts.get_child_inline_instance(model)
- view_on_site_url = self.opts.get_view_on_site_url(original)
-
- yield PolymorphicInlineAdminForm(
- formset=self.formset,
- form=form,
- fieldsets=self.get_child_fieldsets(child_inline),
- prepopulated_fields=self.get_child_prepopulated_fields(child_inline),
- original=original,
- readonly_fields=self.get_child_readonly_fields(child_inline),
- model_admin=child_inline,
- view_on_site_url=view_on_site_url,
- )
-
- # Extra rows, and empty prefixed forms.
- for form in self.formset.extra_forms + self.formset.empty_forms:
- model = form._meta.model
- child_inline = self.opts.get_child_inline_instance(model)
- yield PolymorphicInlineAdminForm(
- formset=self.formset,
- form=form,
- fieldsets=self.get_child_fieldsets(child_inline),
- prepopulated_fields=self.get_child_prepopulated_fields(child_inline),
- original=None,
- readonly_fields=self.get_child_readonly_fields(child_inline),
- model_admin=child_inline,
- )
-
- def get_child_fieldsets(self, child_inline):
- return list(child_inline.get_fieldsets(self.request, self.obj) or ())
-
- def get_child_readonly_fields(self, child_inline):
- return list(child_inline.get_readonly_fields(self.request, self.obj))
-
- def get_child_prepopulated_fields(self, child_inline):
- fields = self.prepopulated_fields.copy()
- fields.update(child_inline.get_prepopulated_fields(self.request, self.obj))
- return fields
-
- def inline_formset_data(self):
- """
- A JavaScript data structure for the JavaScript code
- This overrides the default Django version to add the ``childTypes`` data.
- """
- verbose_name = self.opts.verbose_name
- return json.dumps(
- {
- "name": "#%s" % self.formset.prefix,
- "options": {
- "prefix": self.formset.prefix,
- "addText": gettext("Add another %(verbose_name)s")
- % {"verbose_name": capfirst(verbose_name)},
- "childTypes": [
- {
- "type": model._meta.model_name,
- "name": force_str(model._meta.verbose_name),
- }
- for model in self.formset.child_forms.keys()
- ],
- "deleteText": gettext("Remove"),
- },
- }
- )
-
-
- class PolymorphicInlineSupportMixin:
- """
- A Mixin to add to the regular admin, so it can work with our polymorphic inlines.
-
- This mixin needs to be included in the admin that hosts the ``inlines``.
- It makes sure the generated admin forms have different fieldsets/fields
- depending on the polymorphic type of the form instance.
-
- This is achieved by overwriting :func:`get_inline_formsets` to return
- an :class:`PolymorphicInlineAdminFormSet` instead of a standard Django
- :class:`~django.contrib.admin.helpers.InlineAdminFormSet` for the polymorphic formsets.
- """
-
- def get_inline_formsets(self, request, formsets, inline_instances, obj=None, *args, **kwargs):
- """
- Overwritten version to produce the proper admin wrapping for the
- polymorphic inline formset. This fixes the media and form appearance
- of the inline polymorphic models.
- """
- inline_admin_formsets = super().get_inline_formsets(
- request, formsets, inline_instances, obj=obj
- )
-
- for admin_formset in inline_admin_formsets:
- if isinstance(admin_formset.formset, BasePolymorphicModelFormSet):
- # This is a polymorphic formset, which belongs to our inline.
- # Downcast the admin wrapper that generates the form fields.
- admin_formset.__class__ = PolymorphicInlineAdminFormSet
- admin_formset.request = request
- admin_formset.obj = obj
- return inline_admin_formsets
|