X-Git-Url: http://git.code-monkey.de/?a=blobdiff_plain;f=resources%2Flib%2FNavigation.py;h=1643706551e3992606cf7d6b74ff98c349892939;hb=7afec310e04492b290523718b7f7ef8df0ef7ccf;hp=c1e9397278dcd0fc97f22e8d839014c8ca281cb9;hpb=85447a5fdfc7dff80e2272d77a5f94c0eddff1af;p=plugin.video.netflix.git diff --git a/resources/lib/Navigation.py b/resources/lib/Navigation.py index c1e9397..1643706 100644 --- a/resources/lib/Navigation.py +++ b/resources/lib/Navigation.py @@ -4,22 +4,21 @@ # Created on: 13.01.2017 import urllib -import time +import urllib2 +import json +import ast +from xbmcaddon import Addon from urlparse import parse_qsl -from utils import noop -from utils import log +from utils import noop, log class Navigation: """Routes to the correct subfolder, dispatches actions & acts as a controller for the Kodi view & the Netflix model""" - def __init__ (self, netflix_session, kodi_helper, library, base_url, log_fn=noop): + def __init__ (self, kodi_helper, library, base_url, log_fn=noop): """Takes the instances & configuration options needed to drive the plugin Parameters ---------- - netflix_session : :obj:`NetflixSession` - instance of the NetflixSession class - kodi_helper : :obj:`KodiHelper` instance of the KodiHelper class @@ -32,7 +31,6 @@ class Navigation: log_fn : :obj:`fn` optional log function """ - self.netflix_session = netflix_session self.kodi_helper = kodi_helper self.library = library self.base_url = base_url @@ -49,15 +47,20 @@ class Navigation: """ params = self.parse_paramters(paramstring=paramstring) + # open foreign settings dialog + if 'mode' in params.keys() and params['mode'] == 'openSettings': + return self.open_settings(params['url']) + # log out the user if 'action' in params.keys() and params['action'] == 'logout': - return self.netflix_session.logout() + return self.call_netflix_service({'method': 'logout'}) # check login & try to relogin if necessary account = self.kodi_helper.get_credentials() - if self.netflix_session.is_logged_in(account=account) != True: - if self.establish_session(account=account) != True: - return self.kodi_helper.show_login_failed_notification() + if account['email'] != '' and account['password'] != '': + if self.call_netflix_service({'method': 'is_logged_in'}) != True: + if self.establish_session(account=account) != True: + return self.kodi_helper.show_login_failed_notification() # check if we need to execute any actions before the actual routing # gives back a dict of options routes might need @@ -68,7 +71,7 @@ class Navigation: return False if 'action' not in params.keys(): # show the profiles - self.show_profiles() + return self.show_profiles() elif params['action'] == 'video_lists': # list lists that contain other lists (starting point with recommendations, search, etc.) return self.show_video_lists() @@ -78,29 +81,32 @@ class Navigation: return self.show_video_list(video_list_id=params['video_list_id'], type=type) elif params['action'] == 'season_list': # list of seasons for a show - return self.show_seasons(show_id=params['show_id']) + return self.show_seasons(show_id=params['show_id'], tvshowtitle=params['tvshowtitle']) elif params['action'] == 'episode_list': # list of episodes for a season - return self.show_episode_list(season_id=params['season_id']) + return self.show_episode_list(season_id=params['season_id'], tvshowtitle=params['tvshowtitle']) elif params['action'] == 'rating': return self.rate_on_netflix(video_id=params['id']) elif params['action'] == 'remove_from_list': # removes a title from the users list on Netflix + self.kodi_helper.invalidate_memcache() return self.remove_from_list(video_id=params['id']) elif params['action'] == 'add_to_list': # adds a title to the users list on Netflix + self.kodi_helper.invalidate_memcache() return self.add_to_list(video_id=params['id']) + elif params['action'] == 'export': + # adds a title to the users list on Netflix + alt_title = self.kodi_helper.show_add_to_library_title_dialog(original_title=urllib.unquote(params['title']).decode('utf8')) + return self.export_to_library(video_id=params['id'], alt_title=alt_title) + elif params['action'] == 'remove': + # adds a title to the users list on Netflix + return self.remove_from_library(video_id=params['id']) elif params['action'] == 'user-items' and params['type'] != 'search': # display the lists (recommendations, genres, etc.) return self.show_user_list(type=params['type']) elif params['action'] == 'play_video': - # play a video, check for adult pin if needed - adult_pin = None - if self.check_for_adult_pin(params=params): - adult_pin = self.kodi_helper.show_adult_pin_dialog() - if self.netflix_session.send_adult_pin(adult_pin=adult_pin) != True: - return self.kodi_helper.show_wrong_adult_pin_notification() - self.play_video(video_id=params['video_id'], start_offset=params['start_offset']) + self.play_video(video_id=params['video_id'], start_offset=params.get('start_offset', -1), infoLabels=params['infoLabels']) elif params['action'] == 'user-items' and params['type'] == 'search': # if the user requested a search, ask for the term term = self.kodi_helper.show_search_term_dialog() @@ -110,7 +116,7 @@ class Navigation: return True @log - def play_video (self, video_id, start_offset): + def play_video (self, video_id, start_offset, infoLabels): """Starts video playback Note: This is just a dummy, inputstream is needed to play the vids @@ -122,11 +128,18 @@ class Navigation: start_offset : :obj:`str` Offset to resume playback from (in seconds) + + infoLabels : :obj:`str` + the listitem's infoLabels """ - # widevine esn - esn = self.netflix_session.esn - return self.kodi_helper.play_item(esn=esn, video_id=video_id, start_offset=start_offset) + try: + infoLabels = ast.literal_eval(infoLabels) + except: + infoLabels= {} + esn = self.call_netflix_service({'method': 'get_esn'}) + return self.kodi_helper.play_item(esn=esn, video_id=video_id, start_offset=start_offset, infoLabels=infoLabels) + @log def show_search_results (self, term): """Display a list of search results @@ -140,30 +153,11 @@ class Navigation: bool If no results are available """ - has_search_results = False - search_results_raw = self.netflix_session.fetch_search_results(term=term) + user_data = self.call_netflix_service({'method': 'get_user_data'}) + search_contents = self.call_netflix_service({'method': 'search', 'term': term, 'guid': user_data['guid'], 'cache': True}) # check for any errors - if self._is_dirty_response(response=search_results_raw): + if self._is_dirty_response(response=search_contents): return False - - # determine if we found something - if 'search' in search_results_raw['value']: - for key in search_results_raw['value']['search'].keys(): - if self.netflix_session._is_size_key(key=key) == False: - has_search_results = search_results_raw['value']['search'][key]['titles']['length'] > 0 - - # display that we haven't found a thing - if has_search_results == False: - return self.kodi_helper.build_no_search_results_available(build_url=self.build_url, action='search') - - # list the search results - search_results = self.netflix_session.parse_search_results(response_data=search_results_raw) - # add more menaingful data to the search results - raw_search_contents = self.netflix_session.fetch_video_list_information(video_ids=search_results.keys()) - # check for any errors - if self._is_dirty_response(response=raw_search_contents): - return False - search_contents = self.netflix_session.parse_video_list(response_data=raw_search_contents) actions = {'movie': 'play_video', 'show': 'season_list'} return self.kodi_helper.build_search_result_listing(video_list=search_contents, actions=actions, build_url=self.build_url) @@ -175,38 +169,41 @@ class Navigation: user_list_id : :obj:`str` Type of list to display """ - video_list_ids_raw = self.netflix_session.fetch_video_list_ids() + # determine if we´re in kids mode + user_data = self.call_netflix_service({'method': 'get_user_data'}) + video_list_ids = self.call_netflix_service({'method': 'fetch_video_list_ids', 'guid': user_data['guid'], 'cache': True}) # check for any errors - if self._is_dirty_response(response=video_list_ids_raw): + if self._is_dirty_response(response=video_list_ids): return False - video_list_ids = self.netflix_session.parse_video_list_ids(response_data=video_list_ids_raw) return self.kodi_helper.build_user_sub_listing(video_list_ids=video_list_ids[type], type=type, action='video_list', build_url=self.build_url) - def show_episode_list (self, season_id): + def show_episode_list (self, season_id, tvshowtitle): """Lists all episodes for a given season Parameters ---------- season_id : :obj:`str` ID of the season episodes should be displayed for + + tvshowtitle : :obj:`str` + title of the show (for listitems' infolabels) """ - raw_episode_list = self.netflix_session.fetch_episodes_by_season(season_id=season_id) + user_data = self.call_netflix_service({'method': 'get_user_data'}) + episode_list = self.call_netflix_service({'method': 'fetch_episodes_by_season', 'season_id': season_id, 'guid': user_data['guid'], 'cache': True}) # check for any errors - if self._is_dirty_response(response=raw_episode_list): + if self._is_dirty_response(response=episode_list): return False - # parse the raw Netflix data - episode_list = self.netflix_session.parse_episodes_by_season(response_data=raw_episode_list) - # sort seasons by number (they´re coming back unsorted from the api) episodes_sorted = [] for episode_id in episode_list: + episode_list[episode_id]['tvshowtitle'] = tvshowtitle episodes_sorted.append(int(episode_list[episode_id]['episode'])) episodes_sorted.sort() # list the episodes return self.kodi_helper.build_episode_listing(episodes_sorted=episodes_sorted, episode_list=episode_list, build_url=self.build_url) - def show_seasons (self, show_id): + def show_seasons (self, show_id, tvshowtitle): """Lists all seasons for a given show Parameters @@ -214,24 +211,26 @@ class Navigation: show_id : :obj:`str` ID of the show seasons should be displayed for + tvshowtitle : :obj:`str` + title of the show (for listitems' infolabels) Returns ------- bool If no seasons are available """ - season_list_raw = self.netflix_session.fetch_seasons_for_show(id=show_id); + user_data = self.call_netflix_service({'method': 'get_user_data'}) + season_list = self.call_netflix_service({'method': 'fetch_seasons_for_show', 'show_id': show_id, 'guid': user_data['guid'], 'cache': True}) # check for any errors - if self._is_dirty_response(response=season_list_raw): + if self._is_dirty_response(response=season_list): return False # check if we have sesons, announced shows that are not available yet have none - if 'seasons' not in season_list_raw['value']: + if len(season_list) == 0: return self.kodi_helper.build_no_seasons_available() - # parse the seasons raw response from Netflix - season_list = self.netflix_session.parse_seasons(id=show_id, response_data=season_list_raw) # sort seasons by index by default (they´re coming back unsorted from the api) seasons_sorted = [] for season_id in season_list: - seasons_sorted.append(int(season_list[season_id]['shortName'].split(' ')[1])) + season_list[season_id]['tvshowtitle'] = tvshowtitle + seasons_sorted.append(int(season_list[season_id]['idx'])) seasons_sorted.sort() return self.kodi_helper.build_season_listing(seasons_sorted=seasons_sorted, season_list=season_list, build_url=self.build_url) @@ -246,34 +245,33 @@ class Navigation: type : :obj:`str` None or 'queue' f.e. when it´s a special video lists """ - raw_video_list = self.netflix_session.fetch_video_list(list_id=video_list_id) + user_data = self.call_netflix_service({'method': 'get_user_data'}) + video_list = self.call_netflix_service({'method': 'fetch_video_list', 'list_id': video_list_id, 'guid': user_data['guid'] ,'cache': True}) # check for any errors - if self._is_dirty_response(response=raw_video_list): + if self._is_dirty_response(response=video_list): return False - # parse the video list ids - video_list = self.netflix_session.parse_video_list(response_data=raw_video_list) actions = {'movie': 'play_video', 'show': 'season_list'} return self.kodi_helper.build_video_listing(video_list=video_list, actions=actions, type=type, build_url=self.build_url) def show_video_lists (self): """List the users video lists (recommendations, my list, etc.)""" - # fetch video lists - raw_video_list_ids = self.netflix_session.fetch_video_list_ids() + user_data = self.call_netflix_service({'method': 'get_user_data'}) + video_list_ids = self.call_netflix_service({'method': 'fetch_video_list_ids', 'guid': user_data['guid'], 'cache': True}) # check for any errors - if self._is_dirty_response(response=raw_video_list_ids): + if self._is_dirty_response(response=video_list_ids): return False - # parse the video list ids - video_list_ids = self.netflix_session.parse_video_list_ids(response_data=raw_video_list_ids) # defines an order for the user list, as Netflix changes the order at every request user_list_order = ['queue', 'continueWatching', 'topTen', 'netflixOriginals', 'trendingNow', 'newRelease', 'popularTitles'] # define where to route the user actions = {'recommendations': 'user-items', 'genres': 'user-items', 'search': 'user-items', 'default': 'video_list'} return self.kodi_helper.build_main_menu_listing(video_list_ids=video_list_ids, user_list_order=user_list_order, actions=actions, build_url=self.build_url) + @log def show_profiles (self): """List the profiles for the active account""" - self.netflix_session.refresh_session_data(account=self.kodi_helper.get_credentials()) - profiles = self.netflix_session.profiles + profiles = self.call_netflix_service({'method': 'list_profiles'}) + if len(profiles) == 0: + return self.kodi_helper.show_login_failed_notification() return self.kodi_helper.build_profiles_listing(profiles=profiles, action='video_lists', build_url=self.build_url) @log @@ -286,7 +284,7 @@ class Navigation: ID of the video list that should be displayed """ rating = self.kodi_helper.show_rating_dialog() - return self.netflix_session.rate_video(video_id=video_id, rating=rating) + return self.call_netflix_service({'method': 'rate_video', 'video_id': video_id, 'rating': rating}) @log def remove_from_list (self, video_id): @@ -297,7 +295,7 @@ class Navigation: video_list_id : :obj:`str` ID of the video list that should be displayed """ - self.netflix_session.remove_from_list(video_id=video_id) + self.call_netflix_service({'method': 'remove_from_list', 'video_id': video_id}) return self.kodi_helper.refresh() @log @@ -309,7 +307,57 @@ class Navigation: video_list_id : :obj:`str` ID of the video list that should be displayed """ - self.netflix_session.add_to_list(video_id=video_id) + self.call_netflix_service({'method': 'add_to_list', 'video_id': video_id}) + return self.kodi_helper.refresh() + + @log + def export_to_library (self, video_id, alt_title): + """Adds an item to the local library + + Parameters + ---------- + video_id : :obj:`str` + ID of the movie or show + + alt_title : :obj:`str` + Alternative title (for the folder written to disc) + """ + metadata = self.call_netflix_service({'method': 'fetch_metadata', 'video_id': video_id}) + # check for any errors + if self._is_dirty_response(response=metadata): + return False + video = metadata['video'] + + if video['type'] == 'movie': + self.library.add_movie(title=video['title'], alt_title=alt_title, year=video['year'], video_id=video_id, build_url=self.build_url) + if video['type'] == 'show': + episodes = [] + for season in video['seasons']: + for episode in season['episodes']: + episodes.append({'season': season['seq'], 'episode': episode['seq'], 'id': episode['id']}) + + self.library.add_show(title=video['title'], alt_title=alt_title, episodes=episodes, build_url=self.build_url) + return self.kodi_helper.refresh() + + @log + def remove_from_library (self, video_id, season=None, episode=None): + """Removes an item from the local library + + Parameters + ---------- + video_id : :obj:`str` + ID of the movie or show + """ + metadata = self.call_netflix_service({'method': 'fetch_metadata', 'video_id': video_id}) + # check for any errors + if self._is_dirty_response(response=metadata): + return False + video = metadata['video'] + + if video['type'] == 'movie': + self.library.remove_movie(title=video['title'], year=video['year']) + if video['type'] == 'show': + self.library.remove_show(title=video['title']) return self.kodi_helper.refresh() @log @@ -326,10 +374,8 @@ class Navigation: bool If we don't have an active session & the user couldn't be logged in """ - if self.netflix_session.is_logged_in(account=account): - return True - else: - return self.netflix_session.login(account=account) + is_logged_in = self.call_netflix_service({'method': 'is_logged_in'}) + return True if is_logged_in else self.call_netflix_service({'method': 'login', 'email': account['email'], 'password': account['password']}) @log def before_routing_action (self, params): @@ -355,16 +401,23 @@ class Navigation: if credentials['email'] == '': email = self.kodi_helper.show_email_dialog() self.kodi_helper.set_setting(key='email', value=email) + credentials['email'] = email if credentials['password'] == '': password = self.kodi_helper.show_password_dialog() self.kodi_helper.set_setting(key='password', value=password) + credentials['password'] = password # persist & load main menu selection if 'type' in params: self.kodi_helper.set_main_menu_selection(type=params['type']) options['main_menu_selection'] = self.kodi_helper.get_main_menu_selection() # check and switch the profile if needed if self.check_for_designated_profile_change(params=params): - self.netflix_session.switch_profile(profile_id=params['profile_id'], account=credentials) + self.kodi_helper.invalidate_memcache() + profile_id = params.get('profile_id', None) + if profile_id == None: + user_data = self.call_netflix_service({'method': 'get_user_data'}) + profile_id = user_data['guid'] + self.call_netflix_service({'method': 'switch_profile', 'profile_id': profile_id}) # check login, in case of main menu if 'action' not in params: self.establish_session(account=credentials) @@ -384,26 +437,15 @@ class Navigation: Profile should be switched or not """ # check if we need to switch the user - if 'guid' not in self.netflix_session.user_data: + user_data = self.call_netflix_service({'method': 'get_user_data'}) + profiles = self.call_netflix_service({'method': 'list_profiles'}) + if 'guid' not in user_data: return False - current_profile_id = self.netflix_session.user_data['guid'] + current_profile_id = user_data['guid'] + if profiles.get(current_profile_id).get('isKids', False) == True: + return True return 'profile_id' in params and current_profile_id != params['profile_id'] - def check_for_adult_pin (self, params): - """Checks if an adult pin is given in the query params - - Parameters - ---------- - params : :obj:`dict` of :obj:`str` - Url query params - - Returns - ------- - bool - Adult pin parameter exists or not - """ - return (True, False)[params['pin'] == 'True'] - def parse_paramters (self, paramstring): """Tiny helper to convert a url paramstring into a dictionary @@ -453,7 +495,9 @@ class Navigation: if self._is_expired_session(response=response): if self.establish_session(account=self.kodi_helper.get_credentials()): return True - self.log(msg='[ERROR]: ' + response['message'] + '::' + str(response['code'])) + message = response['message'] if 'message' in response else '' + code = response['code'] if 'code' in response else '' + self.log(msg='[ERROR]: ' + message + '::' + str(code)) return True return False @@ -471,3 +515,47 @@ class Navigation: Url + querystring based on the param """ return self.base_url + '?' + urllib.urlencode(query) + + def get_netflix_service_url (self): + """Returns URL & Port of the internal Netflix HTTP Proxy service + + Returns + ------- + str + Url + Port + """ + return 'http://127.0.0.1:' + str(self.kodi_helper.get_addon().getSetting('netflix_service_port')) + + def call_netflix_service (self, params): + """Makes a GET request to the internal Netflix HTTP proxy and returns the result + + Parameters + ---------- + params : :obj:`dict` of :obj:`str` + List of paramters to be url encoded + + Returns + ------- + :obj:`dict` + Netflix Service RPC result + """ + url_values = urllib.urlencode(params) + # check for cached items + if self.kodi_helper.has_cached_item(cache_id=url_values) and params.get('cache', False) == True: + self.log(msg='Fetching item from cache: (cache_id=' + url_values + ')') + return self.kodi_helper.get_cached_item(cache_id=url_values) + url = self.get_netflix_service_url() + full_url = url + '?' + url_values + data = urllib2.urlopen(full_url).read() + parsed_json = json.loads(data) + result = parsed_json.get('result', None) + if params.get('cache', False) == True: + self.log(msg='Adding item to cache: (cache_id=' + url_values + ')') + self.kodi_helper.add_cached_item(cache_id=url_values, contents=result) + return result + + def open_settings(self, url): + """Opens a foreign settings dialog""" + is_addon = self.kodi_helper.get_inputstream_addon() + url = is_addon if url == 'is' else url + return Addon(url).openSettings()