From ee1c90cd9f6f29dba7858dbafa22088950c08b90 Mon Sep 17 00:00:00 2001
From: Chris Jones <chris@brack3t.com>
Date: Thu, 18 Jul 2013 16:51:16 -0700
Subject: [PATCH] Working model history view.

---
 djadmin2/views.py | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/djadmin2/views.py b/djadmin2/views.py
index da5f472..3a1bfe4 100644
--- a/djadmin2/views.py
+++ b/djadmin2/views.py
@@ -3,19 +3,21 @@ from __future__ import division, absolute_import, unicode_literals
 
 import operator
 
+from django.contrib.auth import get_user_model
 from django.contrib.auth.forms import (PasswordChangeForm,
                                        AdminPasswordChangeForm)
 from django.contrib.auth.views import (logout as auth_logout,
                                        login as auth_login)
-from django.contrib.auth import get_user_model
+from django.contrib.contenttypes.models import ContentType
 from django.core.urlresolvers import reverse, reverse_lazy
-from django.utils.translation import ugettext_lazy
 from django.db import models
+from django.db.models.fields import FieldDoesNotExist
 from django.http import HttpResponseRedirect
+from django.shortcuts import get_object_or_404
 from django.utils.encoding import force_text
 from django.utils.text import capfirst
+from django.utils.translation import ugettext_lazy
 from django.views import generic
-from django.db.models.fields import FieldDoesNotExist
 
 import extra_views
 
@@ -302,12 +304,29 @@ class ModelDeleteView(AdminModel2Mixin, generic.DeleteView):
         return super(ModelDeleteView, self).delete(request, *args, **kwargs)
 
 
-class ModelHistoryView(Admin2Mixin, generic.ListView):
+class ModelHistoryView(AdminModel2Mixin, generic.ListView):
+    default_template_name = "model_history.html"
+    permission_classes = (
+        permissions.IsStaffPermission,
+        permissions.ModelChangePermission
+    )
+
     def get_context_data(self, **kwargs):
         context = super(ModelHistoryView, self).get_context_data(**kwargs)
         context['model'] = self.get_model()
+        context['object'] = self.get_object()
         return context
 
+    def get_object(self):
+        return get_object_or_404(self.get_model(), pk=self.kwargs.get('pk'))
+
+    def get_queryset(self):
+        content_type = ContentType.objects.get_for_model(self.get_object())
+        return LogEntry.objects.filter(
+            content_type=content_type,
+            object_id=self.get_object().id
+        )
+
 
 class PasswordChangeView(Admin2Mixin, generic.UpdateView):