VimPlug/django-stubs
1
0 Fork 0
mirror of https://github.com/davidhalter/django-stubs.git synced 2026-10-19 04:21:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

make ValuesQuerySet have proper Collection generic type (#140)

Browse Source
This commit is contained in:
Maxim Kurnikov authored and GitHub committed 2019-08-24 18:24:21 +03:00
1 parent 5fc39ff110
commit fc9a335dfd
11 files changed
+93 -88
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

mypy_django_plugin/django/context.py
+9 -1
View File
@@ -135,7 +135,15 @@ class DjangoFieldsContext:
related_model_cls = field.field.model
if isinstance(related_model_cls, str):
related_model_cls = self.django_context.apps_registry.get_model(related_model_cls)
if related_model_cls == 'self':
# same model
related_model_cls = field.model
elif '.' not in related_model_cls:
# same file model
related_model_fullname = field.model.__module__ + '.' + related_model_cls
related_model_cls = self.django_context.get_model_class_by_fullname(related_model_fullname)
else:
related_model_cls = self.django_context.apps_registry.get_model(related_model_cls)
return related_model_cls