micropython-stubber 1.20.5__py3-none-any.whl → 1.20.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (152) hide show
  1. {micropython_stubber-1.20.5.dist-info → micropython_stubber-1.20.6.dist-info}/LICENSE +30 -30
  2. {micropython_stubber-1.20.5.dist-info → micropython_stubber-1.20.6.dist-info}/METADATA +1 -1
  3. micropython_stubber-1.20.6.dist-info/RECORD +159 -0
  4. mpflash/README.md +184 -184
  5. mpflash/libusb_flash.ipynb +203 -203
  6. mpflash/mpflash/add_firmware.py +98 -98
  7. mpflash/mpflash/ask_input.py +236 -236
  8. mpflash/mpflash/bootloader/__init__.py +37 -36
  9. mpflash/mpflash/bootloader/manual.py +102 -102
  10. mpflash/mpflash/bootloader/micropython.py +10 -10
  11. mpflash/mpflash/bootloader/touch1200.py +45 -45
  12. mpflash/mpflash/cli_download.py +129 -129
  13. mpflash/mpflash/cli_flash.py +219 -219
  14. mpflash/mpflash/cli_group.py +98 -98
  15. mpflash/mpflash/cli_list.py +81 -81
  16. mpflash/mpflash/cli_main.py +41 -41
  17. mpflash/mpflash/common.py +164 -164
  18. mpflash/mpflash/config.py +47 -47
  19. mpflash/mpflash/connected.py +74 -74
  20. mpflash/mpflash/download.py +360 -360
  21. mpflash/mpflash/downloaded.py +129 -129
  22. mpflash/mpflash/errors.py +9 -9
  23. mpflash/mpflash/flash.py +52 -52
  24. mpflash/mpflash/flash_esp.py +59 -59
  25. mpflash/mpflash/flash_stm32.py +24 -24
  26. mpflash/mpflash/flash_stm32_cube.py +111 -111
  27. mpflash/mpflash/flash_stm32_dfu.py +101 -101
  28. mpflash/mpflash/flash_uf2.py +67 -67
  29. mpflash/mpflash/flash_uf2_boardid.py +15 -15
  30. mpflash/mpflash/flash_uf2_linux.py +123 -123
  31. mpflash/mpflash/flash_uf2_macos.py +34 -34
  32. mpflash/mpflash/flash_uf2_windows.py +34 -34
  33. mpflash/mpflash/list.py +89 -89
  34. mpflash/mpflash/logger.py +41 -41
  35. mpflash/mpflash/mpboard_id/__init__.py +93 -93
  36. mpflash/mpflash/mpboard_id/add_boards.py +255 -255
  37. mpflash/mpflash/mpboard_id/board.py +37 -37
  38. mpflash/mpflash/mpboard_id/board_id.py +86 -86
  39. mpflash/mpflash/mpboard_id/store.py +43 -43
  40. mpflash/mpflash/mpremoteboard/__init__.py +221 -221
  41. mpflash/mpflash/mpremoteboard/mpy_fw_info.py +141 -141
  42. mpflash/mpflash/mpremoteboard/runner.py +140 -140
  43. mpflash/mpflash/uf2disk.py +12 -12
  44. mpflash/mpflash/vendor/basicgit.py +288 -288
  45. mpflash/mpflash/vendor/click_aliases.py +91 -91
  46. mpflash/mpflash/vendor/dfu.py +165 -165
  47. mpflash/mpflash/vendor/pydfu.py +605 -605
  48. mpflash/mpflash/vendor/readme.md +2 -2
  49. mpflash/mpflash/vendor/versions.py +119 -117
  50. mpflash/mpflash/worklist.py +170 -170
  51. mpflash/poetry.lock +1588 -1588
  52. mpflash/pyproject.toml +60 -60
  53. mpflash/stm32_udev_rules.md +62 -62
  54. stubber/__init__.py +3 -3
  55. stubber/basicgit.py +294 -288
  56. stubber/board/board_info.csv +193 -193
  57. stubber/board/boot.py +34 -34
  58. stubber/board/createstubs.py +986 -986
  59. stubber/board/createstubs_db.py +825 -825
  60. stubber/board/createstubs_db_min.py +331 -331
  61. stubber/board/createstubs_db_mpy.mpy +0 -0
  62. stubber/board/createstubs_lvgl.py +741 -741
  63. stubber/board/createstubs_lvgl_min.py +741 -741
  64. stubber/board/createstubs_mem.py +766 -766
  65. stubber/board/createstubs_mem_min.py +306 -306
  66. stubber/board/createstubs_mem_mpy.mpy +0 -0
  67. stubber/board/createstubs_min.py +294 -294
  68. stubber/board/createstubs_mpy.mpy +0 -0
  69. stubber/board/fw_info.py +141 -141
  70. stubber/board/info.py +183 -183
  71. stubber/board/main.py +19 -19
  72. stubber/board/modulelist.txt +247 -247
  73. stubber/board/pyrightconfig.json +34 -34
  74. stubber/bulk/mcu_stubber.py +454 -454
  75. stubber/codemod/_partials/__init__.py +48 -48
  76. stubber/codemod/_partials/db_main.py +147 -147
  77. stubber/codemod/_partials/lvgl_main.py +77 -77
  78. stubber/codemod/_partials/modules_reader.py +80 -80
  79. stubber/codemod/add_comment.py +53 -53
  80. stubber/codemod/add_method.py +65 -65
  81. stubber/codemod/board.py +317 -317
  82. stubber/codemod/enrich.py +145 -145
  83. stubber/codemod/merge_docstub.py +284 -284
  84. stubber/codemod/modify_list.py +54 -54
  85. stubber/codemod/utils.py +57 -57
  86. stubber/commands/build_cmd.py +94 -94
  87. stubber/commands/cli.py +51 -51
  88. stubber/commands/clone_cmd.py +66 -66
  89. stubber/commands/config_cmd.py +29 -29
  90. stubber/commands/enrich_folder_cmd.py +70 -70
  91. stubber/commands/get_core_cmd.py +69 -69
  92. stubber/commands/get_docstubs_cmd.py +87 -87
  93. stubber/commands/get_frozen_cmd.py +112 -112
  94. stubber/commands/get_mcu_cmd.py +56 -56
  95. stubber/commands/merge_cmd.py +66 -66
  96. stubber/commands/publish_cmd.py +119 -119
  97. stubber/commands/stub_cmd.py +30 -30
  98. stubber/commands/switch_cmd.py +54 -54
  99. stubber/commands/variants_cmd.py +48 -48
  100. stubber/cst_transformer.py +178 -178
  101. stubber/data/board_info.csv +193 -193
  102. stubber/data/board_info.json +1729 -1729
  103. stubber/data/micropython_tags.csv +15 -15
  104. stubber/data/requirements-core-micropython.txt +38 -38
  105. stubber/data/requirements-core-pycopy.txt +39 -39
  106. stubber/downloader.py +36 -36
  107. stubber/freeze/common.py +68 -68
  108. stubber/freeze/freeze_folder.py +69 -69
  109. stubber/freeze/freeze_manifest_2.py +113 -113
  110. stubber/freeze/get_frozen.py +127 -127
  111. stubber/get_cpython.py +101 -101
  112. stubber/get_lobo.py +59 -59
  113. stubber/minify.py +418 -418
  114. stubber/publish/bump.py +86 -86
  115. stubber/publish/candidates.py +262 -262
  116. stubber/publish/database.py +18 -18
  117. stubber/publish/defaults.py +45 -45
  118. stubber/publish/enums.py +24 -24
  119. stubber/publish/helpers.py +29 -29
  120. stubber/publish/merge_docstubs.py +130 -130
  121. stubber/publish/missing_class_methods.py +49 -49
  122. stubber/publish/package.py +146 -146
  123. stubber/publish/pathnames.py +51 -51
  124. stubber/publish/publish.py +120 -120
  125. stubber/publish/pypi.py +38 -38
  126. stubber/publish/stubpackage.py +1029 -1029
  127. stubber/rst/__init__.py +9 -9
  128. stubber/rst/classsort.py +77 -77
  129. stubber/rst/lookup.py +530 -530
  130. stubber/rst/output_dict.py +401 -401
  131. stubber/rst/reader.py +822 -822
  132. stubber/rst/report_return.py +69 -69
  133. stubber/rst/rst_utils.py +540 -540
  134. stubber/stubber.py +38 -38
  135. stubber/stubs_from_docs.py +90 -90
  136. stubber/tools/manifestfile.py +610 -610
  137. stubber/tools/readme.md +5 -5
  138. stubber/update_fallback.py +117 -117
  139. stubber/update_module_list.py +123 -123
  140. stubber/utils/__init__.py +5 -5
  141. stubber/utils/config.py +127 -127
  142. stubber/utils/makeversionhdr.py +54 -54
  143. stubber/utils/manifest.py +92 -92
  144. stubber/utils/post.py +79 -79
  145. stubber/utils/repos.py +157 -154
  146. stubber/utils/stubmaker.py +139 -139
  147. stubber/utils/typed_config_toml.py +77 -77
  148. stubber/utils/versions.py +128 -120
  149. stubber/variants.py +106 -106
  150. micropython_stubber-1.20.5.dist-info/RECORD +0 -159
  151. {micropython_stubber-1.20.5.dist-info → micropython_stubber-1.20.6.dist-info}/WHEEL +0 -0
  152. {micropython_stubber-1.20.5.dist-info → micropython_stubber-1.20.6.dist-info}/entry_points.txt +0 -0
@@ -1,102 +1,102 @@
1
- """Manual bootloader mode entry for various MCUs."""
2
-
3
- from click.exceptions import Abort
4
- from rich.console import Console
5
- from rich.highlighter import RegexHighlighter
6
- from rich.panel import Panel
7
- from rich.prompt import Confirm
8
- from rich.theme import Theme
9
-
10
- # from mpflash.logger import console, log
11
- from mpflash.mpremoteboard import MPRemoteBoard
12
-
13
-
14
- class MCUHighlighter(RegexHighlighter):
15
- """Apply style to anything that looks like an email."""
16
-
17
- base_style = "mcu."
18
- highlights = [
19
- r"(?P<bold>Method[\s\d\:]*)",
20
- r"(?P<bold> \d.)", # numbered items
21
- r"(?P<bold> - )", # bullets
22
- r"(?P<pad>GPIO[\d]*)",
23
- r"(?P<pad>GPI[\d]*)",
24
- r"(?P<pad>IO[\d]*)",
25
- r"(?P<pad>RUN)",
26
- r"(?P<pad>GND)",
27
- r"(?P<pad>VCC)",
28
- r"(?P<pad>3.3V)",
29
- r"(?P<pad>5V)",
30
- # buttons
31
- r"(?P<button>BOOTSEL)",
32
- r"(?P<button>RESET)",
33
- r"(?P<button>reset)",
34
- # other
35
- r"(?P<cable>USB)",
36
- # r"(?P<mcu>SAMD[\d]*)",
37
- # r"(?P<mcu>ESP[\d]*)",
38
- # r"(?P<mcu>rp2)",
39
- # r"(?P<mcu>rp2040)",
40
- ]
41
-
42
-
43
- # https://rich.readthedocs.io/en/stable/appendix/colors.html?highlight=colors#standard-colors
44
- mcu_theme = Theme(
45
- {
46
- "mcu.bold": "orange3",
47
- "mcu.mcu": "orange3",
48
- "mcu.button": "bold green",
49
- "mcu.pad": "dodger_blue2",
50
- "mcu.cable": "dodger_blue2",
51
- }
52
- )
53
-
54
-
55
- def enter_bootloader_manual(mcu: MPRemoteBoard, timeout: int = 10):
56
-
57
- message: str
58
- if mcu.port == "rp2":
59
- message = f"""\
60
- Please put your {" ".join([mcu.port,mcu.board])} device into bootloader mode by either:
61
- Method 1:
62
- 1. Unplug the USB cable,
63
- 2. Press and hold the BOOTSEL button on the device,
64
- 3. Plug the USB cable back in.
65
- 4. Release the BOOTSEL button.
66
-
67
- Method 2:
68
- 1. Press and hold the BOOTSEL button on the device,
69
- 2. Reset the device by either:
70
- - pressing the RESET button on the device
71
- - by power-cycling the device,
72
- - by briefly connecting the RUN pin to GND
73
- 3. Release the BOOTSEL button.
74
- """
75
- elif mcu.port == "samd":
76
- message = f"""\
77
- Please put your {mcu.port.upper()} device into bootloader mode by:
78
- - Pressing or sliding the RESET button twice in fast succession
79
- """
80
- else:
81
- message = f"""\
82
- Please put your {mcu.port.upper()} device into bootloader mode by:
83
- - Pressing the RESET button on the device
84
- """
85
-
86
- # todo: would be nice to re-use the console instance from logger
87
- console = Console(highlighter=MCUHighlighter(), theme=mcu_theme)
88
- message += "\nIf you are unsure how to enter bootloader mode, please refer to the device documentation."
89
- console.print(
90
- Panel(
91
- message,
92
- highlight=True,
93
- title="Manual Bootloader",
94
- title_align="left",
95
- expand=False,
96
- )
97
- )
98
- try:
99
- answer = Confirm.ask("Press Enter to continue", default="y")
100
- except Abort:
101
- return False
102
- return answer in ["y", "Y", True]
1
+ """Manual bootloader mode entry for various MCUs."""
2
+
3
+ from click.exceptions import Abort
4
+ from rich.console import Console
5
+ from rich.highlighter import RegexHighlighter
6
+ from rich.panel import Panel
7
+ from rich.prompt import Confirm
8
+ from rich.theme import Theme
9
+
10
+ # from mpflash.logger import console, log
11
+ from mpflash.mpremoteboard import MPRemoteBoard
12
+
13
+
14
+ class MCUHighlighter(RegexHighlighter):
15
+ """Apply style to anything that looks like an email."""
16
+
17
+ base_style = "mcu."
18
+ highlights = [
19
+ r"(?P<bold>Method[\s\d\:]*)",
20
+ r"(?P<bold> \d.)", # numbered items
21
+ r"(?P<bold> - )", # bullets
22
+ r"(?P<pad>GPIO[\d]*)",
23
+ r"(?P<pad>GPI[\d]*)",
24
+ r"(?P<pad>IO[\d]*)",
25
+ r"(?P<pad>RUN)",
26
+ r"(?P<pad>GND)",
27
+ r"(?P<pad>VCC)",
28
+ r"(?P<pad>3.3V)",
29
+ r"(?P<pad>5V)",
30
+ # buttons
31
+ r"(?P<button>BOOTSEL)",
32
+ r"(?P<button>RESET)",
33
+ r"(?P<button>reset)",
34
+ # other
35
+ r"(?P<cable>USB)",
36
+ # r"(?P<mcu>SAMD[\d]*)",
37
+ # r"(?P<mcu>ESP[\d]*)",
38
+ # r"(?P<mcu>rp2)",
39
+ # r"(?P<mcu>rp2040)",
40
+ ]
41
+
42
+
43
+ # https://rich.readthedocs.io/en/stable/appendix/colors.html?highlight=colors#standard-colors
44
+ mcu_theme = Theme(
45
+ {
46
+ "mcu.bold": "orange3",
47
+ "mcu.mcu": "orange3",
48
+ "mcu.button": "bold green",
49
+ "mcu.pad": "dodger_blue2",
50
+ "mcu.cable": "dodger_blue2",
51
+ }
52
+ )
53
+
54
+
55
+ def enter_bootloader_manual(mcu: MPRemoteBoard, timeout: int = 10):
56
+
57
+ message: str
58
+ if mcu.port == "rp2":
59
+ message = f"""\
60
+ Please put your {" ".join([mcu.port,mcu.board])} device into bootloader mode by either:
61
+ Method 1:
62
+ 1. Unplug the USB cable,
63
+ 2. Press and hold the BOOTSEL button on the device,
64
+ 3. Plug the USB cable back in.
65
+ 4. Release the BOOTSEL button.
66
+
67
+ Method 2:
68
+ 1. Press and hold the BOOTSEL button on the device,
69
+ 2. Reset the device by either:
70
+ - pressing the RESET button on the device
71
+ - by power-cycling the device,
72
+ - by briefly connecting the RUN pin to GND
73
+ 3. Release the BOOTSEL button.
74
+ """
75
+ elif mcu.port == "samd":
76
+ message = f"""\
77
+ Please put your {mcu.port.upper()} device into bootloader mode by:
78
+ - Pressing or sliding the RESET button twice in fast succession
79
+ """
80
+ else:
81
+ message = f"""\
82
+ Please put your {mcu.port.upper()} device into bootloader mode by:
83
+ - Pressing the RESET button on the device
84
+ """
85
+
86
+ # todo: would be nice to re-use the console instance from logger
87
+ console = Console(highlighter=MCUHighlighter(), theme=mcu_theme)
88
+ message += "\nIf you are unsure how to enter bootloader mode, please refer to the device documentation."
89
+ console.print(
90
+ Panel(
91
+ message,
92
+ highlight=True,
93
+ title="Manual Bootloader",
94
+ title_align="left",
95
+ expand=False,
96
+ )
97
+ )
98
+ try:
99
+ answer = Confirm.ask("Press Enter to continue", default="y")
100
+ except Abort:
101
+ return False
102
+ return answer in ["y", "Y", True]
@@ -1,10 +1,10 @@
1
- """Module for handling the bootloader mode for micropython boards"""
2
-
3
- from mpflash.mpremoteboard import MPRemoteBoard
4
-
5
-
6
- def enter_bootloader_mpy(mcu: MPRemoteBoard, timeout: int = 10):
7
- """Enter the bootloader mode for the board using mpremote and micropython on the board"""
8
- mcu.run_command("bootloader", timeout=timeout)
9
- # todo: check if mpremote command was successful
10
- return True
1
+ """Module for handling the bootloader mode for micropython boards"""
2
+
3
+ from mpflash.mpremoteboard import MPRemoteBoard
4
+
5
+
6
+ def enter_bootloader_mpy(mcu: MPRemoteBoard, timeout: int = 10):
7
+ """Enter the bootloader mode for the board using mpremote and micropython on the board"""
8
+ mcu.run_command("bootloader", timeout=timeout)
9
+ # todo: check if mpremote command was successful
10
+ return True
@@ -1,45 +1,45 @@
1
- import sys
2
- import time
3
-
4
- import serial
5
-
6
- from mpflash.errors import MPFlashError
7
- from mpflash.logger import log
8
- from mpflash.mpremoteboard import MPRemoteBoard
9
-
10
- from .manual import enter_bootloader_manual
11
-
12
-
13
- def enter_bootloader_cdc_1200bps(mcu: MPRemoteBoard, timeout: int = 10):
14
- if sys.platform == "win32":
15
- log.warning("Touch 1200bps method is currently not supported on Windows, switching to manual")
16
- return enter_bootloader_manual(mcu, timeout=timeout)
17
-
18
- log.info(f"Entering bootloader on {mcu.board} on {mcu.serialport} using CDC 1200bps")
19
- # if port argument is present perform soft reset
20
- if not mcu.serialport:
21
- raise MPFlashError("No serial port specified")
22
- # try to initiate serial port connection on PORT with 1200 baudrate
23
- try:
24
- with serial.Serial(
25
- port=mcu.serialport,
26
- baudrate=1200,
27
- parity=serial.PARITY_NONE,
28
- stopbits=serial.STOPBITS_ONE,
29
- bytesize=serial.EIGHTBITS,
30
- rtscts=True,
31
- ) as connection:
32
- print("Connection established")
33
- connection.rts = True
34
- connection.dtr = False
35
- time.sleep(0.4)
36
-
37
- except serial.SerialException as e:
38
- log.exception(e)
39
- raise MPFlashError("pySerial error: " + str(e) + "\n") from e
40
- except Exception as e:
41
- log.exception(e)
42
- raise MPFlashError("Error: " + str(e) + "\n") from e
43
-
44
- # be optimistic
45
- return True
1
+ import sys
2
+ import time
3
+
4
+ import serial
5
+
6
+ from mpflash.errors import MPFlashError
7
+ from mpflash.logger import log
8
+ from mpflash.mpremoteboard import MPRemoteBoard
9
+
10
+ from .manual import enter_bootloader_manual
11
+
12
+
13
+ def enter_bootloader_cdc_1200bps(mcu: MPRemoteBoard, timeout: int = 10):
14
+ if sys.platform == "win32":
15
+ log.warning("Touch 1200bps method is currently not supported on Windows, switching to manual")
16
+ return enter_bootloader_manual(mcu, timeout=timeout)
17
+
18
+ log.info(f"Entering bootloader on {mcu.board} on {mcu.serialport} using CDC 1200bps")
19
+ # if port argument is present perform soft reset
20
+ if not mcu.serialport:
21
+ raise MPFlashError("No serial port specified")
22
+ # try to initiate serial port connection on PORT with 1200 baudrate
23
+ try:
24
+ with serial.Serial(
25
+ port=mcu.serialport,
26
+ baudrate=1200,
27
+ parity=serial.PARITY_NONE,
28
+ stopbits=serial.STOPBITS_ONE,
29
+ bytesize=serial.EIGHTBITS,
30
+ rtscts=True,
31
+ ) as connection:
32
+ print("Connection established")
33
+ connection.rts = True
34
+ connection.dtr = False
35
+ time.sleep(0.4)
36
+
37
+ except serial.SerialException as e:
38
+ log.exception(e)
39
+ raise MPFlashError("pySerial error: " + str(e) + "\n") from e
40
+ except Exception as e:
41
+ log.exception(e)
42
+ raise MPFlashError("Error: " + str(e) + "\n") from e
43
+
44
+ # be optimistic
45
+ return True
@@ -1,129 +1,129 @@
1
- """CLI to Download MicroPython firmware for specific ports, boards and versions."""
2
-
3
- from pathlib import Path
4
-
5
- import rich_click as click
6
- from loguru import logger as log
7
-
8
- from mpflash.connected import connected_ports_boards
9
- from mpflash.errors import MPFlashError
10
- from mpflash.mpboard_id import find_known_board
11
- from mpflash.vendor.versions import clean_version
12
-
13
- from .ask_input import ask_missing_params
14
- from .cli_group import cli
15
- from .common import DownloadParams
16
- from .config import config
17
- from .download import download
18
-
19
-
20
- @cli.command(
21
- "download",
22
- help="Download MicroPython firmware for specific ports, boards and versions.",
23
- )
24
- @click.option(
25
- "--destination",
26
- "-d",
27
- "fw_folder",
28
- type=click.Path(file_okay=False, dir_okay=True, path_type=Path),
29
- default=config.firmware_folder,
30
- show_default=True,
31
- help="The folder to download the firmware to.",
32
- )
33
- @click.option(
34
- "--version",
35
- "-v",
36
- "versions",
37
- default=["stable"],
38
- multiple=True,
39
- show_default=True,
40
- help="The version of MicroPython to to download.",
41
- metavar="SEMVER, 'stable', 'preview' or '?'",
42
- )
43
- @click.option(
44
- "--board",
45
- "-b",
46
- "boards",
47
- multiple=True,
48
- default=[],
49
- show_default=True,
50
- help="The board(s) to download the firmware for.",
51
- metavar="BOARD_ID or ?",
52
- )
53
- @click.option(
54
- "--serial",
55
- "--serial-port",
56
- "-s",
57
- "serial",
58
- default=["*"],
59
- show_default=True,
60
- multiple=True,
61
- help="Which serial port(s) to flash",
62
- metavar="SERIALPORT",
63
- )
64
- @click.option(
65
- "--ignore",
66
- "-i",
67
- is_eager=True,
68
- help="Serial port(s) to ignore. Defaults to MPFLASH_IGNORE.",
69
- multiple=True,
70
- default=[],
71
- envvar="MPFLASH_IGNORE",
72
- show_default=True,
73
- metavar="SERIALPORT",
74
- )
75
- @click.option(
76
- "--clean/--no-clean",
77
- default=True,
78
- show_default=True,
79
- help="""Remove dates and hashes from the downloaded firmware filenames.""",
80
- )
81
- @click.option(
82
- "--force",
83
- "-f",
84
- default=False,
85
- is_flag=True,
86
- show_default=True,
87
- help="""Force download of firmware even if it already exists.""",
88
- )
89
- def cli_download(**kwargs) -> int:
90
- params = DownloadParams(**kwargs)
91
- params.versions = list(params.versions)
92
- params.boards = list(params.boards)
93
- params.serial = list(params.serial)
94
- params.ignore = list(params.ignore)
95
-
96
- # all_boards: List[MPRemoteBoard] = []
97
- if params.boards:
98
- if not params.ports:
99
- # no ports specified - resolve ports from specified boards by resolving board IDs
100
- for board in params.boards:
101
- if board != "?":
102
- try:
103
- board_ = find_known_board(board)
104
- params.ports.append(board_.port)
105
- except MPFlashError as e:
106
- log.error(f"{e}")
107
- else:
108
- # no boards specified - detect connected ports and boards
109
- params.ports, params.boards, _ = connected_ports_boards(include=params.serial, ignore=params.ignore)
110
-
111
- params = ask_missing_params(params)
112
- if not params: # Cancelled by user
113
- return 2
114
- params.versions = [clean_version(v, drop_v=True) for v in params.versions]
115
- assert isinstance(params, DownloadParams)
116
-
117
- try:
118
- download(
119
- params.fw_folder,
120
- params.ports,
121
- params.boards,
122
- params.versions,
123
- params.force,
124
- params.clean,
125
- )
126
- return 0
127
- except MPFlashError as e:
128
- log.error(f"{e}")
129
- return 1
1
+ """CLI to Download MicroPython firmware for specific ports, boards and versions."""
2
+
3
+ from pathlib import Path
4
+
5
+ import rich_click as click
6
+ from loguru import logger as log
7
+
8
+ from mpflash.connected import connected_ports_boards
9
+ from mpflash.errors import MPFlashError
10
+ from mpflash.mpboard_id import find_known_board
11
+ from mpflash.vendor.versions import clean_version
12
+
13
+ from .ask_input import ask_missing_params
14
+ from .cli_group import cli
15
+ from .common import DownloadParams
16
+ from .config import config
17
+ from .download import download
18
+
19
+
20
+ @cli.command(
21
+ "download",
22
+ help="Download MicroPython firmware for specific ports, boards and versions.",
23
+ )
24
+ @click.option(
25
+ "--destination",
26
+ "-d",
27
+ "fw_folder",
28
+ type=click.Path(file_okay=False, dir_okay=True, path_type=Path),
29
+ default=config.firmware_folder,
30
+ show_default=True,
31
+ help="The folder to download the firmware to.",
32
+ )
33
+ @click.option(
34
+ "--version",
35
+ "-v",
36
+ "versions",
37
+ default=["stable"],
38
+ multiple=True,
39
+ show_default=True,
40
+ help="The version of MicroPython to to download.",
41
+ metavar="SEMVER, 'stable', 'preview' or '?'",
42
+ )
43
+ @click.option(
44
+ "--board",
45
+ "-b",
46
+ "boards",
47
+ multiple=True,
48
+ default=[],
49
+ show_default=True,
50
+ help="The board(s) to download the firmware for.",
51
+ metavar="BOARD_ID or ?",
52
+ )
53
+ @click.option(
54
+ "--serial",
55
+ "--serial-port",
56
+ "-s",
57
+ "serial",
58
+ default=["*"],
59
+ show_default=True,
60
+ multiple=True,
61
+ help="Which serial port(s) to flash",
62
+ metavar="SERIALPORT",
63
+ )
64
+ @click.option(
65
+ "--ignore",
66
+ "-i",
67
+ is_eager=True,
68
+ help="Serial port(s) to ignore. Defaults to MPFLASH_IGNORE.",
69
+ multiple=True,
70
+ default=[],
71
+ envvar="MPFLASH_IGNORE",
72
+ show_default=True,
73
+ metavar="SERIALPORT",
74
+ )
75
+ @click.option(
76
+ "--clean/--no-clean",
77
+ default=True,
78
+ show_default=True,
79
+ help="""Remove dates and hashes from the downloaded firmware filenames.""",
80
+ )
81
+ @click.option(
82
+ "--force",
83
+ "-f",
84
+ default=False,
85
+ is_flag=True,
86
+ show_default=True,
87
+ help="""Force download of firmware even if it already exists.""",
88
+ )
89
+ def cli_download(**kwargs) -> int:
90
+ params = DownloadParams(**kwargs)
91
+ params.versions = list(params.versions)
92
+ params.boards = list(params.boards)
93
+ params.serial = list(params.serial)
94
+ params.ignore = list(params.ignore)
95
+
96
+ # all_boards: List[MPRemoteBoard] = []
97
+ if params.boards:
98
+ if not params.ports:
99
+ # no ports specified - resolve ports from specified boards by resolving board IDs
100
+ for board in params.boards:
101
+ if board != "?":
102
+ try:
103
+ board_ = find_known_board(board)
104
+ params.ports.append(board_.port)
105
+ except MPFlashError as e:
106
+ log.error(f"{e}")
107
+ else:
108
+ # no boards specified - detect connected ports and boards
109
+ params.ports, params.boards, _ = connected_ports_boards(include=params.serial, ignore=params.ignore)
110
+
111
+ params = ask_missing_params(params)
112
+ if not params: # Cancelled by user
113
+ return 2
114
+ params.versions = [clean_version(v, drop_v=True) for v in params.versions]
115
+ assert isinstance(params, DownloadParams)
116
+
117
+ try:
118
+ download(
119
+ params.fw_folder,
120
+ params.ports,
121
+ params.boards,
122
+ params.versions,
123
+ params.force,
124
+ params.clean,
125
+ )
126
+ return 0
127
+ except MPFlashError as e:
128
+ log.error(f"{e}")
129
+ return 1