From 0662b83e814c41482acc65ccd0fdbf3320edfbed Mon Sep 17 00:00:00 2001 From: Richard Neumann Date: Fri, 25 Dec 2020 23:40:20 +0100 Subject: [PATCH] Refac. --- rcon/rconshell.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/rcon/rconshell.py b/rcon/rconshell.py index 793e503..e191f29 100644 --- a/rcon/rconshell.py +++ b/rcon/rconshell.py @@ -36,7 +36,7 @@ def get_args() -> Namespace: return parser.parse_args() -def run() -> int: +def main() -> None: """Runs the RCON shell.""" args = get_args() @@ -48,13 +48,7 @@ def run() -> int: host = port = passwd = None with ErrorHandler(ERRORS, LOGGER): - return rconcmd(host, port, passwd, prompt=args.prompt) - - -def main(): - """Wraps the run function.""" - - with CommandHistory(LOGGER): - returncode = run() + with CommandHistory(LOGGER): + returncode = rconcmd(host, port, passwd, prompt=args.prompt) exit(returncode)