Beispiel #1
0
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
Beispiel #2
0
 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
Beispiel #3
0
 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: