mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-06 20:24:31 +08:00
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
from typing import Optional
|
|
|
|
from mypy.plugin import ClassDefContext, MethodContext
|
|
from mypy.types import CallableType, Instance, NoneTyp, Type as MypyType, TypeType
|
|
|
|
from mypy_django_plugin.lib import helpers
|
|
|
|
|
|
def make_meta_nested_class_inherit_from_any(ctx: ClassDefContext) -> None:
|
|
meta_node = helpers.get_nested_meta_node_for_current_class(ctx.cls.info)
|
|
if meta_node is None:
|
|
if not ctx.api.final_iteration:
|
|
ctx.api.defer()
|
|
else:
|
|
meta_node.fallback_to_any = True
|
|
|
|
|
|
def get_specified_form_class(object_type: Instance) -> Optional[TypeType]:
|
|
form_class_sym = object_type.type.get('form_class')
|
|
if form_class_sym and isinstance(form_class_sym.type, CallableType):
|
|
return TypeType(form_class_sym.type.ret_type)
|
|
return None
|
|
|
|
|
|
def extract_proper_type_for_get_form(ctx: MethodContext) -> MypyType:
|
|
object_type = ctx.type
|
|
assert isinstance(object_type, Instance)
|
|
|
|
form_class_type = helpers.get_call_argument_type_by_name(ctx, 'form_class')
|
|
if form_class_type is None or isinstance(form_class_type, NoneTyp):
|
|
form_class_type = get_specified_form_class(object_type)
|
|
|
|
if isinstance(form_class_type, TypeType) and isinstance(form_class_type.item, Instance):
|
|
return form_class_type.item
|
|
|
|
if isinstance(form_class_type, CallableType) and isinstance(form_class_type.ret_type, Instance):
|
|
return form_class_type.ret_type
|
|
|
|
return ctx.default_return_type
|
|
|
|
|
|
def extract_proper_type_for_get_form_class(ctx: MethodContext) -> MypyType:
|
|
object_type = ctx.type
|
|
assert isinstance(object_type, Instance)
|
|
|
|
form_class_type = get_specified_form_class(object_type)
|
|
if form_class_type is None:
|
|
return ctx.default_return_type
|
|
|
|
return form_class_type
|