diff options
| author | 2021-05-20 18:30:07 +0530 | |
|---|---|---|
| committer | 2021-05-20 18:30:07 +0530 | |
| commit | d032dd1a0e0fb0f866b1b492f917990d66dae55a (patch) | |
| tree | 0adfc37d1350c7eb0b33ea44d4170a44a017e119 /bot/exts/valentines/movie_generator.py | |
| parent | Merge remote-tracking branch 'origin/feature/command-suggestions' into featur... (diff) | |
| parent | Merge pull request #733 from Icebluewolf/http_status_command_randomness (diff) | |
Merge branch 'main' into feature/command-suggestions
Diffstat (limited to 'bot/exts/valentines/movie_generator.py')
| -rw-r--r-- | bot/exts/valentines/movie_generator.py | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/bot/exts/valentines/movie_generator.py b/bot/exts/valentines/movie_generator.py index 4df9e0d5..0fc5edb4 100644 --- a/bot/exts/valentines/movie_generator.py +++ b/bot/exts/valentines/movie_generator.py @@ -6,6 +6,8 @@ from urllib import parse import discord from discord.ext import commands +from bot.bot import Bot + TMDB_API_KEY = environ.get("TMDB_API_KEY") log = logging.getLogger(__name__) @@ -14,7 +16,7 @@ log = logging.getLogger(__name__) class RomanceMovieFinder(commands.Cog): """A Cog that returns a random romance movie suggestion to a user.""" - def __init__(self, bot: commands.Bot): + def __init__(self, bot: Bot): self.bot = bot @commands.command(name="romancemovie") @@ -52,13 +54,15 @@ class RomanceMovieFinder(commands.Cog): embed.set_thumbnail(url="https://i.imgur.com/LtFtC8H.png") await ctx.send(embed=embed) except KeyError: - warning_message = "A KeyError was raised while fetching information on the movie. The API service" \ - " could be unavailable or the API key could be set incorrectly." + warning_message = ( + "A KeyError was raised while fetching information on the movie. The API service" + " could be unavailable or the API key could be set incorrectly." + ) embed = discord.Embed(title=warning_message) log.warning(warning_message) await ctx.send(embed=embed) -def setup(bot: commands.Bot) -> None: - """Romance movie Cog load.""" +def setup(bot: Bot) -> None: + """Load the Romance movie Cog.""" bot.add_cog(RomanceMovieFinder(bot)) |