Skip to content

Commit 2ec7384

Browse files
authored
Merge pull request #1 from HyreAS/manager-fixes
Manager fixes
2 parents f89a0fb + e2e493f commit 2ec7384

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

mypy_django_plugin/transformers/models.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -273,14 +273,17 @@ def run_with_model_cls(self, model_cls: Type[Model]) -> None:
273273
if isinstance(relation, (ManyToOneRel, ManyToManyRel)):
274274
try:
275275
related_manager_info = self.lookup_typeinfo_or_incomplete_defn_error(fullnames.RELATED_MANAGER_CLASS) # noqa: E501
276-
if 'objects' not in related_model_info.names:
277-
raise helpers.IncompleteDefnException()
278276
except helpers.IncompleteDefnException as exc:
279277
if not self.api.final_iteration:
280278
raise exc
281279
else:
282280
continue
283281

282+
if 'objects' not in related_model_info.names:
283+
self.add_new_node_to_model_class(
284+
attname, Instance(related_manager_info, [Instance(related_model_info, [])]))
285+
continue
286+
284287
# create new RelatedManager subclass
285288
parametrized_related_manager_type = Instance(related_manager_info,
286289
[Instance(related_model_info, [])])

0 commit comments

Comments
 (0)