Skip to content

Commit

Permalink
Merge 4cff144 into 1a9d00d
Browse files Browse the repository at this point in the history
  • Loading branch information
sloria committed Apr 20, 2017
2 parents 1a9d00d + 4cff144 commit 63ee87c
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions typedmodels/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,9 @@
class TypedModelManager(models.Manager):
def get_queryset(self):
qs = super(TypedModelManager, self).get_queryset()
return self._filter_by_type(qs)

def _filter_by_type(self, qs):
if hasattr(self.model, '_typedmodels_type'):
if len(self.model._typedmodels_subtypes) > 1:
qs = qs.filter(type__in=self.model._typedmodels_subtypes)
Expand Down

0 comments on commit 63ee87c

Please sign in to comment.