| """Utility functions for finding modules |
| |
| Utility functions for finding modules on sys.path. |
| |
| """ |
| from __future__ import annotations |
|
|
| |
| |
| |
| |
| |
| |
| |
|
|
| |
| |
| |
|
|
| |
| import importlib |
| import importlib.abc |
| import importlib.util |
| import sys |
|
|
| |
|
|
| |
|
|
|
|
| |
| |
| |
|
|
| |
| |
| |
|
|
| |
| |
| |
|
|
| def find_mod(module_name: str) -> str | None | importlib.abc.Loader: |
| """ |
| Find module `module_name` on sys.path, and return the path to module `module_name`. |
| |
| * If `module_name` refers to a module directory, then return path to `__init__` file. |
| * If `module_name` is a directory without an __init__file, return None. |
| |
| * If module is missing or does not have a `.py` or `.pyw` extension, return None. |
| * Note that we are not interested in running bytecode. |
| |
| * Otherwise, return the fill path of the module. |
| |
| Parameters |
| ---------- |
| module_name : str |
| |
| Returns |
| ------- |
| module_path : str |
| Path to module `module_name`, its __init__.py, or None, |
| depending on above conditions. |
| """ |
| spec = importlib.util.find_spec(module_name) |
| if spec is None: |
| return None |
| module_path = spec.origin |
| if module_path is None: |
| if spec.loader is not None and spec.loader in sys.meta_path: |
| return spec.loader |
| return None |
| else: |
| split_path = module_path.split(".") |
| if split_path[-1] in ["py", "pyw"]: |
| return module_path |
| else: |
| return None |
|
|
