VimPlug/django-stubs
1
0
Fork 0
mirror of https://github.com/davidhalter/django-stubs.git synced 2025-12-08 04:54:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

RefExpr in determining model fullname not supported

Browse Source
This commit is contained in:
Maxim Kurnikov
2018-12-21 01:49:10 +03:00
parent 3c3e9305f4
commit 2f58cdae88
2 changed files with 36 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
mypy_django_plugin/plugins/migrations.py
View File

@@ -1,16 +1,30 @@
from typing import cast
from typing import cast, Optional
from mypy.checker import TypeChecker
from mypy.nodes import TypeInfo
from mypy.nodes import TypeInfo, Expression, StrExpr, NameExpr, RefExpr, Var
from mypy.plugin import MethodContext
from mypy.types import Type, Instance, TypeType
from mypy_django_plugin import helpers
def get_string_value_from_expr(expr: Expression) -> Optional[str]:
if isinstance(expr, StrExpr):
return expr.value
# TODO: somehow figure out other cases
return None
def determine_model_cls_from_string_for_migrations(ctx: MethodContext) -> Type:
app_label = ctx.args[ctx.callee_arg_names.index('app_label')][0].value
model_name = ctx.args[ctx.callee_arg_names.index('model_name')][0].value
app_label_expr = ctx.args[ctx.callee_arg_names.index('app_label')][0]
app_label = get_string_value_from_expr(app_label_expr)
if app_label is None:
return ctx.default_return_type
model_name_expr = ctx.args[ctx.callee_arg_names.index('model_name')][0]
model_name = get_string_value_from_expr(model_name_expr)
if model_name is None:
return ctx.default_return_type
api = cast(TypeChecker, ctx.api)
model_fullname = helpers.get_model_fullname(app_label, model_name, all_modules=api.modules)