def run(self):
startup_delay = kodiUtilities.getSettingAsInt('startup_delay')
if startup_delay:
logger.debug("Delaying startup by %d seconds." % startup_delay)
xbmc.sleep(startup_delay * 1000)
logger.debug("Service thread starting.")
# purge queue before doing anything
self.dispatchQueue.purge()
# setup event driven classes
self.Player = traktPlayer(action=self._dispatchQueue)
self.Monitor = traktMonitor(action=self._dispatchQueue)
# init traktapi class
globals.traktapi = traktAPI()
# init sync thread
self.syncThread = syncThread()
# init scrobbler class
self.scrobbler = Scrobbler(globals.traktapi)
AddonSignals.registerSlot('service.nextup.notification', 'NEXTUPWATCHEDSIGNAL', self.callback)
# start loop for events
while not self.Monitor.abortRequested():
if not kodiUtilities.getSetting('authorization'):
last_reminder = kodiUtilities.getSettingAsInt('last_reminder')
now = int(time.time())
if last_reminder >= 0 and last_reminder < now - (24 * 60 * 60):
gui_utils.get_pin()
while len(self.dispatchQueue) and (not self.Monitor.abortRequested()):
data = self.dispatchQueue.get()
logger.debug("Queued dispatch: %s" % data)
self._dispatch(data)
if xbmc.Player().isPlayingVideo():
self.scrobbler.transitionCheck()
if self.Monitor.waitForAbort(1):
# Abort was requested while waiting. We should exit
break
# we are shutting down
logger.debug("Beginning shut down.")
# delete player/monitor
del self.Player
del self.Monitor
# check if sync thread is running, if so, join it.
if self.syncThread.isAlive():
self.syncThread.join()
def run(self):
startup_delay = utilities.getSettingAsInt('startup_delay')
if startup_delay:
logger.debug("Delaying startup by %d seconds." % startup_delay)
xbmc.sleep(startup_delay * 1000)
logger.debug("Service thread starting.")
# purge queue before doing anything
self.dispatchQueue.purge()
# setup event driven classes
self.Player = traktPlayer(action=self._dispatchQueue)
self.Monitor = traktMonitor(action=self._dispatchQueue)
# init traktapi class
globals.traktapi = traktAPI()
# init sync thread
self.syncThread = syncThread()
# init scrobbler class
self.scrobbler = Scrobbler(globals.traktapi)
# start loop for events
while not self.Monitor.abortRequested():
if not utilities.getSetting('authorization'):
last_reminder = utilities.getSettingAsInt('last_reminder')
now = int(time.time())
if last_reminder >= 0 and last_reminder < now - (24 * 60 * 60):
gui_utils.get_pin()
while len(self.dispatchQueue) and (
not self.Monitor.abortRequested()):
data = self.dispatchQueue.get()
logger.debug("Queued dispatch: %s" % data)
self._dispatch(data)
if xbmc.Player().isPlayingVideo():
self.scrobbler.transitionCheck()
if self.Monitor.waitForAbort(1):
# Abort was requested while waiting. We should exit
break
# we are shutting down
logger.debug("Beginning shut down.")
# delete player/monitor
del self.Player
del self.Monitor
# check if sync thread is running, if so, join it.
if self.syncThread.isAlive():
self.syncThread.join()
def run(self):
startup_delay = utilities.getSettingAsInt('startup_delay')
if startup_delay:
logger.debug("Delaying startup by %d seconds." % startup_delay)
xbmc.sleep(startup_delay * 1000)
logger.debug("Service thread starting.")
# purge queue before doing anything
self.dispatchQueue.purge()
# setup event driven classes
self.Player = traktPlayer(action=self._dispatchQueue)
self.Monitor = traktMonitor(action=self._dispatchQueue)
# init traktapi class
globals.traktapi = traktAPI()
# init sync thread
self.syncThread = syncThread()
# init scrobbler class
self.scrobbler = Scrobbler(globals.traktapi)
# start loop for events
while not xbmc.abortRequested:
if not utilities.getSetting('authorization'):
last_reminder = utilities.getSettingAsInt('last_reminder')
now = int(time.time())
if last_reminder >= 0 and last_reminder < now - (24 * 60 * 60):
gui_utils.get_pin()
while len(self.dispatchQueue) and (not xbmc.abortRequested):
data = self.dispatchQueue.get()
logger.debug("Queued dispatch: %s" % data)
self._dispatch(data)
if xbmc.Player().isPlayingVideo():
self.scrobbler.transitionCheck()
xbmc.sleep(500)
# we are shutting down
logger.debug("Beginning shut down.")
# delete player/monitor
del self.Player
del self.Monitor
# check if sync thread is running, if so, join it.
if self.syncThread.isAlive():
self.syncThread.join()
def Main():
args = __getArguments()
data = {}
if args['action'] == 'pin_info':
xbmc.executebuiltin(
'Dialog.Close(all, true)'
) #FIXME stupid hack we need to do as the AddonInfo dialog won't go away
xbmc.executebuiltin('Dialog.Close(all, true)')
gui_utils.get_pin()
if args['action'] == 'contextmenu':
buttons = []
media_type = utils.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 = utils.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
elif utils.isShow(media_type):
tvshow_id = data['dbid']
elif utils.isSeason(media_type):
result = utils.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 utils.isEpisode(media_type):
result = utils.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 utils.isShow(media_type) or utils.isSeason(
media_type) or utils.isEpisode(media_type):
result = utils.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 = utils.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)")
elif args['action'] == 'addtowatchlist':
media_type = utils.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:
data['id'] = result['imdbnumber']
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:
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 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'])
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:
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'] == 'pin_info':
xbmc.executebuiltin('Dialog.Close(all, true)')
gui_utils.get_pin()
if args['action'] == 'contextmenu':
buttons = []
media_type = utils.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 = utils.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
elif utils.isShow(media_type):
tvshow_id = data['dbid']
elif utils.isSeason(media_type):
result = utils.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 utils.isEpisode(media_type):
result = utils.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 utils.isShow(media_type) or utils.isSeason(media_type) or utils.isEpisode(media_type):
result = utils.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 = utils.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)
