def handle_info(tmdb_id):
"""
add information about movie from tmdb to database
information only from tmdb_movie.info() request
:param tmdb_id: id from tmdb
:return: Movie object
"""
tmdb_movie = tmdb.Movies(tmdb_id)
try:
response = tmdb_movie.info(language=LANG)
except HTTPError as e:
print(e)
return
if len(response['title']) > 128:
return
# add main movie data
main_data = {
'title':
response['title'],
'release_date':
response['release_date'],
'release_status':
RELEASE_STATUSES.get(response['status'], response['status']),
'tmdb_id':
response['id']
}
movie = Movie(**main_data)
movie.save()
info_data = {
'original_title': response['original_title'],
'budget': response['budget'],
'revenue': response['revenue'],
'homepage': response['homepage'],
'original_language': response['original_language'],
'runtime': response['runtime'],
'imdb_id': response['imdb_id'],
'overview_ua': response['overview'],
'tagline': response['tagline'],
'translated': False
}
if info_data['overview_ua']:
info_data['overview'], info_data['translated'] = info_data[
'overview_ua'], True
else:
# additional query
response = tmdb_movie.info()
info_data['overview'], info_data['tagline'] = response[
'overview'], response['tagline']
info = Info(movie=movie, **info_data)
info.save()
genres_ids = [item['id'] for item in response['genres']]
movie.genres.add(*genres_ids)
# add production companies
companies_ids = []
for item in response['production_companies']:
try:
company = ProductionCompany.objects.get(id=item['id'])
except ProductionCompany.DoesNotExist:
company = ProductionCompany.objects.create(name=item['name'],
id=item['id'])
companies_ids.append(company.id)
movie.production_companies.add(*companies_ids)
# add countries
countries_ids = []
for item in response['production_countries']:
try:
country = Country.objects.get(iso_3166_1=item['iso_3166_1'])
except Country.DoesNotExist:
country = Country.objects.create(name=item['name'],
iso_3166_1=item['iso_3166_1'])
countries_ids.append(country.id)
movie.production_countries.add(*countries_ids)
# main poster (english)
poster = response.get('poster_path', '')
if poster:
image = Image.upload_image_from_tmdb(poster)
movie.poster = image
movie.save()
movie.images.add(*[image.id])
return movie
def handle(self, *args, **options):
if Movie.objects.exists() or Actor.objects.exists(
) or Director.objects.exists() or Writer.objects.exists():
print('Movie data already loaded...exiting.')
print(ALREADY_LOADED_ERROR_MESSAGE)
return
print(
"\nLoading Movie data for Movies available in movies_metadata.csv")
i = 1
for row in DictReader(open('./movies_metadata.csv')):
print(i, end=', ')
i += 1
movie = Movie()
movie.id = row['id']
movie.title = row['title']
movie.budget = 0 if row['budget'] == '' else row['budget']
# genres is a list of dictionaries
genres_raw = row['genres']
genres_list = eval(genres_raw)
genres_name_list = list()
for genre in genres_list:
genres_name_list.append(genre['name'])
movie.genres = ','.join(genres_name_list)
movie.language = row['original_language']
movie.overview = row['overview']
# companies is a list of dictionaries
companies_raw = row['production_companies']
companies_list = eval(companies_raw)
companies_name_list = list()
for company in companies_list:
companies_name_list.append(company['name'])
movie.companies = ','.join(companies_name_list)
# countries is a list of dictionaries
countries_raw = row['production_countries']
countries_list = eval(countries_raw)
countries_name_list = list()
for country in countries_list:
countries_name_list.append(country['name'])
movie.countries = ','.join(countries_name_list)
movie.release_date = row['release_date']
movie.revenue = 0 if row['revenue'] == '' else row['revenue']
movie.runtime = 0 if row['runtime'] == '' else row['runtime']
movie.vote_average = 0 if row['vote_average'] == '' else row[
'vote_average']
movie.vote_count = 0 if row['vote_count'] == '' else row[
'vote_count']
try:
# get the right picture for movie
api_req = requests.get("https://api.themoviedb.org/3/movie/" +
str(row['id']) + "?api_key=" +
str(os.getenv('API_KEY')) +
"&language=en-US")
if api_req.json()['poster_path'] == None:
raise ValueError
movie.poster = str(api_req.json()['poster_path'])
except:
print('failed at loading poster path from: ' +
"https://api.themoviedb.org/3/movie/" + str(row['id']) +
"?api_key=" + str(os.getenv('API_KEY')) +
"&language=en-US")
movie.poster = row['poster_path']
movie.save()
print(
"\nLoading Actor, Director, Writer data for Credits available in credits.csv"
)
SEX_CHOICES = {1: 'F', 2: 'M', 0: ''}
i = 1
for row in DictReader(open('./credits.csv')):
print(i, end=', ')
i += 1
# import Actors
actors_raw = row['cast']
actors_list = eval(actors_raw)
for each_actor in actors_list[:5]:
# if we have the actor just add movie id to it
try:
actor = Actor.objects.get(actor_id=each_actor['id'])
actor.movie_ids = actor.movie_ids + ',' + row['id']
except:
actor = Actor()
actor.actor_id = each_actor['id']
actor.name = each_actor['name']
gender_raw = each_actor['gender']
actor.gender = SEX_CHOICES[gender_raw]
actor.movie_ids = row['id']
if each_actor['profile_path']:
actor.pic = each_actor['profile_path']
actor.save()
# import Directors, Writers from crew
crews_raw = row['crew']
crews_list = eval(crews_raw)
# get only 1 writer
flag_one_writer = False
# get only 1 director
flag_one_director = False
for crew in crews_list:
# import Director
if flag_one_director == False and crew['job'] == 'Director':
try:
director = Director.objects.get(director_id=crew['id'])
director.movie_ids = director.movie_ids + ',' + row[
'id']
except:
director = Director()
director.director_id = crew['id']
director.name = crew['name']
gender_raw = crew['gender']
director.gender = SEX_CHOICES[gender_raw]
director.movie_ids = row['id']
director.save()
flag_one_director = True
# import Writer
elif flag_one_writer == False and crew[
'department'] == 'Writing':
try:
writer = Writer.objects.get(writer_id=crew['id'])
writer.movie_ids = writer.movie_ids + ',' + row['id']
except:
writer = Writer()
writer.writer_id = crew['id']
writer.name = crew['name']
gender_raw = crew['gender']
writer.gender = SEX_CHOICES[gender_raw]
writer.movie_ids = row['id']
writer.save()
flag_one_writer = True
if flag_one_writer == True and flag_one_director == True:
break
m = m[0]
sv = False
if (not m.plot) & (mv['Plot'] != u''):
m.plot = mv['Plot']
sv = True
print "Adding plot to %s" % (m.title)
if (not m.rating) & (mv['Rating'] != u''):
m.plot = mv['Rating']
sv = True
print "Adding rating to %s" % (m.title)
if (not m.metacritic_rating == int(mv['Metacritic'])):
m.metacritic_rating = mv['Metacritic']
sv = True
print "Adding metacritic to %s" % (m.title != '')
if (not m.poster) & (mv['Poster'] != ''):
m.poster = mv['Poster']
sv = True
print "Adding poster to %s (%s)" % (m.title,
mv['Poster'])
if m.date.year != int(mv["Released"][0:4]):
m.date = date(int(mv["Released"][0:4]),
int(mv["Released"][5:7]),
int(mv["Released"][8:10]))
sv = True
print "Adding date to %s" % (m.title)
if float(m.imdb_rating) != dec2(mv['imdbRating']):
m.imdb_votes = int2(mv['imdbVotes'])
m.imdb_rating = dec2(mv['imdbRating'])
sv = True
print "Adding imdb ratings to %s" % (m.title)
# else: