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