VimPlug/django-stubs
1
0
Fork 0
mirror of https://github.com/davidhalter/django-stubs.git synced 2026-02-26 19:47:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

self reference support for foreignkeys

Browse Source
This commit is contained in:
Maxim Kurnikov
2018-12-21 23:46:10 +03:00
parent 783e427d64
commit f070098ce5
4 changed files with 33 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mypy_django_plugin/plugins/related_fields.py
View File

@@ -2,7 +2,7 @@ import typing
from typing import Optional, cast
from mypy.checker import TypeChecker
from mypy.nodes import StrExpr, TypeInfo, Context
from mypy.nodes import StrExpr, TypeInfo
from mypy.plugin import FunctionContext
from mypy.types import Type, CallableType, Instance, AnyType, TypeOfAny
@@ -31,8 +31,12 @@ def get_valid_to_value_or_none(ctx: FunctionContext) -> Optional[Instance]:
if not isinstance(to_arg_expr, StrExpr):
# not string, not supported
return None
model_fullname = helpers.get_model_fullname_from_string(to_arg_expr,
all_modules=api.modules)
try:
model_fullname = helpers.get_model_fullname_from_string(to_arg_expr.value,
all_modules=api.modules)
except helpers.SelfReference:
model_fullname = api.tscope.classes[-1].fullname()
if model_fullname is None:
return None
model_info = helpers.lookup_fully_qualified_generic(model_fullname,