From e625e9eecc6e37df470e5acb11543f361446fe2e Mon Sep 17 00:00:00 2001 From: jm33-m0 Date: Sat, 6 Jun 2020 09:39:06 +0800 Subject: [PATCH] fix #42 --- lib/cli/cmd.py | 4 ++-- lib/cli/console.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/lib/cli/cmd.py b/lib/cli/cmd.py index 6fce12a..f80f658 100644 --- a/lib/cli/cmd.py +++ b/lib/cli/cmd.py @@ -237,7 +237,7 @@ def run_target(**kwargs): return if target not in os.listdir(session.init_dir + '/data'): - console.print_error("[-] Target list file not found") + console.print_error(f"[-] Target list file '{target}' not found") return colors.colored_print( @@ -508,7 +508,7 @@ def cmd_handler(session, user_cmd): # parse user_cmd try: - split_cmd = str(user_cmd).lower().strip().split() + split_cmd = str(user_cmd).strip().split() user_cmd = split_cmd[0] args = split_cmd[1:] except IndexError: diff --git a/lib/cli/console.py b/lib/cli/console.py index ccc57b6..e0cc792 100644 --- a/lib/cli/console.py +++ b/lib/cli/console.py @@ -40,7 +40,7 @@ def print_banner(ver, exp_cnt): version: {ver} {exp_cnt} exploits -''' + colors.END + colors.GREEN + f''' +''' + colors.END + colors.GREEN + ''' by jm33_m0 /~https://github.com/jm33-m0/mec @@ -147,7 +147,7 @@ def input_check(prompt_info, allow_blank=True, message=input_ps, complete_in_thread=True, complete_while_typing=True, - completer=choice_completer).strip().lower() + completer=choice_completer).strip() if not allow_blank and user_input == '': continue