536 lines
21 KiB
Python
536 lines
21 KiB
Python
"""
|
|
ModelAdmin code to display polymorphic models.
|
|
"""
|
|
from django import forms
|
|
from django.conf.urls import patterns, url
|
|
from django.contrib import admin
|
|
from django.contrib.admin.helpers import AdminForm, AdminErrorList
|
|
from django.contrib.admin.sites import AdminSite
|
|
from django.contrib.admin.widgets import AdminRadioSelect
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.core.exceptions import PermissionDenied
|
|
from django.core.urlresolvers import RegexURLResolver
|
|
from django.http import Http404, HttpResponseRedirect
|
|
from django.shortcuts import render_to_response
|
|
from django.template.context import RequestContext
|
|
from django.utils import six
|
|
from django.utils.encoding import force_text
|
|
from django.utils.http import urlencode
|
|
from django.utils.safestring import mark_safe
|
|
from django.utils.translation import ugettext_lazy as _
|
|
try:
|
|
# Django 1.6 implements this
|
|
from django.contrib.admin.templatetags.admin_urls import add_preserved_filters
|
|
except ImportError:
|
|
def add_preserved_filters(context, form_url):
|
|
return form_url
|
|
|
|
|
|
__all__ = (
|
|
'PolymorphicModelChoiceForm', 'PolymorphicParentModelAdmin',
|
|
'PolymorphicChildModelAdmin', 'PolymorphicChildModelFilter'
|
|
)
|
|
|
|
|
|
class RegistrationClosed(RuntimeError):
|
|
"The admin model can't be registered anymore at this point."
|
|
pass
|
|
|
|
class ChildAdminNotRegistered(RuntimeError):
|
|
"The admin site for the model is not registered."
|
|
pass
|
|
|
|
|
|
class PolymorphicModelChoiceForm(forms.Form):
|
|
"""
|
|
The default form for the ``add_type_form``. Can be overwritten and replaced.
|
|
"""
|
|
#: Define the label for the radiofield
|
|
type_label = _("Type")
|
|
|
|
ct_id = forms.ChoiceField(label=type_label, widget=AdminRadioSelect(attrs={'class': 'radiolist'}))
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
# Allow to easily redefine the label (a commonly expected usecase)
|
|
super(PolymorphicModelChoiceForm, self).__init__(*args, **kwargs)
|
|
self.fields['ct_id'].label = self.type_label
|
|
|
|
|
|
class PolymorphicChildModelFilter(admin.SimpleListFilter):
|
|
"""
|
|
An admin list filter for the PolymorphicParentModelAdmin which enables
|
|
filtering by its child models.
|
|
"""
|
|
title = _('Content type')
|
|
parameter_name = 'polymorphic_ctype'
|
|
|
|
def lookups(self, request, model_admin):
|
|
return model_admin.get_child_type_choices()
|
|
|
|
def queryset(self, request, queryset):
|
|
try:
|
|
value = int(self.value())
|
|
except TypeError:
|
|
value = None
|
|
if value:
|
|
# ensure the content type is allowed
|
|
for choice_value, _ in self.lookup_choices:
|
|
if choice_value == value:
|
|
return queryset.filter(polymorphic_ctype_id=choice_value)
|
|
raise PermissionDenied(
|
|
'Invalid ContentType "{0}". It must be registered as child model.'.format(value))
|
|
return queryset
|
|
|
|
|
|
class PolymorphicParentModelAdmin(admin.ModelAdmin):
|
|
"""
|
|
A admin interface that can displays different change/delete pages, depending on the polymorphic model.
|
|
To use this class, two variables need to be defined:
|
|
|
|
* :attr:`base_model` should
|
|
* :attr:`child_models` should be a list of (Model, Admin) tuples
|
|
|
|
Alternatively, the following methods can be implemented:
|
|
|
|
* :func:`get_child_models` should return a list of (Model, ModelAdmin) tuples
|
|
* optionally, :func:`get_child_type_choices` can be overwritten to refine the choices for the add dialog.
|
|
|
|
This class needs to be inherited by the model admin base class that is registered in the site.
|
|
The derived models should *not* register the ModelAdmin, but instead it should be returned by :func:`get_child_models`.
|
|
"""
|
|
|
|
#: The base model that the class uses
|
|
base_model = None
|
|
|
|
#: The child models that should be displayed
|
|
child_models = None
|
|
|
|
#: Whether the list should be polymorphic too, leave to ``False`` to optimize
|
|
polymorphic_list = False
|
|
|
|
add_type_template = None
|
|
add_type_form = PolymorphicModelChoiceForm
|
|
|
|
#: The regular expression to filter the primary key in the URL.
|
|
#: This accepts only numbers as defensive measure against catch-all URLs.
|
|
#: If your primary key consists of string values, update this regular expression.
|
|
pk_regex = '(\d+)'
|
|
|
|
|
|
def __init__(self, model, admin_site, *args, **kwargs):
|
|
super(PolymorphicParentModelAdmin, self).__init__(model, admin_site, *args, **kwargs)
|
|
self._child_admin_site = AdminSite(name=self.admin_site.name)
|
|
self._is_setup = False
|
|
|
|
|
|
def _lazy_setup(self):
|
|
if self._is_setup:
|
|
return
|
|
|
|
# By not having this in __init__() there is less stress on import dependencies as well,
|
|
# considering an advanced use cases where a plugin system scans for the child models.
|
|
child_models = self.get_child_models()
|
|
for Model, Admin in child_models:
|
|
self.register_child(Model, Admin)
|
|
self._child_models = dict(child_models)
|
|
|
|
# This is needed to deal with the improved ForeignKeyRawIdWidget in Django 1.4 and perhaps other widgets too.
|
|
# The ForeignKeyRawIdWidget checks whether the referenced model is registered in the admin, otherwise it displays itself as a textfield.
|
|
# As simple solution, just make sure all parent admin models are also know in the child admin site.
|
|
# This should be done after all parent models are registered off course.
|
|
complete_registry = self.admin_site._registry.copy()
|
|
complete_registry.update(self._child_admin_site._registry)
|
|
|
|
self._child_admin_site._registry = complete_registry
|
|
self._is_setup = True
|
|
|
|
|
|
def register_child(self, model, model_admin):
|
|
"""
|
|
Register a model with admin to display.
|
|
"""
|
|
# After the get_urls() is called, the URLs of the child model can't be exposed anymore to the Django URLconf,
|
|
# which also means that a "Save and continue editing" button won't work.
|
|
if self._is_setup:
|
|
raise RegistrationClosed("The admin model can't be registered anymore at this point.")
|
|
|
|
if not issubclass(model, self.base_model):
|
|
raise TypeError("{0} should be a subclass of {1}".format(model.__name__, self.base_model.__name__))
|
|
if not issubclass(model_admin, admin.ModelAdmin):
|
|
raise TypeError("{0} should be a subclass of {1}".format(model_admin.__name__, admin.ModelAdmin.__name__))
|
|
|
|
self._child_admin_site.register(model, model_admin)
|
|
|
|
|
|
def get_child_models(self):
|
|
"""
|
|
Return the derived model classes which this admin should handle.
|
|
This should return a list of tuples, exactly like :attr:`child_models` is.
|
|
|
|
The model classes can be retrieved as ``base_model.__subclasses__()``,
|
|
a setting in a config file, or a query of a plugin registration system at your option
|
|
"""
|
|
if self.child_models is None:
|
|
raise NotImplementedError("Implement get_child_models() or child_models")
|
|
|
|
return self.child_models
|
|
|
|
|
|
def get_child_type_choices(self):
|
|
"""
|
|
Return a list of polymorphic types which can be added.
|
|
"""
|
|
choices = []
|
|
for model, _ in self.get_child_models():
|
|
ct = ContentType.objects.get_for_model(model, for_concrete_model=False)
|
|
choices.append((ct.id, model._meta.verbose_name))
|
|
return choices
|
|
|
|
|
|
def _get_real_admin(self, object_id):
|
|
obj = self.model.objects.non_polymorphic().values('polymorphic_ctype').get(pk=object_id)
|
|
return self._get_real_admin_by_ct(obj['polymorphic_ctype'])
|
|
|
|
|
|
def _get_real_admin_by_ct(self, ct_id):
|
|
try:
|
|
ct = ContentType.objects.get_for_id(ct_id)
|
|
except ContentType.DoesNotExist as e:
|
|
raise Http404(e) # Handle invalid GET parameters
|
|
|
|
model_class = ct.model_class()
|
|
if not model_class:
|
|
raise Http404("No model found for '{0}.{1}'.".format(*ct.natural_key())) # Handle model deletion
|
|
|
|
return self._get_real_admin_by_model(model_class)
|
|
|
|
|
|
def _get_real_admin_by_model(self, model_class):
|
|
# In case of a ?ct_id=### parameter, the view is already checked for permissions.
|
|
# Hence, make sure this is a derived object, or risk exposing other admin interfaces.
|
|
if model_class not in self._child_models:
|
|
raise PermissionDenied("Invalid model '{0}', it must be registered as child model.".format(model_class))
|
|
|
|
try:
|
|
# HACK: the only way to get the instance of an model admin,
|
|
# is to read the registry of the AdminSite.
|
|
return self._child_admin_site._registry[model_class]
|
|
except KeyError:
|
|
raise ChildAdminNotRegistered("No child admin site was registered for a '{0}' model.".format(model_class))
|
|
|
|
|
|
def queryset(self, request):
|
|
return self.get_queryset(request)
|
|
|
|
|
|
def get_queryset(self, request):
|
|
# optimize the list display.
|
|
qs = super(PolymorphicParentModelAdmin, self).get_queryset(request)
|
|
if not self.polymorphic_list:
|
|
qs = qs.non_polymorphic()
|
|
return qs
|
|
|
|
|
|
def add_view(self, request, form_url='', extra_context=None):
|
|
"""Redirect the add view to the real admin."""
|
|
ct_id = int(request.GET.get('ct_id', 0))
|
|
if not ct_id:
|
|
# Display choices
|
|
return self.add_type_view(request)
|
|
else:
|
|
real_admin = self._get_real_admin_by_ct(ct_id)
|
|
# rebuild form_url, otherwise libraries below will override it.
|
|
form_url = add_preserved_filters({
|
|
'preserved_filters': urlencode({'ct_id': ct_id}),
|
|
'opts': self.model._meta},
|
|
form_url
|
|
)
|
|
return real_admin.add_view(request, form_url, extra_context)
|
|
|
|
|
|
def change_view(self, request, object_id, *args, **kwargs):
|
|
"""Redirect the change view to the real admin."""
|
|
# between Django 1.3 and 1.4 this method signature differs. Hence the *args, **kwargs
|
|
real_admin = self._get_real_admin(object_id)
|
|
return real_admin.change_view(request, object_id, *args, **kwargs)
|
|
|
|
|
|
def delete_view(self, request, object_id, extra_context=None):
|
|
"""Redirect the delete view to the real admin."""
|
|
real_admin = self._get_real_admin(object_id)
|
|
return real_admin.delete_view(request, object_id, extra_context)
|
|
|
|
|
|
def get_urls(self):
|
|
"""
|
|
Expose the custom URLs for the subclasses and the URL resolver.
|
|
"""
|
|
urls = super(PolymorphicParentModelAdmin, self).get_urls()
|
|
info = self.model._meta.app_label, self.model._meta.module_name
|
|
|
|
# Patch the change URL so it's not a big catch-all; allowing all custom URLs to be added to the end.
|
|
# The url needs to be recreated, patching url.regex is not an option Django 1.4's LocaleRegexProvider changed it.
|
|
new_change_url = url(r'^{0}/$'.format(self.pk_regex), self.admin_site.admin_view(self.change_view), name='{0}_{1}_change'.format(*info))
|
|
for i, oldurl in enumerate(urls):
|
|
if oldurl.name == new_change_url.name:
|
|
urls[i] = new_change_url
|
|
|
|
# Define the catch-all for custom views
|
|
custom_urls = patterns('',
|
|
url(r'^(?P<path>.+)$', self.admin_site.admin_view(self.subclass_view))
|
|
)
|
|
|
|
# At this point. all admin code needs to be known.
|
|
self._lazy_setup()
|
|
|
|
# Add reverse names for all polymorphic models, so the delete button and "save and add" just work.
|
|
# These definitions are masked by the definition above, since it needs special handling (and a ct_id parameter).
|
|
dummy_urls = []
|
|
for model, _ in self.get_child_models():
|
|
admin = self._get_real_admin_by_model(model)
|
|
dummy_urls += admin.get_urls()
|
|
|
|
return urls + custom_urls + dummy_urls
|
|
|
|
|
|
def subclass_view(self, request, path):
|
|
"""
|
|
Forward any request to a custom view of the real admin.
|
|
"""
|
|
ct_id = int(request.GET.get('ct_id', 0))
|
|
if not ct_id:
|
|
# See if the path started with an ID.
|
|
try:
|
|
pos = path.find('/')
|
|
object_id = long(path[0:pos])
|
|
except ValueError:
|
|
raise Http404("No ct_id parameter, unable to find admin subclass for path '{0}'.".format(path))
|
|
|
|
ct_id = self.model.objects.values_list('polymorphic_ctype_id', flat=True).get(pk=object_id)
|
|
|
|
|
|
real_admin = self._get_real_admin_by_ct(ct_id)
|
|
resolver = RegexURLResolver('^', real_admin.urls)
|
|
resolvermatch = resolver.resolve(path) # May raise Resolver404
|
|
if not resolvermatch:
|
|
raise Http404("No match for path '{0}' in admin subclass.".format(path))
|
|
|
|
return resolvermatch.func(request, *resolvermatch.args, **resolvermatch.kwargs)
|
|
|
|
|
|
def add_type_view(self, request, form_url=''):
|
|
"""
|
|
Display a choice form to select which page type to add.
|
|
"""
|
|
if not self.has_add_permission(request):
|
|
raise PermissionDenied
|
|
|
|
extra_qs = ''
|
|
if request.META['QUERY_STRING']:
|
|
extra_qs = '&' + request.META['QUERY_STRING']
|
|
|
|
choices = self.get_child_type_choices()
|
|
if len(choices) == 1:
|
|
return HttpResponseRedirect('?ct_id={0}{1}'.format(choices[0][0], extra_qs))
|
|
|
|
# Create form
|
|
form = self.add_type_form(
|
|
data=request.POST if request.method == 'POST' else None,
|
|
initial={'ct_id': choices[0][0]}
|
|
)
|
|
form.fields['ct_id'].choices = choices
|
|
|
|
if form.is_valid():
|
|
return HttpResponseRedirect('?ct_id={0}{1}'.format(form.cleaned_data['ct_id'], extra_qs))
|
|
|
|
# Wrap in all admin layout
|
|
fieldsets = ((None, {'fields': ('ct_id',)}),)
|
|
adminForm = AdminForm(form, fieldsets, {}, model_admin=self)
|
|
media = self.media + adminForm.media
|
|
opts = self.model._meta
|
|
|
|
context = {
|
|
'title': _('Add %s') % force_text(opts.verbose_name),
|
|
'adminform': adminForm,
|
|
'is_popup': "_popup" in request.REQUEST,
|
|
'media': mark_safe(media),
|
|
'errors': AdminErrorList(form, ()),
|
|
'app_label': opts.app_label,
|
|
}
|
|
return self.render_add_type_form(request, context, form_url)
|
|
|
|
|
|
def render_add_type_form(self, request, context, form_url=''):
|
|
"""
|
|
Render the page type choice form.
|
|
"""
|
|
opts = self.model._meta
|
|
app_label = opts.app_label
|
|
context.update({
|
|
'has_change_permission': self.has_change_permission(request),
|
|
'form_url': mark_safe(form_url),
|
|
'opts': opts,
|
|
'add': True,
|
|
'save_on_top': self.save_on_top,
|
|
})
|
|
if hasattr(self.admin_site, 'root_path'):
|
|
context['root_path'] = self.admin_site.root_path # Django < 1.4
|
|
context_instance = RequestContext(request, current_app=self.admin_site.name)
|
|
return render_to_response(self.add_type_template or [
|
|
"admin/%s/%s/add_type_form.html" % (app_label, opts.object_name.lower()),
|
|
"admin/%s/add_type_form.html" % app_label,
|
|
"admin/polymorphic/add_type_form.html", # added default here
|
|
"admin/add_type_form.html"
|
|
], context, context_instance=context_instance)
|
|
|
|
|
|
@property
|
|
def change_list_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/change_list.html" % (app_label, opts.object_name.lower()),
|
|
"admin/%s/change_list.html" % app_label,
|
|
# Added base class:
|
|
"admin/%s/%s/change_list.html" % (base_app_label, base_opts.object_name.lower()),
|
|
"admin/%s/change_list.html" % base_app_label,
|
|
"admin/change_list.html"
|
|
]
|
|
|
|
|
|
|
|
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`` attribute must be set.
|
|
"""
|
|
base_model = None
|
|
base_form = None
|
|
base_fieldsets = None
|
|
extra_fieldset_title = _("Contents") # Default title for extra fieldset
|
|
|
|
|
|
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 in django 1.6+
|
|
if not self.declared_fieldsets:
|
|
kwargs.setdefault('fields', None)
|
|
|
|
return super(PolymorphicChildModelAdmin, self).get_form(request, obj, **kwargs)
|
|
|
|
|
|
@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 [
|
|
"admin/%s/%s/change_form.html" % (app_label, opts.object_name.lower()),
|
|
"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"
|
|
]
|
|
|
|
|
|
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(PolymorphicChildModelAdmin, self).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(PolymorphicChildModelAdmin, self).delete_view(request, object_id, extra_context)
|
|
|
|
|
|
# ---- Extra: improving the form/fieldset default display ----
|
|
|
|
def get_fieldsets(self, request, obj=None):
|
|
# If subclass declares fieldsets, this is respected
|
|
if self.declared_fieldsets or not self.base_fieldsets:
|
|
return super(PolymorphicChildModelAdmin, self).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 (
|
|
self.base_fieldsets[0],
|
|
(self.extra_fieldset_title, {'fields': other_fields}),
|
|
) + self.base_fieldsets[1:]
|
|
else:
|
|
return self.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(six.iterkeys(form.base_fields)) + list(self.get_readonly_fields(request, obj))
|
|
|
|
# Find which fields are not part of the common fields.
|
|
for fieldset in self.base_fieldsets:
|
|
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
|