Expose _filter_by_type as a method

This makes it easier to change the queryset class returned by a TypedModelManager subclass's get_queryset() method
craigds · Apr 20, 2017 · 4cff144 · 4cff144
1 parent 1a9d00d
commit 4cff144
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)