From d58763504a0a733b42ddff1fb6c1ea725452c488 Mon Sep 17 00:00:00 2001 From: angelblue05 Date: Thu, 20 Oct 2016 19:26:19 -0500 Subject: [PATCH] Fix server detection In the event the connect module encountered an issue --- resources/lib/initialsetup.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/resources/lib/initialsetup.py b/resources/lib/initialsetup.py index 400f6249..354375ab 100644 --- a/resources/lib/initialsetup.py +++ b/resources/lib/initialsetup.py @@ -99,16 +99,17 @@ class InitialSetup(object): # Failed to identify server return True - for server in current_state['Servers']: - if server['Id'] == settings('serverId'): - # Update token - server['UserId'] = settings('userId') or None - server['AccessToken'] = settings('token') or None - self.connectmanager.update_token(current_state['Servers'], server) + elif 'Servers' in current_state: + for server in current_state['Servers']: + if server['Id'] == settings('serverId'): + # Update token + server['UserId'] = settings('userId') or None + server['AccessToken'] = settings('token') or None + self.connectmanager.update_token(current_state['Servers'], server) - server_address = self.connectmanager.get_address(server) - self._set_server(server_address, server) - log.info("Found server!") + server_address = self.connectmanager.get_address(server) + self._set_server(server_address, server) + log.info("Found server!") return True