diff --git a/python/lib/tmdbscraper/tmdb.py b/python/lib/tmdbscraper/tmdb.py index 2d4e429..bb7893e 100644 --- a/python/lib/tmdbscraper/tmdb.py +++ b/python/lib/tmdbscraper/tmdb.py @@ -136,7 +136,7 @@ def _assemble_details(self, movie, movie_fallback, collection, collection_fallba info['duration'] = movie['runtime'] * 60 ratings = {'themoviedb': {'rating': float(movie['vote_average']), 'votes': int(movie['vote_count'])}} - uniqueids = {'tmdb': str(movie['id']), 'imdb': movie['imdb_id']} + uniqueids = _parse_uniqueids(movie) cast = [{ 'name': actor['name'], 'role': actor['character'], @@ -164,6 +164,12 @@ def _parse_media_id(title): return {'type': 'imdb', 'id':title[5:]} return None +def _parse_uniqueids(movie): + uniqueids = {'tmdb': str(movie['id'])} + if movie.get('imdb_id'): + uniqueids['imdb'] = movie['imdb_id'] + return uniqueids + def _get_movie(mid, language=None, search=False): details = None if search else \ 'trailers,images,releases,casts,keywords' if language is not None else \