diff options
| author |  shtlrs <[email protected]> | 2023-05-06 14:06:53 +0100 |
|---|---|---|
| committer | shtlrs <[email protected]> | 2023-05-06 14:06:53 +0100 |
| commit | cd16be11b5686f4d8e63db303b884ed88efb18d1 (patch) | |
| tree | f4fbc6bc4cd93ba2a89b1c4cc6c5c2012549421e /bot/exts/fun | |
| parent | use the new code jam variable (#1267) (diff) | |
Run null check on needed api keys before loading cogs.
This makes sure that all cogs that need keys won't load.
Which helps in avoiding exceptions at runtime, and give a clearer warning at startup time.
Diffstat (limited to 'bot/exts/fun')
| -rw-r--r-- | bot/exts/fun/movie.py | 3 | ||||
| -rw-r--r-- | bot/exts/fun/snakes/__init__.py | 3 |
2 files changed, 6 insertions, 0 deletions
diff --git a/bot/exts/fun/movie.py b/bot/exts/fun/movie.py index 73ef0a3c..21183e0f 100644 --- a/bot/exts/fun/movie.py +++ b/bot/exts/fun/movie.py @@ -208,4 +208,7 @@ class Movie(Cog): async def setup(bot: Bot) -> None: """Load the Movie Cog.""" + if not Tokens.tmdb: + logger.warning("No TMDB token. Not loading Movie Cog.") + return await bot.add_cog(Movie(bot)) diff --git a/bot/exts/fun/snakes/__init__.py b/bot/exts/fun/snakes/__init__.py index 8aa39fb5..37d0e255 100644 --- a/bot/exts/fun/snakes/__init__.py +++ b/bot/exts/fun/snakes/__init__.py @@ -1,6 +1,7 @@ import logging from bot.bot import Bot +from bot.constants import Tokens from bot.exts.fun.snakes._snakes_cog import Snakes log = logging.getLogger(__name__) @@ -8,4 +9,6 @@ log = logging.getLogger(__name__) async def setup(bot: Bot) -> None: """Load the Snakes Cog.""" + if not Tokens.giphy: + log.warning("No Youtube token. All youtube related commands in Snakes cog won't work.") await bot.add_cog(Snakes(bot)) |