From f0d1cf5a25bfb435a19eca5fb16da490596cc429 Mon Sep 17 00:00:00 2001 From: ks123 Date: Fri, 14 Feb 2020 19:52:12 +0200 Subject: Replaced check from Enum members to try-except block. --- bot/seasons/evergreen/movie.py | 9 ++++----- 1 file 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) -- cgit v1.2.3