1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import requests
import random
from os import environ
from discord.ext import commands
from discord import Embed
TMDB_API_KEY = environ.get('TMDB_API_KEY')
TMDB_TOKEN = environ.get('TMDB_TOKEN')
class Movie:
"""
Selects a random scary movie and embeds info into discord chat
"""
def __init__(self, bot):
self.bot = bot
@commands.command(name='movie', alias=['tmdb'], brief='Pick a scary movie')
async def random_movie(self, ctx):
selection = await self.select_movie()
movie_details = await self.format_metadata(selection)
await ctx.send(embed=movie_details)
@staticmethod
async def select_movie():
"""
Selects a random movie and returns a json of movie details from TMDb
"""
url = 'https://api.themoviedb.org/4/discover/movie'
params = {
'with_genres': '27',
'vote_count.gte': '5'
}
headers = {
'Authorization': 'Bearer ' + TMDB_TOKEN,
'Content-Type': 'application/json;charset=utf-8'
}
# Get total page count of horror movies
response = requests.get(url=url, params=params, headers=headers)
total_pages = response.json().get('total_pages')
# Get movie details from one random result on a random page
params['page'] = random.randint(1, total_pages)
response = requests.get(url=url, params=params, headers=headers)
selection_id = random.choice(response.json().get('results')).get('id')
# Get full details and credits
selection = requests.get(url='https://api.themoviedb.org/3/movie/' + str(selection_id),
params={'api_key': TMDB_API_KEY, 'append_to_response': 'credits'})
return selection.json()
@staticmethod
async def format_metadata(movie):
"""
Formats raw TMDb data to be embedded in discord chat
"""
tmdb_url = 'https://www.themoviedb.org/movie/' + str(movie.get('id'))
poster = 'https://image.tmdb.org/t/p/original' + movie.get('poster_path')
cast = []
for actor in movie.get('credits').get('cast')[:3]:
cast.append(actor.get('name'))
director = movie.get('credits').get('crew')[0].get('name')
rating_count = movie.get('vote_average') / 2
rating = ''
for i in range(int(rating_count)):
rating += ':skull:'
if (rating_count % 1) >= .5:
rating += ':bat:'
embed = Embed(
colour=0x01d277,
title='**' + movie.get('title') + '**',
url=tmdb_url,
description=movie.get('overview')
)
embed.set_image(url=poster)
embed.add_field(name='Starring', value=', '.join(cast))
embed.add_field(name='Directed by', value=director)
embed.add_field(name='Year', value=movie.get('release_date')[:4])
embed.add_field(name='Runtime', value=str(movie.get('runtime')) + ' min')
embed.add_field(name='Spooky Rating', value=rating)
embed.set_footer(text='powered by themoviedb.org')
return embed
def setup(bot):
bot.add_cog(Movie(bot))
|
