Explicitly get queryset when using query methods

jazzband · Oct 31, 2014 · 242cfb4 · 242cfb4
1 parent c5ed822
commit 242cfb4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/simple_history/manager.py b/simple_history/manager.py
@@ -53,7 +53,7 @@ def most_recent(self):
                 tmp.append(field.name)
         fields = tuple(tmp)
         try:
-            values = self.values_list(*fields)[0]
+            values = self.get_queryset().values_list(*fields)[0]
         except IndexError:
             raise self.instance.DoesNotExist("%s has no historical record." %
                                              self.instance._meta.object_name)
@@ -68,7 +68,7 @@ def as_of(self, date):
         """
         if not self.instance:
             return self._as_of_set(date)
-        queryset = self.filter(history_date__lte=date)
+        queryset = self.get_queryset().filter(history_date__lte=date)
         try:
             history_obj = queryset[0]
         except IndexError:
@@ -84,7 +84,7 @@ def as_of(self, date):
     def _as_of_set(self, date):
         model = type(self.model().instance)  # a bit of a hack to get the model
         pk_attr = model._meta.pk.name
-        queryset = self.filter(history_date__lte=date)
+        queryset = self.get_queryset().filter(history_date__lte=date)
         for original_pk in set(
                 queryset.order_by().values_list(pk_attr, flat=True)):
             changes = queryset.filter(**{pk_attr: original_pk})