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

add more test folders

Browse Source
This commit is contained in:
Maxim Kurnikov
2019-02-13 19:44:25 +03:00
parent 587c2c484b
commit 79ebe20f2e
19 changed files with 148 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mypy_django_plugin/main.py
View File

@@ -62,7 +62,11 @@ def return_user_model_hook(ctx: FunctionContext) -> Type:
if setting_expr is None:
return ctx.default_return_type
app_label, _, model_class_name = get_string_value_from_expr(setting_expr).rpartition('.')
model_path = get_string_value_from_expr(setting_expr)
if model_path is None:
return ctx.default_return_type
app_label, _, model_class_name = model_path.rpartition('.')
if app_label is None:
return ctx.default_return_type

11
mypy_django_plugin/plugins/init_create.py
View File

@@ -69,11 +69,14 @@ def redefine_and_typecheck_model_init(ctx: FunctionContext) -> Type:
def redefine_and_typecheck_model_create(ctx: MethodContext) -> Type:
api = cast(TypeChecker, ctx.api)
if isinstance(ctx.type, Instance) and len(ctx.type.args) > 0:
model: TypeInfo = ctx.type.args[0].type
model_generic_arg = ctx.type.args[0]
else:
if isinstance(ctx.default_return_type, AnyType):
return ctx.default_return_type
model: TypeInfo = ctx.default_return_type.type
model_generic_arg = ctx.default_return_type
if isinstance(model_generic_arg, AnyType):
return ctx.default_return_type
model: TypeInfo = model_generic_arg.type
# extract name of base models for _ptr
base_pointer_args = extract_base_pointer_args(model)