Skip to content

Commit 997b319

Browse files
plotskimueslo
authored andcommitted
Map region codes to movie titles in ImdbResult.summary["titles"]
1 parent 3e45b18 commit 997b319

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

pythonbits/imdb.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ def url(self):
5151
def summary(self):
5252
return {
5353
'title': get(self.movie, 'base', 'title'),
54+
'titles': get(self.movie, 'titles'),
5455
'directors': get(self.movie, 'credits', 'director', default=[]),
5556
'runtime': self.runtime,
5657
'rating': (get(self.movie, 'ratings', 'rating'), 10),
@@ -107,4 +108,8 @@ def search(self, title):
107108
movie = AttrDict(self.imdb.get_title(imdb_id))
108109
movie.credits = self.imdb.get_title_credits(imdb_id)['credits']
109110
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}
110115
return ImdbResult(movie)

0 commit comments

Comments
 (0)