From 1ef290b2890966e698f6d31a0bf25a30a0ddc493 Mon Sep 17 00:00:00 2001 From: Randall Winkhart Date: Sat, 13 May 2023 19:18:00 -0400 Subject: [PATCH] Set success_flag when running the new tweak menu Former-commit-id: b799bdc3b81c7e38cfad2fab46042bf25de7c157 Former-commit-id: a3e9b01e5b3c4f5f8bfb70ee94b26ea6d82cb031 --- lib/sshyp.py | 2 ++ lib/stweak.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/lib/sshyp.py b/lib/sshyp.py index 301fac5..bae1d0d 100755 --- a/lib/sshyp.py +++ b/lib/sshyp.py @@ -802,6 +802,7 @@ if __name__ == "__main__": success_flag = True read_shortcut() elif arguments[0] == 'tweak': + success_flag = True from stweak import global_menu global_menu(False) @@ -819,6 +820,7 @@ if __name__ == "__main__": success_flag = True whitelist_setup() elif arg_count == 1 and arguments[0] == 'tweak': + success_flag = True from stweak import global_menu global_menu(False) elif arg_count > 2 and arguments[1] in ('add', 'del'): diff --git a/lib/stweak.py b/lib/stweak.py index 00a1e36..abdd735 100644 --- a/lib/stweak.py +++ b/lib/stweak.py @@ -88,7 +88,7 @@ def install_type(): _dev_type = 'server' Path(f"{home}/.config/sshyp/deleted").mkdir(mode=0o700, exist_ok=True) Path(f"{home}/.config/sshyp/whitelist").mkdir(mode=0o700, exist_ok=True) - curses_terminate('\nmake sure the ssh service is running and properly configured') + curses_terminate('\nmake sure the ssh service is running and properly configured\n') else: _dev_type = 'client' if _install_type == 2: