Allow to run from_queryset() with BaseManager, Manager (#271)

* allow to run from_queryset() with BaseManager, Manager

* fix tests
This commit is contained in:
Maksim Kurnikov authored and GitHub committed 2019-12-13 20:16:33 +03:00
1 parent 1c31e71ffc
commit d0c25e3bce
8 files changed
+59 -14

No files matched your search

+1 -1
View File
@@ -144,7 +144,7 @@ class AddManagers(ModelClassInitializer):
return False
def is_any_parametrized_manager(self, typ: Instance) -> bool:
return typ.type.fullname == fullnames.MANAGER_CLASS_FULLNAME and isinstance(typ.args[0], AnyType)
return typ.type.fullname in fullnames.MANAGER_CLASSES and isinstance(typ.args[0], AnyType)
def get_generated_manager_mappings(self, base_manager_fullname: str) -> Dict[str, str]:
base_manager_info = self.lookup_typeinfo(base_manager_fullname)