# -*- coding: utf-8 -*- ################################################################################################# import logging import xbmc import xbmcgui import clientinfo import connectmanager import connect.connectionmanager as connectionmanager import userclient from utils import settings, language as lang, passwordsXML ################################################################################################# log = logging.getLogger("EMBY."+__name__) STATE = connectionmanager.ConnectionState ################################################################################################# class InitialSetup(object): def __init__(self): self.addon_id = clientinfo.ClientInfo().get_addon_id() self.user_client = userclient.UserClient() self.connectmanager = connectmanager.ConnectManager() def setup(self): # Check server, user, direct paths, music, direct stream if not direct path. addon_id = self.addon_id dialog = xbmcgui.Dialog() ##### SERVER INFO ##### log.debug("Initial setup called.") ###$ Begin migration $### if settings('server') == "": current_server = self.user_client.get_server() if current_server is not None: server = self.connectmanager.get_server(current_server) log.info(server) settings('ServerId', value=server['Id']) self.user_client.get_userid() self.user_client.get_token() ###$ End migration $### if settings('server'): current_state = self.connectmanager.get_state() for server in current_state['Servers']: if server['Id'] == settings('serverId'): server_address = self.connectmanager.get_address(server) self._set_server(server_address, server) self._set_user(server['UserId'], server['AccessToken']) break return try: server = self.connectmanager.select_servers() log.info("Server: %s", server) except RuntimeError as error: log.exception(error) xbmc.executebuiltin('Addon.OpenSettings(%s)' % addon_id) return else: server_address = self.connectmanager.get_address(server) self._set_server(server_address, server) if not server.get('AccessToken') and not server.get('UserId'): try: user = self.connectmanager.login(server) log.info("User authenticated: %s", user) except RuntimeError as error: log.exception(error) xbmc.executebuiltin('Addon.OpenSettings(%s)' % addon_id) return settings('username', value=user['User']['Name']) self._set_user(user['User']['Id'], user['AccessToken']) else: user = self.connectmanager.get_state() settings('connectUsername', value=user['ConnectUser']['Name']) self._set_user(server['UserId'], server['AccessToken']) ##### ADDITIONAL PROMPTS ##### direct_paths = dialog.yesno(heading=lang(30511), line1=lang(33035), nolabel=lang(33036), yeslabel=lang(33037)) if direct_paths: log.info("User opted to use direct paths.") settings('useDirectPaths', value="1") # ask for credentials credentials = dialog.yesno(heading=lang(30517), line1=lang(33038)) if credentials: log.info("Presenting network credentials dialog.") passwordsXML() music_disabled = dialog.yesno(heading=lang(29999), line1=lang(33039)) if music_disabled: log.info("User opted to disable Emby music library.") settings('enableMusic', value="false") else: # Only prompt if the user didn't select direct paths for videos if not direct_paths: music_access = dialog.yesno(heading=lang(29999), line1=lang(33040)) if music_access: log.info("User opted to direct stream music.") settings('streamMusic', value="true") @classmethod def _set_server(cls, server_address, server): settings('serverName', value=server['Name']) settings('serverId', value=server['Id']) settings('server', value=server_address) @classmethod def _set_user(cls, user_id, token): settings('userId', value=user_id) settings('token', value=token)