mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-12 23:16:31 +08:00
9bd8aed1e1
This fallback to value.__class__ seems to be doing more harm than good; see #312 and #1162. Replace it with a clear error message that suggests a way to fix the problem rather than incompletely papering over it. Signed-off-by: Anders Kaseorg <andersk@mit.edu> Signed-off-by: Anders Kaseorg <andersk@mit.edu>
46 lines
1.9 KiB
Python
46 lines
1.9 KiB
Python
from mypy.nodes import MemberExpr
|
|
from mypy.plugin import AttributeContext, FunctionContext
|
|
from mypy.types import AnyType, Instance
|
|
from mypy.types import Type as MypyType
|
|
from mypy.types import TypeOfAny, TypeType
|
|
|
|
from mypy_django_plugin.django.context import DjangoContext
|
|
from mypy_django_plugin.lib import helpers
|
|
|
|
|
|
def get_user_model_hook(ctx: FunctionContext, django_context: DjangoContext) -> MypyType:
|
|
auth_user_model = django_context.settings.AUTH_USER_MODEL
|
|
model_cls = django_context.apps_registry.get_model(auth_user_model)
|
|
model_cls_fullname = helpers.get_class_fullname(model_cls)
|
|
|
|
model_info = helpers.lookup_fully_qualified_typeinfo(helpers.get_typechecker_api(ctx), model_cls_fullname)
|
|
if model_info is None:
|
|
return AnyType(TypeOfAny.unannotated)
|
|
|
|
return TypeType(Instance(model_info, []))
|
|
|
|
|
|
def get_type_of_settings_attribute(ctx: AttributeContext, django_context: DjangoContext) -> MypyType:
|
|
assert isinstance(ctx.context, MemberExpr)
|
|
setting_name = ctx.context.name
|
|
|
|
typechecker_api = helpers.get_typechecker_api(ctx)
|
|
|
|
# first look for the setting in the project settings file, then global settings
|
|
settings_module = typechecker_api.modules.get(django_context.django_settings_module)
|
|
global_settings_module = typechecker_api.modules.get("django.conf.global_settings")
|
|
for module in [settings_module, global_settings_module]:
|
|
if module is not None:
|
|
sym = module.names.get(setting_name)
|
|
if sym is not None:
|
|
if sym.type is None:
|
|
ctx.api.fail(
|
|
f"Import cycle from Django settings module prevents type inference for {setting_name!r}",
|
|
ctx.context,
|
|
)
|
|
return ctx.default_attr_type
|
|
return sym.type
|
|
|
|
ctx.api.fail(f"'Settings' object has no attribute {setting_name!r}", ctx.context)
|
|
return ctx.default_attr_type
|