Пример #1
0
    def __init__(self, nx_common):
        """
        Takes the instances & configuration options needed to drive the plugin

        Parameters
        ----------
        kodi_helper : :obj:`KodiHelper`
            instance of the KodiHelper class

        library : :obj:`Library`
            instance of the Library class

        base_url : :obj:`str`
            plugin base url

        log_fn : :obj:`fn`
             optional log function
        """
        self.nx_common = nx_common
        self.library = Library(nx_common=nx_common)

        self.kodi_helper = KodiHelper(
            nx_common=nx_common,
            library=self.library)

        self.library.set_kodi_helper(kodi_helper=self.kodi_helper)

        self.base_url = self.nx_common.base_url
        self.log = self.nx_common.log
Пример #2
0
# Created on: 13.01.2017
# License: MIT https://goo.gl/5bMj3H
"""Kodi plugin for Netflix (https://netflix.com)"""

import sys
from resources.lib.KodiHelper import KodiHelper
from resources.lib.Navigation import Navigation
from resources.lib.Library import Library

# Setup plugin
PLUGIN_HANDLE = int(sys.argv[1])
BASE_URL = sys.argv[0]
# We use string slicing to trim the leading ? from the plugin call paramstring
REQUEST_PARAMS = sys.argv[2][1:]

# init plugin libs
KODI_HELPER = KodiHelper(plugin_handle=PLUGIN_HANDLE, base_url=BASE_URL)
LIBRARY = Library(root_folder=KODI_HELPER.base_data_path,
                  library_settings=KODI_HELPER.get_custom_library_settings(),
                  log_fn=KODI_HELPER.log)
NAVIGATION = Navigation(kodi_helper=KODI_HELPER,
                        library=LIBRARY,
                        base_url=BASE_URL,
                        log_fn=KODI_HELPER.log)
KODI_HELPER.set_library(library=LIBRARY)

if __name__ == '__main__':
    # Call the router function and pass the plugin call parameters to it.
    KODI_HELPER.log('Started (Version ' + KODI_HELPER.version + ')')
    NAVIGATION.router(paramstring=REQUEST_PARAMS)
Пример #3
0
class Navigation(object):
    """
    Routes to the correct subfolder,
    dispatches actions & acts as a controller
    for the Kodi view & the Netflix model
    """

    def __init__(self, nx_common):
        """
        Takes the instances & configuration options needed to drive the plugin

        Parameters
        ----------
        kodi_helper : :obj:`KodiHelper`
            instance of the KodiHelper class

        library : :obj:`Library`
            instance of the Library class

        base_url : :obj:`str`
            plugin base url

        log_fn : :obj:`fn`
             optional log function
        """
        self.nx_common = nx_common
        self.library = Library(nx_common=nx_common)

        self.kodi_helper = KodiHelper(
            nx_common=nx_common,
            library=self.library)

        self.library.set_kodi_helper(kodi_helper=self.kodi_helper)

        self.base_url = self.nx_common.base_url
        self.log = self.nx_common.log

    @log
    def router(self, paramstring):
        """
        Route to the requested subfolder & dispatch actions along the way

        Parameters
        ----------
        paramstring : :obj:`str`
            Url query params
        """
        params = self.parse_paramters(paramstring=paramstring)
        action = params.get('action', None)
        p_type = params.get('type', None)
        p_type_not_search_export = p_type != 'search' and p_type != 'exported'

        # 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 == 'logout':
            logout = self._check_response(self.call_netflix_service({
                'method': 'logout'}))
            return logout
        # switch user account
        if action == 'switch_account':
            return self.switch_account()

        # check if we need to execute any actions before the actual routing
        # gives back a dict of options routes might need
        options = self.before_routing_action(params=params)

        # switch user account
        if action == 'toggle_adult_pin':
            adult_pin = self.kodi_helper.dialogs.show_adult_pin_dialog()
            pin_correct = self._check_response(self.call_netflix_service({
                'method': 'send_adult_pin',
                'pin': adult_pin}))
            if pin_correct is not True:
                return self.kodi_helper.dialogs.show_invalid_pin_notify()
            return self.kodi_helper.toggle_adult_pin()

        # check if one of the before routing options decided to killthe routing
        if 'exit' in options:
            self.nx_common.log(msg='exit in options')
            return False
        if 'action' not in params.keys():
            # show the profiles
            if self.nx_common.get_setting('autologin_enable') == 'true':
                profile_id = self.nx_common.get_setting('autologin_id')
                if profile_id != '':
                    self.call_netflix_service({
                        'method': 'switch_profile',
                        'profile_id': profile_id})
                    return self.show_video_lists()
            return self.show_profiles()
        elif action == 'save_autologin':
            # save profile id and name to settings for autologin
            autologin = self.kodi_helper.save_autologin_data(
                autologin_id=params.get('autologin_id'),
                autologin_user=params.get('autologin_user'))
            return autologin
        elif action == 'video_lists':
            # list lists that contain other lists
            # (starting point with recommendations, search, etc.)
            return self.show_video_lists()
        elif action == 'video_list':
            # show a list of shows/movies
            type = None if 'type' not in params.keys() else params['type']
            start = 0 if 'start' not in params.keys() else int(params['start'])
            video_list = self.show_video_list(
                video_list_id=params.get('video_list_id'),
                type=type,
                start=start)
            return video_list
        elif action == 'season_list':
            # list of seasons for a show
            seasons = self.show_seasons(
                show_id=params.get('show_id'),
                tvshowtitle=params.get('tvshowtitle'))
            return seasons
        elif action == 'episode_list':
            # list of episodes for a season
            episode_list = self.show_episode_list(
                season_id=params.get('season_id'),
                tvshowtitle=params.get('tvshowtitle'))
            return episode_list
        elif action == 'rating':
            return self.rate_on_netflix(video_id=params['id'])
        elif 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 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 action == 'export':
            # adds a title to the users list on Netflix
            alt_title = self.kodi_helper.dialogs.show_add_library_title_dialog(
                original_title=urllib.unquote(params['title']).decode('utf8'))
            self.export_to_library(video_id=params['id'], alt_title=alt_title)
            return self.kodi_helper.refresh()
        elif action == 'remove':
            # adds a title to the users list on Netflix
            self.remove_from_library(video_id=params['id'])
            return self.kodi_helper.refresh()
        elif action == 'update':
            # adds a title to the users list on Netflix
            self.remove_from_library(video_id=params['id'])
            alt_title = self.kodi_helper.dialogs.show_add_library_title_dialog(
                original_title=urllib.unquote(params['title']).decode('utf8'))
            self.export_to_library(video_id=params['id'], alt_title=alt_title)
            return self.kodi_helper.refresh()
        elif action == 'removeexported':
            # adds a title to the users list on Netflix
            term = self.kodi_helper.dialogs.show_finally_remove_modal(
                title=params.get('title'),
                year=params.get('year'))
            if params['type'] == 'movie' and str(term) == '1':
                self.library.remove_movie(
                    title=params['title'].decode('utf-8'),
                    year=int(params['year']))
                self.kodi_helper.refresh()
            if params['type'] == 'show' and str(term) == '1':
                self.library.remove_show(title=params['title'].decode('utf-8'))
                self.kodi_helper.refresh()
            return True
        elif action == 'export-new-episodes':
            return self.export_new_episodes(params.get('inbackground', False))
        elif action == 'updatedb':
            # adds a title to the users list on Netflix
            self.library.updatedb_from_exported()
            self.kodi_helper.dialogs.show_db_updated_notify()
            return True
        elif action == 'user-items' and p_type_not_search_export:
            # display the lists (recommendations, genres, etc.)
            return self.show_user_list(type=params['type'])
        elif action == 'play_video':
            # play a video, check for adult pin if needed
            adult_pin = None
            adult_setting = self.nx_common.get_setting('adultpin_enable')
            ask_for_adult_pin = adult_setting.lower() == 'true'
            if ask_for_adult_pin is True:
                if self.check_for_adult_pin(params=params):
                    pin = self.kodi_helper.dialogs.show_adult_pin_dialog()
                    pin_response = self.call_netflix_service({
                        'method': 'send_adult_pin',
                        'pin': pin})
                    pin_correct = self._check_response(pin_response)
                    if pin_correct is not True:
                        self.kodi_helper.dialogs.show_invalid_pin_notify()
                        return True
            self.play_video(
                video_id=params['video_id'],
                start_offset=params.get('start_offset', -1),
                infoLabels=params.get('infoLabels', {}))
            return True
        elif action == 'user-items' and params['type'] == 'search':
            # if the user requested a search, ask for the term
            term = self.kodi_helper.dialogs.show_search_term_dialog()
            if term:
                result_folder = self.kodi_helper.build_search_result_folder(
                    build_url=self.build_url,
                    term=term)
                return self.kodi_helper.set_location(url=result_folder)
        elif action == 'search_result':
            return self.show_search_results(params.get('term'))
        elif action == 'user-items' and params['type'] == 'exported':
            # update local db from exported media
            self.library.updatedb_from_exported()
            # list exported movies/shows
            exported = self.kodi_helper.build_video_listing_exported(
                content=self.library.list_exported_media(),
                build_url=self.build_url)
            return exported
        else:
            raise ValueError('Invalid paramstring: {0}!'.format(paramstring))
        xbmc.executebuiltin('Container.Refresh')
        return True

    @log
    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

        Parameters
        ----------
        video_id : :obj:`str`
            ID of the video that should be played

        start_offset : :obj:`str`
            Offset to resume playback from (in seconds)

        infoLabels : :obj:`str`
            the listitem's infoLabels
        """
        try:
            infoLabels = ast.literal_eval(infoLabels)
        except:
            infoLabels = {}

        play = self.kodi_helper.play_item(
            video_id=video_id,
            start_offset=start_offset,
            infoLabels=infoLabels)
        return play

    @log
    def show_search_results(self, term):
        """Display a list of search results

        Parameters
        ----------
        term : :obj:`str`
            String to lookup

        Returns
        -------
        bool
            If no results are available
        """
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        if user_data:
            search_contents = self._check_response(self.call_netflix_service({
                'method': 'search',
                'term': term, 'guid':
                user_data['guid'],
                'cache': True}))
            if search_contents and len(search_contents) != 0:
                actions = {'movie': 'play_video', 'show': 'season_list'}
                results = self.kodi_helper.build_search_result_listing(
                    video_list=search_contents,
                    actions=actions,
                    build_url=self.build_url)
                return results
        self.kodi_helper.dialogs.show_no_search_results_notify()
        return False

    def show_user_list(self, type):
        """
        List the users lists for shows/movies for
        recommendations/genres based on the given type

        Parameters
        ----------
        user_list_id : :obj:`str`
            Type of list to display
        """
        # determine if we´re in kids mode
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        if user_data:
            video_list_ids = self._check_response(self.call_netflix_service({
                'method': 'fetch_video_list_ids',
                'guid': user_data['guid'],
                'cache': True}))
            if video_list_ids:
                sub_list = self.kodi_helper.build_user_sub_listing(
                    video_list_ids=video_list_ids[type],
                    type=type,
                    action='video_list',
                    build_url=self.build_url)
                return sub_list
        return False

    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)
        """
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        if user_data:
            episode_list = self._check_response(self.call_netflix_service({
                'method': 'fetch_episodes_by_season',
                'season_id': season_id,
                'guid': user_data['guid'],
                'cache': True}))
            if episode_list:
                # Extract episode numbers and associated keys.
                d = [(v['episode'], k) for k, v in episode_list.items()]
                # sort episodes by number
                # (they´re coming back unsorted from the api)
                episodes_sorted = [episode_list[k] for (_, k) in sorted(d)]
                for episode in episodes_sorted:
                    episode['tvshowtitle'] = tvshowtitle
                # list the episodes
                episodes_list = self.kodi_helper.build_episode_listing(
                    episodes_sorted=episodes_sorted,
                    build_url=self.build_url)
                return episodes_list
        return False

    def show_seasons(self, show_id, tvshowtitle):
        """Lists all seasons for a given show

        Parameters
        ----------
        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
        """
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        if user_data:
            season_list = self._check_response(self.call_netflix_service({
                'method': 'fetch_seasons_for_show',
                'show_id': show_id,
                'guid': user_data['guid'],
                'cache': True}))
            if season_list:
                # check if we have sesons,
                # announced shows that are not available yet have none
                if len(season_list) == 0:
                    return self.kodi_helper.build_no_seasons_available()
                # Extract episode numbers and associated keys.
                d = [(v['idx'], k) for k, v in season_list.items()]
                # sort seasons by index by default
                #  (they´re coming back unsorted from the api)
                seasons_sorted = [season_list[k] for (_, k) in sorted(d)]
                for season in seasons_sorted:
                    season['tvshowtitle'] = tvshowtitle
                season_list = self.kodi_helper.build_season_listing(
                    seasons_sorted=seasons_sorted,
                    build_url=self.build_url)
                return season_list
        return False

    def show_video_list(self, video_list_id, type, start=0):
        """List shows/movies based on the given video list id

        Parameters
        ----------
        video_list_id : :obj:`str`
            ID of the video list that should be displayed

        type : :obj:`str`
            None or 'queue' f.e. when it´s a special video lists

        start : :obj:`int`
            Starting point
        """
        end = start + Netflix.FETCH_VIDEO_REQUEST_COUNT
        video_list = {}
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        if user_data:
            user_list = ['queue', 'topTen', 'netflixOriginals', 'continueWatching',
                         'trendingNow', 'newRelease', 'popularTitles']
            if str(type) in user_list and video_list_id is None:
                video_list_id = self.list_id_for_type(type)
            for i in range(0, 4):
                items = self._check_response(self.call_netflix_service({
                    'method': 'fetch_video_list',
                    'list_id': video_list_id,
                    'list_from': start,
                    'list_to': end,
                    'guid': user_data['guid'],
                    'cache': True}))
                if items is False and i == 0:
                    self.nx_common.log('show_video_list response is dirty')
                    return False
                elif len(items) == 0:
                    if i == 0:
                        self.nx_common.log('show_video_list items=0')
                        return False
                    break
                req_count = Netflix.FETCH_VIDEO_REQUEST_COUNT
                video_list.update(items)
                start = end + 1
                end = start + req_count
            has_more = len(video_list) == (req_count + 1) * 4
            actions = {'movie': 'play_video', 'show': 'season_list'}
            listing = self.kodi_helper.build_video_listing(
                video_list=video_list,
                actions=actions,
                type=type,
                build_url=self.build_url,
                has_more=has_more,
                start=start,
                current_video_list_id=video_list_id)
            return listing
        return False

    def show_video_lists(self):
        """List the users video lists (recommendations, my list, etc.)"""
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        if user_data:
            video_list_ids = self._check_response(self.call_netflix_service({
                'method': 'fetch_video_list_ids',
                'guid': user_data['guid'],
                'cache': True}))
            if 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',
                    'exported': 'user-items',
                    'default': 'video_list'
                }
                listing = 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)
                return listing
        return False

    @log
    def list_id_for_type(self, type):
        """Get the list_ids for a given type"""
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        video_list_ids = self._check_response(self.call_netflix_service({
            'method': 'fetch_video_list_ids',
            'guid': user_data['guid'],
            'cache': True}))
        if video_list_ids:
            for video_list_id in video_list_ids['user']:
                if video_list_ids['user'][video_list_id]['name'] == type:
                    return str(video_list_ids['user'][video_list_id]['id'])

    @log
    def show_profiles(self):
        """List the profiles for the active account"""
        profiles = self._check_response(self.call_netflix_service({
            'method': 'list_profiles'}))
        if profiles and len(profiles) != 0:
            listing = self.kodi_helper.build_profiles_listing(
                profiles=profiles.values(),
                action='video_lists',
                build_url=self.build_url)
            return listing
        return self.kodi_helper.dialogs.show_login_failed_notify()

    @log
    def rate_on_netflix(self, video_id):
        """Rate a show/movie/season/episode on Netflix

        Parameters
        ----------
        video_list_id : :obj:`str`
            ID of the video list that should be displayed
        """
        rating = self.kodi_helper.dialogs.show_rating_dialog()
        result = self._check_response(self.call_netflix_service({
            'method': 'rate_video',
            'video_id': video_id,
            'rating': rating}))
        if result is False:
            self.kodi_helper.dialogs.show_request_error_notify()
        return result

    @log
    def remove_from_list(self, video_id):
        """Remove an item from 'My List' & refresh the view

        Parameters
        ----------
        video_list_id : :obj:`str`
            ID of the video list that should be displayed
        """
        result = self._check_response(self.call_netflix_service({
            'method': 'remove_from_list',
            'video_id': video_id}))
        if result:
            return self.kodi_helper.refresh()
        return self.kodi_helper.dialogs.show_request_error_notify()

    @log
    def add_to_list(self, video_id):
        """Add an item to 'My List' & refresh the view

        Parameters
        ----------
        video_list_id : :obj:`str`
            ID of the video list that should be displayed
        """
        result = self._check_response(self.call_netflix_service({
            'method': 'add_to_list',
            'video_id': video_id}))
        if result:
            return self.kodi_helper.refresh()
        return self.kodi_helper.dialogs.show_request_error_notify()

    @log
    def export_to_library(self, video_id, alt_title, in_background=False):
        """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._check_response(self.call_netflix_service({
            'method': 'fetch_metadata',
            'video_id': video_id}))
        if metadata:
            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']:
                    if not self._download_episode_metadata(
                            season['id'], video['title']):
                        self.log(msg=('Failed to download episode metadata '
                                      'for {} season {}')
                                 .format(video['title'], season['id']),
                                 level=xbmc.LOGERROR)
                    for episode in season['episodes']:
                        episodes.append({
                            'season': season['seq'],
                            'episode': episode['seq'],
                            'id': episode['id']})
                self.library.add_show(
                    netflix_id=video_id,
                    title=video['title'],
                    alt_title=alt_title,
                    episodes=episodes,
                    build_url=self.build_url,
                    in_background=in_background)
            return True
        self.kodi_helper.dialogs.show_no_metadata_notify()
        return False

    def _download_episode_metadata(self, season_id, tvshowtitle):
        user_data = (
            self._check_response(
                self.call_netflix_service(
                    {'method': 'get_user_data'})))
        episode_list = (
            self._check_response(
                self.call_netflix_service({
                    'method': 'fetch_episodes_by_season',
                    'season_id': season_id,
                    'guid': user_data['guid'],
                    'cache': True})))
        if episode_list:
            for episode in episode_list.itervalues():
                episode['tvshowtitle'] = tvshowtitle
                self.kodi_helper._generate_art_info(entry=episode)
                self.kodi_helper._generate_entry_info(
                    entry=episode,
                    base_info={'mediatype': 'episode'})
            return True
        return False

    @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._check_response(self.call_netflix_service({
            'method': 'fetch_metadata',
            'video_id': video_id}))
        if metadata:
            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 True
        self.kodi_helper.dialogs.show_no_metadata_notify()
        return False

    @log
    def export_new_episodes(self, in_background):
        update_started_at = datetime.today().strftime('%Y-%m-%d %H:%M')
        self.nx_common.set_setting('update_running', update_started_at)
        for title, meta in self.library.list_exported_shows().iteritems():
            try:
                netflix_id = meta.get('netflix_id',
                                      self._get_netflix_id(meta['alt_title']))
            except KeyError:
                self.log(
                    ('Cannot determine netflix id for {}. '
                     'Remove and re-add to library to fix this.')
                    .format(title.encode('utf-8')), xbmc.LOGERROR)
                continue
            self.log('Exporting new episodes of {} (id={})'
                     .format(title.encode('utf-8'), netflix_id))
            self.export_to_library(video_id=netflix_id, alt_title=title,
                                   in_background=in_background)
        xbmc.executebuiltin(
            'UpdateLibrary(video, {})'.format(self.library.tvshow_path))
        self.nx_common.set_setting('update_running', 'false')
        self.nx_common.set_setting('last_update', update_started_at[0:10])
        return True

    def _get_netflix_id(self, showtitle):
        show_dir = self.nx_common.check_folder_path(
            path=os.path.join(self.library.tvshow_path, showtitle))
        try:
            filepath = next(os.path.join(show_dir, fn)
                            for fn in self.nx_common.list_dir(show_dir)[1]
                            if 'strm' in fn)
        except StopIteration:
            raise KeyError

        self.log('Reading contents of {}'.format(filepath.encode('utf-8')))
        buf = self.nx_common.load_file(data_path='', filename=filepath)
        episode_id = re.search(r'video_id=(\d+)', buf).group(1)
        show_metadata = self._check_response(self.call_netflix_service({
            'method': 'fetch_metadata',
            'video_id': episode_id}))
        return show_metadata['video']['id']

    @log
    def establish_session(self, account):
        """
        Checks if we have an cookie with an active sessions,
        otherwise tries to login the user

        Parameters
        ----------
        account : :obj:`dict` of :obj:`str`
            Dict containing an email & a password property

        Returns
        -------
        bool
            If we don't have an active session & the user couldn't be logged in
        """
        is_logged_in = self._check_response(self.call_netflix_service({
            'method': 'is_logged_in'}))
        if is_logged_in is True:
            return True
        else:
            check = self._check_response(self.call_netflix_service({
                'method': 'login',
                'email': account['email'],
                'password': account['password']}))
            return check

    def switch_account(self):
        """
        Deletes all current account data & prompts with dialogs for new ones
        """
        self._check_response(self.call_netflix_service({'method': 'logout'}))
        self.nx_common.set_credentials('', '')

        raw_email = self.kodi_helper.dialogs.show_email_dialog()
        raw_password = self.kodi_helper.dialogs.show_password_dialog()
        self.nx_common.set_credentials(raw_email, raw_password)

        account = {
            'email': raw_email,
            'password': raw_password,
        }
        if self.establish_session(account=account) is not True:
            self.nx_common.set_credentials('', '')
            return self.kodi_helper.dialogs.show_login_failed_notify()
        return True

    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.get('pin') == 'True']

    @log
    def before_routing_action(self, params):
        """Executes actions before the actual routing takes place:

            - Check if account data has been stored, if not, asks for it
            - Check if the profile should be changed (and changes if so)
            - Establishes a session if no action route is given

        Parameters
        ----------
        params : :obj:`dict` of :obj:`str`
            Url query params

        Returns
        -------
        :obj:`dict` of :obj:`str`
            Options that can be provided by this hook &
            used later in the routing process
        """
        options = {}
        # check login & try to relogin if necessary
        logged_in = self._check_response(self.call_netflix_service({
            'method': 'is_logged_in'}))
        if logged_in is False:
            credentials = self.nx_common.get_credentials()
            # check if we have user settings, if not, set em
            if credentials['email'] == '':
                email = self.kodi_helper.dialogs.show_email_dialog()
                credentials['email'] = email
            if credentials['password'] == '':
                password = self.kodi_helper.dialogs.show_password_dialog()
                credentials['password'] = password

            self.nx_common.set_credentials(credentials['email'], credentials['password'])

            if self.establish_session(account=credentials) is not True:
                self.nx_common.set_credentials('', '')
                self.kodi_helper.dialogs.show_login_failed_notify()

        # persist & load main menu selection
        if 'type' in params:
            self.kodi_helper.set_main_menu_selection(type=params['type'])
            main_menu = self.kodi_helper.get_main_menu_selection()
            options['main_menu_selection'] = main_menu
        # check and switch the profile if needed
        if self.check_for_designated_profile_change(params=params):
            self.kodi_helper.invalidate_memcache()
            profile_id = params.get('profile_id', None)
            if profile_id is None:
                user_data = self._check_response(self.call_netflix_service({
                    'method': 'get_user_data'}))
                if user_data:
                    profile_id = user_data['guid']
            if profile_id:
                self.call_netflix_service({
                    'method': 'switch_profile',
                    'profile_id': profile_id})
        return options

    def check_for_designated_profile_change(self, params):
        """Checks if the profile needs to be switched

        Parameters
        ----------
        params : :obj:`dict` of :obj:`str`
            Url query params

        Returns
        -------
        bool
            Profile should be switched or not
        """
        # check if we need to switch the user
        user_data = self._check_response(self.call_netflix_service({
            'method': 'get_user_data'}))
        profiles = self._check_response(self.call_netflix_service({
            'method': 'list_profiles'}))
        if user_data and profiles:
            if 'guid' not in user_data:
                return False
            current_profile_id = user_data['guid']
            if profiles.get(current_profile_id).get('isKids', False) is True:
                return True
            has_id = 'profile_id' in params
            return has_id and current_profile_id != params['profile_id']
        self.kodi_helper.dialogs.show_request_error_notify()
        return False

    def parse_paramters(self, paramstring):
        """Tiny helper to convert a url paramstring into a dictionary

        Parameters
        ----------
        paramstring : :obj:`str`
            Url query params (in url string notation)

        Returns
        -------
        :obj:`dict` of :obj:`str`
            Url query params (as a dictionary)
        """
        return dict(parse_qsl(paramstring))

    def _is_expired_session(self, response):
        """Checks if a response error is based on an invalid session

        Parameters
        ----------
        response : :obj:`dict` of :obj:`str`
            Error response object

        Returns
        -------
        bool
            Error is based on an invalid session
        """
        has_error = 'error' in response
        has_code = 'code' in response
        return has_error and has_code and str(response['code']) == '401'

    def _check_response(self, response):
        """
        Checks if a response contains an error & if the error
        is based on an invalid session, it tries a relogin

        Parameters
        ----------
        response : :obj:`dict` of :obj:`str`
            Success response object or Error response object

        Returns
        -------
        :obj:`dict` or :bool:False
            Response if no error or False
        """
        # check for any errors
        if type(response) is dict and 'error' in response:
            # check if we do not have a valid session,
            # in case that happens: (re)login
            if self._is_expired_session(response=response):
                account = self.nx_common.get_credentials()
                self.establish_session(account=account)
            message = response['message'] if 'message' in response else ''
            code = response['code'] if 'code' in response else ''
            self.log(msg='[ERROR]: ' + message + '::' + str(code))
            return False
        return response

    def build_url(self, query):
        """Tiny helper to transform a dict into a url + querystring

        Parameters
        ----------
        query : :obj:`dict` of  :obj:`str`
            List of paramters to be url encoded

        Returns
        -------
        str
            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:' +
                self.nx_common.get_setting('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
        """
        cache = params.pop('cache', None)
        values = urllib.urlencode(params)
        # check for cached items
        if cache:
            cached_value = self.kodi_helper.get_cached_item(
                cache_id=values)

            # Cache lookup successful?
            if cached_value is not None:
                self.log(
                    msg='Fetched item from cache: (cache_id=' + values + ')')
                return cached_value

        url = self.get_netflix_service_url()
        full_url = url + '?' + values
        # don't use proxy for localhost
        if urlparse(url).hostname in ('localhost', '127.0.0.1', '::1'):
            opener = urllib2.build_opener(urllib2.ProxyHandler({}))
            urllib2.install_opener(opener)
        data = urllib2.urlopen(full_url).read()
        parsed_json = json.loads(data, object_pairs_hook=OrderedDict)
        if 'error' in parsed_json:
            result = {'error': parsed_json.get('error')}
            return result
        result = parsed_json.get('result', None)
        if result and cache:
            self.log(msg='Adding item to cache: (cache_id=' + values + ')')
            self.kodi_helper.add_cached_item(cache_id=values, contents=result)
        return result

    def open_settings(self, url):
        """Opens a foreign settings dialog"""
        url = 'inputstream.adaptive' if url == 'is' else url
        from xbmcaddon import Addon
        return Addon(url).openSettings()
Пример #4
0
from resources.lib.KodiHelper import KodiHelper
from resources.lib.Navigation import Navigation
from resources.lib.Library import Library

# Setup plugin
plugin_handle = int(sys.argv[1])
base_url = sys.argv[0]

# init plugin libs
kodi_helper = KodiHelper(
    plugin_handle=plugin_handle,
    base_url=base_url
)
library = Library(
    root_folder=kodi_helper.base_data_path,
    library_settings=kodi_helper.get_custom_library_settings(),
    log_fn=kodi_helper.log
)
navigation = Navigation(
    kodi_helper=kodi_helper,
    library=library,
    base_url=base_url,
    log_fn=kodi_helper.log
)
kodi_helper.set_library(library=library)

if __name__ == '__main__':
    # Call the router function and pass the plugin call parameters to it.
    # We use string slicing to trim the leading '?' from the plugin call paramstring
    kodi_helper.log('Started (Version ' + kodi_helper.version + ')')
    navigation.router(paramstring=sys.argv[2][1:])