From bb27d0a18b24e96a60b5d7ad44f983039b3bf609 Mon Sep 17 00:00:00 2001 From: Ruud Date: Sun, 11 Mar 2012 00:03:59 +0100 Subject: [PATCH] Remove non-used items from imdbapi --- couchpotato/core/providers/movie/imdbapi/main.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/couchpotato/core/providers/movie/imdbapi/main.py b/couchpotato/core/providers/movie/imdbapi/main.py index 1c81727..158359b 100644 --- a/couchpotato/core/providers/movie/imdbapi/main.py +++ b/couchpotato/core/providers/movie/imdbapi/main.py @@ -47,8 +47,9 @@ class IMDBAPI(MovieProvider): if cached: result = self.parseMovie(cached) - log.info('Found: %s' % result['titles'][0] + ' (' + str(result['year']) + ')') - return result + if result.get('titles') and len(result.get('titles')) > 0: + log.info('Found: %s' % result['titles'][0] + ' (' + str(result['year']) + ')') + return result return {} @@ -60,6 +61,11 @@ class IMDBAPI(MovieProvider): if isinstance(movie, (str, unicode)): movie = json.loads(movie) + tmp_movie = movie.copy() + for key in tmp_movie: + if tmp_movie.get(key).lower() == 'n/a': + del movie[key] + movie_data = { 'titles': [movie.get('Title', '')], 'original_title': movie.get('Title', ''),