def step_check_install_code(self, ui): if ui.get_key() is not None and ui.get_key() == 'tryAgain': if ui.get_variable().get() == 'configureLater': return self.step_menu_main(ui) elif ui.get_variable().get() == 'configProxy': ui.set_key("installCode") return self.step_configure_proxy_type(ui) elif ui.get_variable().get() == 'reEnterCode': return self.step_menu_agent_install_key_selected(ui) msg = resources.get_message('checkInstallCode') self._uinterface.wait_message(msg) key = self._install_code.get() try: self.install_key(key) msg = user_interface.Message( resources.get_message('configureKeyInstalled')) msg.next_step(self.step_menu_main) return msg except Exception as e: s = str(e) if s == "INVALID_CODE": chs = user_interface.Chooser() chs.set_key("tryAgain") chs.set_message(resources.get_message('errorInvalidCode')) chs.add("reEnterCode", resources.get_message('reEnterCode')) chs.add("configureLater", resources.get_message('configureLater')) chs.set_variable(user_interface.VarString("reEnterCode")) chs.prev_step(self.step_menu_agent_install_key_selected) chs.next_step(self.step_check_install_code) return chs elif s == "CONNECT_ERROR": chs = user_interface.Chooser() chs.set_key("tryAgain") chs.set_message( resources.get_message('errorConnectionQuestion')) chs.add("configProxy", resources.get_message('yes')) chs.add("noTryAgain", resources.get_message('noTryAgain')) chs.add("configureLater", resources.get_message('configureLater')) chs.set_variable(user_interface.VarString("noTryAgain")) chs.prev_step(self.step_menu_agent_install_key_selected) chs.next_step(self.step_check_install_code) return chs elif s == "REQUEST_TIMEOUT": return user_interface.ErrorDialog( resources.get_message('configureErrorConnection')) else: return user_interface.ErrorDialog(s)
def step_config_password(self, ui): if ui.get_variable().get() == 'configureSetPassword': self._change_pwd = user_interface.VarString("", True) self._change_repwd = user_interface.VarString("", True) ipt = user_interface.Inputs() ipt.set_key("set_password") ipt.set_message(resources.get_message('configurePassword')) ipt.add('password', resources.get_message('password'), self._change_pwd, True) ipt.add('rePassword', resources.get_message('rePassword'), self._change_repwd, True) ipt.prev_step(self.step_menu_password) ipt.next_step(self.step_config_password_procede) return ipt elif ui.get_variable().get() == 'configureRemovePassword': chs = user_interface.Chooser() chs.set_key("remove_password") chs.set_message( resources.get_message('configureRemovePasswordQuestion')) chs.add("yes", resources.get_message('yes')) chs.add("no", resources.get_message('no')) chs.set_variable(user_interface.VarString("no")) chs.set_accept_key("yes") chs.prev_step(self.step_menu_password) chs.next_step(self.step_config_password_procede) return chs
def step_menu_agent(self, ui): try: self._install_code.set("") key = self.get_config("key") if key == "": return self.step_menu_agent_install_key_selected(ui) else: chs = user_interface.Chooser() chs.set_message( resources.get_message('configureChooseOperation')) chs.add("configureChangeInstallKey", resources.get_message('configureChangeInstallKey')) if self.is_agent_enable(): chs.add("configureDisableAgent", resources.get_message('configureDisableAgent')) else: chs.add("configureEnableAgent", resources.get_message('configureEnableAgent')) chs.set_variable(self._agent_menu_sel) chs.prev_step(self.step_menu_main) chs.next_step(self.step_menu_agent_selected) return chs except: return user_interface.ErrorDialog( resources.get_message('configureErrorConnection'))
def step_menu_monitor(self, ui): chs = user_interface.Chooser() chs.set_message(resources.get_message('configureChooseOperation')) chs.add("configureTrayIconVisibility", resources.get_message('configureTrayIconVisibility')) chs.set_variable( user_interface.VarString("configureTrayIconVisibility")) chs.prev_step(self.step_menu_main) chs.next_step(self.step_menu_monitor_selected) return chs
def step_menu_agent_disable(self, ui): chs = user_interface.Chooser() chs.set_message(resources.get_message('configureDisableAgentQuestion')) chs.add("yes", resources.get_message('yes')) chs.add("no", resources.get_message('no')) chs.set_variable(user_interface.VarString("no")) chs.set_accept_key("yes") chs.prev_step(self.step_menu_agent) chs.next_step(self.step_menu_agent_disable_procede) return chs
def step_menu_agent_install_key(self, ui): chs = user_interface.Chooser() chs.set_message(resources.get_message('configureUninstallKeyQuestion')) chs.add("yes", resources.get_message('yes')) chs.add("no", resources.get_message('no')) chs.set_variable(user_interface.VarString("no")) chs.set_accept_key("yes") chs.prev_step(self.step_menu_agent) chs.next_step(self.step_menu_agent_remove_key_selected) return chs
def step_menu_password(self, ui): chs = user_interface.Chooser() chs.set_message(resources.get_message('configureChooseOperation')) chs.add("configureSetPassword", resources.get_message('configureSetPassword')) chs.add("configureRemovePassword", resources.get_message('configureRemovePassword')) chs.set_variable(self._password_menu_sel) chs.prev_step(self.step_menu_main) chs.next_step(self.step_config_password) return chs
def step_configure_proxy_set(self, ui): ar = ui.get_key().split('_') ui.set_key(ar[0]) if len(ar) == 2 and ar[1] == 'tryAgain': if ui.get_variable() is not None and ui.get_variable().get( ) == 'configureLater': if ui.get_key() == "menuProxy": return self.step_menu_main(ui) elif ui.get_key() == "installCode": return self.step_menu_agent_install_key_selected(ui) try: if self._proxy_type.get() == 'HTTP' or self._proxy_type.get( ) == 'SOCKS4' or self._proxy_type.get( ) == 'SOCKS4A' or self._proxy_type.get() == 'SOCKS5': try: int(self._proxy_port.get()) except: return user_interface.ErrorDialog( resources.get_message("validInteger").format( resources.get_message('proxyPort'))) sret = self.invoke_req( "set_proxy", { 'type': self._proxy_type.get(), 'host': self._proxy_host.get(), 'port': self._proxy_port.get(), 'user': self._proxy_user.get(), 'password': self._proxy_password.get() }) if sret != "OK": raise Exception(sret[6:]) except: chs = user_interface.Chooser() chs.set_key(ui.get_key() + "_tryAgain") chs.set_message(resources.get_message('errorConnectionConfig')) chs.add("noTryAgain", resources.get_message('noTryAgain')) chs.add("configureLater", resources.get_message('configureLater')) chs.set_variable(user_interface.VarString("noTryAgain")) if ui.get_key() == "menuProxy": chs.prev_step(self.step_menu_main) elif ui.get_key() == "installCode": chs.prev_step(self.step_menu_agent_install_key_selected) chs.next_step(self.step_configure_proxy_set) return chs if ui.get_key() == "menuProxy": msg = user_interface.Message( resources.get_message('configureProxyEnd')) msg.next_step(self.step_menu_main) return msg elif ui.get_key() == "installCode": return self.step_check_install_code(ui)
def step_menu_main(self, ui): try: self.read_proxy_info() except: return user_interface.ErrorDialog( resources.get_message('configureErrorConnection')) chs = user_interface.Chooser() chs.set_message(resources.get_message('configureChooseOperation')) chs.add("configureAgent", resources.get_message('configureAgent')) chs.add("configureProxy", resources.get_message('configureProxy')) #chs.add("configureMonitor", resources.get_message('configureMonitor')) chs.add("configurePassword", resources.get_message('configurePassword')) chs.add("configureExit", resources.get_message('configureExit')) chs.set_variable(self._main_menu_sel) chs.next_step(self.step_menu_main_selected) return chs
def step_menu_monitor_selected(self, ui): try: chs = user_interface.Chooser() chs.set_message( resources.get_message( 'configureChooseMonitorTrayIconVisibility')) chs.add("yes", resources.get_message('yes')) chs.add("no", resources.get_message('no')) if self.get_config("monitor_tray_icon") == "True": chs.set_variable(user_interface.VarString("yes")) else: chs.set_variable(user_interface.VarString("no")) chs.prev_step(self.step_menu_monitor) chs.next_step(self.step_menu_monitor_procede) return chs except: return user_interface.ErrorDialog( resources.get_message('configureErrorConnection'))
def step_configure_proxy_type(self, ui): chs = user_interface.Chooser() chs.set_key(ui.get_key()) chs.set_message(resources.get_message('chooseProxyType')) chs.add("SYSTEM", resources.get_message('proxySystem')) chs.add("HTTP", resources.get_message('proxyHttp')) chs.add("SOCKS4", resources.get_message('proxySocks4')) chs.add("SOCKS4A", resources.get_message('proxySocks4a')) chs.add("SOCKS45", resources.get_message('proxySocks5')) chs.add("NONE", resources.get_message('proxyNone')) chs.set_variable(self._proxy_type) if ui.get_key() == "menuProxy": chs.prev_step(self.step_menu_main) elif ui.get_key() == "installCode": chs.prev_step(self.step_menu_agent_install_key_selected) chs.next_step(self.step_configure_proxy_info) return chs