Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-12-08 06:10:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

PyREPL module completion: check for already imported modules

Browse source
This commit is contained in:
Loïc Simon 2025-09-30 21:38:18 +02:00
parent 98a41af5b0
commit 6ed27763e9
2 changed files with 86 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

20
Lib/_pyrepl/_module_completer.py
View file

@ -107,7 +107,25 @@ def _find_modules(self, path: str, prefix: str) -> list[str]:
if path is None:
return []
modules: Iterable[pkgutil.ModuleInfo] = self.global_cache
modules: Iterable[pkgutil.ModuleInfo]
imported_module = sys.modules.get(path.split('.')[0])
if imported_module:
# Module already imported: only look for its submodules,
# even if a module with the same name would be higher in path
imported_path = (imported_module.__spec__
and imported_module.__spec__.origin)
if imported_path:
if os.path.basename(imported_path) == "__init__.py": # package
imported_path = os.path.dirname(imported_path)
import_location = os.path.dirname(imported_path)
modules = list(pkgutil.iter_modules([import_location]))
else:
# Module already imported but without spec/origin:
# propose no suggestions
modules = []
else:
modules = self.global_cache
is_stdlib_import: bool | None = None
for segment in path.split('.'):
modules = [mod_info for mod_info in modules