Added ModelAdmin.save_model() and ModelAdmin.save_formset() methods to allow for easier modification of objects/inlines at admin-save time. Refs #6002.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@8266 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Jacob Kaplan-Moss 2008-08-09 17:12:48 +00:00
parent 6ba6489662
commit f53e4d80b4
1 changed files with 18 additions and 7 deletions

View File

@ -452,6 +452,19 @@ class ModelAdmin(BaseModelAdmin):
""" """
request.user.message_set.create(message=message) request.user.message_set.create(message=message)
def save_model(self, request, form, change):
"""
Save and return a model given a ModelForm. ``change`` is True if the
object is being changed, and False if it's being added.
"""
return form.save(commit=True)
def save_formset(self, request, form, formset, change):
"""
Save an inline formset attached to the object.
"""
formset.save()
def save_add(self, request, form, formsets, post_url_continue): def save_add(self, request, form, formsets, post_url_continue):
""" """
Saves the object in the "add" stage and returns an HttpResponseRedirect. Saves the object in the "add" stage and returns an HttpResponseRedirect.
@ -459,14 +472,12 @@ class ModelAdmin(BaseModelAdmin):
`form` is a bound Form instance that's verified to be valid. `form` is a bound Form instance that's verified to be valid.
""" """
opts = self.model._meta opts = self.model._meta
new_object = form.save(commit=True)
new_object = self.save_model(request, form, change=False)
if formsets: if formsets:
for formset in formsets: for formset in formsets:
# HACK: it seems like the parent obejct should be passed into
# a method of something, not just set as an attribute
formset.instance = new_object formset.instance = new_object
formset.save() self.save_formset(request, form, formset, change=False)
pk_value = new_object._get_pk_val() pk_value = new_object._get_pk_val()
self.log_addition(request, new_object) self.log_addition(request, new_object)
@ -509,12 +520,12 @@ class ModelAdmin(BaseModelAdmin):
`formsets` is a sequence of InlineFormSet instances that are verified to be valid. `formsets` is a sequence of InlineFormSet instances that are verified to be valid.
""" """
opts = self.model._meta opts = self.model._meta
new_object = form.save(commit=True) new_object = self.save_model(request, form, change=True)
pk_value = new_object._get_pk_val() pk_value = new_object._get_pk_val()
if formsets: if formsets:
for formset in formsets: for formset in formsets:
formset.save() self.save_formset(request, form, formset, change=True)
change_message = self.construct_change_message(request, form, formsets) change_message = self.construct_change_message(request, form, formsets)
self.log_change(request, new_object, change_message) self.log_change(request, new_object, change_message)