126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
|
|
from django.contrib.admin.exceptions import DisallowedModelAdminToField
|
|
from django.contrib.admin.utils import flatten_fieldsets, unquote
|
|
from django.core.exceptions import PermissionDenied
|
|
from django.template.response import TemplateResponse
|
|
from django.contrib.admin.options import TO_FIELD_VAR
|
|
from django.utils.translation import gettext as _
|
|
|
|
|
|
class AdminConfirmMixin(object):
|
|
"""Generic AdminConfirm Mixin"""
|
|
|
|
change_needs_confirmation = False
|
|
|
|
# Custom templates (designed to be over-ridden in subclasses)
|
|
change_confirmation_template = None
|
|
|
|
def render_change_confirmation(self, request, context):
|
|
opts = self.model._meta
|
|
app_label = opts.app_label
|
|
|
|
request.current_app = self.admin_site.name
|
|
context.update(
|
|
media=self.media,
|
|
)
|
|
|
|
return TemplateResponse(
|
|
request,
|
|
self.change_confirmation_template
|
|
or [
|
|
"admin/{}/{}/change_confirmation.html".format(
|
|
app_label, opts.model_name
|
|
),
|
|
"admin/{}/change_confirmation.html".format(app_label),
|
|
"admin/change_confirmation.html",
|
|
],
|
|
context,
|
|
)
|
|
|
|
def change_view(self, request, object_id=None, form_url="", extra_context=None):
|
|
# self.message_user(request, f"{request.POST}")
|
|
if request.method == "POST" and request.POST.get("_change_needs_confirmation"):
|
|
return self._change_confirmation_view(
|
|
request, object_id, form_url, extra_context
|
|
)
|
|
|
|
extra_context = {
|
|
**(extra_context or {}),
|
|
'change_needs_confirmation': self.change_needs_confirmation
|
|
}
|
|
return super().change_view(request, object_id, form_url, extra_context)
|
|
|
|
def _change_confirmation_view(self, request, object_id, form_url, extra_context):
|
|
# Do we need any of this code?
|
|
to_field = request.POST.get(
|
|
TO_FIELD_VAR, request.GET.get(TO_FIELD_VAR)
|
|
)
|
|
if to_field and not self.to_field_allowed(request, to_field):
|
|
raise DisallowedModelAdminToField(
|
|
"The field %s cannot be referenced." % to_field
|
|
)
|
|
|
|
model = self.model
|
|
opts = model._meta
|
|
|
|
add = object_id is None
|
|
|
|
obj = self.get_object(request, unquote(object_id), to_field)
|
|
|
|
if obj is None:
|
|
return self._get_obj_does_not_exist_redirect(request, opts, object_id)
|
|
|
|
if not self.has_view_or_change_permission(request, obj):
|
|
raise PermissionDenied
|
|
|
|
fieldsets = self.get_fieldsets(request, obj)
|
|
ModelForm = self.get_form(
|
|
request, obj, change=not add, fields=flatten_fieldsets(fieldsets)
|
|
)
|
|
|
|
# Should we be validating the data here? Or just pass it to super?
|
|
form = ModelForm(request.POST, request.FILES, obj)
|
|
form_validated = form.is_valid()
|
|
if form_validated:
|
|
new_object = self.save_form(request, form, change=not add)
|
|
else:
|
|
new_object = form.instance
|
|
|
|
if add:
|
|
title = _("Add %s")
|
|
elif self.has_change_permission(request, obj):
|
|
title = _("Change %s")
|
|
|
|
# Parse the original save action from request
|
|
save_action = None
|
|
for action in ["_save", "_saveasnew", "_addanother", "_continue"]:
|
|
if action in request.POST:
|
|
save_action = action
|
|
break
|
|
|
|
# Parse raw form data from POST
|
|
form_data = {}
|
|
for key in request.POST:
|
|
if key.startswith("_") or key == 'csrfmiddlewaretoken':
|
|
continue
|
|
|
|
form_data[key] = request.POST.get(key)
|
|
|
|
context = {
|
|
**self.admin_site.each_context(request),
|
|
"title": title % opts.verbose_name,
|
|
"subtitle": str(obj),
|
|
"object_name": str(obj),
|
|
"object_id": object_id,
|
|
"original": obj,
|
|
"new_object": new_object,
|
|
"app_label": opts.app_label,
|
|
"model_name": opts.model_name,
|
|
"opts": opts,
|
|
"preserved_filters": self.get_preserved_filters(request),
|
|
"form_data": form_data,
|
|
"submit_name": save_action,
|
|
**(extra_context or {}),
|
|
}
|
|
return self.render_change_confirmation(request, context)
|