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 __init__(self): self.handle = int(sys.argv[1]) self.paramstring = try_decode(sys.argv[2][1:]) self.params = parse_paramstring(self.paramstring) self.parent_params = self.params # self.container_path = u'{}?{}'.format(sys.argv[0], self.paramstring) self.update_listing = False self.plugin_category = '' self.container_content = '' self.container_update = None self.container_refresh = False self.item_type = None self.kodi_db = None self.kodi_db_tv = {} self.library = None self.tmdb_cache_only = True self.tmdb_api = TMDb() self.trakt_watchedindicators = ADDON.getSettingBool('trakt_watchedindicators') self.trakt_api = TraktAPI() self.is_widget = True if self.params.pop('widget', '').lower() == 'true' else False self.hide_watched = ADDON.getSettingBool('widgets_hidewatched') if self.is_widget else False self.flatten_seasons = ADDON.getSettingBool('flatten_seasons') self.ftv_forced_lookup = self.params.pop('fanarttv', '').lower() self.ftv_api = FanartTV(cache_only=self.ftv_is_cache_only()) self.filter_key = self.params.pop('filter_key', None) self.filter_value = split_items(self.params.pop('filter_value', None))[0] self.exclude_key = self.params.pop('exclude_key', None) self.exclude_value = split_items(self.params.pop('exclude_value', None))[0] self.pagination = self.pagination_is_allowed() self.params = reconfigure_legacy_params(**self.params)
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 = u'{} ({})'.format(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 get_item_details(tmdb_type, tmdb_id, season=None, episode=None): details = TMDb().get_details(tmdb_type, tmdb_id, season, episode) if not details: return details = ListItem(**details) details.infolabels['mediatype'] == 'movie' if tmdb_type == 'movie' else 'episode' details.set_details(details=get_external_ids(details, season=season, episode=episode)) return details
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 __init__(self): self.properties = set() self.index_properties = set() self.trakt_api = TraktAPI() self.tmdb_api = TMDb() self.fanarttv = FanartTV() self.omdb_api = OMDb() if ADDON.getSettingString('omdb_apikey') else None self.imdb_top250 = {} self.property_prefix = 'ListItem'
def _get_query(tmdb_type, method, query=None, header=None, use_details=False): item = TMDb().get_tmdb_id_from_query( tmdb_type=tmdb_type, query=query or try_decode(xbmcgui.Dialog().input(header)), header=header or u'{} {}'.format(ADDON.getLocalizedString(32276), tmdb_type), use_details=use_details, get_listitem=True) if item and item.getUniqueID('tmdb'): return {'label': item.getLabel(), 'value': item.getUniqueID('tmdb'), 'method': method}
def get_item_details(tmdb_type, tmdb_id, season=None, episode=None, language=None): tmdb_api = TMDb(language=language) if language else TMDb() details = tmdb_api.get_details(tmdb_type, tmdb_id, season, episode) del tmdb_api if not details: return details = ListItem(**details) details.infolabels['mediatype'] == 'movie' if tmdb_type == 'movie' else 'episode' details.set_details(details=get_external_ids(details, season=season, episode=episode)) return details
def _get_language_details(tmdb_type, tmdb_id, season=None, episode=None, language=None): details = TMDb().get_request_lc(tmdb_type, tmdb_id, 'translations') if not details or not details.get('translations'): return item = {} for i in details['translations']: if i.get('iso_639_1') == language: item['title'] = i.get('data', {}).get('title') or i.get('data', {}).get('name') item['plot'] = i.get('data', {}).get('overview') return item
def _translate_discover_params(tmdb_type, params): lookup_keyword = None if params.get('with_id') and params.get('with_id') != 'False' else 'keyword' lookup_company = None if params.get('with_id') and params.get('with_id') != 'False' else 'company' lookup_person = None if params.get('with_id') and params.get('with_id') != 'False' else 'person' lookup_genre = None if params.get('with_id') and params.get('with_id') != 'False' else 'genre' with_separator = params.get('with_separator') if params.get('with_genres'): params['with_genres'] = TMDb().get_translated_list( split_items(params.get('with_genres')), lookup_genre, separator=with_separator) if params.get('without_genres'): params['without_genres'] = TMDb().get_translated_list( split_items(params.get('without_genres')), lookup_genre, separator=with_separator) if params.get('with_keywords'): params['with_keywords'] = TMDb().get_translated_list( split_items(params.get('with_keywords')), lookup_keyword, separator=with_separator) if params.get('without_keywords'): params['without_keywords'] = TMDb().get_translated_list( split_items(params.get('without_keywords')), lookup_keyword, separator=with_separator) if params.get('with_companies'): params['with_companies'] = TMDb().get_translated_list( split_items(params.get('with_companies')), lookup_company, separator='NONE') if params.get('with_people'): params['with_people'] = TMDb().get_translated_list( split_items(params.get('with_people')), lookup_person, separator=with_separator) if params.get('with_cast'): params['with_cast'] = TMDb().get_translated_list( split_items(params.get('with_cast')), lookup_person, separator=with_separator) if params.get('with_crew'): params['with_crew'] = TMDb().get_translated_list( split_items(params.get('with_crew')), lookup_person, separator=with_separator) if params.get('with_release_type'): params['with_release_type'] = TMDb().get_translated_list( split_items(params.get('with_release_type')), None, separator='OR') # Translate relative dates based upon today's date for i in RELATIVE_DATES: datecode = params.get(i, '') datecode = datecode.lower() if not datecode or all(x not in datecode for x in ['t-', 't+']): continue # No value or not a relative date so skip elif 't-' in datecode: days = try_int(datecode.replace('t-', '')) date = get_datetime_now() - get_timedelta(days=days) elif 't+' in datecode: days = try_int(datecode.replace('t+', '')) date = get_datetime_now() + get_timedelta(days=days) params[i] = date.strftime("%Y-%m-%d") return params
def list_discover(self, tmdb_type, **kwargs): kwargs.pop('info', None) items = TMDb().get_discover_list(tmdb_type, **_translate_discover_params(tmdb_type, kwargs)) self.kodi_db = self.get_kodi_database(tmdb_type) self.library = convert_type(tmdb_type, 'library') self.container_content = convert_type(tmdb_type, 'container') return items
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 = u'{} ({})'.format(details.get('name'), details['first_air_date'][:4]) if folder == name: return # Skip if already converted # Convert name basedir = BASEDIR_TV.replace('\\', '/') old_folder = u'{}{}/'.format(basedir, validify_filename(folder)) new_folder = u'{}{}/'.format(basedir, validify_filename(name)) xbmcvfs.rename(old_folder, new_folder)
def log_request(**kwargs): with busy_dialog(): kwargs['response'] = None if not kwargs.get('url'): kwargs['url'] = xbmcgui.Dialog().input('URL') if not kwargs['url']: return if kwargs.get('log_request').lower() == 'trakt': kwargs['response'] = TraktAPI().get_response_json(kwargs['url']) else: kwargs['response'] = TMDb().get_response_json(kwargs['url']) if not kwargs['response']: xbmcgui.Dialog().ok(kwargs['log_request'].capitalize(), u'{}\nNo Response!'.format(kwargs['url'])) return filename = validify_filename(u'{}_{}.json'.format( kwargs['log_request'], kwargs['url'])) dumps_to_file(kwargs, 'log_request', filename) xbmcgui.Dialog().ok( kwargs['log_request'].capitalize(), u'[B]{}[/B]\n\n{}\n{}\n{}'.format( kwargs['url'], xbmc.translatePath('special://profile/addon_data/'), 'plugin.video.themoviedb.helper/log_request', filename)) xbmcgui.Dialog().textviewer(filename, dumps(kwargs['response'], indent=2))
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): name = u'{}{}'.format( self.details.get('name'), u' ({})'.format(self.details['first_air_date'][:4]) if self.details.get('first_air_date') else '') 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'))
def add_query(self, query, tmdb_type): with busy_dialog(): query = try_decode(query) tmdb_id = TMDb().get_tmdb_id_from_query(tmdb_type, query, header=query, use_details=True, auto_single=True) if not tmdb_id: xbmcgui.Dialog().notification('TMDbHelper', ADDON.getLocalizedString(32310).format(query)) return url = 'plugin://plugin.video.themoviedb.helper/?info=details&tmdb_type={}&tmdb_id={}' url = url.format(tmdb_type, tmdb_id) return self.add_path(url)
def refresh_details(tmdb_id=None, tmdb_type=None, season=None, episode=None, confirm=True, **kwargs): if not tmdb_id or not tmdb_type: return with busy_dialog(): details = TMDb().get_details(tmdb_type, tmdb_id, season, episode, cache_refresh=True) if details and confirm: xbmcgui.Dialog().ok( 'TMDbHelper', ADDON.getLocalizedString(32234).format(tmdb_type, tmdb_id)) container_refresh() return details
def refresh_details(tmdb_id=None, tmdb_type=None, season=None, episode=None, **kwargs): if not tmdb_id or not tmdb_type: return with busy_dialog(): details = TMDb().get_details(tmdb_type, tmdb_id, season, episode, cache_refresh=True) if details: xbmcgui.Dialog().ok( 'TMDbHelper', ADDON.getLocalizedString(32234).format(tmdb_type, tmdb_id)) xbmc.executebuiltin('Container.Refresh') xbmc.executebuiltin( 'UpdateLibrary(video,/fake/path/to/force/refresh/on/home)')
class Container(TMDbLists, BaseDirLists, SearchLists, UserDiscoverLists, TraktLists): def __init__(self): self.handle = int(sys.argv[1]) self.paramstring = try_decode(sys.argv[2][1:]) self.params = parse_paramstring(self.paramstring) self.parent_params = self.params # self.container_path = u'{}?{}'.format(sys.argv[0], self.paramstring) self.update_listing = False self.plugin_category = '' self.container_content = '' self.container_update = None self.container_refresh = False self.item_type = None self.kodi_db = None self.kodi_db_tv = {} self.library = None self.tmdb_cache_only = True self.tmdb_api = TMDb() self.trakt_watchedindicators = ADDON.getSettingBool('trakt_watchedindicators') self.trakt_api = TraktAPI() self.is_widget = True if self.params.pop('widget', '').lower() == 'true' else False self.hide_watched = ADDON.getSettingBool('widgets_hidewatched') if self.is_widget else False self.flatten_seasons = ADDON.getSettingBool('flatten_seasons') self.ftv_forced_lookup = self.params.pop('fanarttv', '').lower() self.ftv_api = FanartTV(cache_only=self.ftv_is_cache_only()) self.filter_key = self.params.pop('filter_key', None) self.filter_value = split_items(self.params.pop('filter_value', None))[0] self.exclude_key = self.params.pop('exclude_key', None) self.exclude_value = split_items(self.params.pop('exclude_value', None))[0] self.pagination = self.pagination_is_allowed() self.params = reconfigure_legacy_params(**self.params) def pagination_is_allowed(self): if self.params.pop('nextpage', '').lower() == 'false': return False if self.is_widget and not ADDON.getSettingBool('widgets_nextpage'): return False return True def ftv_is_cache_only(self): if self.ftv_forced_lookup == 'true': return False if self.ftv_forced_lookup == 'false': return True if self.is_widget and ADDON.getSettingBool('widget_fanarttv_lookup'): return False if not self.is_widget and ADDON.getSettingBool('fanarttv_lookup'): return False return True def _add_item(self, x, li, tmdb_cache_only): cache_only = True if tmdb_cache_only and not self.ftv_api else False li.set_details(details=self.get_tmdb_details(li, cache_only=cache_only)) # Quick because only get cached li.set_details(details=self.get_ftv_artwork(li), reverse=True) # Slow when not cache only self.items_queue[x] = li def add_items(self, items=None, pagination=True, parent_params=None, property_params=None, kodi_db=None, tmdb_cache_only=True): if not items: return self.check_is_aired = parent_params.get('info') not in NO_LABEL_FORMATTING # Build empty queue and thread pool self.items_queue, pool = [None] * len(items), [None] * len(items) # Start item build threads for x, i in enumerate(items): if not pagination and 'next_page' in i: continue if self.item_is_excluded(i): continue li = ListItem(parent_params=parent_params, **i) pool[x] = Thread(target=self._add_item, args=[x, li, tmdb_cache_only]) pool[x].start() # Wait to join threads in pool first before adding item to directory for x, i in enumerate(pool): if not i: continue i.join() li = self.items_queue[x] if not li: continue li.set_episode_label() if self.check_is_aired and li.is_unaired(): return li.set_details(details=self.get_kodi_details(li), reverse=True) # Quick because local db li.set_playcount(playcount=self.get_playcount_from_trakt(li)) # Quick because of agressive caching of Trakt object and pre-emptive dict comprehension if self.hide_watched and try_int(li.infolabels.get('playcount')) != 0: continue li.set_context_menu() # Set the context menu items li.set_uids_to_info() # Add unique ids to properties so accessible in skins li.set_params_reroute(self.ftv_forced_lookup, self.flatten_seasons) # Reroute details to proper end point li.set_params_to_info(self.plugin_category) # Set path params to properties for use in skins li.infoproperties.update(property_params or {}) xbmcplugin.addDirectoryItem( handle=self.handle, url=li.get_url(), listitem=li.get_listitem(), isFolder=li.is_folder) def set_params_to_container(self, **kwargs): params = {} for k, v in viewitems(kwargs): if not k or not v: continue try: k = u'Param.{}'.format(k) v = u'{}'.format(v) params[k] = v xbmcplugin.setProperty(self.handle, k, v) # Set params to container properties except Exception as exc: kodi_log(u'Error: {}\nUnable to set param {} to {}'.format(exc, k, v), 1) return params def finish_container(self, update_listing=False, plugin_category='', container_content=''): xbmcplugin.setPluginCategory(self.handle, plugin_category) # Container.PluginCategory xbmcplugin.setContent(self.handle, container_content) # Container.Content xbmcplugin.endOfDirectory(self.handle, updateListing=update_listing) def item_is_excluded(self, item): if self.filter_key and self.filter_value: if self.filter_key in item.get('infolabels', {}): if filtered_item(item['infolabels'], self.filter_key, self.filter_value): return True elif self.filter_key in item.get('infoproperties', {}): if filtered_item(item['infoproperties'], self.filter_key, self.filter_value): return True if self.exclude_key and self.exclude_value: if self.exclude_key in item.get('infolabels', {}): if filtered_item(item['infolabels'], self.exclude_key, self.exclude_value, True): return True elif self.exclude_key in item.get('infoproperties', {}): if filtered_item(item['infoproperties'], self.exclude_key, self.exclude_value, True): return True def get_tmdb_details(self, li, cache_only=True): if not self.tmdb_api: return return self.tmdb_api.get_details( li.get_tmdb_type(), li.unique_ids.get('tvshow.tmdb') if li.infolabels.get('mediatype') in ['season', 'episode'] else li.unique_ids.get('tmdb'), li.infolabels.get('season') if li.infolabels.get('mediatype') in ['season', 'episode'] else None, li.infolabels.get('episode') if li.infolabels.get('mediatype') == 'episode' else None, cache_only=cache_only) def get_ftv_artwork(self, li): if not self.ftv_api: return artwork = self.ftv_api.get_all_artwork(li.get_ftv_id(), li.get_ftv_type()) if not artwork: return if li.infolabels.get('mediatype') in ['season', 'episode']: artwork = {u'tvshow.{}'.format(k): v for k, v in viewitems(artwork) if v} return {'art': artwork} def get_playcount_from_trakt(self, li): if not self.trakt_watchedindicators: return if li.infolabels.get('mediatype') == 'movie': return self.trakt_api.get_movie_playcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb'))) if li.infolabels.get('mediatype') == 'episode': return self.trakt_api.get_episode_playcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tvshow.tmdb')), season=li.infolabels.get('season'), episode=li.infolabels.get('episode')) if li.infolabels.get('mediatype') == 'tvshow': li.infolabels['episode'] = self.trakt_api.get_episodes_airedcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb'))) return self.trakt_api.get_episodes_watchcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb'))) if li.infolabels.get('mediatype') == 'season': li.infolabels['episode'] = self.trakt_api.get_episodes_airedcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb')), season=li.infolabels.get('season')) return self.trakt_api.get_episodes_watchcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb')), season=li.infolabels.get('season')) def get_kodi_database(self, tmdb_type): if ADDON.getSettingBool('local_db'): return get_kodi_library(tmdb_type) def get_kodi_parent_dbid(self, li): if not self.kodi_db: return if li.infolabels.get('mediatype') in ['movie', 'tvshow']: return self.kodi_db.get_info( info='dbid', imdb_id=li.unique_ids.get('imdb'), tmdb_id=li.unique_ids.get('tmdb'), tvdb_id=li.unique_ids.get('tvdb'), originaltitle=li.infolabels.get('originaltitle'), title=li.infolabels.get('title'), year=li.infolabels.get('year')) if li.infolabels.get('mediatype') in ['season', 'episode']: return self.kodi_db.get_info( info='dbid', imdb_id=li.unique_ids.get('tvshow.imdb'), tmdb_id=li.unique_ids.get('tvshow.tmdb'), tvdb_id=li.unique_ids.get('tvshow.tvdb'), title=li.infolabels.get('tvshowtitle')) def get_kodi_details(self, li): if not self.kodi_db: return dbid = self.get_kodi_parent_dbid(li) if not dbid: return if li.infolabels.get('mediatype') == 'movie': return get_movie_details(dbid) if li.infolabels.get('mediatype') == 'tvshow': return get_tvshow_details(dbid) if li.infolabels.get('mediatype') == 'season': return set_show(self.get_kodi_tvchild_details( tvshowid=dbid, season=li.infolabels.get('season'), is_season=True) or get_empty_item(), get_tvshow_details(dbid)) if li.infolabels.get('mediatype') == 'episode': return set_show(self.get_kodi_tvchild_details( tvshowid=dbid, season=li.infolabels.get('season'), episode=li.infolabels.get('episode')) or get_empty_item(), get_tvshow_details(dbid)) def get_kodi_tvchild_details(self, tvshowid, season=None, episode=None, is_season=False): if not tvshowid or not season or (not episode and not is_season): return library = 'season' if is_season else 'episode' self.kodi_db_tv[tvshowid] = self.kodi_db_tv.get(tvshowid) or get_kodi_library(library, tvshowid) if not self.kodi_db_tv[tvshowid].database: return dbid = self.kodi_db_tv[tvshowid].get_info('dbid', season=season, episode=episode) if not dbid: return details = get_season_details(dbid) if is_season else get_episode_details(dbid) details['infoproperties']['tvshow.dbid'] = tvshowid return details def get_container_content(self, tmdb_type, season=None, episode=None): if tmdb_type == 'tv' and season and episode: return convert_type('episode', 'container') elif tmdb_type == 'tv' and season: return convert_type('season', 'container') return convert_type(tmdb_type, 'container') def list_randomised_trakt(self, **kwargs): kwargs['info'] = RANDOMISED_TRAKT.get(kwargs.get('info'), {}).get('info') kwargs['randomise'] = True self.parent_params = kwargs return self.get_items(**kwargs) def list_randomised(self, **kwargs): params = merge_two_dicts(kwargs, RANDOMISED_LISTS.get(kwargs.get('info'), {}).get('params')) item = random_from_list(self.get_items(**params)) if not item: return self.plugin_category = item.get('label') self.parent_params = item.get('params', {}) return self.get_items(**item.get('params', {})) def get_tmdb_id(self, info, **kwargs): if info == 'collection': kwargs['tmdb_type'] = 'collection' return self.tmdb_api.get_tmdb_id(**kwargs) def _noop(self): return None def _get_items(self, func, **kwargs): return func['lambda'](getattr(self, func['getattr']), **kwargs) def get_items(self, **kwargs): info = kwargs.get('info') # Check routes that don't require ID lookups first route = ROUTE_NO_ID route.update(TRAKT_LIST_OF_LISTS) route.update(RANDOMISED_LISTS) route.update(RANDOMISED_TRAKT) # Early exit if we have a route if info in route: return self._get_items(route[info]['route'], **kwargs) # Check routes that require ID lookups second route = ROUTE_TMDB_ID route.update(TMDB_BASIC_LISTS) route.update(TRAKT_BASIC_LISTS) route.update(TRAKT_SYNC_LISTS) # Early exit to basedir if no route found if info not in route: return self.list_basedir(info) # Lookup up our TMDb ID if not kwargs.get('tmdb_id'): kwargs['tmdb_id'] = self.get_tmdb_id(**kwargs) return self._get_items(route[info]['route'], **kwargs) def get_directory(self): items = self.get_items(**self.params) if not items: return self.add_items( items, pagination=self.pagination, parent_params=self.parent_params, property_params=self.set_params_to_container(**self.params), kodi_db=self.kodi_db, tmdb_cache_only=self.tmdb_cache_only) self.finish_container( update_listing=self.update_listing, plugin_category=self.plugin_category, container_content=self.container_content) if self.container_update: xbmc.executebuiltin(try_encode(u'Container.Update({})'.format(self.container_update))) if self.container_refresh: xbmc.executebuiltin('Container.Refresh') def play_external(self, **kwargs): kodi_log(['lib.container.router - Attempting to play item\n', kwargs], 1) if not kwargs.get('tmdb_id'): kwargs['tmdb_id'] = self.tmdb_api.get_tmdb_id(**kwargs) Players(**kwargs).play(handle=self.handle if self.handle != -1 else None) def context_related(self, **kwargs): if not kwargs.get('tmdb_id'): kwargs['tmdb_id'] = self.tmdb_api.get_tmdb_id(**kwargs) kwargs['container_update'] = True related_lists(include_play=True, **kwargs) def router(self): if self.params.get('info') == 'play': return self.play_external(**self.params) if self.params.get('info') == 'related': return self.context_related(**self.params) return self.get_directory()
class CommonMonitorFunctions(object): def __init__(self): self.properties = set() self.index_properties = set() self.trakt_api = TraktAPI() self.tmdb_api = TMDb() self.fanarttv = FanartTV() self.omdb_api = OMDb() if ADDON.getSettingString( 'omdb_apikey') else None self.imdb_top250 = {} self.property_prefix = 'ListItem' @try_except_log('lib.monitor.common clear_property') def clear_property(self, key): key = u'{}.{}'.format(self.property_prefix, key) get_property(key, clear_property=True) @try_except_log('lib.monitor.common set_property') def set_property(self, key, value): key = u'{}.{}'.format(self.property_prefix, key) if value is None: get_property(key, clear_property=True) else: get_property(key, set_property=u'{0}'.format(try_decode(value))) def set_iter_properties(self, dictionary, keys): if not isinstance(dictionary, dict): dictionary = {} for k in keys: try: v = dictionary.get(k, None) if isinstance(v, list): try: v = ' / '.join(v) except Exception as exc: kodi_traceback( exc, u'\nlib.monitor.common set_iter_properties\nk: {} v: {}' .format(k, v)) self.properties.add(k) self.set_property(k, v) except Exception as exc: kodi_traceback( exc, u'\nlib.monitor.common set_iter_properties\nk: {}'.format( k)) def set_indexed_properties(self, dictionary): if not isinstance(dictionary, dict): return index_properties = set() for k, v in viewitems(dictionary): if k in self.properties or k in SETPROP_RATINGS or k in SETMAIN_ARTWORK: continue try: v = v or '' self.set_property(k, v) index_properties.add(k) except Exception as exc: kodi_traceback( exc, u'\nlib.monitor.common set_indexed_properties\nk: {} v: {}' .format(k, v)) for k in (self.index_properties - index_properties): self.clear_property(k) self.index_properties = index_properties.copy() @try_except_log('lib.monitor.common set_list_properties') def set_list_properties(self, items, key, prop): if not isinstance(items, list): return joinlist = [i[key] for i in items[:10] if i.get(key)] joinlist = ' / '.join(joinlist) self.properties.add(prop) self.set_property(prop, joinlist) @try_except_log('lib.monitor.common set_time_properties') def set_time_properties(self, duration): minutes = duration // 60 % 60 hours = duration // 60 // 60 totalmin = duration // 60 self.set_property('Duration', totalmin) self.set_property('Duration_H', hours) self.set_property('Duration_M', minutes) self.set_property('Duration_HHMM', u'{0:02d}:{1:02d}'.format(hours, minutes)) self.properties.update( ['Duration', 'Duration_H', 'Duration_M', 'Duration_HHMM']) def set_properties(self, item): self.set_iter_properties(item, SETMAIN) self.set_iter_properties(item.get('infolabels', {}), SETINFO) self.set_iter_properties(item.get('infoproperties', {}), SETPROP) self.set_time_properties(item.get('infolabels', {}).get('duration', 0)) self.set_list_properties(item.get('cast', []), 'name', 'cast') if xbmc.getCondVisibility( "!Skin.HasSetting(TMDbHelper.DisableExtendedProperties)"): self.set_indexed_properties(item.get('infoproperties', {})) @try_except_log('lib.monitor.common get_tmdb_id') def get_tmdb_id(self, tmdb_type, imdb_id=None, query=None, year=None, episode_year=None): if imdb_id and imdb_id.startswith('tt'): return self.tmdb_api.get_tmdb_id(tmdb_type=tmdb_type, imdb_id=imdb_id) return self.tmdb_api.get_tmdb_id(tmdb_type=tmdb_type, query=query, year=year, episode_year=episode_year) def get_fanarttv_artwork(self, item, tmdb_type=None, tmdb_id=None, tvdb_id=None): if not self.fanarttv or tmdb_type not in ['movie', 'tv']: return item lookup_id = None if tmdb_type == 'tv': lookup_id = tvdb_id or item.get('unique_ids', {}).get('tvshow.tvdb') or item.get( 'unique_ids', {}).get('tvdb') func = self.fanarttv.get_tv_all_artwork elif tmdb_type == 'movie': lookup_id = tmdb_id or item.get('unique_ids', {}).get('tmdb') func = self.fanarttv.get_movies_all_artwork if lookup_id: item['art'] = merge_two_dicts(item.get('art', {}), func(lookup_id)) return item def get_trakt_ratings(self, item, trakt_type, season=None, episode=None): ratings = self.trakt_api.get_ratings( trakt_type=trakt_type, imdb_id=item.get('unique_ids', {}).get('tvshow.imdb') or item.get('unique_ids', {}).get('imdb'), trakt_id=item.get('unique_ids', {}).get('tvshow.trakt') or item.get('unique_ids', {}).get('trakt'), slug_id=item.get('unique_ids', {}).get('tvshow.slug') or item.get('unique_ids', {}).get('slug'), season=season, episode=episode) if not ratings: return item item['infoproperties'] = merge_two_dicts( item.get('infoproperties', {}), ratings) return item def get_imdb_top250_rank(self, item): if not self.imdb_top250: self.imdb_top250 = self.trakt_api.get_imdb_top250(id_type='tmdb') try: item['infolabels']['top250'] = self.imdb_top250.index( try_int(item.get('unique_ids', {}).get('tmdb'))) + 1 except Exception: pass return item def get_omdb_ratings(self, item, cache_only=False): if not self.omdb_api: return item imdb_id = item.get('infolabels', {}).get('imdbnumber') if not imdb_id or not imdb_id.startswith('tt'): imdb_id = item.get('unique_ids', {}).get('imdb') if not imdb_id or not imdb_id.startswith('tt'): imdb_id = item.get('unique_ids', {}).get('tvshow.imdb') if not imdb_id: return item ratings = self.omdb_api.get_ratings_awards(imdb_id=imdb_id, cache_only=cache_only) item['infoproperties'] = merge_two_dicts( item.get('infoproperties', {}), ratings.get('infoproperties', {})) return item def clear_properties(self, ignore_keys=None): ignore_keys = ignore_keys or set() for k in self.properties - ignore_keys: self.clear_property(k) self.properties = set() for k in self.index_properties: self.clear_property(k) self.index_properties = set() def clear_property_list(self, properties): for k in properties: self.clear_property(k)
def _get_genre(tmdb_type, method): data_list = TMDb().get_request_lc('genre', tmdb_type, 'list') if data_list and data_list.get('genres'): return _select_properties(data_list['genres'], method, header=ADDON.getLocalizedString(32112))
def wrapper(*args, **kwargs): with busy_dialog(): if not kwargs.get('tmdb_id'): kwargs['tmdb_id'] = TMDb().get_tmdb_id(**kwargs) return func(*args, **kwargs)
class Container(TMDbLists, BaseDirLists, SearchLists, UserDiscoverLists, TraktLists): def __init__(self): self.handle = int(sys.argv[1]) self.paramstring = try_decode(sys.argv[2][1:]) self.params = parse_paramstring(self.paramstring) self.parent_params = self.params self.container_path = '{}?{}'.format(sys.argv[0], self.paramstring) self.update_listing = False self.plugin_category = '' self.container_content = '' self.container_update = None self.container_refresh = False self.item_type = None self.kodi_db = None self.kodi_db_tv = {} self.library = None self.tmdb_cache_only = True self.tmdb_api = TMDb() self.trakt_watchedindicators = ADDON.getSettingBool( 'trakt_watchedindicators') self.trakt_api = TraktAPI() self.is_widget = True if self.params.pop( 'widget', '').lower() == 'true' else False self.hide_watched = ADDON.getSettingBool( 'widgets_hidewatched') if self.is_widget else False self.flatten_seasons = ADDON.getSettingBool('flatten_seasons') self.ftv_forced_lookup = self.params.pop('fanarttv', '').lower() self.ftv_api = FanartTV(cache_only=self.ftv_is_cache_only()) self.filter_key = self.params.pop('filter_key', None) self.filter_value = split_items(self.params.pop('filter_value', None))[0] self.exclude_key = self.params.pop('exclude_key', None) self.exclude_value = split_items(self.params.pop( 'exclude_value', None))[0] self.pagination = self.pagination_is_allowed() self.params = reconfigure_legacy_params(**self.params) def pagination_is_allowed(self): if self.params.pop('nextpage', '').lower() == 'false': return False if self.is_widget and not ADDON.getSettingBool('widgets_nextpage'): return False return True def ftv_is_cache_only(self): if self.ftv_forced_lookup == 'true': return False if self.ftv_forced_lookup == 'false': return True if self.is_widget and ADDON.getSettingBool('widget_fanarttv_lookup'): return False if not self.is_widget and ADDON.getSettingBool('fanarttv_lookup'): return False return True def add_items(self, items=None, pagination=True, parent_params=None, kodi_db=None, tmdb_cache_only=True): if not items: return check_is_aired = parent_params.get('info') not in NO_LABEL_FORMATTING for i in items: if not pagination and 'next_page' in i: continue if self.item_is_excluded(i): continue li = ListItem(parent_params=parent_params, **i) li.set_details(details=self.get_tmdb_details( li, cache_only=tmdb_cache_only)) # Quick because only get cached li.set_episode_label() if check_is_aired and li.is_unaired(): continue li.set_details(details=self.get_ftv_artwork(li), reverse=True) # Slow when not cache only li.set_details(details=self.get_kodi_details(li), reverse=True) # Quick because local db li.set_playcount( playcount=self.get_playcount_from_trakt(li) ) # Quick because of agressive caching of Trakt object and pre-emptive dict comprehension if self.hide_watched and try_int( li.infolabels.get('playcount')) != 0: continue li.set_context_menu() # Set the context menu items li.set_uids_to_info( ) # Add unique ids to properties so accessible in skins li.set_params_reroute( self.ftv_forced_lookup, self.flatten_seasons) # Reroute details to proper end point li.set_params_to_info( self.plugin_category ) # Set path params to properties for use in skins xbmcplugin.addDirectoryItem(handle=self.handle, url=li.get_url(), listitem=li.get_listitem(), isFolder=li.is_folder) def set_params_to_container(self, **kwargs): for k, v in viewitems(kwargs): if not k or not v: continue try: xbmcplugin.setProperty( self.handle, u'Param.{}'.format(k), u'{}'.format(v)) # Set params to container properties except Exception as exc: kodi_log( u'Error: {}\nUnable to set Param.{} to {}'.format( exc, k, v), 1) def finish_container(self, update_listing=False, plugin_category='', container_content=''): xbmcplugin.setPluginCategory( self.handle, plugin_category) # Container.PluginCategory xbmcplugin.setContent(self.handle, container_content) # Container.Content xbmcplugin.endOfDirectory(self.handle, updateListing=update_listing) def item_is_excluded(self, item): if self.filter_key and self.filter_value: if self.filter_key in item.get('infolabels', {}): if filtered_item(item['infolabels'], self.filter_key, self.filter_value): return True elif self.filter_key in item.get('infoproperties', {}): if filtered_item(item['infoproperties'], self.filter_key, self.filter_value): return True if self.exclude_key and self.exclude_value: if self.exclude_key in item.get('infolabels', {}): if filtered_item(item['infolabels'], self.exclude_key, self.exclude_value, True): return True elif self.exclude_key in item.get('infoproperties', {}): if filtered_item(item['infoproperties'], self.exclude_key, self.exclude_value, True): return True def get_tmdb_details(self, li, cache_only=True): if not self.tmdb_api: return return self.tmdb_api.get_details( li.get_tmdb_type(), li.unique_ids.get('tvshow.tmdb') if li.infolabels.get('mediatype') in ['season', 'episode'] else li.unique_ids.get('tmdb'), li.infolabels.get('season') if li.infolabels.get('mediatype') in ['season', 'episode'] else None, li.infolabels.get('episode') if li.infolabels.get('mediatype') == 'episode' else None, cache_only=cache_only) def get_ftv_artwork(self, li): if not self.ftv_api: return artwork = self.ftv_api.get_all_artwork(li.get_ftv_id(), li.get_ftv_type()) if not artwork: return if li.infolabels.get('mediatype') in ['season', 'episode']: artwork = { u'tvshow.{}'.format(k): v for k, v in viewitems(artwork) if v } return {'art': artwork} def get_playcount_from_trakt(self, li): if not self.trakt_watchedindicators: return if li.infolabels.get('mediatype') == 'movie': return self.trakt_api.get_movie_playcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb'))) if li.infolabels.get('mediatype') == 'episode': return self.trakt_api.get_episode_playcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tvshow.tmdb')), season=li.infolabels.get('season'), episode=li.infolabels.get('episode')) if li.infolabels.get('mediatype') == 'tvshow': li.infolabels['episode'] = self.trakt_api.get_episodes_airedcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb'))) return self.trakt_api.get_episodes_watchcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb'))) if li.infolabels.get('mediatype') == 'season': li.infolabels['episode'] = self.trakt_api.get_episodes_airedcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb')), season=li.infolabels.get('season')) return self.trakt_api.get_episodes_watchcount( id_type='tmdb', unique_id=try_int(li.unique_ids.get('tmdb')), season=li.infolabels.get('season')) def get_kodi_database(self, tmdb_type): if ADDON.getSettingBool('local_db'): return get_kodi_library(tmdb_type) def get_kodi_parent_dbid(self, li): if not self.kodi_db: return if li.infolabels.get('mediatype') in ['movie', 'tvshow']: return self.kodi_db.get_info( info='dbid', imdb_id=li.unique_ids.get('imdb'), tmdb_id=li.unique_ids.get('tmdb'), tvdb_id=li.unique_ids.get('tvdb'), originaltitle=li.infolabels.get('originaltitle'), title=li.infolabels.get('title'), year=li.infolabels.get('year')) if li.infolabels.get('mediatype') in ['season', 'episode']: return self.kodi_db.get_info( info='dbid', imdb_id=li.unique_ids.get('tvshow.imdb'), tmdb_id=li.unique_ids.get('tvshow.tmdb'), tvdb_id=li.unique_ids.get('tvshow.tvdb'), title=li.infolabels.get('tvshowtitle')) def get_kodi_details(self, li): if not self.kodi_db: return dbid = self.get_kodi_parent_dbid(li) if not dbid: return if li.infolabels.get('mediatype') == 'movie': return get_movie_details(dbid) if li.infolabels.get('mediatype') == 'tvshow': return get_tvshow_details(dbid) if li.infolabels.get('mediatype') == 'season': return set_show( self.get_kodi_tvchild_details( tvshowid=dbid, season=li.infolabels.get('season'), is_season=True) or get_empty_item(), get_tvshow_details(dbid)) if li.infolabels.get('mediatype') == 'episode': return set_show( self.get_kodi_tvchild_details( tvshowid=dbid, season=li.infolabels.get('season'), episode=li.infolabels.get('episode')) or get_empty_item(), get_tvshow_details(dbid)) def get_kodi_tvchild_details(self, tvshowid, season=None, episode=None, is_season=False): if not tvshowid or not season or (not episode and not is_season): return library = 'season' if is_season else 'episode' self.kodi_db_tv[tvshowid] = self.kodi_db_tv.get( tvshowid) or get_kodi_library(library, tvshowid) if not self.kodi_db_tv[tvshowid].database: return dbid = self.kodi_db_tv[tvshowid].get_info('dbid', season=season, episode=episode) if not dbid: return details = get_season_details( dbid) if is_season else get_episode_details(dbid) details['infoproperties']['tvshow.dbid'] = tvshowid return details def get_container_content(self, tmdb_type, season=None, episode=None): if tmdb_type == 'tv' and season and episode: return convert_type('episode', 'container') elif tmdb_type == 'tv' and season: return convert_type('season', 'container') return convert_type(tmdb_type, 'container') def list_randomised_trakt(self, **kwargs): kwargs['info'] = RANDOMISED_TRAKT.get(kwargs.get('info')) kwargs['randomise'] = True self.parent_params = kwargs return self.get_items(**kwargs) def list_randomised(self, **kwargs): params = merge_two_dicts(kwargs, RANDOMISED_LISTS.get(kwargs.get('info'))) item = random_from_list(self.get_items(**params)) if not item: return self.plugin_category = item.get('label') return self.get_items(**item.get('params', {})) def get_tmdb_id(self, info, **kwargs): if info == 'collection': kwargs['tmdb_type'] = 'collection' return self.tmdb_api.get_tmdb_id(**kwargs) def get_items(self, **kwargs): info = kwargs.get('info') if info == 'pass': return if info == 'dir_search': return self.list_searchdir_router(**kwargs) if info == 'search': return self.list_search(**kwargs) if info == 'user_discover': return self.list_userdiscover(**kwargs) if info == 'dir_discover': return self.list_discoverdir_router(**kwargs) if info == 'discover': return self.list_discover(**kwargs) if info == 'all_items': return self.list_all_items(**kwargs) if info == 'trakt_userlist': return self.list_userlist(**kwargs) if info in ['trakt_becauseyouwatched', 'trakt_becausemostwatched']: return self.list_becauseyouwatched(**kwargs) if info == 'trakt_inprogress': return self.list_inprogress(**kwargs) if info == 'trakt_nextepisodes': return self.list_nextepisodes(**kwargs) if info == 'trakt_calendar': return self.list_trakt_calendar(**kwargs) if info == 'library_nextaired': return self.list_trakt_calendar(library=True, **kwargs) if info in TRAKT_LIST_OF_LISTS: return self.list_lists(**kwargs) if info in RANDOMISED_LISTS: return self.list_randomised(**kwargs) if info in RANDOMISED_TRAKT: return self.list_randomised_trakt(**kwargs) if info == 'trakt_sortby': return self.list_trakt_sortby(**kwargs) if info and not kwargs.get('tmdb_id'): kwargs['tmdb_id'] = self.get_tmdb_id(**kwargs) if info == 'details': return self.list_details(**kwargs) if info == 'seasons': return self.list_seasons(**kwargs) if info == 'flatseasons': return self.list_flatseasons(**kwargs) if info == 'episodes': return self.list_episodes(**kwargs) if info == 'episode_groups': return self.list_episode_groups(**kwargs) if info == 'episode_group_seasons': return self.list_episode_group_seasons(**kwargs) if info == 'episode_group_episodes': return self.list_episode_group_episodes(**kwargs) if info == 'cast': return self.list_cast(**kwargs) if info == 'crew': return self.list_crew(**kwargs) if info == 'trakt_upnext': return self.list_upnext(**kwargs) if info in TMDB_BASIC_LISTS: return self.list_tmdb(**kwargs) if info in TRAKT_BASIC_LISTS: return self.list_trakt(**kwargs) if info in TRAKT_SYNC_LISTS: return self.list_sync(**kwargs) return self.list_basedir(info) def get_directory(self): items = self.get_items(**self.params) if not items: return self.add_items(items, pagination=self.pagination, parent_params=self.parent_params, kodi_db=self.kodi_db, tmdb_cache_only=self.tmdb_cache_only) self.finish_container(update_listing=self.update_listing, plugin_category=self.plugin_category, container_content=self.container_content) self.set_params_to_container(**self.params) if self.container_update: xbmc.executebuiltin( try_encode(u'Container.Update({})'.format( self.container_update))) if self.container_refresh: xbmc.executebuiltin('Container.Refresh') def play_external(self, **kwargs): kodi_log(['lib.container.router - Attempting to play item\n', kwargs], 1) if not kwargs.get('tmdb_id'): kwargs['tmdb_id'] = self.tmdb_api.get_tmdb_id(**kwargs) Players(**kwargs).play( handle=self.handle if self.handle != -1 else None) def context_related(self, **kwargs): if not kwargs.get('tmdb_id'): kwargs['tmdb_id'] = self.tmdb_api.get_tmdb_id(**kwargs) kwargs['container_update'] = True related_lists(include_play=True, **kwargs) def router(self): if self.params.get('info') == 'play': return self.play_external(**self.params) if self.params.get('info') == 'related': return self.context_related(**self.params) return self.get_directory()