Added PolymorphicManager.from_queryset() to properly set the queryset

fix_request_path_info
Diederik van der Boor 2016-02-17 12:36:48 +01:00
parent a5267e95e6
commit 65de1f74ab
1 changed files with 7 additions and 0 deletions

View File

@ -53,3 +53,10 @@ class PolymorphicManager(models.Manager):
def get_real_instances(self, base_result_objects=None): def get_real_instances(self, base_result_objects=None):
return self.all().get_real_instances(base_result_objects=base_result_objects) return self.all().get_real_instances(base_result_objects=base_result_objects)
if django.VERSION >= (1, 7):
@classmethod
def from_queryset(cls, queryset_class, class_name=None):
manager = super(PolymorphicManager, cls).from_queryset(queryset_class, class_name=class_name)
manager.queryset_class = queryset_class # also set our version, Django uses _queryset_class
return manager