zhaarey/unshackle
1
0
Fork 0
mirror of https://github.com/unshackle-dl/unshackle.git synced 2025-10-23 15:11:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(tags): Enhance tag_file function to prioritize provided TMDB ID if --tmdb is used

Browse Source
This commit is contained in:
Andy
2025-08-06 22:15:16 +00:00
parent 41d203aaba
commit 9eb6bdbe12
1 changed files with 48 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
unshackle/core/utils/tags.py
View File

@@ -305,6 +305,11 @@ def tag_file(path: Path, title: Title, tmdb_id: Optional[int] | None = None) ->
return return
if config.tag_imdb_tmdb: if config.tag_imdb_tmdb:
# If tmdb_id is provided (via --tmdb), skip Simkl and use TMDB directly
if tmdb_id is not None:
log.debug("Using provided TMDB ID %s for tags", tmdb_id)
else:
# Try Simkl first for automatic lookup
simkl_data, simkl_title, simkl_tmdb_id = search_simkl(name, year, kind) simkl_data, simkl_title, simkl_tmdb_id = search_simkl(name, year, kind)
if simkl_data and simkl_title and fuzzy_match(simkl_title, name): if simkl_data and simkl_title and fuzzy_match(simkl_title, name):
@@ -319,7 +324,8 @@ def tag_file(path: Path, title: Title, tmdb_id: Optional[int] | None = None) ->
standard_tags["TVDB"] = f"https://thetvdb.com/dereferrer/series/{show_ids['tvdb']}" standard_tags["TVDB"] = f"https://thetvdb.com/dereferrer/series/{show_ids['tvdb']}"
if show_ids.get("tmdbtv"): if show_ids.get("tmdbtv"):
standard_tags["TMDB"] = f"https://www.themoviedb.org/tv/{show_ids['tmdbtv']}" standard_tags["TMDB"] = f"https://www.themoviedb.org/tv/{show_ids['tmdbtv']}"
else:
# Use TMDB API for additional metadata (either from provided ID or Simkl lookup)
api_key = _api_key() api_key = _api_key()
if not api_key: if not api_key:
log.debug("No TMDB API key set; applying basic tags only") log.debug("No TMDB API key set; applying basic tags only")