diff options
| -rw-r--r-- | bot/exts/__init__.py | 6 | ||||
| -rw-r--r-- | bot/utils/persist.py | 4 |
2 files changed, 5 insertions, 5 deletions
diff --git a/bot/exts/__init__.py b/bot/exts/__init__.py index ceb472d2..03a686df 100644 --- a/bot/exts/__init__.py +++ b/bot/exts/__init__.py @@ -3,12 +3,12 @@ import pkgutil from pathlib import Path from typing import List -__all__ = ("get_season_names", "get_extensions") +__all__ = ("get_package_names", "get_extensions") log = logging.getLogger(__name__) -def get_season_names() -> List[str]: +def get_package_names() -> List[str]: """Return names of all packages located in /bot/exts/.""" seasons = [ package.name @@ -29,7 +29,7 @@ def get_extensions() -> List[str]: base_path = Path(__path__[0]) extensions = [] - for season in get_season_names(): + for season in get_package_names(): for module in pkgutil.iter_modules([base_path.joinpath(season)]): extensions.append(f"bot.exts.{season}.{module.name}") diff --git a/bot/utils/persist.py b/bot/utils/persist.py index 6f6507fa..d78e5420 100644 --- a/bot/utils/persist.py +++ b/bot/utils/persist.py @@ -2,7 +2,7 @@ import sqlite3 from pathlib import Path from shutil import copyfile -from bot.exts import get_season_names +from bot.exts import get_package_names DIRECTORY = Path("data") # directory that has a persistent volume mapped to it @@ -41,7 +41,7 @@ def make_persistent(file_path: Path) -> Path: raise OSError(f"File not found at {file_path}.") # detect season in datafile path for assigning to subdirectory - season = next((s for s in get_season_names() if s in file_path.parts), None) + season = next((s for s in get_package_names() if s in file_path.parts), None) if season: # make sure subdirectory exists first |