Skip to content

Commit ca0ddc9

Browse files
plotskimueslo
authored andcommitted
Add IMDB.get_info()
This allows us to get an IMDBResult if we already know the IMDb ID.
1 parent 57f87d8 commit ca0ddc9

1 file changed

Lines changed: 12 additions & 10 deletions

File tree

pythonbits/imdb.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -103,13 +103,15 @@ def search(self, title):
103103
except IndexError:
104104
pass
105105
else:
106-
imdb_id = result['imdb_id']
107-
log.debug("Found IMDb item {}", imdb_id)
108-
movie = AttrDict(self.imdb.get_title(imdb_id))
109-
movie.credits = self.imdb.get_title_credits(imdb_id)['credits']
110-
movie.genres = self.imdb.get_title_genres(imdb_id)['genres']
111-
title_versions = self.imdb.get_title_versions(imdb_id)
112-
movie.titles = {item["region"]: item["title"]
113-
for item in title_versions['alternateTitles']
114-
if "region" in item and "title" in item}
115-
return ImdbResult(movie)
106+
log.debug("Found IMDb item {}", result['imdb_id'])
107+
return self.get_info(result['imdb_id'])
108+
109+
def get_info(self, imdb_id):
110+
movie = AttrDict(self.imdb.get_title(imdb_id))
111+
movie.credits = self.imdb.get_title_credits(imdb_id)['credits']
112+
movie.genres = self.imdb.get_title_genres(imdb_id)['genres']
113+
title_versions = self.imdb.get_title_versions(imdb_id)
114+
movie.titles = {item["region"]: item["title"]
115+
for item in title_versions['alternateTitles']
116+
if "region" in item and "title" in item}
117+
return ImdbResult(movie)

0 commit comments

Comments
 (0)