Add ability to edit history

Author: buddylindsey

simple_history/admin.py

@@ -6,6 +6,7 @@
 from django.contrib.admin import helpers
 from django.contrib.contenttypes.models import ContentType
 from django.core.urlresolvers import reverse
+from django.http import HttpResponseRedirect
 from django.shortcuts import get_object_or_404, render
 from django.utils.text import capfirst
 from django.utils.html import mark_safe
@@ -75,6 +76,24 @@ def history_view(self, request, object_id, extra_context=None):
         return render(request, template_name=self.object_history_template,
                       dictionary=context, current_app=self.admin_site.name)
 
+    def response_change(self, request, obj):
+        if '_change_history' in request.POST:
+            verbose_name = obj._meta.verbose_name
+
+            msg = _('The %(name)s "%(obj)s" was changed successfully.') % {
+                'name': force_text(verbose_name),
+                'obj': force_text(obj)
+            }
+
+            self.message_user(
+                request, "%s - %s" % (msg, _("You may edit it again below")),
+                fail_silently=True)
+
+            return HttpResponseRedirect(request.path)
+        else:
+            return super(SimpleHistoryAdmin, self).response_change(
+                request, obj)
+
     def history_form_view(self, request, object_id, version_id):
         original_opts = self.model._meta
         model = getattr(
@@ -89,6 +108,9 @@ def history_form_view(self, request, object_id, version_id):
         if not self.has_change_permission(request, obj):
             raise PermissionDenied
 
+        if '_change_history' in request.POST:
+            obj = obj.history.get(pk=version_id)
+
         formsets = []
         form_class = self.get_form(request, obj)
         if request.method == 'POST':

simple_history/templates/simple_history/object_history_form.html

@@ -13,13 +13,10 @@
   </div>
 {% endblock %}
 
-{% comment %}Hack to remove "Save as New" and "Save and Continue" buttons {% endcomment %}
-{% block content %}
-  {% with 1 as is_popup %}
-    {{block.super}}
-  {% endwith %}
+{% block submit_buttons_bottom %}
+  {% include "simple_history/submit_line.html" %}
 {% endblock %}
 
 {% block form_top %}
-  <p>{% blocktrans %}Press the save button below to revert to this version of the object.{% endblocktrans %}</p>
+  <p>{% blocktrans %}Press the 'Revert' button below to revert to this version of the object. Or press the 'Change History' button to edit the history.{% endblocktrans %}</p>
 {% endblock %}

simple_history/templates/simple_history/submit_line.html

@@ -0,0 +1,5 @@
+{% load i18n %}
+<div class="submit-row">
+<input type="submit" value="{% trans 'Revert' %}" class="default" name="_save" {{ onclick_attrib }}/>
+<input type="submit" value="{% trans 'Change History' %}" class="default" name="_change_history" {{ onclick_attrib }}/>
+</div>