mirror of
https://github.com/meisnate12/Plex-Meta-Manager
synced 2024-11-22 12:33:10 +00:00
imdb convert fix
This commit is contained in:
parent
914082aea4
commit
677d2f312d
1 changed files with 13 additions and 10 deletions
|
@ -1,7 +1,7 @@
|
|||
import logging
|
||||
from modules import util
|
||||
from modules.util import Failed
|
||||
from tmdbapis import TMDbAPIs, TMDbException
|
||||
from tmdbapis import TMDbAPIs, TMDbException, NotFound
|
||||
|
||||
logger = logging.getLogger("Plex Meta Manager")
|
||||
|
||||
|
@ -82,15 +82,18 @@ class TMDb:
|
|||
return results.tv_results[0].id
|
||||
|
||||
def convert_imdb_to(self, imdb_id):
|
||||
results = self.TMDb.find_by_id(imdb_id=imdb_id)
|
||||
if results.movie_results:
|
||||
return results.movie_results[0].id, "movie"
|
||||
elif results.tv_results:
|
||||
return results.tv_results[0].id, "show"
|
||||
elif results.tv_episode_results:
|
||||
item = results.tv_episode_results[0]
|
||||
return f"{item.tv_id}_{item.season_number}_{item.episode_number}", "episode"
|
||||
else:
|
||||
try:
|
||||
results = self.TMDb.find_by_id(imdb_id=imdb_id)
|
||||
if results.movie_results:
|
||||
return results.movie_results[0].id, "movie"
|
||||
elif results.tv_results:
|
||||
return results.tv_results[0].id, "show"
|
||||
elif results.tv_episode_results:
|
||||
item = results.tv_episode_results[0]
|
||||
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}")
|
||||
|
||||
def get_movie_show_or_collection(self, tmdb_id, is_movie):
|
||||
|
|
Loading…
Reference in a new issue