diff options
| author |  ks123 <[email protected]> | 2020-02-14 19:52:12 +0200 |
|---|---|---|
| committer | ks123 <[email protected]> | 2020-02-14 19:52:12 +0200 |
| commit | f0d1cf5a25bfb435a19eca5fb16da490596cc429 (patch) | |
| tree | 15097771febf9a86582dd13b084e7b3e69620b9a | |
| parent | Made .movies command genres getting ID and name check to Enum, also made thin... (diff) | |
Replaced check from Enum members to try-except block.
| -rw-r--r-- | bot/seasons/evergreen/movie.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/bot/seasons/evergreen/movie.py b/bot/seasons/evergreen/movie.py index 8ae49cd2..e7c4c73c 100644 --- a/bot/seasons/evergreen/movie.py +++ b/bot/seasons/evergreen/movie.py @@ -208,14 +208,13 @@ class Movie(Cog): # Capitalize genre for getting data from Enum genre = genre.capitalize() - # If invalid genre, send help message - if genre not in MovieGenres.__members__: + # Try to fetch pages and embed, when invalid genre, show help + try: + pages, embed = await get_random_movies(self.http_session, amount, MovieGenres[genre].value, genre) + except KeyError: await ctx.send_help('movies') return - # Get pages and embed of movies - pages, embed = await get_random_movies(self.http_session, amount, MovieGenres[genre].value, genre) - # Send movies, paginate await ImagePaginator.paginate(pages, ctx, embed) |