tools/mpremote: Allow user configuration on Windows.

Josverl · Josverl · commit 8dabeb8f11fb · 2023-07-19T22:05:58.000+02:00
Signed-off-by: Jos Verlinde &lt;jos_verlinde@hotmail.com&gt;
diff --git a/docs/reference/mpremote.rst b/docs/reference/mpremote.rst
@@ -430,9 +430,7 @@ Shortcuts can be defined using the macro system.  Built-in shortcuts are:
 Additional shortcuts can be defined by in user-configuration files, which is
 located at ``.config/mpremote/config.py`` relative to the ``XDG_CONFIG_HOME`` or ``HOME`` environment variable on unix systems
 , or on Windows relative to ``HOME``, ``USERPROFILE`` or ``APPDATA``.
-
-For example:
- This file should define a dictionary named ``commands``. The keys of this dictionary are the shortcuts
+This file should define a dictionary named ``commands``. The keys of this dictionary are the shortcuts
 and the values are either a string or a list-of-strings:
 
 .. code-block:: python3
diff --git a/tools/mpremote/mpremote/main.py b/tools/mpremote/mpremote/main.py
@@ -61,7 +61,9 @@ def print_commands_help(cmds, help_key):
     print("See https://docs.micropython.org/en/latest/reference/mpremote.html")
 
     print("\nList of commands:")
-    print_commands_help(_COMMANDS, lambda x: x[1]().description)  # extract description from argparse
+    print_commands_help(
+        _COMMANDS, lambda x: x[1]().description
+    )  # extract description from argparse
 
     print("\nList of shortcuts:")
     print_commands_help(_command_expansions, lambda x: x[2])  # (args, sub, help_message)
@@ -95,7 +97,9 @@ def _bool_flag(cmd_parser, name, short_name, default, description):
 
 def argparse_connect():
     cmd_parser = argparse.ArgumentParser(description="connect to given device")
-    cmd_parser.add_argument("device", nargs=1, help="Either list, auto, id:x, port:x, or any valid device name/path")
+    cmd_parser.add_argument(
+        "device", nargs=1, help="Either list, auto, id:x, port:x, or any valid device name/path"
+    )
     return cmd_parser
 
 
@@ -133,7 +137,9 @@ def argparse_repl():
         required=False,
         help="saves a copy of the REPL session to the specified path",
     )
-    cmd_parser.add_argument("--inject-code", type=str, required=False, help="code to be run when Ctrl-J is pressed")
+    cmd_parser.add_argument(
+        "--inject-code", type=str, required=False, help="code to be run when Ctrl-J is pressed"
+    )
     cmd_parser.add_argument(
         "--inject-file",
         type=str,
@@ -151,14 +157,18 @@ def argparse_eval():
 
 def argparse_exec():
     cmd_parser = argparse.ArgumentParser(description="execute the string")
-    _bool_flag(cmd_parser, "follow", "f", True, "follow output until the expression completes (default)")
+    _bool_flag(
+        cmd_parser, "follow", "f", True, "follow output until the expression completes (default)"
+    )
     cmd_parser.add_argument("expr", nargs=1, help="expression to execute")
     return cmd_parser
 
 
 def argparse_run():
     cmd_parser = argparse.ArgumentParser(description="run the given local script")
-    _bool_flag(cmd_parser, "follow", "f", True, "follow output until the script completes (default)")
+    _bool_flag(
+        cmd_parser, "follow", "f", True, "follow output until the script completes (default)"
+    )
     cmd_parser.add_argument("path", nargs=1, help="path to script to execute")
     return cmd_parser
 
@@ -179,15 +189,21 @@ def argparse_filesystem():
         None,
         "enable verbose output (defaults to True for all commands except cat)",
     )
-    cmd_parser.add_argument("command", nargs=1, help="filesystem command (e.g. cat, cp, ls, rm, touch)")
+    cmd_parser.add_argument(
+        "command", nargs=1, help="filesystem command (e.g. cat, cp, ls, rm, touch)"
+    )
     cmd_parser.add_argument("path", nargs="+", help="local and remote paths")
     return cmd_parser
 
 
 def argparse_mip():
-    cmd_parser = argparse.ArgumentParser(description="install packages from micropython-lib or third-party sources")
+    cmd_parser = argparse.ArgumentParser(
+        description="install packages from micropython-lib or third-party sources"
+    )
     _bool_flag(cmd_parser, "mpy", "m", True, "download as compiled .mpy files (default)")
-    cmd_parser.add_argument("--target", type=str, required=False, help="destination direction on the device")
+    cmd_parser.add_argument(
+        "--target", type=str, required=False, help="destination direction on the device"
+    )
     cmd_parser.add_argument(
         "--index",
         type=str,
@@ -330,10 +346,10 @@ def argparse_none(description):
 for port_num in range(4):
     for prefix, port in [("a", "/dev/ttyACM"), ("u", "/dev/ttyUSB"), ("c", "COM")]:
         if port_num == 0 and port == "COM":
-            continue  # skip COM0 as it does not exist
-        _BUILTIN_COMMAND_EXPANSIONS[f"{prefix}{port_num}"] = {
-            "command": f"connect {port}{port_num}",
-            "help": f'connect to serial port "{port}{port_num}"',
+            continue  # skip COM0 as it does not exist on Windows
+        _BUILTIN_COMMAND_EXPANSIONS["{}{}".format(prefix, port_num)] = {
+            "command": "connect {}{}".format(port, port_num),
+            "help": 'connect to serial port "{}{}"'.format(port, port_num),
         }
 
 
@@ -357,7 +373,6 @@ def load_user_config():
             break
     if not path:
         return config
-
     config_file = os.path.join(path, "config.py")
     # Check if config file exists.
     if not os.path.exists(config_file):
@@ -514,7 +529,9 @@ def main():
             cmd_parser = parser_func()
             cmd_parser.prog = cmd
             # Catch all for unhandled positional arguments (this is the next command).
-            cmd_parser.add_argument("next_command", nargs=argparse.REMAINDER, help=f"Next {_PROG} command")
+            cmd_parser.add_argument(
+                "next_command", nargs=argparse.REMAINDER, help=f"Next {_PROG} command"
+            )
             args = cmd_parser.parse_args(command_args)
 
             # Execute command.
