Merge pull request #36 from cos-forks/filter-by-type

Expose _filter_by_type as a method
craigds · Apr 20, 2017 · bcd443b · bcd443b
2 parents 250e12a + 4cff144
commit bcd443b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/typedmodels/models.py b/typedmodels/models.py
@@ -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)