tools/mpy-tool: Support calling main() from an external script.

vshymanskyy · projectgus · commit 91e8a2f954c8 · 2025-03-04T16:47:27.000+11:00
Signed-off-by: Volodymyr Shymanskyy &lt;vshymanskyi@gmail.com&gt;
Signed-off-by: Angus Gratton &lt;angus@redyak.com.au&gt;
diff --git a/tools/mpy-tool.py b/tools/mpy-tool.py
@@ -1765,7 +1765,7 @@ def copy_section(file, offset, offset2):
             f.write(merged_mpy)
 
 
-def main():
+def main(args=None):
     global global_qstrs
 
     import argparse
@@ -1797,7 +1797,7 @@ def main():
     )
     cmd_parser.add_argument("-o", "--output", default=None, help="output file")
     cmd_parser.add_argument("files", nargs="+", help="input .mpy files")
-    args = cmd_parser.parse_args()
+    args = cmd_parser.parse_args(args)
 
     # set config values relevant to target machine
     config.MICROPY_LONGINT_IMPL = {
