# Module: KodiHelper
# Created on: 13.01.2017
-import os
-import urllib
import xbmcplugin
import xbmcgui
-import xbmcaddon
import xbmc
import json
-import uuid
+from MSL import MSL
+from os import remove
+from os.path import join, isfile
+from urllib import urlencode
+from xbmcaddon import Addon
+from uuid import uuid4
+from utils import get_user_agent_for_current_platform
from UniversalAnalytics import Tracker
try:
import cPickle as pickle
class KodiHelper:
"""Consumes all the configuration data from Kodi as well as turns data into lists of folders and videos"""
- def __init__ (self, plugin_handle, base_url):
+ def __init__ (self, plugin_handle=None, base_url=None):
"""Fetches all needed info from Kodi & configures the baseline of the plugin
Parameters
base_url : :obj:`str`
Plugin base url
"""
+ addon = self.get_addon()
self.plugin_handle = plugin_handle
self.base_url = base_url
- self.addon = xbmcaddon.Addon()
- self.plugin = self.addon.getAddonInfo('name')
- self.base_data_path = xbmc.translatePath(self.addon.getAddonInfo('profile'))
+ self.plugin = addon.getAddonInfo('name')
+ self.version = addon.getAddonInfo('version')
+ self.base_data_path = xbmc.translatePath(addon.getAddonInfo('profile'))
self.home_path = xbmc.translatePath('special://home')
- self.plugin_path = self.addon.getAddonInfo('path')
+ self.plugin_path = addon.getAddonInfo('path')
self.cookie_path = self.base_data_path + 'COOKIE'
self.data_path = self.base_data_path + 'DATA'
- self.config_path = os.path.join(self.base_data_path, 'config')
+ self.config_path = join(self.base_data_path, 'config')
self.msl_data_path = xbmc.translatePath('special://profile/addon_data/service.msl').decode('utf-8') + '/'
- self.verb_log = self.addon.getSetting('logging') == 'true'
- self.default_fanart = self.addon.getAddonInfo('fanart')
+ self.verb_log = addon.getSetting('logging') == 'true'
+ self.default_fanart = addon.getAddonInfo('fanart')
self.library = None
self.setup_memcache()
+ def get_addon (self):
+ """Returns a fresh addon instance"""
+ return Addon()
+
def refresh (self):
"""Refresh the current list"""
return xbmc.executebuiltin('Container.Refresh')
bool
Setting could be set or not
"""
- return self.addon.setSetting(key, value)
+ return self.get_addon().setSetting(key, value)
def get_credentials (self):
"""Returns the users stored credentials
The users stored account data
"""
return {
- 'email': self.addon.getSetting('email'),
- 'password': self.addon.getSetting('password')
+ 'email': self.get_addon().getSetting('email'),
+ 'password': self.get_addon().getSetting('password')
}
+ def get_esn(self):
+ """
+ Returns the esn from settings
+ """
+ self.log(msg='Is FILE: ' + str(isfile(self.msl_data_path + 'msl_data.json')))
+ self.log(msg=self.get_addon().getSetting('esn'))
+ return self.get_addon().getSetting('esn')
+
+ def set_esn(self, esn):
+ """
+ Returns the esn from settings
+ """
+ stored_esn = self.get_esn()
+ if not stored_esn and esn:
+ self.set_setting('esn', esn)
+ self.delete_manifest_data()
+ return esn
+ return stored_esn
+
+ def delete_manifest_data(self):
+ if isfile(self.msl_data_path + 'msl_data.json'):
+ remove(self.msl_data_path + 'msl_data.json')
+ if isfile(self.msl_data_path + 'manifest.json'):
+ remove(self.msl_data_path + 'manifest.json')
+ msl = MSL(kodi_helper=self)
+ msl.perform_key_handshake()
+ msl.save_msl_data()
+
def get_dolby_setting(self):
"""
Returns if the dolby sound is enabled
:return: True|False
"""
- return self.addon.getSetting('enable_dolby_sound') == 'true'
+ return self.get_addon().getSetting('enable_dolby_sound') == 'true'
def get_custom_library_settings (self):
"""Returns the settings in regards to the custom library folder(s)
The users library settings
"""
return {
- 'enablelibraryfolder': self.addon.getSetting('enablelibraryfolder'),
- 'customlibraryfolder': self.addon.getSetting('customlibraryfolder')
+ 'enablelibraryfolder': self.get_addon().getSetting('enablelibraryfolder'),
+ 'customlibraryfolder': self.get_addon().getSetting('customlibraryfolder')
}
def get_ssl_verification_setting (self):
bool
Verify or not
"""
- return self.addon.getSetting('ssl_verification') == 'true'
+ return self.get_addon().getSetting('ssl_verification') == 'true'
def set_main_menu_selection (self, type):
"""Persist the chosen main menu entry in memory
"""Invalidates the memory cache"""
xbmcgui.Window(xbmcgui.getCurrentWindowId()).setProperty('memcache', pickle.dumps({}))
- def has_cached_item (self, cache_id):
- """Checks if the requested item is in memory cache
-
- Parameters
- ----------
- cache_id : :obj:`str`
- ID of the cache entry
-
- Returns
- -------
- bool
- Item is cached
- """
- cached_items = pickle.loads(xbmcgui.Window(xbmcgui.getCurrentWindowId()).getProperty('memcache'))
- return cache_id in cached_items.keys()
-
def get_cached_item (self, cache_id):
"""Returns an item from the in memory cache
Contents of the requested cache item or none
"""
cached_items = pickle.loads(xbmcgui.Window(xbmcgui.getCurrentWindowId()).getProperty('memcache'))
- if self.has_cached_item(cache_id) != True:
- return None
- return cached_items[cache_id]
+
+ return cached_items.get(cache_id)
def add_cached_item (self, cache_id, contents):
"""Adds an item to the in memory cache
Parameters
----------
- profiles : :obj:`dict` of :obj:`str`
+ profiles : :obj:`list` of :obj:`dict` of :obj:`str`
List of user profiles
action : :obj:`str`
bool
List could be build
"""
- for profile_id in profiles:
- profile = profiles[profile_id]
- url = build_url({'action': action, 'profile_id': profile_id})
+ for profile in profiles:
+ url = build_url({'action': action, 'profile_id': profile['id']})
li = xbmcgui.ListItem(label=profile['profileName'], iconImage=profile['avatar'])
li.setProperty('fanart_image', self.default_fanart)
xbmcplugin.addDirectoryItem(handle=self.plugin_handle, url=url, listitem=li, isFolder=True)
Dictionary of actions to build subsequent routes
type : :obj:`str`
- None or 'queue' f.e. when it´s a special video lists
+ None or 'queue' f.e. when it´s a special video lists
build_url : :obj:`fn`
Function to build the subsequent routes
"""
for video_list_id in video_list:
video = video_list[video_list_id]
- if type != 'queue' or (type == 'queue' and video['in_my_list'] == True):
- li = xbmcgui.ListItem(label=video['title'])
- # add some art to the item
- li = self._generate_art_info(entry=video, li=li)
+ li = xbmcgui.ListItem(label=video['title'])
+ # add some art to the item
+ li = self._generate_art_info(entry=video, li=li)
+ # add list item info
+ li, infos = self._generate_entry_info(entry=video, li=li)
+ li = self._generate_context_menu_items(entry=video, li=li)
+ # lists can be mixed with shows & movies, therefor we need to check if its a movie, so play it right away
+ if video_list[video_list_id]['type'] == 'movie':
+ # it´s a movie, so we need no subfolder & a route to play it
+ isFolder = False
+ url = build_url({'action': 'play_video', 'video_id': video_list_id, 'infoLabels': infos})
+ else:
# it´s a show, so we need a subfolder & route (for seasons)
isFolder = True
- url = build_url({'action': actions[video['type']], 'show_id': video_list_id})
- # lists can be mixed with shows & movies, therefor we need to check if its a movie, so play it right away
- if video_list[video_list_id]['type'] == 'movie':
- # it´s a movie, so we need no subfolder & a route to play it
- isFolder = False
- url = build_url({'action': 'play_video', 'video_id': video_list_id})
- # add list item info
- li = self._generate_entry_info(entry=video, li=li)
- li = self._generate_context_menu_items(entry=video, li=li)
- xbmcplugin.addDirectoryItem(handle=self.plugin_handle, url=url, listitem=li, isFolder=isFolder)
+ params = {'action': actions[video['type']], 'show_id': video_list_id}
+ if 'tvshowtitle' in infos:
+ params['tvshowtitle'] = infos.get('tvshowtitle', '').encode('utf-8')
+ url = build_url(params)
+ xbmcplugin.addDirectoryItem(handle=self.plugin_handle, url=url, listitem=li, isFolder=isFolder)
xbmcplugin.addSortMethod(handle=self.plugin_handle, sortMethod=xbmcplugin.SORT_METHOD_LABEL)
xbmcplugin.addSortMethod(handle=self.plugin_handle, sortMethod=xbmcplugin.SORT_METHOD_TITLE)
xbmcplugin.endOfDirectory(self.plugin_handle)
return True
- def build_season_listing (self, seasons_sorted, season_list, build_url):
+ def build_season_listing (self, seasons_sorted, build_url):
"""Builds the season list screen for a show
Parameters
----------
- seasons_sorted : :obj:`list` of :obj:`str`
- Sorted season indexes
-
- season_list : :obj:`dict` of :obj:`str`
- List of season entries
+ seasons_sorted : :obj:`list` of :obj:`dict` of :obj:`str`
+ Sorted list of season entries
build_url : :obj:`fn`
Function to build the subsequent routes
bool
List could be build
"""
- for index in seasons_sorted:
- for season_id in season_list:
- season = season_list[season_id]
- if int(season['idx']) == index:
- li = xbmcgui.ListItem(label=season['text'])
- # add some art to the item
- li = self._generate_art_info(entry=season, li=li)
- # add list item info
- li = self._generate_entry_info(entry=season, li=li, base_info={'mediatype': 'season'})
- li = self._generate_context_menu_items(entry=season, li=li)
- url = build_url({'action': 'episode_list', 'season_id': season_id})
- xbmcplugin.addDirectoryItem(handle=self.plugin_handle, url=url, listitem=li, isFolder=True)
+ for season in seasons_sorted:
+ li = xbmcgui.ListItem(label=season['text'])
+ # add some art to the item
+ li = self._generate_art_info(entry=season, li=li)
+ # add list item info
+ li, infos = self._generate_entry_info(entry=season, li=li, base_info={'mediatype': 'season'})
+ li = self._generate_context_menu_items(entry=season, li=li)
+ params = {'action': 'episode_list', 'season_id': season['id']}
+ if 'tvshowtitle' in infos:
+ params['tvshowtitle'] = infos.get('tvshowtitle', '').encode('utf-8')
+ url = build_url(params)
+ xbmcplugin.addDirectoryItem(handle=self.plugin_handle, url=url, listitem=li, isFolder=True)
xbmcplugin.addSortMethod(handle=self.plugin_handle, sortMethod=xbmcplugin.SORT_METHOD_NONE)
xbmcplugin.addSortMethod(handle=self.plugin_handle, sortMethod=xbmcplugin.SORT_METHOD_VIDEO_YEAR)
xbmcplugin.endOfDirectory(self.plugin_handle)
return True
- def build_episode_listing (self, episodes_sorted, episode_list, build_url):
+ def build_episode_listing (self, episodes_sorted, build_url):
"""Builds the episode list screen for a season of a show
Parameters
----------
- episodes_sorted : :obj:`list` of :obj:`str`
- Sorted episode indexes
-
- episode_list : :obj:`dict` of :obj:`str`
- List of episode entries
+ episodes_sorted : :obj:`list` of :obj:`dict` of :obj:`str`
+ Sorted list of episode entries
build_url : :obj:`fn`
Function to build the subsequent routes
bool
List could be build
"""
- for index in episodes_sorted:
- for episode_id in episode_list:
- episode = episode_list[episode_id]
- if int(episode['episode']) == index:
- li = xbmcgui.ListItem(label=episode['title'])
- # add some art to the item
- li = self._generate_art_info(entry=episode, li=li)
- # add list item info
- li = self._generate_entry_info(entry=episode, li=li, base_info={'mediatype': 'episode'})
- li = self._generate_context_menu_items(entry=episode, li=li)
- url = build_url({'action': 'play_video', 'video_id': episode_id, 'start_offset': episode['bookmark']})
- xbmcplugin.addDirectoryItem(handle=self.plugin_handle, url=url, listitem=li, isFolder=False)
+ for episode in episodes_sorted:
+ li = xbmcgui.ListItem(label=episode['title'])
+ # add some art to the item
+ li = self._generate_art_info(entry=episode, li=li)
+ # add list item info
+ li, infos = self._generate_entry_info(entry=episode, li=li, base_info={'mediatype': 'episode'})
+ li = self._generate_context_menu_items(entry=episode, li=li)
+ url = build_url({'action': 'play_video', 'video_id': episode['id'], 'start_offset': episode['bookmark'], 'infoLabels': infos})
+ xbmcplugin.addDirectoryItem(handle=self.plugin_handle, url=url, listitem=li, isFolder=False)
xbmcplugin.addSortMethod(handle=self.plugin_handle, sortMethod=xbmcplugin.SORT_METHOD_EPISODE)
xbmcplugin.addSortMethod(handle=self.plugin_handle, sortMethod=xbmcplugin.SORT_METHOD_NONE)
xbmcplugin.endOfDirectory(self.plugin_handle)
return True
- def play_item (self, esn, video_id, start_offset=-1):
+ def play_item (self, esn, video_id, start_offset=-1, infoLabels={}):
"""Plays a video
Parameters
start_offset : :obj:`str`
Offset to resume playback from (in seconds)
+
+ infoLabels : :obj:`str`
+ the listitem's infoLabels
Returns
-------
bool
List could be build
"""
+ self.set_esn(esn)
+ addon = self.get_addon()
inputstream_addon = self.get_inputstream_addon()
if inputstream_addon == None:
self.show_missing_inputstream_addon_notification()
self.track_event('playVideo')
# check esn in settings
- settings_esn = str(self.addon.getSetting('esn'))
+ settings_esn = str(addon.getSetting('esn'))
if len(settings_esn) == 0:
- self.addon.setSetting('esn', str(esn))
+ addon.setSetting('esn', str(esn))
# inputstream addon properties
- msl_service_url = 'http://localhost:' + str(self.addon.getSetting('msl_service_port'))
+ msl_service_url = 'http://localhost:' + str(addon.getSetting('msl_service_port'))
play_item = xbmcgui.ListItem(path=msl_service_url + '/manifest?id=' + video_id)
+ play_item.setContentLookup(False)
+ play_item.setMimeType('application/dash+xml')
+ play_item.setProperty(inputstream_addon + '.stream_headers', 'user-agent=' + get_user_agent_for_current_platform())
play_item.setProperty(inputstream_addon + '.license_type', 'com.widevine.alpha')
play_item.setProperty(inputstream_addon + '.manifest_type', 'mpd')
play_item.setProperty(inputstream_addon + '.license_key', msl_service_url + '/license?id=' + video_id + '||b{SSM}!b{SID}|')
# check if we have a bookmark e.g. start offset position
if int(start_offset) > 0:
play_item.setProperty('StartOffset', str(start_offset) + '.0')
+ # set infoLabels
+ if len(infoLabels) > 0:
+ play_item.setInfo('video', infoLabels)
return xbmcplugin.setResolvedUrl(self.plugin_handle, True, listitem=play_item)
def _generate_art_info (self, entry, li):
if 'mpaa' in entry_keys:
infos.update({'mpaa': entry['mpaa']})
else:
- infos.update({'mpaa': str(entry['maturity']['board']) + '-' + str(entry['maturity']['value'])})
+ if entry.get('maturity', None) is not None:
+ if entry['maturity']['board'] is not None and entry['maturity']['value'] is not None:
+ infos.update({'mpaa': str(entry['maturity']['board'].encode('utf-8')) + '-' + str(entry['maturity']['value'].encode('utf-8'))})
if 'rating' in entry_keys:
infos.update({'rating': int(entry['rating']) * 2})
if 'synopsis' in entry_keys:
if 'type' in entry_keys:
if entry['type'] == 'movie' or entry['type'] == 'episode':
li.setProperty('IsPlayable', 'true')
+ elif entry['type'] == 'show':
+ infos.update({'tvshowtitle': entry['title']})
if 'mediatype' in entry_keys:
if entry['mediatype'] == 'movie' or entry['mediatype'] == 'episode':
li.setProperty('IsPlayable', 'true')
if entry['quality'] == '1080':
quality = {'width': '1920', 'height': '1080'}
li.addStreamInfo('video', quality)
+ if 'tvshowtitle' in entry_keys:
+ infos.update({'tvshowtitle': entry.get('tvshowtitle', '').encode('utf-8')})
li.setInfo('video', infos)
- return li
+ return li, infos
def _generate_context_menu_items (self, entry, li):
"""Adds context menue items to a Kodi list item
entry_keys = entry.keys()
# action item templates
- encoded_title = urllib.urlencode({'title': entry['title'].encode('utf-8')}) if 'title' in entry else ''
+ encoded_title = urlencode({'title': entry['title'].encode('utf-8')}) if 'title' in entry else ''
url_tmpl = 'XBMC.RunPlugin(' + self.base_url + '?action=%action%&id=' + str(entry['id']) + '&' + encoded_title + ')'
actions = [
['export_to_library', self.get_local_string(30018), 'export'],
:obj:`str`
Requested string or empty string
"""
- src = xbmc if string_id < 30000 else self.addon
+ src = xbmc if string_id < 30000 else self.get_addon()
locString = src.getLocalizedString(string_id)
if isinstance(locString, unicode):
locString = locString.encode('utf-8')
:param event: the string idetifier of the event
:return: None
"""
+ addon = self.get_addon()
# Check if tracking is enabled
- enable_tracking = (self.addon.getSetting('enable_tracking') == 'true')
+ enable_tracking = (addon.getSetting('enable_tracking') == 'true')
if enable_tracking:
#Get or Create Tracking id
- tracking_id = self.addon.getSetting('tracking_id')
+ tracking_id = addon.getSetting('tracking_id')
if tracking_id is '':
- tracking_id = str(uuid.uuid4())
- self.addon.setSetting('tracking_id', tracking_id)
+ tracking_id = str(uuid4())
+ addon.setSetting('tracking_id', tracking_id)
# Send the tracking event
tracker = Tracker.create('UA-46081640-5', client_id=tracking_id)
- tracker.send('event', event)
+ tracker.send('event', event)
\ No newline at end of file