From 18c908bf98ebb93e50494f7a1697d86cf4132160 Mon Sep 17 00:00:00 2001 From: Maxim Kurnikov Date: Tue, 5 Mar 2019 20:15:46 +0300 Subject: [PATCH] set plugin_generated on new symbol nodes --- mypy_django_plugin/transformers/models.py | 2 +- mypy_django_plugin/transformers/settings.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mypy_django_plugin/transformers/models.py b/mypy_django_plugin/transformers/models.py index 32c5e3f..5db4dbb 100644 --- a/mypy_django_plugin/transformers/models.py +++ b/mypy_django_plugin/transformers/models.py @@ -42,7 +42,7 @@ class ModelClassInitializer(metaclass=ABCMeta): var._fullname = self.model_classdef.info.fullname() + '.' + name var.is_inferred = True var.is_initialized_in_class = True - self.model_classdef.info.names[name] = SymbolTableNode(MDEF, var) + self.model_classdef.info.names[name] = SymbolTableNode(MDEF, var, plugin_generated=True) @abstractmethod def run(self) -> None: diff --git a/mypy_django_plugin/transformers/settings.py b/mypy_django_plugin/transformers/settings.py index cef5893..739a41e 100644 --- a/mypy_django_plugin/transformers/settings.py +++ b/mypy_django_plugin/transformers/settings.py @@ -59,7 +59,7 @@ def load_settings_from_names(settings_classdef: ClassDef, else: var = Var(name, AnyType(TypeOfAny.unannotated)) var.info = api.named_type('__builtins__.object').type # outer class type - settings_classdef.info.names[name] = SymbolTableNode(sym.kind, var) + settings_classdef.info.names[name] = SymbolTableNode(sym.kind, var, plugin_generated=True) settings_metadata[name] = module.fullname()