Ticket #1125: admin_stages_extra_context.patch

django/contrib/admin/views/main.py

@@ -230 +233 @@
             lookup_params.update(opts.one_to_one_field.rel.limit_choices_to)
         self.lookup_params = lookup_params
 
-def change_list(request, app_label, module_name):
+def change_list(request, app_label, module_name, extra_context={}):
     try:
         cl = ChangeList(request, app_label, module_name)
     except IncorrectLookupParameters:
@@ -242 +245 @@
         'cl' : cl
     })
     c.update({'has_add_permission': c['perms'][app_label][cl.opts.get_add_permission()]}),
+    for key, value in extra_context.items():
+        if callable(value):
+            c[key] = value()
+        else:
+            c[key] = value
+            
     return render_to_response(['admin/%s/%s/change_list' % (app_label, cl.opts.object_name.lower()),
                                'admin/%s/change_list' % app_label,
                                'admin/change_list'], context_instance=c)
@@ -392 +401 @@
     pk_value = getattr(new_object, opts.pk.attname)
     log.log_action(user.id, opts.get_content_type_id(), pk_value, str(new_object), log.ADDITION)
 
-def add_stage(request, app_label, module_name, show_delete=False, form_url='', post_url='../', post_url_continue='../%s/', object_id_override=None):
+def add_stage(request, app_label, module_name, show_delete=False, form_url='', post_url='../', 
+        post_url_continue='../%s/', object_id_override=None, extra_context={}):
     mod, opts = _get_mod_opts(app_label, module_name)
     if not request.user.has_perm(app_label + '.' + opts.get_add_permission()):
         raise PermissionDenied
@@ -445 +455 @@
     })
     if object_id_override is not None:
         c['object_id'] = object_id_override
+    for key, value in extra_context.items():
+        if callable(value):
+            c[key] = value()
+        else:
+            c[key] = value
 
     return render_change_form(opts, manipulator, app_label, c, add=True)
 add_stage = staff_member_required(add_stage)
@@ -464 +479 @@
         change_message = _('No fields changed.')
     log.log_action(user.id, opts.get_content_type_id(), pk_value, str(new_object), log.CHANGE, change_message)
 
-def change_stage(request, app_label, module_name, object_id):
+def change_stage(request, app_label, module_name, object_id, extra_context={}):
     mod, opts = _get_mod_opts(app_label, module_name)
     if not request.user.has_perm(app_label + '.' + opts.get_change_permission()):
         raise PermissionDenied
@@ -539 +554 @@
         'original': manipulator.original_object,
         'is_popup' : request.REQUEST.has_key('_popup')
     })
+    for key, value in extra_context.items():
+        if callable(value):
+            c[key] = value()
+        else:
+            c[key] = value
 
     return render_change_form(opts,manipulator, app_label, c, change=True)