diff options
| -rw-r--r-- | bot/exts/__init__.py | 16 |
1 files changed, 6 insertions, 10 deletions
diff --git a/bot/exts/__init__.py b/bot/exts/__init__.py index 03a686df..4f19c167 100644 --- a/bot/exts/__init__.py +++ b/bot/exts/__init__.py @@ -1,22 +1,18 @@ import logging import pkgutil from pathlib import Path -from typing import List +from typing import Iterator, List __all__ = ("get_package_names", "get_extensions") log = logging.getLogger(__name__) -def get_package_names() -> List[str]: - """Return names of all packages located in /bot/exts/.""" - seasons = [ - package.name - for package in pkgutil.iter_modules(__path__) - if package.ispkg - ] - - return seasons +def get_package_names() -> Iterator[str]: + """Iterate names of all packages located in /bot/exts/.""" + for package in pkgutil.iter_modules(__path__): + if package.ispkg: + yield package.name def get_extensions() -> List[str]: |