import xbmcvfs import tmdbhelper.lib.api.kodi.rpc as rpc from xbmcgui import DialogProgressBG from tmdbhelper.lib.addon.plugin import get_setting, get_localized, set_setting from jurialmunkey.parser import try_int from tmdbhelper.lib.addon.tmdate import is_unaired_timestamp, get_current_date_time from tmdbhelper.lib.files.futils import validify_filename, get_tmdb_id_nfo from tmdbhelper.lib.update.logger import _LibraryLogger from tmdbhelper.lib.update.update import BASEDIR_MOVIE, BASEDIR_TV, STRM_MOVIE, STRM_EPISODE, create_file, create_nfo, get_unique_folder from tmdbhelper.lib.update.cacher import _TVShowCache from tmdbhelper.lib.update.common import LibraryCommonFunctions from tmdbhelper.lib.api.tmdb.api import TMDb def add_to_library(info, busy_spinner=True, library_adder=None, finished=True, **kwargs): if not info: return if not library_adder: library_adder = LibraryAdder(busy_spinner) library_adder._start() if not get_setting('legacy_conversion'): library_adder.legacy_conversion() if info == 'movie' and kwargs.get('tmdb_id'): library_adder.add_movie(**kwargs) elif info == 'tv' and kwargs.get('tmdb_id'): library_adder.add_tvshow(**kwargs) elif info == 'trakt' and kwargs.get('list_slug'): library_adder.add_userlist(**kwargs) elif info == 'update': library_adder.update_tvshows(**kwargs) if not finished: return library_adder library_adder._finish() del library_adder class LibraryAdder(LibraryCommonFunctions): def __init__(self, busy_spinner=True): self.kodi_db_movies = rpc.get_kodi_library('movie', cache_refresh=True) self.kodi_db_tv = rpc.get_kodi_library('tv', cache_refresh=True) self.p_dialog = DialogProgressBG() if busy_spinner else None self.auto_update = get_setting('auto_update') self._log = _LibraryLogger() self.tv = None self.hide_unaired = get_setting('hide_unaired_episodes') self.hide_nodate = get_setting('nodate_is_unaired') # self.debug_logging = get_setting('debug_logging') self.debug_logging = True self.clean_library = False self._msg_start = get_localized(32166) self._msg_title = 'TMDbHelper Library' def get_tv_folder_nfos(self): nfos = [] nfos_append = nfos.append # For speed since we can't do a list comp easily here for f in xbmcvfs.listdir(BASEDIR_TV)[0]: tmdb_id = get_tmdb_id_nfo(BASEDIR_TV, f) nfos_append({'tmdb_id': tmdb_id, 'folder': f}) if tmdb_id else None return nfos def _legacy_conversion(self, folder, tmdb_id): # Get details details = TMDb().get_request_sc('tv', tmdb_id, append_to_response='external_ids') if not details or not details.get('first_air_date'): return # Skip shows without details/year # Get new name and compare to old name name = f'{details.get("name")} ({details["first_air_date"][:4]})' if folder == name: return # Skip if already converted # Convert name basedir = BASEDIR_TV.replace('\\', '/') old_folder = f'{basedir}{validify_filename(folder)}/' new_folder = f'{basedir}{validify_filename(name)}/' xbmcvfs.rename(old_folder, new_folder) def legacy_conversion(self, confirm=True): """ Converts old style tvshow folders without years so that they have years """ nfos = self.get_tv_folder_nfos() # Update each show in folder nfos_total = len(nfos) for x, i in enumerate(nfos): folder, tmdb_id = i['folder'], i['tmdb_id'] self._update(x, nfos_total, message=f'{get_localized(32167)} {folder}...') self._legacy_conversion(folder, tmdb_id) # Mark as complete and set to clean library set_setting('legacy_conversion', True) self.clean_library = True def update_tvshows(self, force=False, **kwargs): nfos = self.get_tv_folder_nfos() # Update each show in folder nfos_total = len(nfos) for x, i in enumerate(nfos): self._update(x, nfos_total, message=f'{get_localized(32167)} {i["folder"]}...') self.add_tvshow(tmdb_id=i['tmdb_id'], force=force) # Update last updated stamp set_setting('last_autoupdate', f'Last updated {get_current_date_time()}', 'str') def add_movie(self, tmdb_id=None, **kwargs): if not tmdb_id: return # Get movie details details = TMDb().get_request_sc('movie', tmdb_id, append_to_response='external_ids') if not details or not details.get('title'): return imdb_id = details.get('external_ids', {}).get('imdb_id') name = f'{details["title"]} ({details["release_date"][:4]})' if details.get('release_date') else details['title'] # Only add strm if not in library file = self.kodi_db_movies.get_info(info='file', imdb_id=imdb_id, tmdb_id=tmdb_id) if not file: file = create_file(STRM_MOVIE.format(tmdb_id), name, name, basedir=BASEDIR_MOVIE) create_nfo('movie', tmdb_id, name, basedir=BASEDIR_MOVIE) self._log._add('movie', tmdb_id, 'added strm file', path=file) else: self._log._add('movie', tmdb_id, 'item in library', path=file) # Return our playlist rule return ('filename', file.replace('\\', '/').split('/')[-1]) def add_tvshow(self, tmdb_id=None, force=False, **kwargs): self.tv = _TVShow(tmdb_id, force) # Return playlist rule if we don't need to check show this time if self._log._add('tv', tmdb_id, self.tv._cache.get_next_check()): return ('title', self.tv._cache.cache_info.get('name')) if not self.tv.get_details(): return # Skip if no details found on TMDb if not self.tv.get_name(): return # Skip if we don't have a folder name for some reason self.tv.get_dbid() self.tv.make_nfo() self.tv.set_next() # Add seasons for x, season in enumerate(self.tv.get_seasons()): self._update(x, self.tv.s_total, message=f'{get_localized(32167)} {self.tv.details.get("name")} - {get_localized(20373)} {season.get("season_number", 0)}...') # Update our progress dialog self._add_season(season) # Store details about what we did into the cache self.tv._cache.set_cache() # Return our playlist rule tuple return ('title', self.tv.details.get('name')) def _add_season(self, season, blacklist=[0]): number = season.get('season_number', 0) folder = f'Season {number}' # Skip blacklisted seasons if try_int(number) in blacklist: # TODO: Optional whitelist also self._log._add('tv', self.tv.tmdb_id, 'skipped special season', season=number) return # Skip if we've added season before and it isn't the most recent season # We still add most recent season even if we added it before because it might currently be airing if self._log._add('tv', self.tv.tmdb_id, self.tv._cache.is_added_season(number), season=number): return # Add our episodes for x, episode in enumerate(self.tv.get_episodes(number), 1): self._add_episode(episode, number, folder) self._update(x, self.tv.e_total) # Store a season value of where we got up to if self.tv.e_total > 2 and season.get('air_date') and not is_unaired_timestamp(season.get('air_date'), self.hide_nodate): self.tv._cache.my_history['latest_season'] = try_int(number) def _add_episode(self, episode, season, folder): number = episode.get('episode_number') filename = validify_filename(f'S{try_int(season):02d}E{try_int(number):02d} - {episode.get("name")}') self.tv._cache.my_history['episodes'].append(filename) # Skip episodes we added in the past if self._log._add('tv', self.tv.tmdb_id, self.tv._cache.is_added_episode(filename), season=season, episode=number): return # Skip future episodes if self.hide_unaired and is_unaired_timestamp(episode.get('air_date'), self.hide_nodate): self.tv._cache.my_history['skipped'].append(filename) self._log._add('tv', self.tv.tmdb_id, 'unaired episode', season=season, episode=number, air_date=episode.get('air_date')) return # Check if item has already been added file = self.tv.get_episode_db_info(season, number, info='file') if file: self._log._add('tv', self.tv.tmdb_id, 'found in library', season=season, episode=number, path=file) return # Add our strm file file = create_file(STRM_EPISODE.format(self.tv.tmdb_id, season, number), filename, self.tv.name, folder, basedir=BASEDIR_TV) self._log._add('tv', self.tv.tmdb_id, 'added strm file', season=season, episode=number, path=file) class _TVShow(): def __init__(self, tmdb_id, force=False): self._cache = _TVShowCache(tmdb_id, force) self.tmdb_id = tmdb_id self.details = None self.name = None def get_details(self): self.details = TMDb().get_request_sc('tv', self.tmdb_id, append_to_response='external_ids') if not self.details: return self.tvdb_id = self.details.get('external_ids', {}).get('tvdb_id') self.imdb_id = self.details.get('external_ids', {}).get('imdb_id') return self.details def get_name(self): date = f' ({self.details["first_air_date"][:4]})' if self.details.get('first_air_date') else '' name = f'{self.details.get("name")}{date}' self.name = get_unique_folder(name, self.tmdb_id, BASEDIR_TV) return self.name def get_dbid(self, kodi_db=None): kodi_db = kodi_db or rpc.get_kodi_library('tv') self.dbid = kodi_db.get_info(info='dbid', imdb_id=self.imdb_id, tmdb_id=self.tmdb_id, tvdb_id=self.tvdb_id) return self.dbid def get_episode_db_info(self, season, episode, info='dbid'): if not self.dbid: return return rpc.KodiLibrary(dbtype='episode', tvshowid=self.dbid, logging=False).get_info( info=info, season=season, episode=episode) def get_seasons(self): self.seasons = self.details.get('seasons', []) self.s_total = len(self.seasons) return self.seasons def get_episodes(self, season): self.e_total = 0 self.season_details = TMDb().get_request('tv', self.tmdb_id, 'season', season, cache_refresh=True) if not self.season_details: return [] self.episodes = [i for i in self.season_details.get('episodes', []) if i.get('episode_number', 0) != 0] self.e_total = len(self.episodes) return self.episodes def make_nfo(self): create_nfo('tv', self.tmdb_id, self.name, basedir=BASEDIR_TV) def set_next(self): self._cache.create_new_cache(self.details.get('name', '')) self._cache.set_next_check( next_aired=self.details.get('next_episode_to_air', {}), last_aired=self.details.get('last_episode_to_air', {}), status=self.details.get('status'))