Find frozen stdlib modules

Author: DanielNoord

astroid/interpreter/_import/spec.py

@@ -5,13 +5,15 @@
 import abc
 import collections
 import enum
+import importlib
 import importlib.machinery
 import importlib.util
 import os
 import sys
 import zipimport
 from functools import lru_cache
 from pathlib import Path
+from typing import List, Optional
 
 from . import util
 
@@ -60,7 +62,13 @@ def __init__(self, path=None):
         self._path = path or sys.path
 
     @abc.abstractmethod
-    def find_module(self, modname, module_parts, processed, submodule_path):
+    def find_module(
+        self,
+        modname: str,
+        module_parts: List[str],
+        processed: List[str],
+        submodule_path: Optional[List[str]],
+    ) -> Optional[ModuleSpec]:
         """Find the given module
 
         Each finder is responsible for each protocol of finding, as long as
@@ -91,9 +99,13 @@ class ImportlibFinder(Finder):
         + [(s, ModuleType.PY_COMPILED) for s in importlib.machinery.BYTECODE_SUFFIXES]
     )
 
-    def find_module(self, modname, module_parts, processed, submodule_path):
-        if not isinstance(modname, str):
-            raise TypeError(f"'modname' must be a str, not {type(modname)}")
+    def find_module(
+        self,
+        modname: str,
+        module_parts: List[str],
+        processed: List[str],
+        submodule_path: Optional[List[str]],
+    ) -> Optional[ModuleSpec]:
         if submodule_path is not None:
             submodule_path = list(submodule_path)
         else:
@@ -109,7 +121,7 @@ def find_module(self, modname, module_parts, processed, submodule_path):
                     if spec.loader is importlib.machinery.FrozenImporter:
                         return ModuleSpec(
                             name=modname,
-                            location=None,
+                            location=getattr(spec.loader_state, "filename", None),
                             module_type=ModuleType.PY_FROZEN,
                         )
             except ValueError:

astroid/manager.py

@@ -7,6 +7,7 @@
 from various source and using a cache of built modules)
 """
 
+
 import os
 import types
 import zipimport
@@ -188,7 +189,11 @@ def ast_from_module_name(self, modname, context_file=None):
                     modname, found_spec.submodule_search_locations
                 )
             elif found_spec.type == spec.ModuleType.PY_FROZEN:
-                return self._build_stub_module(modname)
+                if found_spec.location is None:
+                    return self._build_stub_module(modname)
+                # For stdlib frozen modules we can determine the location and
+                # can therefore create a module from the source file
+                return self.ast_from_file(found_spec.location, modname, fallback=False)
 
             if found_spec.location is None:
                 raise AstroidImportError(