Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-03-16 20:00:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor to make sure get_subclass() is on QuerySet

Browse source 
There's an edge case where you might want to call get_subclass() on a
QuerySet the same way you can call get() after you have already called
various filter/exclude methods.
This commit is contained in:
Travis Swicegood 2013-10-07 18:15:41 -05:00
parent 5a33ff760a
commit 9111843752
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
model_utils/managers.py
View file

@ -96,6 +96,8 @@ class InheritanceQuerySet(QuerySet):
else:
return node
def get_subclass(self, *args, **kwargs):
return self.select_subclasses().get(*args, **kwargs)
class InheritanceManager(models.Manager):
@ -108,8 +110,7 @@ class InheritanceManager(models.Manager):
return self.get_query_set().select_subclasses(*subclasses)
def get_subclass(self, *args, **kwargs):
return self.get_query_set().select_subclasses().get(*args, **kwargs)
return self.get_query_set().get_subclass(*args, **kwargs)
class QueryManager(models.Manager):