# Author: Nic Wolfe # URL: http://code.google.com/p/sickbeard/ # # This file is part of SickGear. # # SickGear is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # SickGear is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with SickGear. If not, see . import base64 import requests import sickbeard from sickbeard import logger, common from sickbeard.exceptions import ex from sickbeard.encodingKludge import fixStupidEncodings try: import json except ImportError: from lib import simplejson as json class KODINotifier: sg_logo_url = 'https://raw.githubusercontent.com/SickGear/SickGear/master/gui/slick/images/ico/apple-touch-icon' \ '-precomposed.png' def _notify_kodi(self, message, title='SickGear', host=None, username=None, password=None, force=False): # fill in omitted parameters if not host: host = sickbeard.KODI_HOST if not username: username = sickbeard.KODI_USERNAME if not password: password = sickbeard.KODI_PASSWORD # suppress notifications if the notifier is disabled but the notify options are checked if not sickbeard.USE_KODI and not force: return False result = '' for curHost in [x.strip() for x in host.split(',')]: logger.log(u'KODI: Sending Kodi notification to "%s" - %s' % (curHost, message), logger.MESSAGE) command = {'jsonrpc': '2.0', 'method': 'GUI.ShowNotification', 'params': {'title': title, 'message': message, 'image': self.sg_logo_url}, 'id': 1} notifyResult = self._send_to_kodi(command, curHost, username, password) if notifyResult: result += '%s:%s' % (curHost, notifyResult['result']) else: if sickbeard.KODI_ALWAYS_ON or force: result += '%s:False' % curHost return result def _send_to_kodi(self, command, host=None, username=None, password=None): # fill in omitted parameters if not username: username = sickbeard.KODI_USERNAME if not password: password = sickbeard.KODI_PASSWORD if not host: logger.log(u'KODI: No host specified, check your settings', logger.ERROR) return False data = json.dumps(command) logger.log(u'KODI: JSON command: %s' % data, logger.DEBUG) url = 'http://%s/jsonrpc' % host headers = {'Content-type': 'application/json'} # if we have a password, use authentication if password: base64string = base64.encodestring('%s:%s' % (username, password))[:-1] authheader = 'Basic %s' % base64string headers['Authorization'] = authheader logger.log(u'KODI: Contacting (with auth header) via url: %s' % fixStupidEncodings(url), logger.DEBUG) else: logger.log(u'KODI: Contacting via url: %s' % fixStupidEncodings(url), logger.DEBUG) try: response = requests.post(url, data=data, headers=headers) except Exception as e: logger.log(u'KODI: Warning: Couldn\'t contact Kodi at %s - %s' % (host, ex(e)), logger.WARNING) return False if response.status_code == 401: logger.log(u'KODI: Invalid login credentials', logger.ERROR) return False # parse the json result try: result = response.json() logger.log(u'KODI: JSON response: %s' % result, logger.DEBUG) return result # need to return response for parsing except ValueError as e: logger.log(u'KODI: Unable to decode JSON response: %s' % response.text, logger.WARNING) return False def _update_library(self, host=None, showName=None): if not host: logger.log(u'KODI: No host specified, check your settings', logger.DEBUG) return False logger.log(u'KODI: Updating library on host: %s' % host, logger.MESSAGE) # if we're doing per-show if showName: logger.log(u'KODI: Updating library for show %s' % showName, logger.DEBUG) # get tvshowid by showName showsCommand = {'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShows', 'id': 1} showsResponse = self._send_to_kodi(showsCommand, host) try: shows = showsResponse['result']['tvshows'] except: logger.log(u'KODI: No TV shows in Kodi TV show list', logger.DEBUG) return False try: tvshowid = next((show['tvshowid'] for show in shows if show['label'] == showName)) except StopIteration: logger.log(u'XBMC: Exact show name not matched in XBMC TV show list', logger.DEBUG) return False # lookup tv-show path pathCommand = {'jsonrpc': '2.0', 'method': 'VideoLibrary.GetTVShowDetails', 'params': {'tvshowid': tvshowid, 'properties': ['file']}, 'id': 1} pathResponse = self._send_to_kodi(pathCommand, host) path = pathResponse['result']['tvshowdetails']['file'] logger.log(u'KODI: Received Show: %s with ID: %s Path: %s' % (showName, tvshowid, path), logger.DEBUG) if len(path) < 1: logger.log(u'KODI: No valid path found for %s with ID: %s on %s' % (showName, tvshowid, host), logger.WARNING) return False logger.log(u'KODI: Updating %s on %s at %s' % (showName, host, path), logger.DEBUG) updateCommand = {'jsonrpc': '2.0', 'method': 'VideoLibrary.Scan', 'params': {'directory': json.dumps(path)}, 'id': 1} request = self._send_to_kodi(updateCommand, host) if not request: logger.log(u'KODI: Update of show directory failed on %s on %s at %s' % (showName, host, path), logger.ERROR) return False # catch if there was an error in the returned request for r in request: if 'error' in r: logger.log(u'KODI: Error while attempting to update show directory for %s on %s at %s' % (showName, host, path), logger.ERROR) return False # do a full update if requested else: logger.log(u'KODI: Performing full library update on host: %s' % host, logger.DEBUG) updateCommand = {'jsonrpc': '2.0', 'method': 'VideoLibrary.Scan', 'id': 1} request = self._send_to_kodi(updateCommand, host, sickbeard.KODI_USERNAME, sickbeard.KODI_PASSWORD) if not request: logger.log(u'KODI: Full library update failed on host: %s' % host, logger.ERROR) return False return True def notify_snatch(self, ep_name): if sickbeard.KODI_NOTIFY_ONSNATCH: self._notify_kodi(ep_name, common.notifyStrings[common.NOTIFY_SNATCH]) def notify_download(self, ep_name): if sickbeard.KODI_NOTIFY_ONDOWNLOAD: self._notify_kodi(ep_name, common.notifyStrings[common.NOTIFY_DOWNLOAD]) def notify_subtitle_download(self, ep_name, lang): if sickbeard.KODI_NOTIFY_ONSUBTITLEDOWNLOAD: self._notify_kodi('%s: %s' % (ep_name, lang), common.notifyStrings[common.NOTIFY_SUBTITLE_DOWNLOAD]) def notify_git_update(self, new_version='??'): if sickbeard.USE_KODI: update_text = common.notifyStrings[common.NOTIFY_GIT_UPDATE_TEXT] title = common.notifyStrings[common.NOTIFY_GIT_UPDATE] self._notify_kodi('%s %s' % (update_text, new_version), title) def test_notify(self, host, username, password): return self._notify_kodi( 'Testing Kodi notifications from SickGear', 'Test', host, username, password, force=True) def update_library(self, showName=None): if sickbeard.USE_KODI and sickbeard.KODI_UPDATE_LIBRARY: if not sickbeard.KODI_HOST: logger.log(u'KODI: No host specified, check your settings', logger.DEBUG) return False # either update each host, or only attempt to update first only result = 0 for host in [x.strip() for x in sickbeard.KODI_HOST.split(',')]: if self._update_library(host, showName): if sickbeard.KODI_UPDATE_ONLYFIRST: logger.log( u'KODI: Update first host successful on host %s , stopped sending library update commands' % host, logger.DEBUG) return True else: if sickbeard.KODI_ALWAYS_ON: result = result + 1 # needed for the 'update kodi' submenu command as it only cares of the final result vs the individual ones if result == 0: return True else: return False notifier = KODINotifier