Add the abilty to resend code though bluetooth in the same session

pybricksdev/cli/__init__.py

@@ -16,6 +16,7 @@
 
 import argcomplete
 from argcomplete.completers import FilesCompleter
+import simple_term_menu
 
 from pybricksdev import __name__ as MODULE_NAME
 from pybricksdev import __version__ as MODULE_VERSION
@@ -160,6 +161,13 @@ def add_parser(self, subparsers: argparse._SubParsersAction):
             default=True,
         )
 
+        parser.add_argument(
+            "--resend",
+            help="Add a menu option to resend the code with bluetooth instead of disconnecting from the robot after the program ends.",
+            action=argparse.BooleanOptionalAction,
+            default=False,
+        )
+
     async def run(self, args: argparse.Namespace):
 
         # Pick the right connection
@@ -213,11 +221,22 @@ def is_pybricks_usb(dev):
         # Connect to the address and run the script
         await hub.connect()
         try:
-            with _get_script_path(args.file) as script_path:
-                if args.start:
-                    await hub.run(script_path, args.wait)
-                else:
-                    await hub.download(script_path)
+            while True:
+                with _get_script_path(args.file) as script_path:
+                    if args.start:
+                        await hub.run(script_path, args.wait)
+                    else:
+                        await hub.download(script_path)
+
+                if args.conntype == "usb" or not args.wait or not args.resend:
+                    break
+
+                menu = simple_term_menu.TerminalMenu(["Resend Code", "Exit"])
+                entry = menu.show()
+
+                if entry:
+                    break
+
         finally:
             await hub.disconnect()
 

pyproject.toml

@@ -48,6 +48,7 @@ typing-extensions = ">=4.3.0"
 reactivex = {version = ">=4.0.4", python = "<4"}
 hidapi = ">=0.14.0"
 pybricks = {version = ">=3", allow-prereleases = true, python = "<4"}
+simple-term-menu = {version = ">=1.6.6", python = "<4"}
 
 [tool.poetry.group.lint.dependencies]
 black = ">=23,<25"