From f915b253a74a2207498192c809448ae74cc7e059 Mon Sep 17 00:00:00 2001 From: Dobin Date: Mon, 19 Feb 2024 19:32:21 +0000 Subject: [PATCH] refactor: project.show_command_output to def:ShowCommandOutput --- defs.py | 1 + helper.py | 2 +- supermega.py | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/defs.py b/defs.py index b675c5c..7b3f08e 100644 --- a/defs.py +++ b/defs.py @@ -7,6 +7,7 @@ class FilePath(str): # with shellcodes/createfile.bin VerifyFilename: FilePath = r'C:\Temp\a' +ShowCommandOutput: bool = False # Correlated with real template files # in plugins/ diff --git a/helper.py b/helper.py index a3b5e56..93d1299 100644 --- a/helper.py +++ b/helper.py @@ -54,7 +54,7 @@ def run_process_checkret(args, check=True): if ret.stderr != None: logger.info(ret.stderr.decode('utf-8')) raise Exception("Command failed: " + " ".join(args)) - if project.show_command_output: + if ShowCommandOutput: logger.info("> " + " ".join(args)) if ret.stdout != None: logger.info(ret.stdout.decode('utf-8')) diff --git a/supermega.py b/supermega.py index ae17531..a79784e 100644 --- a/supermega.py +++ b/supermega.py @@ -40,7 +40,7 @@ def main(): args = parser.parse_args() if args.show: - project.show_command_output = True + ShowCommandOutput = True if args.verify: project.payload_path = "shellcodes/createfile.bin"