diff --git a/wo/cli/plugins/secure.py b/wo/cli/plugins/secure.py index fe50e28..e71ef5e 100644 --- a/wo/cli/plugins/secure.py +++ b/wo/cli/plugins/secure.py @@ -148,20 +148,16 @@ class WOSecureController(CementBaseController): @expose(hide=True) def secure_ssh(self): """Harden ssh security""" - pargs = self.app.pargs - if pargs.user_input: - current_ssh_port = pargs.user_input - else: - if os.path.isfile('/etc/ssh/sshd_config'): - for line in open('/etc/ssh/sshd_config', encoding='utf-8'): - if 'Port' in line: - ssh_line = line.strip() + if os.path.isfile('/etc/ssh/sshd_config'): + for line in open('/etc/ssh/sshd_config', encoding='utf-8'): + if 'Port' in line: + ssh_line = line.strip() break - else: - port = (ssh_line).split(' ') - current_ssh_port = port[1] else: - Log.error(self, "SSH config file not found") + port = (ssh_line).split(' ') + current_ssh_port = port[1] + else: + Log.error(self, "SSH config file not found") if not current_ssh_port: current_ssh_port = '22' data = dict(sshport=current_ssh_port, allowpass='no')