diff --git a/src/main.py b/src/main.py index 892addd..dd8a020 100644 --- a/src/main.py +++ b/src/main.py @@ -177,6 +177,7 @@ def get_user_file() -> str: if __name__ == "__main__": parse = ArgumentParser() parse.add_argument("-f", "--file", help="File to obfuscate", type=str) + parse.add_argument("-o", "--output", help="Output file location", type=str) parse.add_argument( "-nu", "--no-utf-16-bom", help="No UTF-16 BOM", action="store_true" ) @@ -191,6 +192,7 @@ def get_user_file() -> str: args.file, double_click_check=Settings.double_click_check, utf_16_bom=Settings.utf_16_bom, + output=args.output, ) sys.exit(0) AutoUpdate(__version__) diff --git a/src/util/obfuscation/obfuscate.py b/src/util/obfuscation/obfuscate.py index 7a09c1f..d729282 100644 --- a/src/util/obfuscation/obfuscate.py +++ b/src/util/obfuscation/obfuscate.py @@ -36,8 +36,12 @@ def __init__( file: str, double_click_check: bool = True, utf_16_bom: bool = True, + output: str = None, ) -> None: - self.new_file = f"{file[:-4]}_obf.bat" + if output: + self.new_file = output + else: + self.new_file = f"{file[:-4]}_obf.bat" self.double_click = double_click_check self.utf_16_bom = utf_16_bom