Deseasonify: allow `get_extensions` to find cogs in `seasons`

Add a conditional check to handle cases where `package` is, in fact,
just a plain module, and should be loaded directly rather than being
searched for modules.

The naming here could be improved: `package` does not necessarily have
to be a package.

1 files changed, 7 insertions, 3 deletions

diff --git a/bot/seasons/__init__.py b/bot/seasons/__init__.py
index 1569f29a..1292485c 100644
--- a/bot/seasons/__init__.py
+++ b/bot/seasons/__init__.py
@@ -29,10 +29,14 @@ def get_extensions() -> List[str]:
     extensions = []
 
     for package in pkgutil.iter_modules([base_path]):
-        package_path = base_path.joinpath(package.name)
 
-        for module in pkgutil.iter_modules([package_path]):
-            extensions.append(f"bot.seasons.{package.name}.{module.name}")
+        if package.ispkg:
+            package_path = base_path.joinpath(package.name)
+
+            for module in pkgutil.iter_modules([package_path]):
+                extensions.append(f"bot.seasons.{package.name}.{module.name}")
+        else:
+            extensions.append(f"bot.seasons.{package.name}")
 
     return extensions