Skip to content

Commit fd69967

Browse files
committed
don't save IMDB ID if it is empty
1 parent cdfbdc2 commit fd69967

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

python/lib/tmdbscraper/tmdb.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def _assemble_details(self, movie, movie_fallback, collection, collection_fallba
136136
info['duration'] = movie['runtime'] * 60
137137

138138
ratings = {'themoviedb': {'rating': float(movie['vote_average']), 'votes': int(movie['vote_count'])}}
139-
uniqueids = {'tmdb': str(movie['id']), 'imdb': movie['imdb_id']}
139+
uniqueids = _parse_uniqueids(movie)
140140
cast = [{
141141
'name': actor['name'],
142142
'role': actor['character'],
@@ -164,6 +164,12 @@ def _parse_media_id(title):
164164
return {'type': 'imdb', 'id':title[5:]}
165165
return None
166166

167+
def _parse_uniqueids(movie):
168+
uniqueids = {'tmdb': str(movie['id'])}
169+
if movie.get('imdb_id'):
170+
uniqueids['imdb'] = movie['imdb_id']
171+
return uniqueids
172+
167173
def _get_movie(mid, language=None, search=False):
168174
details = None if search else \
169175
'trailers,images,releases,casts,keywords' if language is not None else \

0 commit comments

Comments
 (0)