def Main():
args = getArguments()
if args['action'] == 'sync':
# set property for service to initiate a manual sync
utils.setProperty('traktManualSync', 'True')
elif args['action'] in ['rate', 'unrate']:
utils.Debug(str(args))
data = {}
data['action'] = args['action']
media_type = None
if 'media_type' in args and 'dbid' in args:
media_type = args['media_type']
try:
data['dbid'] = int(args['dbid'])
except ValueError:
utils.Debug("Manual %s triggered for library item, but DBID is invalid." % args['action'])
return
elif 'media_type' in args and 'remoteid' in args:
media_type = args['media_type']
data['remoteid'] = args['remoteid']
if 'season' in args:
if not 'episode' in args:
utils.Debug("Manual %s triggered for non-library episode, but missing episode number." % args['action'])
return
try:
data['season'] = int(args['season'])
data['episode'] = int(args['episode'])
except ValueError:
utilities.Debug("Error parsing season or episode for manual %s" % args['action'])
return
else:
media_type = getMediaType()
if not utils.isValidMediaType(media_type):
utils.Debug("Error, not in video library.")
return
data['dbid'] = int(xbmc.getInfoLabel('ListItem.DBID'))
if media_type is None:
utils.Debug("Manual %s triggered on an unsupported content container." % args['action'])
elif utils.isValidMediaType(media_type):
data['media_type'] = media_type
if 'dbid' in data:
utils.Debug("Manual %s of library '%s' with an ID of '%s'." % (args['action'], media_type, data['dbid']))
else:
if 'season' in data:
utils.Debug("Manual %s of non-library '%s' S%02dE%02d, with an ID of '%s'." % (args['action'], media_type, data['season'], data['episode'], data['remoteid']))
else:
utils.Debug("Manual %s of non-library '%s' with an ID of '%s'." % (args['action'], media_type, data['remoteid']))
if args['action'] == 'rate' and 'rating' in args:
if args['rating'] in ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10']:
data['rating'] = int(args['rating'])
utils.setProperty('traktManualRateData', json.dumps(data))
utils.setProperty('traktManualRate', 'True')
else:
utils.Debug("Manual %s of '%s' is unsupported." % (args['action'], media_type))
def doManualRating(self, data):
action = data['action']
media_type = data['media_type']
summaryInfo = None
if not utilities.isValidMediaType(media_type):
logger.debug("doManualRating(): Invalid media type '%s' passed for manual %s." % (media_type, action))
return
if not data['action'] in ['rate', 'unrate']:
logger.debug("doManualRating(): Unknown action passed.")
return
logger.debug("Getting data for manual %s of %s: video_id: |%s| dbid: |%s|" % (action, media_type, data.get('video_id'), data.get('dbid')))
id_type = utilities.parseIdToTraktIds(str(data['video_id']), media_type)[1]
if not id_type:
logger.debug("doManualRating(): Unrecognized id_type: |%s|-|%s|." % (media_type, data['video_id']))
return
ids = globals.traktapi.getIdLookup(data['video_id'], id_type)
if not ids:
logger.debug("doManualRating(): No Results for: |%s|-|%s|." % (media_type, data['video_id']))
return
trakt_id = dict(ids[0].keys)['trakt']
if utilities.isEpisode(media_type):
summaryInfo = globals.traktapi.getEpisodeSummary(trakt_id, data['season'], data['episode'])
userInfo = globals.traktapi.getEpisodeRatingForUser(trakt_id, data['season'], data['episode'], 'trakt')
elif utilities.isSeason(media_type):
summaryInfo = globals.traktapi.getShowSummary(trakt_id)
userInfo = globals.traktapi.getSeasonRatingForUser(trakt_id, data['season'], 'trakt')
elif utilities.isShow(media_type):
summaryInfo = globals.traktapi.getShowSummary(trakt_id)
userInfo = globals.traktapi.getShowRatingForUser(trakt_id, 'trakt')
elif utilities.isMovie(media_type):
summaryInfo = globals.traktapi.getMovieSummary(trakt_id)
userInfo = globals.traktapi.getMovieRatingForUser(trakt_id, 'trakt')
if summaryInfo is not None:
summaryInfo = summaryInfo.to_dict()
summaryInfo['user'] = {'ratings': userInfo}
if utilities.isEpisode(media_type):
summaryInfo['season'] = data['season']
summaryInfo['number'] = data['episode']
elif utilities.isSeason(media_type):
summaryInfo['season'] = data['season']
if action == 'rate':
if not 'rating' in data:
rateMedia(media_type, [summaryInfo])
else:
rateMedia(media_type, [summaryInfo], rating=data['rating'])
else:
logger.debug("doManualRating(): Summary info was empty, possible problem retrieving data from Trakt.tv")
def doManualRating(self, data):
action = data['action']
media_type = data['media_type']
summaryInfo = None
if not utilities.isValidMediaType(media_type):
logger.debug("doManualRating(): Invalid media type '%s' passed for manual %s." % (media_type, action))
return
if not data['action'] in ['rate', 'unrate']:
logger.debug("doManualRating(): Unknown action passed.")
return
logger.debug("Getting data for manual %s of %s: video_id: |%s| dbid: |%s|" % (action, media_type, data.get('video_id'), data.get('dbid')))
id_type = utilities.parseIdToTraktIds(str(data['video_id']), media_type)[1]
if not id_type:
logger.debug("doManualRating(): Unrecognized id_type: |%s|-|%s|." % (media_type, data['video_id']))
return
ids = globals.traktapi.getIdLookup(data['video_id'], id_type)
if not ids:
logger.debug("doManualRating(): No Results for: |%s|-|%s|." % (media_type, data['video_id']))
return
trakt_id = dict(ids[0].keys)['trakt']
if utilities.isEpisode(media_type):
summaryInfo = globals.traktapi.getEpisodeSummary(trakt_id, data['season'], data['episode'])
userInfo = globals.traktapi.getEpisodeRatingForUser(trakt_id, data['season'], data['episode'], 'trakt')
elif utilities.isSeason(media_type):
summaryInfo = globals.traktapi.getShowSummary(trakt_id)
userInfo = globals.traktapi.getSeasonRatingForUser(trakt_id, data['season'], 'trakt')
elif utilities.isShow(media_type):
summaryInfo = globals.traktapi.getShowSummary(trakt_id)
userInfo = globals.traktapi.getShowRatingForUser(trakt_id, 'trakt')
elif utilities.isMovie(media_type):
summaryInfo = globals.traktapi.getMovieSummary(trakt_id)
userInfo = globals.traktapi.getMovieRatingForUser(trakt_id, 'trakt')
if summaryInfo is not None:
summaryInfo = summaryInfo.to_dict()
summaryInfo['user'] = {'ratings': userInfo}
if utilities.isEpisode(media_type):
summaryInfo['season'] = data['season']
summaryInfo['number'] = data['episode']
elif utilities.isSeason(media_type):
summaryInfo['season'] = data['season']
if action == 'rate':
if not 'rating' in data:
rateMedia(media_type, [summaryInfo])
else:
rateMedia(media_type, [summaryInfo], rating=data['rating'])
else:
logger.debug("doManualRating(): Summary info was empty, possible problem retrieving data from Trakt.tv")
def doManualRating(self, data):
action = data['action']
media_type = data['media_type']
summaryInfo = None
if not utilities.isValidMediaType(media_type):
utilities.Debug(
"doManualRating(): Invalid media type '%s' passed for manual %s."
% (media_type, action))
return
if not data['action'] in ['rate', 'unrate']:
utilities.Debug("doManualRating(): Unknown action passed.")
return
if 'dbid' in data:
utilities.Debug(
"Getting data for manual %s of library '%s' with ID of '%s'" %
(action, media_type, data['dbid']))
elif 'remoteitd' in data:
if 'season' in data:
utilities.Debug(
"Getting data for manual %s of non-library '%s' S%02dE%02d, with ID of '%s'."
% (action, media_type, data['season'], data['episode'],
data['remoteid']))
else:
utilities.Debug(
"Getting data for manual %s of non-library '%s' with ID of '%s'"
% (action, media_type, data['remoteid']))
if utilities.isEpisode(media_type):
summaryInfo = globals.traktapi.getEpisodeSummary(
data['tvdb_id'], data['season'], data['episode'])
elif utilities.isShow(media_type):
summaryInfo = globals.traktapi.getShowSummary(data['imdbnumber'])
elif utilities.isMovie(media_type):
summaryInfo = globals.traktapi.getMovieSummary(data['imdbnumber'])
if not summaryInfo is None:
if utilities.isMovie(media_type) or utilities.isShow(media_type):
summaryInfo['xbmc_id'] = data['dbid']
if action == 'rate':
if not 'rating' in data:
rateMedia(media_type, summaryInfo)
else:
rateMedia(media_type, summaryInfo, rating=data['rating'])
elif action == 'unrate':
rateMedia(media_type, summaryInfo, unrate=True)
else:
utilities.Debug(
"doManualRating(): Summary info was empty, possible problem retrieving data from trakt.tv"
)
def doManualRating(self, data):
action = data['action']
media_type = data['media_type']
summaryInfo = None
if not utilities.isValidMediaType(media_type):
logger.debug("doManualRating(): Invalid media type '%s' passed for manual %s." % (media_type, action))
return
if not data['action'] in ['rate', 'unrate']:
logger.debug("doManualRating(): Unknown action passed.")
return
logger.debug("Getting data for manual %s of %s: imdb: |%s| dbid: |%s|" % (action, media_type, data.get('remoteid'), data.get('dbid')))
if utilities.isEpisode(media_type):
summaryInfo = globals.traktapi.getEpisodeSummary(data['imdbnumber'], data['season'], data['episode'])
userInfo = globals.traktapi.getEpisodeRatingForUser(data['imdbnumber'], data['season'], data['episode'], 'imdb')
elif utilities.isSeason(media_type):
summaryInfo = globals.traktapi.getShowSummary(data['imdbnumber'])
userInfo = globals.traktapi.getSeasonRatingForUser(data['imdbnumber'], data['season'], 'imdb')
elif utilities.isShow(media_type):
summaryInfo = globals.traktapi.getShowSummary(data['imdbnumber'])
userInfo = globals.traktapi.getShowRatingForUser(data['imdbnumber'], 'imdb')
elif utilities.isMovie(media_type):
summaryInfo = globals.traktapi.getMovieSummary(data['imdbnumber'])
userInfo = globals.traktapi.getMovieRatingForUser(data['imdbnumber'])
if summaryInfo is not None:
summaryInfo = summaryInfo.to_dict()
summaryInfo['user'] = {'ratings': userInfo}
if utilities.isEpisode(media_type):
summaryInfo['season'] = data['season']
summaryInfo['number'] = data['episode']
elif utilities.isSeason(media_type):
summaryInfo['season'] = data['season']
if action == 'rate':
if not 'rating' in data:
rateMedia(media_type, [summaryInfo])
else:
rateMedia(media_type, [summaryInfo], rating=data['rating'])
else:
logger.debug("doManualRating(): Summary info was empty, possible problem retrieving data from Trakt.tv")
def doManualRating(self, data):
action = data['action']
media_type = data['media_type']
summaryInfo = None
if not utilities.isValidMediaType(media_type):
utilities.Debug("doManualRating(): Invalid media type '%s' passed for manual %s." % (media_type, action))
return
if not data['action'] in ['rate', 'unrate']:
utilities.Debug("doManualRating(): Unknown action passed.")
return
if 'dbid' in data:
utilities.Debug("Getting data for manual %s of library '%s' with ID of '%s'" % (action, media_type, data['dbid']))
elif 'remoteitd' in data:
if 'season' in data:
utilities.Debug("Getting data for manual %s of non-library '%s' S%02dE%02d, with ID of '%s'." % (action, media_type, data['season'], data['episode'], data['remoteid']))
else:
utilities.Debug("Getting data for manual %s of non-library '%s' with ID of '%s'" % (action, media_type, data['remoteid']))
if utilities.isEpisode(media_type):
summaryInfo = globals.traktapi.getEpisodeSummary(data['tvdb_id'], data['season'], data['episode'])
elif utilities.isShow(media_type):
summaryInfo = globals.traktapi.getShowSummary(data['imdbnumber'])
elif utilities.isMovie(media_type):
summaryInfo = globals.traktapi.getMovieSummary(data['imdbnumber'])
if not summaryInfo is None:
if utilities.isMovie(media_type) or utilities.isShow(media_type):
summaryInfo['xbmc_id'] = data['dbid']
if action == 'rate':
if not 'rating' in data:
rateMedia(media_type, summaryInfo)
else:
rateMedia(media_type, summaryInfo, rating=data['rating'])
elif action == 'unrate':
rateMedia(media_type, summaryInfo, unrate=True)
else:
utilities.Debug("doManualRating(): Summary info was empty, possible problem retrieving data from trakt.tv")
def Main():
args = __getArguments()
data = {}
if args["action"] == "pin_info":
xbmc.executebuiltin("Dialog.Close(all, true)")
pinInfo = xbmcgui.WindowXMLDialog("PinInfoWindow.xml", __addon__.getAddonInfo("path"))
pinInfo.doModal()
del pinInfo
if args["action"] == "contextmenu":
buttons = []
media_type = __getMediaType()
if media_type in ["movie", "show", "season", "episode"]:
buttons.append("rate")
if media_type in ["movie", "show", "season", "episode"]:
buttons.append("togglewatched")
buttons.append("sync")
contextMenu = traktContextMenu(media_type=media_type, buttons=buttons)
contextMenu.doModal()
_action = contextMenu.action
del contextMenu
if _action is None:
return
logger.debug("'%s' selected from trakt.tv action menu" % _action)
args["action"] = _action
if args["action"] == "sync":
data = {"action": "manualSync", "silent": False}
if "silent" in args:
data["silent"] = args["silent"].lower() == "true"
data["library"] = "all"
if "library" in args and args["library"] in ["episodes", "movies"]:
data["library"] = args["library"]
elif args["action"] in ["rate", "unrate"]:
data = {"action": args["action"]}
media_type = None
if "media_type" in args and "dbid" in args:
media_type = args["media_type"]
try:
data["dbid"] = int(args["dbid"])
except ValueError:
logger.debug("Manual %s triggered for library item, but DBID is invalid." % args["action"])
return
elif "media_type" in args and "remoteid" in args:
media_type = args["media_type"]
data["remoteid"] = args["remoteid"]
try:
data["season"] = int(args["season"])
data["episode"] = int(args["episode"])
except ValueError:
logger.debug("Error parsing season or episode for manual %s" % args["action"])
return
except KeyError:
pass
else:
media_type = __getMediaType()
if not utils.isValidMediaType(media_type):
logger.debug("Error, not in video library.")
return
data["dbid"] = int(xbmc.getInfoLabel("ListItem.DBID"))
if media_type is None:
logger.debug("Manual %s triggered on an unsupported content container." % args["action"])
elif utils.isValidMediaType(media_type):
data["media_type"] = media_type
if "dbid" in data:
logger.debug(
"Manual %s of library '%s' with an ID of '%s'." % (args["action"], media_type, data["dbid"])
)
if utils.isMovie(media_type):
result = utils.getMovieDetailsFromKodi(data["dbid"], ["imdbnumber", "title", "year"])
if not result:
logger.debug("No data was returned from Kodi, aborting manual %s." % args["action"])
return
data["imdbnumber"] = result["imdbnumber"]
elif utils.isShow(media_type) or utils.isSeason(media_type):
result = utils.getShowDetailsFromKodi(data["dbid"], ["imdbnumber", "tag"])
if not result:
logger.debug("No data was returned from Kodi, aborting manual %s." % args["action"])
return
data["imdbnumber"] = result["imdbnumber"]
data["tag"] = result["tag"]
elif utils.isEpisode(media_type):
result = utils.getEpisodeDetailsFromKodi(
data["dbid"], ["showtitle", "season", "episode", "tvshowid", "uniqueid", "file", "playcount"]
)
if not result:
logger.debug("No data was returned from Kodi, aborting manual %s." % args["action"])
return
data["imdbnumber"] = result["episodeid"]
data["season"] = result["season"]
data["episode"] = result["episode"]
else:
data["imdbnumber"] = data["remoteid"]
if "season" in data and "episode" in data:
logger.debug(
"Manual %s of non-library '%s' S%02dE%02d, with an ID of '%s'."
% (args["action"], media_type, data["season"], data["episode"], data["remoteid"])
)
elif "season" in data:
logger.debug(
"Manual %s of non-library '%s' S%02d, with an ID of '%s'."
% (args["action"], media_type, data["season"], data["remoteid"])
)
else:
logger.debug(
"Manual %s of non-library '%s' with an ID of '%s'."
% (args["action"], media_type, data["remoteid"])
)
if args["action"] == "rate" and "rating" in args:
if args["rating"] in ["1", "2", "3", "4", "5", "6", "7", "8", "9", "10"]:
data["rating"] = int(args["rating"])
data = {"action": "manualRating", "ratingData": data}
else:
logger.debug("Manual %s of '%s' is unsupported." % (args["action"], media_type))
elif args["action"] == "togglewatched":
media_type = __getMediaType()
if media_type in ["movie", "show", "season", "episode"]:
data = {"media_type": media_type}
if utils.isMovie(media_type):
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getMovieDetailsFromKodi(dbid, ["imdbnumber", "title", "year", "playcount"])
if result:
if result["playcount"] == 0:
data["id"] = result["imdbnumber"]
else:
logger.debug("Movie alread marked as watched in Kodi.")
else:
logger.debug("Error getting movie details from Kodi.")
return
elif utils.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getEpisodeDetailsFromKodi(
dbid, ["showtitle", "season", "episode", "tvshowid", "playcount"]
)
if result:
if result["playcount"] == 0:
data["id"] = result["imdbnumber"]
data["season"] = result["season"]
data["number"] = result["episode"]
data["title"] = result["showtitle"]
else:
logger.debug("Episode already marked as watched in Kodi.")
else:
logger.debug("Error getting episode details from Kodi.")
return
elif utils.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel("ListItem.TVShowTitle")
result = utils.kodiJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShows",
"params": {"properties": ["title", "imdbnumber", "year"]},
"id": 0,
}
)
if result and "tvshows" in result:
for show in result["tvshows"]:
if show["title"] == showTitle:
showID = show["tvshowid"]
data["id"] = show["imdbnumber"]
data["title"] = show["title"]
break
else:
logger.debug("Error getting TV shows from Kodi.")
return
season = xbmc.getInfoLabel("ListItem.Season")
if season == "":
season = 0
else:
season = int(season)
result = utils.kodiJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetEpisodes",
"params": {
"tvshowid": showID,
"season": season,
"properties": ["season", "episode", "playcount"],
},
"id": 0,
}
)
if result and "episodes" in result:
episodes = []
for episode in result["episodes"]:
if episode["playcount"] == 0:
episodes.append(episode["episode"])
if len(episodes) == 0:
logger.debug("'%s - Season %d' is already marked as watched." % (showTitle, season))
return
data["season"] = season
data["episodes"] = episodes
else:
logger.debug("Error getting episodes from '%s' for Season %d" % (showTitle, season))
return
elif utils.isShow(media_type):
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getShowDetailsFromKodi(dbid, ["year", "imdbnumber"])
if not result:
logger.debug("Error getting show details from Kodi.")
return
showTitle = result["label"]
data["id"] = result["imdbnumber"]
result = utils.kodiJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetEpisodes",
"params": {"tvshowid": dbid, "properties": ["season", "episode", "playcount", "showtitle"]},
"id": 0,
}
)
if result and "episodes" in result:
i = 0
s = {}
for e in result["episodes"]:
data["title"] = e["showtitle"]
season = str(e["season"])
if not season in s:
s[season] = []
if e["playcount"] == 0:
s[season].append(e["episode"])
i += 1
if i == 0:
logger.debug("'%s' is already marked as watched." % showTitle)
return
data["seasons"] = dict((k, v) for k, v in s.iteritems() if v)
else:
logger.debug("Error getting episode details for '%s' from Kodi." % showTitle)
return
if len(data) > 1:
logger.debug(
"Marking '%s' with the following data '%s' as watched on Trakt.tv" % (media_type, str(data))
)
data["action"] = "markWatched"
# execute toggle watched action
xbmc.executebuiltin("Action(ToggleWatched)")
q = sqlitequeue.SqliteQueue()
if "action" in data:
logger.debug("Queuing for dispatch: %s" % data)
q.append(data)
def rateMedia(media_type, itemsToRate, unrate=False, rating=None):
"""Launches the rating dialog"""
for summary_info in itemsToRate:
if not utils.isValidMediaType(media_type):
logger.debug("Not a valid media type")
return
elif 'user' not in summary_info:
logger.debug("No user data")
return
s = utils.getFormattedItemName(media_type, summary_info)
logger.debug("Summary Info %s" % summary_info)
if unrate:
rating = None
if summary_info['user']['ratings']['rating'] > 0:
rating = 0
if not rating is None:
logger.debug("'%s' is being unrated." % s)
__rateOnTrakt(rating, media_type, summary_info, unrate=True)
else:
logger.debug("'%s' has not been rated, so not unrating." % s)
return
rerate = utils.getSettingAsBool('rate_rerate')
if rating is not None:
if summary_info['user']['ratings']['rating'] == 0:
logger.debug("Rating for '%s' is being set to '%d' manually." % (s, rating))
__rateOnTrakt(rating, media_type, summary_info)
else:
if rerate:
if not summary_info['user']['ratings']['rating'] == rating:
logger.debug("Rating for '%s' is being set to '%d' manually." % (s, rating))
__rateOnTrakt(rating, media_type, summary_info)
else:
utils.notification(utils.getString(32043), s)
logger.debug("'%s' already has a rating of '%d'." % (s, rating))
else:
utils.notification(utils.getString(32041), s)
logger.debug("'%s' is already rated." % s)
return
if summary_info['user']['ratings'] and summary_info['user']['ratings']['rating']:
if not rerate:
logger.debug("'%s' has already been rated." % s)
utils.notification(utils.getString(32041), s)
return
else:
logger.debug("'%s' is being re-rated." % s)
xbmc.executebuiltin('Dialog.Close(all, true)')
gui = RatingDialog(
"script-trakt-RatingDialog.xml",
__addon__.getAddonInfo('path'),
media_type=media_type,
media=summary_info,
rerate=rerate
)
gui.doModal()
if gui.rating:
rating = gui.rating
if rerate:
rating = gui.rating
if summary_info['user']['ratings'] and summary_info['user']['ratings']['rating'] > 0 and rating == summary_info['user']['ratings']['rating']:
rating = 0
if rating == 0 or rating == "unrate":
__rateOnTrakt(rating, gui.media_type, gui.media, unrate=True)
else:
__rateOnTrakt(rating, gui.media_type, gui.media)
else:
logger.debug("Rating dialog was closed with no rating.")
del gui
#Reset rating and unrate for multi part episodes
unrate=False
rating=None
def Main():
args = __getArguments()
data = {}
if args['action'] == 'auth_info':
data['action'] = 'auth_info'
if args['action'] == 'contextmenu':
buttons = []
media_type = kodiUtilities.getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
buttons.append("rate")
buttons.append("togglewatched")
buttons.append("addtowatchlist")
buttons.append("sync")
contextMenu = traktContextMenu(media_type=media_type, buttons=buttons)
contextMenu.doModal()
_action = contextMenu.action
del contextMenu
if _action is None:
return
logger.debug("'%s' selected from trakt.tv action menu" % _action)
args['action'] = _action
if args['action'] == 'sync':
data = {'action': 'manualSync', 'silent': False}
if 'silent' in args:
data['silent'] = (args['silent'].lower() == 'true')
data['library'] = "all"
if 'library' in args and args['library'] in ['episodes', 'movies']:
data['library'] = args['library']
elif args['action'] in ['rate', 'unrate']:
data = {'action': args['action']}
media_type = None
if 'media_type' in args and 'dbid' in args:
media_type = args['media_type']
try:
data['dbid'] = int(args['dbid'])
except ValueError:
logger.debug(
"Manual %s triggered for library item, but DBID is invalid."
% args['action'])
return
elif 'media_type' in args and 'remoteid' in args:
media_type = args['media_type']
data['remoteid'] = args['remoteid']
try:
data['season'] = int(args['season'])
data['episode'] = int(args['episode'])
except ValueError:
logger.debug("Error parsing season or episode for manual %s" %
args['action'])
return
except KeyError:
pass
else:
media_type = kodiUtilities.getMediaType()
if not utilities.isValidMediaType(media_type):
logger.debug("Error, not in video library.")
return
data['dbid'] = int(xbmc.getInfoLabel('ListItem.DBID'))
if media_type is None:
logger.debug(
"Manual %s triggered on an unsupported content container." %
args['action'])
elif utilities.isValidMediaType(media_type):
data['media_type'] = media_type
if 'dbid' in data:
logger.debug("Manual %s of library '%s' with an ID of '%s'." %
(args['action'], media_type, data['dbid']))
if utilities.isMovie(media_type):
result = kodiUtilities.getMovieDetailsFromKodi(
data['dbid'], ['imdbnumber', 'title', 'year'])
if not result:
logger.debug(
"No data was returned from Kodi, aborting manual %s."
% args['action'])
return
elif utilities.isShow(media_type):
tvshow_id = data['dbid']
elif utilities.isSeason(media_type):
result = kodiUtilities.getSeasonDetailsFromKodi(
data['dbid'], ['tvshowid', 'season'])
if not result:
logger.debug(
"No data was returned from Kodi, aborting manual %s."
% args['action'])
return
tvshow_id = result['tvshowid']
data['season'] = result['season']
elif utilities.isEpisode(media_type):
result = kodiUtilities.getEpisodeDetailsFromKodi(
data['dbid'], ['season', 'episode', 'tvshowid'])
if not result:
logger.debug(
"No data was returned from Kodi, aborting manual %s."
% args['action'])
return
tvshow_id = result['tvshowid']
data['season'] = result['season']
data['episode'] = result['episode']
if utilities.isShow(media_type) or utilities.isSeason(
media_type) or utilities.isEpisode(media_type):
result = kodiUtilities.getShowDetailsFromKodi(
tvshow_id, ['imdbnumber'])
if not result:
logger.debug(
"No data was returned from Kodi, aborting manual %s."
% args['action'])
return
data['video_id'] = result['imdbnumber']
else:
data['video_id'] = data['remoteid']
if 'season' in data and 'episode' in data:
logger.debug(
"Manual %s of non-library '%s' S%02dE%02d, with an ID of '%s'."
% (args['action'], media_type, data['season'],
data['episode'], data['remoteid']))
elif 'season' in data:
logger.debug(
"Manual %s of non-library '%s' S%02d, with an ID of '%s'."
% (args['action'], media_type, data['season'],
data['remoteid']))
else:
logger.debug(
"Manual %s of non-library '%s' with an ID of '%s'." %
(args['action'], media_type, data['remoteid']))
if args['action'] == 'rate' and 'rating' in args:
if args['rating'] in [
'1', '2', '3', '4', '5', '6', '7', '8', '9', '10'
]:
data['rating'] = int(args['rating'])
data = {'action': 'manualRating', 'ratingData': data}
else:
logger.debug("Manual %s of '%s' is unsupported." %
(args['action'], media_type))
elif args['action'] == 'togglewatched':
media_type = kodiUtilities.getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
data = {'media_type': media_type}
if utilities.isMovie(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getMovieDetailsFromKodi(
dbid, ['imdbnumber', 'title', 'year', 'playcount'])
if result:
if result['playcount'] == 0:
data['id'] = result['imdbnumber']
else:
logger.debug("Movie alread marked as watched in Kodi.")
else:
logger.debug("Error getting movie details from Kodi.")
return
elif utilities.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getEpisodeDetailsFromKodi(
dbid, [
'showtitle', 'season', 'episode', 'tvshowid',
'playcount'
])
if result:
if result['playcount'] == 0:
data['id'] = result['imdbnumber']
data['season'] = result['season']
data['number'] = result['episode']
data['title'] = result['showtitle']
else:
logger.debug(
"Episode already marked as watched in Kodi.")
else:
logger.debug("Error getting episode details from Kodi.")
return
elif utilities.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel('ListItem.TVShowTitle')
result = kodiUtilities.kodiJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShows',
'params': {
'properties': ['title', 'imdbnumber', 'year']
},
'id': 0
})
if result and 'tvshows' in result:
for show in result['tvshows']:
if show['title'] == showTitle:
showID = show['tvshowid']
data['id'] = show['imdbnumber']
data['title'] = show['title']
break
else:
logger.debug("Error getting TV shows from Kodi.")
return
season = xbmc.getInfoLabel('ListItem.Season')
if season == "":
season = 0
else:
season = int(season)
result = kodiUtilities.kodiJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetEpisodes',
'params': {
'tvshowid': showID,
'season': season,
'properties': ['season', 'episode', 'playcount']
},
'id': 0
})
if result and 'episodes' in result:
episodes = []
for episode in result['episodes']:
if episode['playcount'] == 0:
episodes.append(episode['episode'])
if len(episodes) == 0:
logger.debug(
"'%s - Season %d' is already marked as watched." %
(showTitle, season))
return
data['season'] = season
data['episodes'] = episodes
else:
logger.debug(
"Error getting episodes from '%s' for Season %d" %
(showTitle, season))
return
elif utilities.isShow(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getShowDetailsFromKodi(
dbid, ['year', 'imdbnumber'])
if not result:
logger.debug("Error getting show details from Kodi.")
return
showTitle = result['label']
data['id'] = result['imdbnumber']
result = kodiUtilities.kodiJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetEpisodes',
'params': {
'tvshowid':
dbid,
'properties':
['season', 'episode', 'playcount', 'showtitle']
},
'id': 0
})
if result and 'episodes' in result:
i = 0
s = {}
for e in result['episodes']:
data['title'] = e['showtitle']
season = str(e['season'])
if not season in s:
s[season] = []
if e['playcount'] == 0:
s[season].append(e['episode'])
i += 1
if i == 0:
logger.debug("'%s' is already marked as watched." %
showTitle)
return
data['seasons'] = dict(
(k, v) for k, v in s.iteritems() if v)
else:
logger.debug(
"Error getting episode details for '%s' from Kodi." %
showTitle)
return
if len(data) > 1:
logger.debug(
"Marking '%s' with the following data '%s' as watched on Trakt.tv"
% (media_type, str(data)))
data['action'] = 'markWatched'
# execute toggle watched action
xbmc.executebuiltin("Action(ToggleWatched)")
elif args['action'] == 'addtowatchlist':
media_type = kodiUtilities.getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
data = {'media_type': media_type}
if utilities.isMovie(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getMovieDetailsFromKodi(
dbid, ['imdbnumber', 'title', 'year', 'playcount'])
if result:
data['id'] = result['imdbnumber']
else:
logger.debug("Error getting movie details from Kodi.")
return
elif utilities.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getEpisodeDetailsFromKodi(
dbid, [
'showtitle', 'season', 'episode', 'tvshowid',
'playcount'
])
if result:
data['id'] = result['imdbnumber']
data['season'] = result['season']
data['number'] = result['episode']
data['title'] = result['showtitle']
else:
logger.debug("Error getting episode details from Kodi.")
return
elif utilities.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel('ListItem.TVShowTitle')
result = kodiUtilities.kodiJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShows',
'params': {
'properties': ['title', 'imdbnumber', 'year']
},
'id': 0
})
if result and 'tvshows' in result:
for show in result['tvshows']:
if show['title'] == showTitle:
showID = show['tvshowid']
data['id'] = show['imdbnumber']
data['title'] = show['title']
break
else:
logger.debug("Error getting TV shows from Kodi.")
return
season = xbmc.getInfoLabel('ListItem.Season')
if season == "":
season = 0
else:
season = int(season)
result = kodiUtilities.kodiJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetEpisodes',
'params': {
'tvshowid': showID,
'season': season,
'properties': ['season', 'episode', 'playcount']
},
'id': 0
})
if result and 'episodes' in result:
episodes = []
for episode in result['episodes']:
if episode['playcount'] == 0:
episodes.append(episode['episode'])
data['season'] = season
data['episodes'] = episodes
else:
logger.debug(
"Error getting episodes from '%s' for Season %d" %
(showTitle, season))
return
elif utilities.isShow(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getShowDetailsFromKodi(
dbid, ['year', 'imdbnumber'])
if not result:
logger.debug("Error getting show details from Kodi.")
return
showTitle = result['label']
data['id'] = result['imdbnumber']
result = kodiUtilities.kodiJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetEpisodes',
'params': {
'tvshowid':
dbid,
'properties':
['season', 'episode', 'playcount', 'showtitle']
},
'id': 0
})
if result and 'episodes' in result:
s = {}
for e in result['episodes']:
data['title'] = e['showtitle']
season = str(e['season'])
if not season in s:
s[season] = []
if e['playcount'] == 0:
s[season].append(e['episode'])
data['seasons'] = dict(
(k, v) for k, v in s.iteritems() if v)
else:
logger.debug(
"Error getting episode details for '%s' from Kodi." %
showTitle)
return
if len(data) > 1:
logger.debug(
"Adding '%s' with the following data '%s' to users watchlist on Trakt.tv"
% (media_type, str(data)))
data['action'] = 'addtowatchlist'
q = sqlitequeue.SqliteQueue()
if 'action' in data:
logger.debug("Queuing for dispatch: %s" % data)
q.append(data)
def Main():
args = __getArguments()
data = {}
if args['action'] == 'auth_info':
data['action'] = 'auth_info'
if args['action'] == 'contextmenu':
buttons = []
media_type = kodiUtilities.getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
buttons.append("rate")
buttons.append("togglewatched")
buttons.append("addtowatchlist")
buttons.append("sync")
contextMenu = traktContextMenu(media_type=media_type, buttons=buttons)
contextMenu.doModal()
_action = contextMenu.action
del contextMenu
if _action is None:
return
logger.debug("'%s' selected from trakt.tv action menu" % _action)
args['action'] = _action
if args['action'] == 'sync':
data = {'action': 'manualSync', 'silent': False}
if 'silent' in args:
data['silent'] = (args['silent'].lower() == 'true')
data['library'] = "all"
if 'library' in args and args['library'] in ['episodes', 'movies']:
data['library'] = args['library']
elif args['action'] in ['rate', 'unrate']:
data = {'action': args['action']}
media_type = None
if 'media_type' in args and 'dbid' in args:
media_type = args['media_type']
try:
data['dbid'] = int(args['dbid'])
except ValueError:
logger.debug("Manual %s triggered for library item, but DBID is invalid." % args['action'])
return
elif 'media_type' in args and 'remoteid' in args:
media_type = args['media_type']
data['remoteid'] = args['remoteid']
try:
data['season'] = int(args['season'])
data['episode'] = int(args['episode'])
except ValueError:
logger.debug("Error parsing season or episode for manual %s" % args['action'])
return
except KeyError:
pass
else:
media_type = kodiUtilities.getMediaType()
if not utilities.isValidMediaType(media_type):
logger.debug("Error, not in video library.")
return
data['dbid'] = int(xbmc.getInfoLabel('ListItem.DBID'))
if media_type is None:
logger.debug("Manual %s triggered on an unsupported content container." % args['action'])
elif utilities.isValidMediaType(media_type):
data['media_type'] = media_type
if 'dbid' in data:
logger.debug("Manual %s of library '%s' with an ID of '%s'." % (args['action'], media_type, data['dbid']))
if utilities.isMovie(media_type):
result = kodiUtilities.getMovieDetailsFromKodi(data['dbid'], ['imdbnumber', 'title', 'year'])
if not result:
logger.debug("No data was returned from Kodi, aborting manual %s." % args['action'])
return
elif utilities.isShow(media_type):
tvshow_id = data['dbid']
elif utilities.isSeason(media_type):
result = kodiUtilities.getSeasonDetailsFromKodi(data['dbid'], ['tvshowid', 'season'])
if not result:
logger.debug("No data was returned from Kodi, aborting manual %s." % args['action'])
return
tvshow_id = result['tvshowid']
data['season'] = result['season']
elif utilities.isEpisode(media_type):
result = kodiUtilities.getEpisodeDetailsFromKodi(data['dbid'], ['season', 'episode', 'tvshowid'])
if not result:
logger.debug("No data was returned from Kodi, aborting manual %s." % args['action'])
return
tvshow_id = result['tvshowid']
data['season'] = result['season']
data['episode'] = result['episode']
if utilities.isShow(media_type) or utilities.isSeason(media_type) or utilities.isEpisode(media_type):
result = kodiUtilities.getShowDetailsFromKodi(tvshow_id, ['imdbnumber'])
if not result:
logger.debug("No data was returned from Kodi, aborting manual %s." % args['action'])
return
data['video_id'] = result['imdbnumber']
else:
data['video_id'] = data['remoteid']
if 'season' in data and 'episode' in data:
logger.debug("Manual %s of non-library '%s' S%02dE%02d, with an ID of '%s'." % (args['action'], media_type, data['season'], data['episode'], data['remoteid']))
elif 'season' in data:
logger.debug("Manual %s of non-library '%s' S%02d, with an ID of '%s'." % (args['action'], media_type, data['season'], data['remoteid']))
else:
logger.debug("Manual %s of non-library '%s' with an ID of '%s'." % (args['action'], media_type, data['remoteid']))
if args['action'] == 'rate' and 'rating' in args:
if args['rating'] in ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10']:
data['rating'] = int(args['rating'])
data = {'action': 'manualRating', 'ratingData': data}
else:
logger.debug("Manual %s of '%s' is unsupported." % (args['action'], media_type))
elif args['action'] == 'togglewatched':
media_type = kodiUtilities.getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
data = {'media_type': media_type}
if utilities.isMovie(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getMovieDetailsFromKodi(dbid, ['imdbnumber', 'title', 'year', 'playcount'])
if result:
if result['playcount'] == 0:
data['id'] = result['imdbnumber']
else:
logger.debug("Movie alread marked as watched in Kodi.")
else:
logger.debug("Error getting movie details from Kodi.")
return
elif utilities.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getEpisodeDetailsFromKodi(dbid, ['showtitle', 'season', 'episode', 'tvshowid', 'playcount'])
if result:
if result['playcount'] == 0:
data['id'] = result['imdbnumber']
data['season'] = result['season']
data['number'] = result['episode']
data['title'] = result['showtitle']
else:
logger.debug("Episode already marked as watched in Kodi.")
else:
logger.debug("Error getting episode details from Kodi.")
return
elif utilities.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel('ListItem.TVShowTitle')
result = kodiUtilities.kodiJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShows', 'params': {'properties': ['title', 'imdbnumber', 'year']}, 'id': 0})
if result and 'tvshows' in result:
for show in result['tvshows']:
if show['title'] == showTitle:
showID = show['tvshowid']
data['id'] = show['imdbnumber']
data['title'] = show['title']
break
else:
logger.debug("Error getting TV shows from Kodi.")
return
season = xbmc.getInfoLabel('ListItem.Season')
if season == "":
season = 0
else:
season = int(season)
result = kodiUtilities.kodiJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetEpisodes', 'params': {'tvshowid': showID, 'season': season, 'properties': ['season', 'episode', 'playcount']}, 'id': 0})
if result and 'episodes' in result:
episodes = []
for episode in result['episodes']:
if episode['playcount'] == 0:
episodes.append(episode['episode'])
if len(episodes) == 0:
logger.debug("'%s - Season %d' is already marked as watched." % (showTitle, season))
return
data['season'] = season
data['episodes'] = episodes
else:
logger.debug("Error getting episodes from '%s' for Season %d" % (showTitle, season))
return
elif utilities.isShow(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getShowDetailsFromKodi(dbid, ['year', 'imdbnumber'])
if not result:
logger.debug("Error getting show details from Kodi.")
return
showTitle = result['label']
data['id'] = result['imdbnumber']
result = kodiUtilities.kodiJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetEpisodes', 'params': {'tvshowid': dbid, 'properties': ['season', 'episode', 'playcount', 'showtitle']}, 'id': 0})
if result and 'episodes' in result:
i = 0
s = {}
for e in result['episodes']:
data['title'] = e['showtitle']
season = str(e['season'])
if not season in s:
s[season] = []
if e['playcount'] == 0:
s[season].append(e['episode'])
i += 1
if i == 0:
logger.debug("'%s' is already marked as watched." % showTitle)
return
data['seasons'] = dict((k, v) for k, v in s.iteritems() if v)
else:
logger.debug("Error getting episode details for '%s' from Kodi." % showTitle)
return
if len(data) > 1:
logger.debug("Marking '%s' with the following data '%s' as watched on Trakt.tv" % (media_type, str(data)))
data['action'] = 'markWatched'
# execute toggle watched action
xbmc.executebuiltin("Action(ToggleWatched)")
elif args['action'] == 'addtowatchlist':
media_type = kodiUtilities.getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
data = {'media_type': media_type}
if utilities.isMovie(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getMovieDetailsFromKodi(dbid, ['imdbnumber', 'title', 'year', 'playcount'])
if result:
data['id'] = result['imdbnumber']
else:
logger.debug("Error getting movie details from Kodi.")
return
elif utilities.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getEpisodeDetailsFromKodi(dbid, ['showtitle', 'season', 'episode', 'tvshowid', 'playcount'])
if result:
data['id'] = result['imdbnumber']
data['season'] = result['season']
data['number'] = result['episode']
data['title'] = result['showtitle']
else:
logger.debug("Error getting episode details from Kodi.")
return
elif utilities.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel('ListItem.TVShowTitle')
result = kodiUtilities.kodiJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShows', 'params': {'properties': ['title', 'imdbnumber', 'year']}, 'id': 0})
if result and 'tvshows' in result:
for show in result['tvshows']:
if show['title'] == showTitle:
showID = show['tvshowid']
data['id'] = show['imdbnumber']
data['title'] = show['title']
break
else:
logger.debug("Error getting TV shows from Kodi.")
return
season = xbmc.getInfoLabel('ListItem.Season')
if season == "":
season = 0
else:
season = int(season)
result = kodiUtilities.kodiJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetEpisodes',
'params': {'tvshowid': showID, 'season': season,
'properties': ['season', 'episode', 'playcount']},
'id': 0})
if result and 'episodes' in result:
episodes = []
for episode in result['episodes']:
if episode['playcount'] == 0:
episodes.append(episode['episode'])
data['season'] = season
data['episodes'] = episodes
else:
logger.debug("Error getting episodes from '%s' for Season %d" % (showTitle, season))
return
elif utilities.isShow(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = kodiUtilities.getShowDetailsFromKodi(dbid, ['year', 'imdbnumber'])
if not result:
logger.debug("Error getting show details from Kodi.")
return
showTitle = result['label']
data['id'] = result['imdbnumber']
result = kodiUtilities.kodiJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetEpisodes',
'params': {'tvshowid': dbid, 'properties':
['season', 'episode', 'playcount', 'showtitle']}, 'id': 0})
if result and 'episodes' in result:
s = {}
for e in result['episodes']:
data['title'] = e['showtitle']
season = str(e['season'])
if not season in s:
s[season] = []
if e['playcount'] == 0:
s[season].append(e['episode'])
data['seasons'] = dict((k, v) for k, v in s.iteritems() if v)
else:
logger.debug("Error getting episode details for '%s' from Kodi." % showTitle)
return
if len(data) > 1:
logger.debug("Adding '%s' with the following data '%s' to users watchlist on Trakt.tv"
% (media_type, str(data)))
data['action'] = 'addtowatchlist'
q = sqlitequeue.SqliteQueue()
if 'action' in data:
logger.debug("Queuing for dispatch: %s" % data)
q.append(data)
def doManualRating(self):
data = json.loads(utilities.getProperty('traktManualRateData'))
action = data['action']
media_type = data['media_type']
summaryInfo = None
if not utilities.isValidMediaType(media_type):
utilities.Debug("doManualRating(): Invalid media type '%s' passed for manual %s." % (media_type, action))
return
if not data['action'] in ['rate', 'unrate']:
utilities.Debug("doManualRating(): Unknown action passed.")
return
if 'dbid' in data:
utilities.Debug("Getting data for manual %s of library '%s' with ID of '%s'" % (action, media_type, data['dbid']))
elif 'remoteitd' in data:
if 'season' in data:
utilities.Debug("Getting data for manual %s of non-library '%s' S%02dE%02d, with ID of '%s'." % (action, media_type, data['season'], data['episode'], data['remoteid']))
else:
utilities.Debug("Getting data for manual %s of non-library '%s' with ID of '%s'" % (action, media_type, data['remoteid']))
if utilities.isEpisode(media_type):
result = {}
if 'dbid' in data:
result = utilities.getEpisodeDetailsFromXbmc(data['dbid'], ['showtitle', 'season', 'episode', 'tvshowid'])
if not result:
utilities.Debug("doManualRating(): No data was returned from XBMC, aborting manual %s." % action)
return
else:
result['tvdb_id'] = data['remoteid']
result['season'] = data['season']
result['episode'] = data['episode']
summaryInfo = globals.traktapi.getEpisodeSummary(result['tvdb_id'], result['season'], result['episode'])
elif utilities.isShow(media_type):
result = {}
if 'dbid' in data:
result = utilities.getShowDetailsFromXBMC(data['dbid'], ['imdbnumber'])
if not result:
utilities.Debug("doManualRating(): No data was returned from XBMC, aborting manual %s." % action)
return
else:
result['imdbnumber'] = data['remoteid']
summaryInfo = globals.traktapi.getShowSummary(result['imdbnumber'])
elif utilities.isMovie(media_type):
result = {}
if 'dbid' in data:
result = utilities.getMovieDetailsFromXbmc(data['dbid'], ['imdbnumber', 'title', 'year'])
if not result:
utilities.Debug("doManualRating(): No data was returned from XBMC, aborting manual %s." % action)
return
else:
result['imdbnumber'] = data['remoteid']
summaryInfo = globals.traktapi.getMovieSummary(result['imdbnumber'])
if not summaryInfo is None:
if action == 'rate':
if not 'rating' in data:
rateMedia(media_type, summaryInfo)
else:
rateMedia(media_type, summaryInfo, rating=data['rating'])
elif action == 'unrate':
rateMedia(media_type, summaryInfo, unrate=True)
else:
utilities.Debug("doManualRating(): Summary info was empty, possible problem retrieving data from trakt.tv")
def test_isValidMediaType_Show():
assert utilities.isValidMediaType('show')
def test_isValidMediaType_Season():
assert utilities.isValidMediaType('season')
def test_isValidMediaType_Episode():
assert utilities.isValidMediaType('episode')
def test_isValidMediaType_Show():
assert utilities.isValidMediaType('show')
def test_isValidMediaType_Movie():
assert utilities.isValidMediaType('movie')
def test_isValidMediaType_Movie():
assert utilities.isValidMediaType('movie')
def test_isValidMediaType_Season():
assert utilities.isValidMediaType('season')
def rateMedia(media_type, summary_info, unrate=False, rating=None):
"""Launches the rating dialog"""
if not utils.isValidMediaType(media_type):
return
if utils.isEpisode(media_type):
if 'rating' in summary_info['episode']:
summary_info['rating'] = summary_info['episode']['rating']
if 'rating_advanced' in summary_info['episode']:
summary_info['rating_advanced'] = summary_info['episode']['rating_advanced']
s = utils.getFormattedItemName(media_type, summary_info)
if not globals.traktapi.settings:
globals.traktapi.getAccountSettings()
rating_type = globals.traktapi.settings['viewing']['ratings']['mode']
if unrate:
rating = None
if rating_type == "simple":
if not summary_info['rating'] == "false":
rating = "unrate"
else:
if summary_info['rating_advanced'] > 0:
rating = 0
if not rating is None:
utils.Debug("[Rating] '%s' is being unrated." % s)
rateOnTrakt(rating, media_type, summary_info, unrate=True)
else:
utils.Debug("[Rating] '%s' has not been rated, so not unrating." % s)
return
rerate = utils.getSettingAsBool('rate_rerate')
if not rating is None:
if summary_info['rating_advanced'] == 0:
utils.Debug("[Rating] Rating for '%s' is being set to '%d' manually." % (s, rating))
rateOnTrakt(rating, media_type, summary_info)
else:
if rerate:
if not summary_info['rating_advanced'] == rating:
utils.Debug("[Rating] Rating for '%s' is being set to '%d' manually." % (s, rating))
rateOnTrakt(rating, media_type, summary_info)
else:
utils.notification(utils.getString(1353), s)
utils.Debug("[Rating] '%s' already has a rating of '%d'." % (s, rating))
else:
utils.notification(utils.getString(1351), s)
utils.Debug("[Rating] '%s' is already rated." % s)
return
if summary_info['rating'] or summary_info['rating_advanced']:
if not rerate:
utils.Debug("[Rating] '%s' has already been rated." % s)
utils.notification(utils.getString(1351), s)
return
else:
utils.Debug("[Rating] '%s' is being re-rated." % s)
xbmc.executebuiltin('Dialog.Close(all, true)')
gui = RatingDialog(
"RatingDialog.xml",
__addon__.getAddonInfo('path'),
media_type=media_type,
media=summary_info,
rating_type=rating_type,
rerate=rerate
)
gui.doModal()
if gui.rating:
rating = gui.rating
if rerate:
rating = gui.rating
if rating_type == "simple":
if not summary_info['rating'] == "false" and rating == summary_info['rating']:
rating = "unrate"
else:
if summary_info['rating_advanced'] > 0 and rating == summary_info['rating_advanced']:
rating = 0
if rating == 0 or rating == "unrate":
rateOnTrakt(rating, gui.media_type, gui.media, unrate=True)
else:
rateOnTrakt(rating, gui.media_type, gui.media)
else:
utils.Debug("[Rating] Rating dialog was closed with no rating.")
del gui
def Main():
args = getArguments()
data = {}
if args['action'] == 'sync':
data = {'action': 'manualSync'}
elif args['action'] in ['rate', 'unrate']:
data = {}
data['action'] = args['action']
media_type = None
if 'media_type' in args and 'dbid' in args:
media_type = args['media_type']
try:
data['dbid'] = int(args['dbid'])
except ValueError:
utils.Debug(
"Manual %s triggered for library item, but DBID is invalid."
% args['action'])
return
elif 'media_type' in args and 'remoteid' in args:
media_type = args['media_type']
data['remoteid'] = args['remoteid']
if 'season' in args:
if not 'episode' in args:
utils.Debug(
"Manual %s triggered for non-library episode, but missing episode number."
% args['action'])
return
try:
data['season'] = int(args['season'])
data['episode'] = int(args['episode'])
except ValueError:
utilities.Debug(
"Error parsing season or episode for manual %s" %
args['action'])
return
else:
media_type = getMediaType()
if not utils.isValidMediaType(media_type):
utils.Debug("Error, not in video library.")
return
data['dbid'] = int(xbmc.getInfoLabel('ListItem.DBID'))
if media_type is None:
utils.Debug(
"Manual %s triggered on an unsupported content container." %
args['action'])
elif utils.isValidMediaType(media_type):
data['media_type'] = media_type
if 'dbid' in data:
utils.Debug("Manual %s of library '%s' with an ID of '%s'." %
(args['action'], media_type, data['dbid']))
if utils.isMovie(media_type):
result = utils.getMovieDetailsFromXbmc(
data['dbid'], ['imdbnumber', 'title', 'year'])
if not result:
utils.Debug(
"No data was returned from XBMC, aborting manual %s."
% args['action'])
return
data['imdbnumber'] = result['imdbnumber']
elif utils.isShow(media_type):
result = utils.getShowDetailsFromXBMC(
data['dbid'], ['imdbnumber', 'tag'])
if not result:
utils.Debug(
"No data was returned from XBMC, aborting manual %s."
% args['action'])
return
data['imdbnumber'] = result['imdbnumber']
data['tag'] = result['tag']
elif utils.isEpisode(media_type):
result = utils.getEpisodeDetailsFromXbmc(
data['dbid'],
['showtitle', 'season', 'episode', 'tvshowid'])
if not result:
utils.Debug(
"No data was returned from XBMC, aborting manual %s."
% args['action'])
return
data['tvdb_id'] = result['tvdb_id']
data['season'] = result['season']
data['episode'] = result['episode']
else:
if 'season' in data:
utils.Debug(
"Manual %s of non-library '%s' S%02dE%02d, with an ID of '%s'."
% (args['action'], media_type, data['season'],
data['episode'], data['remoteid']))
data['tvdb_id'] = data['remoteid']
else:
utils.Debug(
"Manual %s of non-library '%s' with an ID of '%s'." %
(args['action'], media_type, data['remoteid']))
data['imdbnumber'] = data['remoteid']
if args['action'] == 'rate' and 'rating' in args:
if args['rating'] in [
'1', '2', '3', '4', '5', '6', '7', '8', '9', '10'
]:
data['rating'] = int(args['rating'])
data = {'action': 'manualRating', 'ratingData': data}
else:
utils.Debug("Manual %s of '%s' is unsupported." %
(args['action'], media_type))
elif args['action'] == 'togglewatched':
media_type = getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
data = {}
data['media_type'] = media_type
if utils.isMovie(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = utils.getMovieDetailsFromXbmc(
dbid, ['imdbnumber', 'title', 'year', 'playcount'])
if result:
if result['playcount'] == 0:
data['id'] = result['imdbnumber']
else:
utils.Debug("Movie alread marked as watched in XBMC.")
else:
utils.Debug("Error getting movie details from XBMC.")
return
elif utils.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = utils.getEpisodeDetailsFromXbmc(
dbid, [
'showtitle', 'season', 'episode', 'tvshowid',
'playcount'
])
if result:
if result['playcount'] == 0:
data['id'] = result['tvdb_id']
data['season'] = result['season']
data['episode'] = result['episode']
else:
utils.Debug(
"Episode already marked as watched in XBMC.")
else:
utils.Debug("Error getting episode details from XBMC.")
return
elif utils.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel('ListItem.TVShowTitle')
result = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShows',
'params': {
'properties': ['title', 'imdbnumber', 'year']
},
'id': 0
})
if result and 'tvshows' in result:
for show in result['tvshows']:
if show['title'] == showTitle:
showID = show['tvshowid']
data['id'] = show['imdbnumber']
break
else:
utils.Debug("Error getting TV shows from XBMC.")
return
season = xbmc.getInfoLabel('ListItem.Season')
if season == "":
season = 0
else:
season = int(season)
result = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetEpisodes',
'params': {
'tvshowid': showID,
'season': season,
'properties': ['season', 'episode', 'playcount']
},
'id': 0
})
if result and 'episodes' in result:
episodes = []
for episode in result['episodes']:
if episode['playcount'] == 0:
episodes.append(episode['episode'])
if len(episodes) == 0:
utils.Debug(
"'%s - Season %d' is already marked as watched." %
(showTitle, season))
return
data['season'] = season
data['episodes'] = episodes
else:
utils.Debug(
"Error getting episodes from '%s' for Season %d" %
(showTitle, season))
return
elif utils.isShow(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = utils.getShowDetailsFromXBMC(dbid,
['year', 'imdbnumber'])
if not result:
utils.Debug("Error getting show details from XBMC.")
return
showTitle = result['label']
data['id'] = result['imdbnumber']
result = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetEpisodes',
'params': {
'tvshowid': dbid,
'properties': ['season', 'episode', 'playcount']
},
'id': 0
})
if result and 'episodes' in result:
i = 0
s = {}
for e in result['episodes']:
season = str(e['season'])
if not season in s:
s[season] = []
if e['playcount'] == 0:
s[season].append(e['episode'])
i = i + 1
if i == 0:
utils.Debug("'%s' is already marked as watched." %
showTitle)
return
data['seasons'] = dict(
(k, v) for k, v in s.iteritems() if v)
else:
utils.Debug(
"Error getting episode details for '%s' from XBMC." %
showTitle)
return
if len(data) > 1:
utils.Debug(
"Marking '%s' with the following data '%s' as watched on trakt.tv"
% (media_type, str(data)))
data['action'] = 'markWatched'
# execute toggle watched action
xbmc.executebuiltin("Action(ToggleWatched)")
elif args['action'] == 'updatetags':
data = {'action': 'updatetags'}
elif args['action'] == 'managelists':
data = {'action': 'managelists'}
elif args['action'] == 'timertest':
utils.Debug("Timing JSON requests.")
import time
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShowDetails',
'params': {
'tvshowid': 254,
'properties': ['tag']
},
'id': 1
})
#data = utils.getShowDetailsFromXBMC(254, ['tag', 'imdbnumber'])
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShowDetails with tags: %0.3f seconds." %
e)
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetMovieDetails',
'params': {
'movieid': 634,
'properties': ['tag']
},
'id': 1
})
#data = utils.getMovieDetailsFromXbmc(634, ['tag', 'imdbnumber', 'title', 'year'])
e = time.time() - t
utils.Debug("VideoLibrary.GetMovieDetails with tags: %0.3f seconds." %
e)
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShows',
'params': {
'properties': ['tag', 'title', 'imdbnumber', 'year']
},
'id': 0
})
e = time.time() - t
utils.Debug(
"VideoLibrary.GetTVShows with tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data['tvshows']), e / len(data['tvshows'])))
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'id': 0,
'method': 'VideoLibrary.GetMovies',
'params': {
'properties': ['tag', 'title', 'imdbnumber', 'year']
}
})
e = time.time() - t
utils.Debug(
"VideoLibrary.GetMovies with tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data['movies']), e / len(data['movies'])))
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShows',
'params': {
'properties': ['title', 'imdbnumber', 'year']
},
'id': 0
})
e = time.time() - t
utils.Debug(
"VideoLibrary.GetTVShows without tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data['tvshows']), e / len(data['tvshows'])))
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'id': 0,
'method': 'VideoLibrary.GetMovies',
'params': {
'properties': ['title', 'imdbnumber', 'year']
}
})
e = time.time() - t
utils.Debug(
"VideoLibrary.GetMovies without tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data['movies']), e / len(data['movies'])))
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShowDetails',
'params': {
'tvshowid': 254,
'properties': ['imdbnumber']
},
'id': 1
})
#data = utils.getShowDetailsFromXBMC(254, ['imdbnumber'])
e = time.time() - t
utils.Debug(
"VideoLibrary.GetTVShowDetails without tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetMovieDetails',
'params': {
'movieid': 634,
'properties': ['imdbnumber', 'title', 'year']
},
'id': 1
})
#data = utils.getMovieDetailsFromXbmc(634, ['imdbnumber', 'title', 'year'])
e = time.time() - t
utils.Debug(
"VideoLibrary.GetMovieDetails without tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShows',
'params': {
'properties': ['title', 'imdbnumber', 'year']
},
'id': 0
})
data = data['tvshows']
for item in data:
item_data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetTVShowDetails',
'params': {
'tvshowid': item['tvshowid'],
'properties': ['tag']
},
'id': 1
})
item['tag'] = item_data['tvshowdetails']['tag']
e = time.time() - t
utils.Debug(
"VideoLibrary.GetTVShows with tags from loop: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data), e / len(data)))
t = time.time()
data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetMovies',
'params': {
'properties': ['title', 'imdbnumber', 'year']
},
'id': 0
})
data = data['movies']
for item in data:
item_data = utils.xbmcJsonRequest({
'jsonrpc': '2.0',
'method': 'VideoLibrary.GetMovieDetails',
'params': {
'movieid': item['movieid'],
'properties': ['tag']
},
'id': 1
})
item['tag'] = item_data['moviedetails']['tag']
e = time.time() - t
utils.Debug(
"VideoLibrary.GetMovies with tags from: %0.3f seconds, %d items at %0.5f seconds per item."
% (e, len(data), e / len(data)))
elif args['action'] in ['itemlists', 'addtolist', 'removefromlist']:
data = {}
data['action'] = args['action']
media_type = None
dbid = None
if 'media_type' in args and 'dbid' in args:
media_type = args['media_type']
try:
dbid = int(args['dbid'])
except ValueError:
utils.Debug(
"'%s' triggered for library item, but DBID is invalid." %
args['action'])
return
else:
media_type = getMediaType()
if not media_type in ['movie', 'show']:
utils.Debug("Error, not in video library.")
return
try:
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
except ValueError:
utils.Debug(
"'%s' triggered for library item, but there is a problem with ListItem.DBID."
% args['action'])
return
if not media_type in ['movie', 'show']:
utils.Debug("'%s' is not a valid media type for '%s'." %
(media_type, args['action']))
return
if args['action'] in ['addtolist', 'removefromlist']:
if 'list' in args:
data['list'] = args['list']
else:
utils.Debug("'%s' requires a list parameter." % data['action'])
data['type'] = media_type
if utils.isMovie(media_type):
result = utils.getMovieDetailsFromXbmc(
dbid, ['imdbnumber', 'title', 'year', 'tag'])
if not result:
utils.Debug("Error getting movie details from XBMC.")
return
data['tag'] = result['tag']
data['movieid'] = result['movieid']
data['title'] = result['title']
data['year'] = result['year']
if result['imdbnumber'].startswith("tt"):
data['imdb_id'] = result['imdbnumber']
elif result['imdbnumber'].isdigit():
data['tmdb_id'] = result['imdbnumber']
elif utils.isShow(media_type):
result = utils.getShowDetailsFromXBMC(
dbid, ['imdbnumber', 'title', 'tag'])
if not result:
utils.Debug("Error getting show details from XBMC.")
return
data['tag'] = result['tag']
data['tvshowid'] = result['tvshowid']
data['title'] = result['title']
if result['imdbnumber'].startswith("tt"):
data['imdb_id'] = result['imdbnumber']
elif result['imdbnumber'].isdigit():
data['tvdb_id'] = result['imdbnumber']
q = queue.SqliteQueue()
if 'action' in data:
utils.Debug("Queuing for dispatch: %s" % data)
q.append(data)
def rateMedia(media_type, itemsToRate, unrate=False, rating=None):
"""Launches the rating dialog"""
for summary_info in itemsToRate:
if not utils.isValidMediaType(media_type):
logger.debug("Not a valid media type")
return
elif 'user' not in summary_info:
logger.debug("No user data")
return
s = utils.getFormattedItemName(media_type, summary_info)
logger.debug("Summary Info %s" % summary_info)
if unrate:
rating = None
if summary_info['user']['ratings']['rating'] > 0:
rating = 0
if not rating is None:
logger.debug("'%s' is being unrated." % s)
__rateOnTrakt(rating, media_type, summary_info, unrate=True)
else:
logger.debug("'%s' has not been rated, so not unrating." % s)
return
rerate = utils.getSettingAsBool('rate_rerate')
if rating is not None:
if summary_info['user']['ratings']['rating'] == 0:
logger.debug("Rating for '%s' is being set to '%d' manually." %
(s, rating))
__rateOnTrakt(rating, media_type, summary_info)
else:
if rerate:
if not summary_info['user']['ratings']['rating'] == rating:
logger.debug(
"Rating for '%s' is being set to '%d' manually." %
(s, rating))
__rateOnTrakt(rating, media_type, summary_info)
else:
utils.notification(utils.getString(32043), s)
logger.debug("'%s' already has a rating of '%d'." %
(s, rating))
else:
utils.notification(utils.getString(32041), s)
logger.debug("'%s' is already rated." % s)
return
if summary_info['user']['ratings'] and summary_info['user']['ratings'][
'rating']:
if not rerate:
logger.debug("'%s' has already been rated." % s)
utils.notification(utils.getString(32041), s)
return
else:
logger.debug("'%s' is being re-rated." % s)
xbmc.executebuiltin('Dialog.Close(all, true)')
gui = RatingDialog("script-trakt-RatingDialog.xml",
__addon__.getAddonInfo('path'),
media_type=media_type,
media=summary_info,
rerate=rerate)
gui.doModal()
if gui.rating:
rating = gui.rating
if rerate:
rating = gui.rating
if summary_info['user']['ratings'] and summary_info['user'][
'ratings']['rating'] > 0 and rating == summary_info[
'user']['ratings']['rating']:
rating = 0
if rating == 0 or rating == "unrate":
__rateOnTrakt(rating, gui.media_type, gui.media, unrate=True)
else:
__rateOnTrakt(rating, gui.media_type, gui.media)
else:
logger.debug("Rating dialog was closed with no rating.")
del gui
#Reset rating and unrate for multi part episodes
unrate = False
rating = None
def Main():
args = getArguments()
data = {}
if args["action"] == "contextmenu":
buttons = []
media_type = getMediaType()
if utils.getSettingAsBool("tagging_enable"):
if utils.isMovie(media_type):
buttons.append("itemlists")
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getMovieDetailsFromXbmc(dbid, ["tag"])
if tagging.hasTraktWatchlistTag(result["tag"]):
buttons.append("removefromlist")
else:
buttons.append("addtolist")
elif utils.isShow(media_type):
buttons.append("itemlists")
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getShowDetailsFromXBMC(dbid, ["tag"])
if tagging.hasTraktWatchlistTag(result["tag"]):
buttons.append("removefromlist")
else:
buttons.append("addtolist")
if media_type in ["movie", "show", "episode"]:
buttons.append("rate")
if media_type in ["movie", "show", "season", "episode"]:
buttons.append("togglewatched")
if utils.getSettingAsBool("tagging_enable"):
buttons.append("managelists")
buttons.append("updatetags")
buttons.append("sync")
contextMenu = traktContextMenu(media_type=media_type, buttons=buttons)
contextMenu.doModal()
_action = contextMenu.action
del contextMenu
if _action is None:
return
utils.Debug("'%s' selected from trakt.tv action menu" % _action)
args["action"] = _action
if _action in ["addtolist", "removefromlist"]:
args["list"] = "watchlist"
if args["action"] == "sync":
data = {"action": "manualSync"}
data["silent"] = False
if "silent" in args:
data["silent"] = args["silent"].lower() == "true"
elif args["action"] == "loadsettings":
data = {"action": "loadsettings", "force": True}
utils.notification(utils.getString(1201), utils.getString(1111))
elif args["action"] in ["rate", "unrate"]:
data = {}
data["action"] = args["action"]
media_type = None
if "media_type" in args and "dbid" in args:
media_type = args["media_type"]
try:
data["dbid"] = int(args["dbid"])
except ValueError:
utils.Debug("Manual %s triggered for library item, but DBID is invalid." % args["action"])
return
elif "media_type" in args and "remoteid" in args:
media_type = args["media_type"]
data["remoteid"] = args["remoteid"]
if "season" in args:
if not "episode" in args:
utils.Debug(
"Manual %s triggered for non-library episode, but missing episode number." % args["action"]
)
return
try:
data["season"] = int(args["season"])
data["episode"] = int(args["episode"])
except ValueError:
utilities.Debug("Error parsing season or episode for manual %s" % args["action"])
return
else:
media_type = getMediaType()
if not utils.isValidMediaType(media_type):
utils.Debug("Error, not in video library.")
return
data["dbid"] = int(xbmc.getInfoLabel("ListItem.DBID"))
if media_type is None:
utils.Debug("Manual %s triggered on an unsupported content container." % args["action"])
elif utils.isValidMediaType(media_type):
data["media_type"] = media_type
if "dbid" in data:
utils.Debug(
"Manual %s of library '%s' with an ID of '%s'." % (args["action"], media_type, data["dbid"])
)
if utils.isMovie(media_type):
result = utils.getMovieDetailsFromXbmc(data["dbid"], ["imdbnumber", "title", "year"])
if not result:
utils.Debug("No data was returned from XBMC, aborting manual %s." % args["action"])
return
data["imdbnumber"] = result["imdbnumber"]
elif utils.isShow(media_type):
result = utils.getShowDetailsFromXBMC(data["dbid"], ["imdbnumber", "tag"])
if not result:
utils.Debug("No data was returned from XBMC, aborting manual %s." % args["action"])
return
data["imdbnumber"] = result["imdbnumber"]
data["tag"] = result["tag"]
elif utils.isEpisode(media_type):
result = utils.getEpisodeDetailsFromXbmc(
data["dbid"], ["showtitle", "season", "episode", "tvshowid"]
)
if not result:
utils.Debug("No data was returned from XBMC, aborting manual %s." % args["action"])
return
data["tvdb_id"] = result["tvdb_id"]
data["season"] = result["season"]
data["episode"] = result["episode"]
else:
if "season" in data:
utils.Debug(
"Manual %s of non-library '%s' S%02dE%02d, with an ID of '%s'."
% (args["action"], media_type, data["season"], data["episode"], data["remoteid"])
)
data["tvdb_id"] = data["remoteid"]
else:
utils.Debug(
"Manual %s of non-library '%s' with an ID of '%s'."
% (args["action"], media_type, data["remoteid"])
)
data["imdbnumber"] = data["remoteid"]
if args["action"] == "rate" and "rating" in args:
if args["rating"] in ["1", "2", "3", "4", "5", "6", "7", "8", "9", "10"]:
data["rating"] = int(args["rating"])
data = {"action": "manualRating", "ratingData": data}
else:
utils.Debug("Manual %s of '%s' is unsupported." % (args["action"], media_type))
elif args["action"] == "togglewatched":
media_type = getMediaType()
if media_type in ["movie", "show", "season", "episode"]:
data = {}
data["media_type"] = media_type
if utils.isMovie(media_type):
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getMovieDetailsFromXbmc(dbid, ["imdbnumber", "title", "year", "playcount"])
if result:
if result["playcount"] == 0:
data["id"] = result["imdbnumber"]
else:
utils.Debug("Movie alread marked as watched in XBMC.")
else:
utils.Debug("Error getting movie details from XBMC.")
return
elif utils.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getEpisodeDetailsFromXbmc(
dbid, ["showtitle", "season", "episode", "tvshowid", "playcount"]
)
if result:
if result["playcount"] == 0:
data["id"] = result["tvdb_id"]
data["season"] = result["season"]
data["episode"] = result["episode"]
else:
utils.Debug("Episode already marked as watched in XBMC.")
else:
utils.Debug("Error getting episode details from XBMC.")
return
elif utils.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel("ListItem.TVShowTitle")
result = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShows",
"params": {"properties": ["title", "imdbnumber", "year"]},
"id": 0,
}
)
if result and "tvshows" in result:
for show in result["tvshows"]:
if show["title"] == showTitle:
showID = show["tvshowid"]
data["id"] = show["imdbnumber"]
break
else:
utils.Debug("Error getting TV shows from XBMC.")
return
season = xbmc.getInfoLabel("ListItem.Season")
if season == "":
season = 0
else:
season = int(season)
result = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetEpisodes",
"params": {
"tvshowid": showID,
"season": season,
"properties": ["season", "episode", "playcount"],
},
"id": 0,
}
)
if result and "episodes" in result:
episodes = []
for episode in result["episodes"]:
if episode["playcount"] == 0:
episodes.append(episode["episode"])
if len(episodes) == 0:
utils.Debug("'%s - Season %d' is already marked as watched." % (showTitle, season))
return
data["season"] = season
data["episodes"] = episodes
else:
utils.Debug("Error getting episodes from '%s' for Season %d" % (showTitle, season))
return
elif utils.isShow(media_type):
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
result = utils.getShowDetailsFromXBMC(dbid, ["year", "imdbnumber"])
if not result:
utils.Debug("Error getting show details from XBMC.")
return
showTitle = result["label"]
data["id"] = result["imdbnumber"]
result = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetEpisodes",
"params": {"tvshowid": dbid, "properties": ["season", "episode", "playcount"]},
"id": 0,
}
)
if result and "episodes" in result:
i = 0
s = {}
for e in result["episodes"]:
season = str(e["season"])
if not season in s:
s[season] = []
if e["playcount"] == 0:
s[season].append(e["episode"])
i = i + 1
if i == 0:
utils.Debug("'%s' is already marked as watched." % showTitle)
return
data["seasons"] = dict((k, v) for k, v in s.iteritems() if v)
else:
utils.Debug("Error getting episode details for '%s' from XBMC." % showTitle)
return
if len(data) > 1:
utils.Debug(
"Marking '%s' with the following data '%s' as watched on trakt.tv" % (media_type, str(data))
)
data["action"] = "markWatched"
# execute toggle watched action
xbmc.executebuiltin("Action(ToggleWatched)")
elif args["action"] == "updatetags":
data = {"action": "updatetags"}
elif args["action"] == "managelists":
data = {"action": "managelists"}
elif args["action"] == "timertest":
utils.Debug("Timing JSON requests.")
import time
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShowDetails",
"params": {"tvshowid": 254, "properties": ["tag"]},
"id": 1,
}
)
# data = utils.getShowDetailsFromXBMC(254, ['tag', 'imdbnumber'])
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShowDetails with tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetMovieDetails",
"params": {"movieid": 634, "properties": ["tag"]},
"id": 1,
}
)
# data = utils.getMovieDetailsFromXbmc(634, ['tag', 'imdbnumber', 'title', 'year'])
e = time.time() - t
utils.Debug("VideoLibrary.GetMovieDetails with tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShows",
"params": {"properties": ["tag", "title", "imdbnumber", "year"]},
"id": 0,
}
)
e = time.time() - t
utils.Debug(
"VideoLibrary.GetTVShows with tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data["tvshows"]), e / len(data["tvshows"]))
)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"id": 0,
"method": "VideoLibrary.GetMovies",
"params": {"properties": ["tag", "title", "imdbnumber", "year"]},
}
)
e = time.time() - t
utils.Debug(
"VideoLibrary.GetMovies with tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data["movies"]), e / len(data["movies"]))
)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShows",
"params": {"properties": ["title", "imdbnumber", "year"]},
"id": 0,
}
)
e = time.time() - t
utils.Debug(
"VideoLibrary.GetTVShows without tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data["tvshows"]), e / len(data["tvshows"]))
)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"id": 0,
"method": "VideoLibrary.GetMovies",
"params": {"properties": ["title", "imdbnumber", "year"]},
}
)
e = time.time() - t
utils.Debug(
"VideoLibrary.GetMovies without tags: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data["movies"]), e / len(data["movies"]))
)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShowDetails",
"params": {"tvshowid": 254, "properties": ["imdbnumber"]},
"id": 1,
}
)
# data = utils.getShowDetailsFromXBMC(254, ['imdbnumber'])
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShowDetails without tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetMovieDetails",
"params": {"movieid": 634, "properties": ["imdbnumber", "title", "year"]},
"id": 1,
}
)
# data = utils.getMovieDetailsFromXbmc(634, ['imdbnumber', 'title', 'year'])
e = time.time() - t
utils.Debug("VideoLibrary.GetMovieDetails without tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShows",
"params": {"properties": ["title", "imdbnumber", "year"]},
"id": 0,
}
)
data = data["tvshows"]
for item in data:
item_data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetTVShowDetails",
"params": {"tvshowid": item["tvshowid"], "properties": ["tag"]},
"id": 1,
}
)
item["tag"] = item_data["tvshowdetails"]["tag"]
e = time.time() - t
utils.Debug(
"VideoLibrary.GetTVShows with tags from loop: %0.3f seconds, %d items at %0.5f seconds per item"
% (e, len(data), e / len(data))
)
t = time.time()
data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetMovies",
"params": {"properties": ["title", "imdbnumber", "year"]},
"id": 0,
}
)
data = data["movies"]
for item in data:
item_data = utils.xbmcJsonRequest(
{
"jsonrpc": "2.0",
"method": "VideoLibrary.GetMovieDetails",
"params": {"movieid": item["movieid"], "properties": ["tag"]},
"id": 1,
}
)
item["tag"] = item_data["moviedetails"]["tag"]
e = time.time() - t
utils.Debug(
"VideoLibrary.GetMovies with tags from: %0.3f seconds, %d items at %0.5f seconds per item."
% (e, len(data), e / len(data))
)
elif args["action"] in ["itemlists", "addtolist", "removefromlist"]:
data = {}
data["action"] = args["action"]
media_type = None
dbid = None
if "media_type" in args and "dbid" in args:
media_type = args["media_type"]
try:
dbid = int(args["dbid"])
except ValueError:
utils.Debug("'%s' triggered for library item, but DBID is invalid." % args["action"])
return
else:
media_type = getMediaType()
if not media_type in ["movie", "show"]:
utils.Debug("Error, not in video library.")
return
try:
dbid = int(xbmc.getInfoLabel("ListItem.DBID"))
except ValueError:
utils.Debug(
"'%s' triggered for library item, but there is a problem with ListItem.DBID." % args["action"]
)
return
if not media_type in ["movie", "show"]:
utils.Debug("'%s' is not a valid media type for '%s'." % (media_type, args["action"]))
return
if args["action"] in ["addtolist", "removefromlist"]:
if "list" in args:
data["list"] = args["list"]
else:
utils.Debug("'%s' requires a list parameter." % data["action"])
data["type"] = media_type
if utils.isMovie(media_type):
result = utils.getMovieDetailsFromXbmc(dbid, ["imdbnumber", "title", "year", "tag"])
if not result:
utils.Debug("Error getting movie details from XBMC.")
return
data["tag"] = result["tag"]
data["movieid"] = result["movieid"]
data["title"] = result["title"]
data["year"] = result["year"]
if result["imdbnumber"].startswith("tt"):
data["imdb_id"] = result["imdbnumber"]
elif result["imdbnumber"].isdigit():
data["tmdb_id"] = result["imdbnumber"]
elif utils.isShow(media_type):
result = utils.getShowDetailsFromXBMC(dbid, ["imdbnumber", "title", "tag"])
if not result:
utils.Debug("Error getting show details from XBMC.")
return
data["tag"] = result["tag"]
data["tvshowid"] = result["tvshowid"]
data["title"] = result["title"]
if result["imdbnumber"].startswith("tt"):
data["imdb_id"] = result["imdbnumber"]
elif result["imdbnumber"].isdigit():
data["tvdb_id"] = result["imdbnumber"]
q = queue.SqliteQueue()
if "action" in data:
utils.Debug("Queuing for dispatch: %s" % data)
q.append(data)
def Main():
args = getArguments()
data = {}
if args['action'] == 'sync':
data = {'action': 'manualSync'}
elif args['action'] in ['rate', 'unrate']:
data = {}
data['action'] = args['action']
media_type = None
if 'media_type' in args and 'dbid' in args:
media_type = args['media_type']
try:
data['dbid'] = int(args['dbid'])
except ValueError:
utils.Debug("Manual %s triggered for library item, but DBID is invalid." % args['action'])
return
elif 'media_type' in args and 'remoteid' in args:
media_type = args['media_type']
data['remoteid'] = args['remoteid']
if 'season' in args:
if not 'episode' in args:
utils.Debug("Manual %s triggered for non-library episode, but missing episode number." % args['action'])
return
try:
data['season'] = int(args['season'])
data['episode'] = int(args['episode'])
except ValueError:
utilities.Debug("Error parsing season or episode for manual %s" % args['action'])
return
else:
media_type = getMediaType()
if not utils.isValidMediaType(media_type):
utils.Debug("Error, not in video library.")
return
data['dbid'] = int(xbmc.getInfoLabel('ListItem.DBID'))
if media_type is None:
utils.Debug("Manual %s triggered on an unsupported content container." % args['action'])
elif utils.isValidMediaType(media_type):
data['media_type'] = media_type
if 'dbid' in data:
utils.Debug("Manual %s of library '%s' with an ID of '%s'." % (args['action'], media_type, data['dbid']))
if utils.isMovie(media_type):
result = utils.getMovieDetailsFromXbmc(data['dbid'], ['imdbnumber', 'title', 'year'])
if not result:
utils.Debug("No data was returned from XBMC, aborting manual %s." % args['action'])
return
data['imdbnumber'] = result['imdbnumber']
elif utils.isShow(media_type):
result = utils.getShowDetailsFromXBMC(data['dbid'], ['imdbnumber', 'tag'])
if not result:
utils.Debug("No data was returned from XBMC, aborting manual %s." % args['action'])
return
data['imdbnumber'] = result['imdbnumber']
data['tag'] = result['tag']
elif utils.isEpisode(media_type):
result = utils.getEpisodeDetailsFromXbmc(data['dbid'], ['showtitle', 'season', 'episode', 'tvshowid'])
if not result:
utils.Debug("No data was returned from XBMC, aborting manual %s." % args['action'])
return
data['tvdb_id'] = result['tvdb_id']
data['season'] = result['season']
data['episode'] = result['episode']
else:
if 'season' in data:
utils.Debug("Manual %s of non-library '%s' S%02dE%02d, with an ID of '%s'." % (args['action'], media_type, data['season'], data['episode'], data['remoteid']))
data['tvdb_id'] = data['remoteid']
else:
utils.Debug("Manual %s of non-library '%s' with an ID of '%s'." % (args['action'], media_type, data['remoteid']))
data['imdbnumber'] = data['remoteid']
if args['action'] == 'rate' and 'rating' in args:
if args['rating'] in ['1', '2', '3', '4', '5', '6', '7', '8', '9', '10']:
data['rating'] = int(args['rating'])
data = {'action': 'manualRating', 'ratingData': data}
else:
utils.Debug("Manual %s of '%s' is unsupported." % (args['action'], media_type))
elif args['action'] == 'togglewatched':
media_type = getMediaType()
if media_type in ['movie', 'show', 'season', 'episode']:
data = {}
data['media_type'] = media_type
if utils.isMovie(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = utils.getMovieDetailsFromXbmc(dbid, ['imdbnumber', 'title', 'year', 'playcount'])
if result:
if result['playcount'] == 0:
data['id'] = result['imdbnumber']
else:
utils.Debug("Movie alread marked as watched in XBMC.")
else:
utils.Debug("Error getting movie details from XBMC.")
return
elif utils.isEpisode(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = utils.getEpisodeDetailsFromXbmc(dbid, ['showtitle', 'season', 'episode', 'tvshowid', 'playcount'])
if result:
if result['playcount'] == 0:
data['id'] = result['tvdb_id']
data['season'] = result['season']
data['episode'] = result['episode']
else:
utils.Debug("Episode already marked as watched in XBMC.")
else:
utils.Debug("Error getting episode details from XBMC.")
return
elif utils.isSeason(media_type):
showID = None
showTitle = xbmc.getInfoLabel('ListItem.TVShowTitle')
result = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShows', 'params': {'properties': ['title', 'imdbnumber', 'year']}, 'id': 0})
if result and 'tvshows' in result:
for show in result['tvshows']:
if show['title'] == showTitle:
showID = show['tvshowid']
data['id'] = show['imdbnumber']
break
else:
utils.Debug("Error getting TV shows from XBMC.")
return
season = xbmc.getInfoLabel('ListItem.Season')
if season == "":
season = 0
else:
season = int(season)
result = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetEpisodes', 'params': {'tvshowid': showID, 'season': season, 'properties': ['season', 'episode', 'playcount']}, 'id': 0})
if result and 'episodes' in result:
episodes = []
for episode in result['episodes']:
if episode['playcount'] == 0:
episodes.append(episode['episode'])
if len(episodes) == 0:
utils.Debug("'%s - Season %d' is already marked as watched." % (showTitle, season))
return
data['season'] = season
data['episodes'] = episodes
else:
utils.Debug("Error getting episodes from '%s' for Season %d" % (showTitle, season))
return
elif utils.isShow(media_type):
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
result = utils.getShowDetailsFromXBMC(dbid, ['year', 'imdbnumber'])
if not result:
utils.Debug("Error getting show details from XBMC.")
return
showTitle = result['label']
data['id'] = result['imdbnumber']
result = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetEpisodes', 'params': {'tvshowid': dbid, 'properties': ['season', 'episode', 'playcount']}, 'id': 0})
if result and 'episodes' in result:
i = 0
s = {}
for e in result['episodes']:
season = str(e['season'])
if not season in s:
s[season] = []
if e['playcount'] == 0:
s[season].append(e['episode'])
i = i + 1
if i == 0:
utils.Debug("'%s' is already marked as watched." % showTitle)
return
data['seasons'] = dict((k, v) for k, v in s.iteritems() if v)
else:
utils.Debug("Error getting episode details for '%s' from XBMC." % showTitle)
return
if len(data) > 1:
utils.Debug("Marking '%s' with the following data '%s' as watched on trakt.tv" % (media_type, str(data)))
data['action'] = 'markWatched'
# execute toggle watched action
xbmc.executebuiltin("Action(ToggleWatched)")
elif args['action'] == 'updatetags':
data = {'action': 'updatetags'}
elif args['action'] == 'managelists':
data = {'action': 'managelists'}
elif args['action'] == 'timertest':
utils.Debug("Timing JSON requests.")
import time
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShowDetails', 'params':{'tvshowid': 254, 'properties': ['tag']}, 'id': 1})
#data = utils.getShowDetailsFromXBMC(254, ['tag', 'imdbnumber'])
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShowDetails with tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetMovieDetails', 'params':{'movieid': 634, 'properties': ['tag']}, 'id': 1})
#data = utils.getMovieDetailsFromXbmc(634, ['tag', 'imdbnumber', 'title', 'year'])
e = time.time() - t
utils.Debug("VideoLibrary.GetMovieDetails with tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShows', 'params': {'properties': ['tag', 'title', 'imdbnumber', 'year']}, 'id': 0})
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShows with tags: %0.3f seconds, %d items at %0.5f seconds per item" % (e, len(data['tvshows']), e / len(data['tvshows'])))
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'id': 0, 'method': 'VideoLibrary.GetMovies', 'params': {'properties': ['tag', 'title', 'imdbnumber', 'year']}})
e = time.time() - t
utils.Debug("VideoLibrary.GetMovies with tags: %0.3f seconds, %d items at %0.5f seconds per item" % (e, len(data['movies']), e / len(data['movies'])))
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShows', 'params': {'properties': ['title', 'imdbnumber', 'year']}, 'id': 0})
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShows without tags: %0.3f seconds, %d items at %0.5f seconds per item" % (e, len(data['tvshows']), e / len(data['tvshows'])))
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'id': 0, 'method': 'VideoLibrary.GetMovies', 'params': {'properties': ['title', 'imdbnumber', 'year']}})
e = time.time() - t
utils.Debug("VideoLibrary.GetMovies without tags: %0.3f seconds, %d items at %0.5f seconds per item" % (e, len(data['movies']), e / len(data['movies'])))
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShowDetails', 'params':{'tvshowid': 254, 'properties': ['imdbnumber']}, 'id': 1})
#data = utils.getShowDetailsFromXBMC(254, ['imdbnumber'])
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShowDetails without tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetMovieDetails', 'params':{'movieid': 634, 'properties': ['imdbnumber', 'title', 'year']}, 'id': 1})
#data = utils.getMovieDetailsFromXbmc(634, ['imdbnumber', 'title', 'year'])
e = time.time() - t
utils.Debug("VideoLibrary.GetMovieDetails without tags: %0.3f seconds." % e)
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShows', 'params': {'properties': ['title', 'imdbnumber', 'year']}, 'id': 0})
data = data['tvshows']
for item in data:
item_data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShowDetails', 'params':{'tvshowid': item['tvshowid'], 'properties': ['tag']}, 'id': 1})
item['tag'] = item_data['tvshowdetails']['tag']
e = time.time() - t
utils.Debug("VideoLibrary.GetTVShows with tags from loop: %0.3f seconds, %d items at %0.5f seconds per item" % (e, len(data), e / len(data)))
t = time.time()
data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetMovies', 'params': {'properties': ['title', 'imdbnumber', 'year']}, 'id': 0})
data = data['movies']
for item in data:
item_data = utils.xbmcJsonRequest({'jsonrpc': '2.0', 'method': 'VideoLibrary.GetMovieDetails', 'params':{'movieid': item['movieid'], 'properties': ['tag']}, 'id': 1})
item['tag'] = item_data['moviedetails']['tag']
e = time.time() - t
utils.Debug("VideoLibrary.GetMovies with tags from: %0.3f seconds, %d items at %0.5f seconds per item." % (e, len(data), e / len(data)))
elif args['action'] in ['itemlists', 'addtolist', 'removefromlist']:
data = {}
data['action'] = args['action']
media_type = None
dbid = None
if 'media_type' in args and 'dbid' in args:
media_type = args['media_type']
try:
dbid = int(args['dbid'])
except ValueError:
utils.Debug("'%s' triggered for library item, but DBID is invalid." % args['action'])
return
else:
media_type = getMediaType()
if not media_type in ['movie', 'show']:
utils.Debug("Error, not in video library.")
return
try:
dbid = int(xbmc.getInfoLabel('ListItem.DBID'))
except ValueError:
utils.Debug("'%s' triggered for library item, but there is a problem with ListItem.DBID." % args['action'])
return
if not media_type in ['movie', 'show']:
utils.Debug("'%s' is not a valid media type for '%s'." % (media_type, args['action']))
return
if args['action'] in ['addtolist', 'removefromlist']:
if 'list' in args:
data['list'] = args['list']
else:
utils.Debug("'%s' requires a list parameter." % data['action'])
data['type'] = media_type
if utils.isMovie(media_type):
result = utils.getMovieDetailsFromXbmc(dbid, ['imdbnumber', 'title', 'year', 'tag'])
if not result:
utils.Debug("Error getting movie details from XBMC.")
return
data['tag'] = result['tag']
data['movieid'] = result['movieid']
data['title'] = result['title']
data['year'] = result['year']
if result['imdbnumber'].startswith("tt"):
data['imdb_id'] = result['imdbnumber']
elif result['imdbnumber'].isdigit():
data['tmdb_id'] = result['imdbnumber']
elif utils.isShow(media_type):
result = utils.getShowDetailsFromXBMC(dbid, ['imdbnumber', 'title', 'tag'])
if not result:
utils.Debug("Error getting show details from XBMC.")
return
data['tag'] = result['tag']
data['tvshowid'] = result['tvshowid']
data['title'] = result['title']
if result['imdbnumber'].startswith("tt"):
data['imdb_id'] = result['imdbnumber']
elif result['imdbnumber'].isdigit():
data['tvdb_id'] = result['imdbnumber']
q = queue.SqliteQueue()
if 'action' in data:
utils.Debug("Queuing for dispatch: %s" % data)
q.append(data)
def test_isValidMediaType_Episode():
assert utilities.isValidMediaType('episode')
