Try idcluding only module itself

Author: ilevkivskyi

mypy/indirection.py

@@ -4,7 +4,6 @@
 
 import mypy.types as types
 from mypy.types import TypeVisitor
-from mypy.util import split_module_names
 
 
 class TypeIndirectionVisitor(TypeVisitor[None]):
@@ -64,10 +63,6 @@ def _visit_type_list(self, typs: list[types.Type]) -> None:
                 self.seen_types.add(typ)
             typ.accept(self)
 
-    def _visit_module_name(self, module_name: str) -> None:
-        if module_name not in self.modules:
-            self.modules.update(split_module_names(module_name))
-
     def visit_unbound_type(self, t: types.UnboundType) -> None:
         self._visit_type_tuple(t.args)
 
@@ -118,7 +113,7 @@ def visit_instance(self, t: types.Instance) -> None:
             # Important optimization: instead of simply recording the definition and
             # recursing into bases, record the MRO and only traverse generic bases.
             for s in t.type.mro:
-                self._visit_module_name(s.module_name)
+                self.modules.add(s.module_name)
                 for base in s.bases:
                     if base.args:
                         self._visit_type_tuple(base.args)
@@ -163,6 +158,6 @@ def visit_type_alias_type(self, t: types.TypeAliasType) -> None:
         # Type alias is named, record its definition and continue digging into
         # components that constitute semantic meaning of this type: target and args.
         if t.alias:
-            self._visit_module_name(t.alias.module)
+            self.modules.add(t.alias.module)
             self._visit(t.alias.target)
         self._visit_type_list(t.args)

mypy/semanal.py

@@ -305,14 +305,7 @@
 )
 from mypy.types_utils import is_invalid_recursive_alias, store_argument_type
 from mypy.typevars import fill_typevars
-from mypy.util import (
-    correct_relative_import,
-    is_dunder,
-    module_prefix,
-    split_module_names,
-    unmangle,
-    unnamed_function,
-)
+from mypy.util import correct_relative_import, is_dunder, module_prefix, unmangle, unnamed_function
 from mypy.visitor import NodeVisitor
 
 T = TypeVar("T")
@@ -6313,8 +6306,8 @@ def record_imported_symbol(self, node: SymbolNode) -> None:
                 return
             while "." in fullname and fullname not in self.modules:
                 fullname = fullname.rsplit(".")[0]
-        if fullname != self.cur_mod_id and fullname not in self.cur_mod_node.module_refs:
-            self.cur_mod_node.module_refs.update(split_module_names(fullname))
+        if fullname != self.cur_mod_id:
+            self.cur_mod_node.module_refs.add(fullname)
 
     def _lookup(
         self, name: str, ctx: Context, suppress_errors: bool = False