Skip to content

Commit

Permalink
Support new get_queryset method for Django 1.8
Browse files Browse the repository at this point in the history
  • Loading branch information
treyhunner committed Apr 14, 2014
1 parent ccac8ca commit 4680d1f
Showing 1 changed file with 12 additions and 3 deletions.
15 changes: 12 additions & 3 deletions simple_history/manager.py
Expand Up @@ -19,15 +19,24 @@ def __init__(self, model, instance=None):
self.model = model
self.instance = instance

def get_query_set(self):
if self.instance is None:
def get_super_queryset(self):
try:
return super(HistoryManager, self).get_queryset()
except AttributeError:
return super(HistoryManager, self).get_query_set()

def get_queryset(self):
qs = self.get_super_queryset()
if self.instance is None:
return qs

if isinstance(self.instance._meta.pk, models.OneToOneField):
filter = {self.instance._meta.pk.name + "_id": self.instance.pk}
else:
filter = {self.instance._meta.pk.name: self.instance.pk}
return super(HistoryManager, self).get_query_set().filter(**filter)
return self.get_super_queryset().filter(**filter)

get_query_set = get_queryset

def most_recent(self):
"""
Expand Down

0 comments on commit 4680d1f

Please sign in to comment.