diff --git a/src/prep.py b/src/prep.py index 1e8225493..21c01e95d 100644 --- a/src/prep.py +++ b/src/prep.py @@ -95,12 +95,11 @@ async def update_metadata_from_tracker(self, tracker_name, tracker_instance, met blu_tmdb, blu_imdb, blu_tvdb, blu_mal, blu_desc, blu_category, meta['ext_torrenthash'], blu_imagelist, blu_filename = await COMMON(self.config).unit3d_torrent_info("BLU", tracker_instance.torrent_url, tracker_instance.search_url, id=meta[tracker_key]) if blu_tmdb not in [None, '0'] or blu_imdb not in [None, '0'] or blu_tvdb not in [None, '0']: console.print(f"[green]Valid data found on {tracker_name}, setting meta values[/green]") - # Prompt user for selection if await self.prompt_user_for_id_selection(blu_tmdb, blu_imdb, blu_tvdb, blu_filename): if blu_tmdb not in [None, '0']: meta['tmdb_manual'] = blu_tmdb if blu_imdb not in [None, '0']: - meta['imdb'] = str(blu_imdb) + meta['imdb'] = str(blu_imdb).zfill(7) # Pad IMDb ID with leading zeros if blu_tvdb not in [None, '0']: meta['tvdb_id'] = blu_tvdb if blu_mal not in [None, '0']: @@ -115,17 +114,20 @@ async def update_metadata_from_tracker(self, tracker_name, tracker_instance, met meta['blu_filename'] = blu_filename # Store the filename in meta for later use found_match = True # Set flag if any relevant data is found else: - console.print(f"[yellow]User skipped the found ID on {tracker_name}[/yellow]") + console.print(f"[yellow]User skipped the found ID on {tracker_name}, moving to the next site.[/yellow]") + return meta, found_match # Return immediately to skip the current site else: console.print(f"[yellow]No valid data found on {tracker_name}[/yellow]") else: meta['imdb'], meta['ext_torrenthash'] = await tracker_instance.get_imdb_from_torrent_id(meta[tracker_key]) if meta['imdb']: + meta['imdb'] = str(meta['imdb']).zfill(7) # Pad IMDb ID with leading zeros if await self.prompt_user_for_id_selection(imdb=meta['imdb']): console.print(f"[green]{tracker_name} IMDb ID found: {meta['imdb']}[/green]") found_match = True else: - console.print(f"[yellow]User skipped the found IMDb ID on {tracker_name}[/yellow]") + console.print(f"[yellow]User skipped the found IMDb ID on {tracker_name}, moving to the next site.[/yellow]") + return meta, found_match # Return immediately to skip the current site else: console.print(f"[yellow]No IMDb ID found on {tracker_name}[/yellow]") else: @@ -146,7 +148,7 @@ async def update_metadata_from_tracker(self, tracker_name, tracker_instance, met if blu_tmdb not in [None, '0']: meta['tmdb_manual'] = blu_tmdb if blu_imdb not in [None, '0']: - meta['imdb'] = str(blu_imdb) + meta['imdb'] = str(blu_imdb).zfill(7) # Pad IMDb ID with leading zeros if blu_tvdb not in [None, '0']: meta['tvdb_id'] = blu_tvdb if blu_mal not in [None, '0']: @@ -160,18 +162,23 @@ async def update_metadata_from_tracker(self, tracker_name, tracker_instance, met if blu_filename: meta['blu_filename'] = blu_filename # Store the filename in meta for later use found_match = True + else: + console.print(f"[yellow]User skipped the found ID on {tracker_name}, moving to the next site.[/yellow]") + return meta, found_match # Return immediately to skip the current site else: console.print(f"[yellow]No valid data found on {tracker_name}[/yellow]") else: imdb = tracker_id = None if imdb: + imdb = str(imdb).zfill(7) # Pad IMDb ID with leading zeros if await self.prompt_user_for_id_selection(imdb=imdb): console.print(f"[green]{tracker_name} IMDb ID found: {imdb}[/green]") - meta['imdb'] = str(imdb) + meta['imdb'] = imdb found_match = True else: - console.print(f"[yellow]User skipped the found IMDb ID on {tracker_name}[/yellow]") + console.print(f"[yellow]User skipped the found IMDb ID on {tracker_name}, moving to the next site.[/yellow]") + return meta, found_match # Return immediately to skip the current site if tracker_id: meta[tracker_key] = tracker_id @@ -374,7 +381,6 @@ async def gather_prep(self, meta, mode): meta = await self.get_tmdb_from_imdb(meta, filename) else: meta['tmdb_manual'] = meta.get('tmdb', None) - # If no tmdb, use imdb for meta if int(meta['tmdb']) == 0: @@ -490,9 +496,6 @@ async def get_disc(self, meta): discs = sorted(discs, key=lambda d: d['name']) return is_disc, videoloc, bdinfo, discs - - - """ Get video files @@ -517,8 +520,6 @@ def get_video(self, videoloc, mode): filelist = sorted(filelist) return video, filelist - - """ Get and parse mediainfo """ @@ -1587,13 +1588,6 @@ def get_romaji(self, tmdb_name, mal): episodes = 0 return romaji, mal_id, eng_title, season_year, episodes - - - - - - - """ Mediainfo/Bdinfo > meta """