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)
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 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()