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