fix: stop exit 0 when being called programmatically

Author: hfudev

espefuse/__init__.py

@@ -214,7 +214,11 @@ def main(argv: list[str] | None = None, esp: esptool.ESPLoader | None = None):
     returned by esptool.get_default_connected_device()
     """
     args = esptool.expand_file_arguments(argv or sys.argv[1:])
-    cli(args=args, esp=esp)
+    try:
+        cli(args=args, esp=esp)
+    except SystemExit as e:
+        if e.code != 0:
+            raise
 
 
 def _main():

espsecure/__init__.py

@@ -1957,7 +1957,11 @@ def main(argv: list[str] | None = None):
     Arguments and their values need to be added as individual items to the list
     e.g. "--port /dev/ttyUSB1" thus becomes ['--port', '/dev/ttyUSB1'].
     """
-    cli(args=argv)
+    try:
+        cli(args=argv)
+    except SystemExit as e:
+        if e.code != 0:
+            raise
 
 
 def _main():

esptool/__init__.py

@@ -1028,7 +1028,11 @@ def main(argv: list[str] | None = None, esp: ESPLoader | None = None):
     returned by get_default_connected_device()
     """
     args = expand_file_arguments(argv or sys.argv[1:])
-    cli(args=args, esp=esp)
+    try:
+        cli(args=args, esp=esp)
+    except SystemExit as e:
+        if e.code != 0:
+            raise
 
 
 def get_port_list(