specify output command

KingKDot · Jan 19, 2025 · 6510f32 · 6510f32
1 parent d29b706
commit 6510f32
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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