From 4cff14430de93ac0cde05ae8c91432c867d8b2d9 Mon Sep 17 00:00:00 2001 From: Steven Loria Date: Thu, 20 Apr 2017 11:08:35 -0400 Subject: [PATCH] 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 --- typedmodels/models.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/typedmodels/models.py b/typedmodels/models.py index f2c769a..7035e53 100644 --- 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)