diff options
Diffstat (limited to 'bot/seasons')
| -rw-r--r-- | bot/seasons/valentines/movie_generator.py | 27 |
1 files changed, 13 insertions, 14 deletions
diff --git a/bot/seasons/valentines/movie_generator.py b/bot/seasons/valentines/movie_generator.py index 394dd0fe..1b823bc0 100644 --- a/bot/seasons/valentines/movie_generator.py +++ b/bot/seasons/valentines/movie_generator.py @@ -7,7 +7,7 @@ import aiohttp import discord from discord.ext import commands -TMDB_API_KEY = environ.get('TMDB_API_KEY') +TMDB_API_KEY = environ.get("TMDB_API_KEY") log = logging.getLogger(__name__) @@ -15,13 +15,12 @@ log = logging.getLogger(__name__) class RomanceMovieFinder: """ A cog that returns a random romance movie suggestion to a user - """ def __init__(self, bot): self.bot = bot - @commands.command(name='romancemovie') + @commands.command(name="romancemovie") async def romance_movie(self, ctx): """ Randomly selects a romance movie and displays information about it @@ -29,13 +28,13 @@ class RomanceMovieFinder: # selecting a random int to parse it to the page parameter random_page = random.randint(0, 20) # TMDB api params - params = {'api_key': TMDB_API_KEY, - 'language': 'en-US', - 'sort_by': 'popularity.desc', - 'include_adult': 'false', - 'include_video': 'false', - 'page': random_page, - 'with_genres': '10749' + params = {"api_key": TMDB_API_KEY, + "language": "en-US", + "sort_by": "popularity.desc", + "include_adult": "false", + "include_video": "false", + "page": random_page, + "with_genres": "10749" } # the api request url request_url = "https://api.themoviedb.org/3/discover/movie?" + parse.urlencode(params) @@ -44,15 +43,15 @@ class RomanceMovieFinder: # loading the json file returned from the api data = await resp.json() # selecting random result from results object in the json file - selected_movie = random.choice(data['results']) + selected_movie = random.choice(data["results"]) embed = discord.Embed( title=f":sparkling_heart: {selected_movie['title']} :sparkling_heart:", - description=selected_movie['overview'], + description=selected_movie["overview"], ) embed.set_image(url=f"http://image.tmdb.org/t/p/w200/{selected_movie['poster_path']}") - embed.add_field(name='Release date :clock1:', value=selected_movie['release_date']) - embed.add_field(name='Rating :star2:', value=selected_movie['vote_average']) + embed.add_field(name="Release date :clock1:", value=selected_movie["release_date"]) + embed.add_field(name="Rating :star2:", value=selected_movie["vote_average"]) await ctx.send(embed=embed) |