Mirrors/Plex-Meta-Manager
2
0
Fork 0
mirror of https://github.com/meisnate12/Plex-Meta-Manager synced 2024-11-22 20:43:07 +00:00
Code Issues Projects Releases Packages Wiki Activity

imdb convert fix

Browse source
This commit is contained in:
meisnate12 2022-01-22 19:09:06 -05:00
parent 914082aea4
commit 677d2f312d
1 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
modules/tmdb.py
View file

@ -1,7 +1,7 @@
import logging import logging
from modules import util from modules import util
from modules.util import Failed from modules.util import Failed
from tmdbapis import TMDbAPIs, TMDbException from tmdbapis import TMDbAPIs, TMDbException, NotFound
logger = logging.getLogger("Plex Meta Manager") logger = logging.getLogger("Plex Meta Manager")
@ -82,15 +82,18 @@ class TMDb:
return results.tv_results[0].id return results.tv_results[0].id
def convert_imdb_to(self, imdb_id): def convert_imdb_to(self, imdb_id):
results = self.TMDb.find_by_id(imdb_id=imdb_id) try:
if results.movie_results: results = self.TMDb.find_by_id(imdb_id=imdb_id)
return results.movie_results[0].id, "movie" if results.movie_results:
elif results.tv_results: return results.movie_results[0].id, "movie"
return results.tv_results[0].id, "show" elif results.tv_results:
elif results.tv_episode_results: return results.tv_results[0].id, "show"
item = results.tv_episode_results[0] elif results.tv_episode_results:
return f"{item.tv_id}_{item.season_number}_{item.episode_number}", "episode" item = results.tv_episode_results[0]
else: return f"{item.tv_id}_{item.season_number}_{item.episode_number}", "episode"
else:
raise NotFound
except NotFound:
raise Failed(f"TMDb Error: No TMDb ID found for IMDb ID {imdb_id}") raise Failed(f"TMDb Error: No TMDb ID found for IMDb ID {imdb_id}")
def get_movie_show_or_collection(self, tmdb_id, is_movie): def get_movie_show_or_collection(self, tmdb_id, is_movie):