Fix issues, linting, use argparse, with other py exes

1 files changed, 18 insertions, 8 deletions

diff --git a/nandgame/assembler/disas_main.py b/nandgame/assembler/disas_main.py
index 05ce466..d9d314f 100644
--- a/nandgame/assembler/disas_main.py
+++ b/nandgame/assembler/disas_main.py
@@ -1,24 +1,34 @@
+import argparse
 import sys
 
 import py_nand_ass.disas as dass  # pyright: ignore[reportImplicitRelativeImport]
 
 
 def main():
-    if len(sys.argv) != 2:
-        print(f"Usage: {sys.argv[0]} [filename]")
-        sys.exit(1)
-
-    filename = sys.argv[1]
+    arg_parser = argparse.ArgumentParser(
+        prog="disas_main.py", description="Nandgame disassembler"
+    )
+
+    g = arg_parser.add_mutually_exclusive_group()
+    g.add_argument("--le", action="store_true")
+    g.add_argument("--be", action="store_true")
+    arg_parser.add_argument("filename")
+    args = arg_parser.parse_args()
+    endianness = "little"
+    if args.be:
+        endianness = "big"
+
+    filename = args.filename
     try:
         with open(filename, "rb") as f:
             while True:
                 insb = f.read(2)
                 if not insb:
                     break
-                ins = int.from_bytes(insb, byteorder=ENDIANNESS)
-                if ENDIANNESS in ["big", "be"]:
+                ins = int.from_bytes(insb, byteorder=endianness)
+                if endianness in ["big", "be"]:
                     raw_ins = f"{insb[0]:02x} {insb[1]:02x}"
-                elif ENDIANNESS in ["little", "le"]:
+                elif endianness in ["little", "le"]:
                     raw_ins = f"{insb[1]:02x} {insb[0]:02x}"
                 else:
                     raw_ins = f"?? ??"