diff options
| author | Pierre Sassoulas <pierre.sassoulas@gmail.com> | 2021-06-18 09:38:10 +0200 |
|---|---|---|
| committer | Pierre Sassoulas <pierre.sassoulas@gmail.com> | 2021-06-19 11:09:32 +0200 |
| commit | b21709c5e65a859f72f94ea71563355268c15eab (patch) | |
| tree | 3e8793aa9f09eeb9bbd7027d91f6715523276311 /astroid/__init__.py | |
| parent | 4f735d546b2588d72ba5db0965b6a6c39c0b57dc (diff) | |
| download | astroid-git-b21709c5e65a859f72f94ea71563355268c15eab.tar.gz | |
Remove use of os in favor of pathlib.Path in astroid/__init__.py
Diffstat (limited to 'astroid/__init__.py')
| -rw-r--r-- | astroid/__init__.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/astroid/__init__.py b/astroid/__init__.py index acd4d920..50bf48a2 100644 --- a/astroid/__init__.py +++ b/astroid/__init__.py @@ -39,7 +39,6 @@ Main modules are: * builder contains the class responsible to build astroid trees """ -import os from importlib import import_module from pathlib import Path @@ -61,7 +60,8 @@ MANAGER = AstroidManager() del AstroidManager # load brain plugins -BRAIN_MODULES_DIR = Path(__file__).with_name("brain") -for module in os.listdir(BRAIN_MODULES_DIR): - if module.endswith(".py"): - import_module(f"astroid.brain.{module[:-3]}") +ASTROID_INSTALL_DIRECTORY = Path(__file__).parent +BRAIN_MODULES_DIRECTORY = ASTROID_INSTALL_DIRECTORY / "brain" +for module in BRAIN_MODULES_DIRECTORY.iterdir(): + if module.suffix == ".py": + import_module(f"astroid.brain.{module.stem}") |