Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-04-14 08:31:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merged in jgsogo/django-model-utils (pull request #5)

Browse source 
patch to issue #16
This commit is contained in:
Carl Meyer 2013-01-26 16:40:26 -08:00
commit 8808aa05b8
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
model_utils/managers.py
View file

@ -6,6 +6,7 @@ from django.db import models
from django.db.models.fields.related import OneToOneField
from django.db.models.manager import Manager
from django.db.models.query import QuerySet
from django.core.exceptions import ObjectDoesNotExist
class InheritanceQuerySet(QuerySet):
@ -33,7 +34,12 @@ class InheritanceQuerySet(QuerySet):
iter = super(InheritanceQuerySet, self).iterator()
if getattr(self, 'subclasses', False):
for obj in iter:
sub_obj = [getattr(obj, s) for s in self.subclasses if getattr(obj, s)] or [obj]
def get_attr(obj, s):
try:
return getattr(obj,s)
except ObjectDoesNotExist:
return None
sub_obj = [getattr(obj, s) for s in self.subclasses if get_attr(obj, s)] or [obj]
sub_obj = sub_obj[0]
if getattr(self, '_annotated', False):
for k in self._annotated: