mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-08 21:14:49 +08:00
33 lines
1.5 KiB
Python
33 lines
1.5 KiB
Python
from typing import cast
|
|
|
|
from django.conf import Settings
|
|
from mypy.nodes import MDEF
|
|
from mypy.plugin import ClassDefContext
|
|
from mypy.semanal import SemanticAnalyzerPass2
|
|
from mypy.types import Instance, AnyType, TypeOfAny
|
|
|
|
from mypy_django_plugin import helpers
|
|
|
|
|
|
class DjangoConfSettingsInitializerHook(object):
|
|
def __init__(self, settings: Settings):
|
|
self.settings = settings
|
|
|
|
def __call__(self, ctx: ClassDefContext) -> None:
|
|
api = cast(SemanticAnalyzerPass2, ctx.api)
|
|
if self.settings:
|
|
for name, value in self.settings.__dict__.items():
|
|
if name.isupper():
|
|
if value is None:
|
|
# TODO: change to Optional[Any] later
|
|
ctx.cls.info.names[name] = helpers.create_new_symtable_node(name, MDEF,
|
|
instance=api.builtin_type('builtins.object'))
|
|
continue
|
|
|
|
type_fullname = helpers.get_obj_type_name(type(value))
|
|
sym = api.lookup_fully_qualified_or_none(type_fullname)
|
|
if sym is not None:
|
|
args = len(sym.node.type_vars) * [AnyType(TypeOfAny.from_omitted_generics)]
|
|
ctx.cls.info.names[name] = helpers.create_new_symtable_node(name, MDEF,
|
|
instance=Instance(sym.node, args))
|