micropython-stubber 1.23.1.post1__py3-none-any.whl → 1.23.2__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.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/LICENSE +30 -30
  2. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/METADATA +4 -4
  3. micropython_stubber-1.23.2.dist-info/RECORD +158 -0
  4. mpflash/README.md +220 -220
  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/basicgit.py +284 -284
  9. mpflash/mpflash/bootloader/__init__.py +2 -2
  10. mpflash/mpflash/bootloader/activate.py +60 -60
  11. mpflash/mpflash/bootloader/detect.py +82 -82
  12. mpflash/mpflash/bootloader/manual.py +101 -101
  13. mpflash/mpflash/bootloader/micropython.py +12 -12
  14. mpflash/mpflash/bootloader/touch1200.py +36 -36
  15. mpflash/mpflash/cli_download.py +129 -129
  16. mpflash/mpflash/cli_flash.py +224 -216
  17. mpflash/mpflash/cli_group.py +111 -111
  18. mpflash/mpflash/cli_list.py +87 -87
  19. mpflash/mpflash/cli_main.py +39 -39
  20. mpflash/mpflash/common.py +210 -166
  21. mpflash/mpflash/config.py +44 -44
  22. mpflash/mpflash/connected.py +96 -77
  23. mpflash/mpflash/download.py +364 -364
  24. mpflash/mpflash/downloaded.py +130 -130
  25. mpflash/mpflash/errors.py +9 -9
  26. mpflash/mpflash/flash/__init__.py +55 -55
  27. mpflash/mpflash/flash/esp.py +59 -59
  28. mpflash/mpflash/flash/stm32.py +19 -19
  29. mpflash/mpflash/flash/stm32_dfu.py +104 -104
  30. mpflash/mpflash/flash/uf2/__init__.py +88 -88
  31. mpflash/mpflash/flash/uf2/boardid.py +15 -15
  32. mpflash/mpflash/flash/uf2/linux.py +136 -130
  33. mpflash/mpflash/flash/uf2/macos.py +42 -42
  34. mpflash/mpflash/flash/uf2/uf2disk.py +12 -12
  35. mpflash/mpflash/flash/uf2/windows.py +43 -43
  36. mpflash/mpflash/flash/worklist.py +170 -170
  37. mpflash/mpflash/list.py +106 -106
  38. mpflash/mpflash/logger.py +41 -41
  39. mpflash/mpflash/mpboard_id/__init__.py +93 -93
  40. mpflash/mpflash/mpboard_id/add_boards.py +251 -251
  41. mpflash/mpflash/mpboard_id/board.py +37 -37
  42. mpflash/mpflash/mpboard_id/board_id.py +86 -86
  43. mpflash/mpflash/mpboard_id/store.py +43 -43
  44. mpflash/mpflash/mpremoteboard/__init__.py +266 -266
  45. mpflash/mpflash/mpremoteboard/mpy_fw_info.py +141 -141
  46. mpflash/mpflash/mpremoteboard/runner.py +140 -140
  47. mpflash/mpflash/vendor/click_aliases.py +91 -91
  48. mpflash/mpflash/vendor/dfu.py +165 -165
  49. mpflash/mpflash/vendor/pydfu.py +605 -605
  50. mpflash/mpflash/vendor/readme.md +2 -2
  51. mpflash/mpflash/versions.py +135 -135
  52. mpflash/poetry.lock +1599 -1599
  53. mpflash/pyproject.toml +65 -65
  54. mpflash/stm32_udev_rules.md +62 -62
  55. stubber/__init__.py +3 -3
  56. stubber/board/board_info.csv +193 -193
  57. stubber/board/boot.py +34 -34
  58. stubber/board/createstubs.py +1004 -986
  59. stubber/board/createstubs_db.py +826 -825
  60. stubber/board/createstubs_db_min.py +332 -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 +767 -766
  65. stubber/board/createstubs_mem_min.py +307 -306
  66. stubber/board/createstubs_mem_mpy.mpy +0 -0
  67. stubber/board/createstubs_min.py +295 -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 +437 -437
  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 +151 -145
  83. stubber/codemod/merge_docstub.py +284 -284
  84. stubber/codemod/modify_list.py +54 -54
  85. stubber/codemod/utils.py +56 -56
  86. stubber/commands/build_cmd.py +94 -94
  87. stubber/commands/cli.py +49 -49
  88. stubber/commands/clone_cmd.py +78 -78
  89. stubber/commands/config_cmd.py +29 -29
  90. stubber/commands/enrich_folder_cmd.py +71 -71
  91. stubber/commands/get_core_cmd.py +71 -71
  92. stubber/commands/get_docstubs_cmd.py +92 -92
  93. stubber/commands/get_frozen_cmd.py +117 -117
  94. stubber/commands/get_mcu_cmd.py +102 -102
  95. stubber/commands/merge_cmd.py +66 -66
  96. stubber/commands/publish_cmd.py +118 -118
  97. stubber/commands/stub_cmd.py +31 -31
  98. stubber/commands/switch_cmd.py +62 -62
  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 +37 -37
  107. stubber/freeze/common.py +72 -72
  108. stubber/freeze/freeze_folder.py +69 -69
  109. stubber/freeze/freeze_manifest_2.py +126 -126
  110. stubber/freeze/get_frozen.py +131 -131
  111. stubber/get_cpython.py +112 -112
  112. stubber/get_lobo.py +59 -59
  113. stubber/minify.py +423 -423
  114. stubber/publish/bump.py +86 -86
  115. stubber/publish/candidates.py +275 -275
  116. stubber/publish/database.py +18 -18
  117. stubber/publish/defaults.py +40 -40
  118. stubber/publish/enums.py +24 -24
  119. stubber/publish/helpers.py +29 -29
  120. stubber/publish/merge_docstubs.py +136 -132
  121. stubber/publish/missing_class_methods.py +51 -51
  122. stubber/publish/package.py +150 -150
  123. stubber/publish/pathnames.py +51 -51
  124. stubber/publish/publish.py +120 -120
  125. stubber/publish/pypi.py +42 -42
  126. stubber/publish/stubpackage.py +1055 -1051
  127. stubber/rst/__init__.py +9 -9
  128. stubber/rst/classsort.py +78 -78
  129. stubber/rst/lookup.py +533 -531
  130. stubber/rst/output_dict.py +401 -401
  131. stubber/rst/reader.py +814 -814
  132. stubber/rst/report_return.py +77 -77
  133. stubber/rst/rst_utils.py +541 -541
  134. stubber/stubber.py +38 -38
  135. stubber/stubs_from_docs.py +90 -90
  136. stubber/tools/manifestfile.py +654 -654
  137. stubber/tools/readme.md +6 -6
  138. stubber/update_fallback.py +117 -117
  139. stubber/update_module_list.py +123 -123
  140. stubber/utils/__init__.py +6 -6
  141. stubber/utils/config.py +137 -137
  142. stubber/utils/makeversionhdr.py +54 -54
  143. stubber/utils/manifest.py +90 -90
  144. stubber/utils/post.py +80 -80
  145. stubber/utils/repos.py +156 -156
  146. stubber/utils/stubmaker.py +139 -139
  147. stubber/utils/typed_config_toml.py +80 -80
  148. stubber/variants.py +106 -106
  149. micropython_stubber-1.23.1.post1.dist-info/RECORD +0 -159
  150. mpflash/basicgit.py +0 -288
  151. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/WHEEL +0 -0
  152. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.2.dist-info}/entry_points.txt +0 -0
stubber/tools/readme.md CHANGED
@@ -1,7 +1,7 @@
1
- tools vendored from micropython
2
-
3
- - manifestfile.py
4
- - Micropython manifest file parser
5
- - https://github.com/micropython/micropython/blob/master/tools/manifestfile.py
6
- - v1.23.0
1
+ tools vendored from micropython
2
+
3
+ - manifestfile.py
4
+ - Micropython manifest file parser
5
+ - https://github.com/micropython/micropython/blob/master/tools/manifestfile.py
6
+ - v1.23.0
7
7
  - https://raw.githubusercontent.com/micropython/micropython/v1.23.0/tools/manifestfile.py
@@ -1,117 +1,117 @@
1
- # """build/update the fallback 'catch-all' stubfolder"""
2
-
3
- # import os
4
- # import shutil
5
- # from distutils.dir_util import copy_tree
6
- # from distutils.errors import DistutilsFileError
7
- # from pathlib import Path
8
- # from typing import List, Optional, Tuple
9
-
10
- # from mpflash.logger import log
11
-
12
- # from mpflash.versions import V_PREVIEW
13
-
14
- # # log = logging.getLogger()
15
-
16
- # RELEASED = "v1_18"
17
-
18
-
19
- # def fallback_sources(version: str, fw_version: Optional[str] = None) -> List[Tuple[str, str]]:
20
- # # sourcery skip: inline-immediately-returned-variable
21
- # """
22
- # list of sources to build/update the fallback 'catch-all' stubfolder
23
- # version : the version to use
24
- # fw_version : version to source the MCU stubs from. Defaults to the version used , but can be lower
25
-
26
- # """
27
- # if not fw_version:
28
- # fw_version = version
29
- # if fw_version == V_PREVIEW:
30
- # fw_version = RELEASED
31
- # SOURCES = [
32
- # ("uasyncio", f"micropython-{fw_version}-esp32"),
33
- # ("umqtt", f"micropython-{fw_version}-esp32"),
34
- # ("_onewire.py*", f"micropython-{fw_version}-esp32"),
35
- # ("_uasyncio.py*", f"micropython-{fw_version}-esp32"),
36
- # ("array.py*", f"micropython-{fw_version}-esp32"),
37
- # ("binascii.py*", f"micropython-{fw_version}-esp32"),
38
- # ("hashlib.py*", f"micropython-{fw_version}-esp32"),
39
- # ("machine.py*", f"micropython-{fw_version}-esp32"),
40
- # ("micropython.py*", f"micropython-{version}-docstubs"), # esp32"),
41
- # ("network.pyi", f"micropython-{version}-docstubs"), # esp32"),
42
- # ("struct.py*", f"micropython-{fw_version}-esp32"),
43
- # ("uarray.py*", f"micropython-{fw_version}-esp32"),
44
- # ("ubinascii.py*", f"micropython-{fw_version}-esp32"),
45
- # ("uctypes.py*", f"micropython-{fw_version}-esp32"),
46
- # ("uerrno.py*", f"micropython-{fw_version}-esp32"),
47
- # ("uhashlib.py*", f"micropython-{fw_version}-esp32"),
48
- # ("uio.py*", f"micropython-{fw_version}-esp32"),
49
- # ("ujson.py*", f"micropython-{fw_version}-esp32"),
50
- # ("uselect.py*", f"micropython-{fw_version}-esp32"),
51
- # ("usocket.py*", f"micropython-{fw_version}-esp32"),
52
- # ("ussl.py*", f"micropython-{fw_version}-esp32"),
53
- # ("ustruct.py*", f"micropython-{fw_version}-esp32"),
54
- # ("sys.py*", f"micropython-{fw_version}-esp32"),
55
- # ("usys.py*", f"micropython-{fw_version}-esp32"),
56
- # ("uzlib.py*", f"micropython-{fw_version}-esp32"),
57
- # ("bluetooth.py*", f"micropython-{version}-docstubs"),
58
- # # esp
59
- # ("esp.py*", f"micropython-{version}-docstubs"), # 8266 fw stub has most functions
60
- # ("esp32.py*", f"micropython-{version}-docstubs"), # esp32"),
61
- # # pyboard == stm32
62
- # ("pyb.py*", f"micropython-{fw_version}-stm32"),
63
- # ("framebuf.py*", f"micropython-{fw_version}-stm32"),
64
- # # rp2
65
- # ("_rp2.py*", f"micropython-{fw_version}-rp2"),
66
- # # stdlib
67
- # ("os.py*", f"micropython-{fw_version}-esp32"), # -> stdlib
68
- # ("uos.py*", f"micropython-{fw_version}-esp32"),
69
- # ("time.py*", f"micropython-{fw_version}-esp32"), # -> stdlib
70
- # ("utime.py*", f"micropython-{fw_version}-esp32"),
71
- # ]
72
- # return SOURCES
73
-
74
-
75
- # def update_fallback(stubpath: Path, fallback_path: Path, version: str = RELEASED):
76
- # "update the fallback stubs from the defined sources"
77
- # # remove all *.py/.pyi files
78
- # if not fallback_path.exists():
79
- # os.makedirs(fallback_path)
80
- # else:
81
- # oldstubs = list(fallback_path.rglob("*.py")) + list(fallback_path.rglob("*.pyi"))
82
- # log.debug(f"deleting {len(oldstubs)} stubs from {fallback_path.as_posix()}")
83
- # for f in oldstubs:
84
- # try:
85
- # os.remove(f)
86
- # except OSError as e:
87
- # log.warning(e)
88
- # added = 0
89
- # for name, source in fallback_sources(version):
90
- # if "." not in name:
91
- # # copy folder
92
- # log.debug(f"add {source} folder")
93
- # try:
94
- # copy_tree(
95
- # (stubpath / source / name).as_posix(),
96
- # (fallback_path / name).as_posix(),
97
- # )
98
- # added += 1
99
- # except DistutilsFileError:
100
- # log.warning(f"{(stubpath / source / name).as_posix()} not found")
101
- # else:
102
- # # copy file(s)
103
- # log.debug(f"add {source}")
104
- # for f in (stubpath / source).glob(name):
105
- # shutil.copyfile(f, fallback_path / f.name)
106
- # added += 1
107
- # return added
108
-
109
-
110
- # if __name__ == "__main__":
111
- # from stubber.utils.config import CONFIG
112
-
113
- # update_fallback(
114
- # CONFIG.stub_path,
115
- # CONFIG.fallback_path,
116
- # version=RELEASED,
117
- # )
1
+ # """build/update the fallback 'catch-all' stubfolder"""
2
+
3
+ # import os
4
+ # import shutil
5
+ # from distutils.dir_util import copy_tree
6
+ # from distutils.errors import DistutilsFileError
7
+ # from pathlib import Path
8
+ # from typing import List, Optional, Tuple
9
+
10
+ # from mpflash.logger import log
11
+
12
+ # from mpflash.versions import V_PREVIEW
13
+
14
+ # # log = logging.getLogger()
15
+
16
+ # RELEASED = "v1_18"
17
+
18
+
19
+ # def fallback_sources(version: str, fw_version: Optional[str] = None) -> List[Tuple[str, str]]:
20
+ # # sourcery skip: inline-immediately-returned-variable
21
+ # """
22
+ # list of sources to build/update the fallback 'catch-all' stubfolder
23
+ # version : the version to use
24
+ # fw_version : version to source the MCU stubs from. Defaults to the version used , but can be lower
25
+
26
+ # """
27
+ # if not fw_version:
28
+ # fw_version = version
29
+ # if fw_version == V_PREVIEW:
30
+ # fw_version = RELEASED
31
+ # SOURCES = [
32
+ # ("uasyncio", f"micropython-{fw_version}-esp32"),
33
+ # ("umqtt", f"micropython-{fw_version}-esp32"),
34
+ # ("_onewire.py*", f"micropython-{fw_version}-esp32"),
35
+ # ("_uasyncio.py*", f"micropython-{fw_version}-esp32"),
36
+ # ("array.py*", f"micropython-{fw_version}-esp32"),
37
+ # ("binascii.py*", f"micropython-{fw_version}-esp32"),
38
+ # ("hashlib.py*", f"micropython-{fw_version}-esp32"),
39
+ # ("machine.py*", f"micropython-{fw_version}-esp32"),
40
+ # ("micropython.py*", f"micropython-{version}-docstubs"), # esp32"),
41
+ # ("network.pyi", f"micropython-{version}-docstubs"), # esp32"),
42
+ # ("struct.py*", f"micropython-{fw_version}-esp32"),
43
+ # ("uarray.py*", f"micropython-{fw_version}-esp32"),
44
+ # ("ubinascii.py*", f"micropython-{fw_version}-esp32"),
45
+ # ("uctypes.py*", f"micropython-{fw_version}-esp32"),
46
+ # ("uerrno.py*", f"micropython-{fw_version}-esp32"),
47
+ # ("uhashlib.py*", f"micropython-{fw_version}-esp32"),
48
+ # ("uio.py*", f"micropython-{fw_version}-esp32"),
49
+ # ("ujson.py*", f"micropython-{fw_version}-esp32"),
50
+ # ("uselect.py*", f"micropython-{fw_version}-esp32"),
51
+ # ("usocket.py*", f"micropython-{fw_version}-esp32"),
52
+ # ("ussl.py*", f"micropython-{fw_version}-esp32"),
53
+ # ("ustruct.py*", f"micropython-{fw_version}-esp32"),
54
+ # ("sys.py*", f"micropython-{fw_version}-esp32"),
55
+ # ("usys.py*", f"micropython-{fw_version}-esp32"),
56
+ # ("uzlib.py*", f"micropython-{fw_version}-esp32"),
57
+ # ("bluetooth.py*", f"micropython-{version}-docstubs"),
58
+ # # esp
59
+ # ("esp.py*", f"micropython-{version}-docstubs"), # 8266 fw stub has most functions
60
+ # ("esp32.py*", f"micropython-{version}-docstubs"), # esp32"),
61
+ # # pyboard == stm32
62
+ # ("pyb.py*", f"micropython-{fw_version}-stm32"),
63
+ # ("framebuf.py*", f"micropython-{fw_version}-stm32"),
64
+ # # rp2
65
+ # ("_rp2.py*", f"micropython-{fw_version}-rp2"),
66
+ # # stdlib
67
+ # ("os.py*", f"micropython-{fw_version}-esp32"), # -> stdlib
68
+ # ("uos.py*", f"micropython-{fw_version}-esp32"),
69
+ # ("time.py*", f"micropython-{fw_version}-esp32"), # -> stdlib
70
+ # ("utime.py*", f"micropython-{fw_version}-esp32"),
71
+ # ]
72
+ # return SOURCES
73
+
74
+
75
+ # def update_fallback(stubpath: Path, fallback_path: Path, version: str = RELEASED):
76
+ # "update the fallback stubs from the defined sources"
77
+ # # remove all *.py/.pyi files
78
+ # if not fallback_path.exists():
79
+ # os.makedirs(fallback_path)
80
+ # else:
81
+ # oldstubs = list(fallback_path.rglob("*.py")) + list(fallback_path.rglob("*.pyi"))
82
+ # log.debug(f"deleting {len(oldstubs)} stubs from {fallback_path.as_posix()}")
83
+ # for f in oldstubs:
84
+ # try:
85
+ # os.remove(f)
86
+ # except OSError as e:
87
+ # log.warning(e)
88
+ # added = 0
89
+ # for name, source in fallback_sources(version):
90
+ # if "." not in name:
91
+ # # copy folder
92
+ # log.debug(f"add {source} folder")
93
+ # try:
94
+ # copy_tree(
95
+ # (stubpath / source / name).as_posix(),
96
+ # (fallback_path / name).as_posix(),
97
+ # )
98
+ # added += 1
99
+ # except DistutilsFileError:
100
+ # log.warning(f"{(stubpath / source / name).as_posix()} not found")
101
+ # else:
102
+ # # copy file(s)
103
+ # log.debug(f"add {source}")
104
+ # for f in (stubpath / source).glob(name):
105
+ # shutil.copyfile(f, fallback_path / f.name)
106
+ # added += 1
107
+ # return added
108
+
109
+
110
+ # if __name__ == "__main__":
111
+ # from stubber.utils.config import CONFIG
112
+
113
+ # update_fallback(
114
+ # CONFIG.stub_path,
115
+ # CONFIG.fallback_path,
116
+ # version=RELEASED,
117
+ # )
@@ -1,123 +1,123 @@
1
- # pragma: no cover
2
- """
3
- generate the list of modules that should be attempted to stub
4
- for this :
5
- - combine the modules from the different texts files
6
- - split the lines into individual module names
7
- - combine them in one set
8
- - remove the ones than cannot be stubbed
9
- - remove test modules, ending in `_test`
10
- - write updates to:
11
- - board/modulelist.txt
12
- - board/createstubs.py
13
- """
14
-
15
- from pathlib import Path
16
- from typing import List, Optional, Set
17
-
18
- from mpflash.logger import log
19
-
20
-
21
- def read_modules(path: Optional[Path] = None) -> Set[str]:
22
- """
23
- read text files with modules per firmware.
24
- each contains the output of help("modules")
25
- - lines starting with # are comments.
26
- - split the other lines at whitespace seperator,
27
- - and add each module to a set
28
- """
29
- path = Path(path or "./data")
30
- assert path
31
- all_modules: Set[str] = set()
32
- for file in path.glob("*.txt"):
33
- log.debug(f"processing: {file.name}")
34
- with file.open("r") as f:
35
- while line := f.readline():
36
- if len(line) > 1 and line[0] != "#":
37
- file_mods = line.split()
38
- log.trace(line[:-1])
39
- log.trace(set(file_mods))
40
- # remove modules ending in _test
41
- file_mods = [m for m in file_mods if not m.endswith("_test")]
42
- all_modules = set(all_modules | set(file_mods))
43
- log.trace(">" * 40)
44
- return {m.replace(".", "/") for m in all_modules}
45
-
46
-
47
- def update_module_list():
48
- """
49
- helper script
50
- generate a few lines of code with all modules to be stubbed by createstubs
51
- """
52
- #######################################################################
53
- # the exceptions
54
- #######################################################################
55
- mods_problematic = {
56
- "upysh",
57
- "webrepl_setup",
58
- "http_client",
59
- "http_client_ssl",
60
- "http_server",
61
- "http_server_ssl",
62
- }
63
- mods_excluded = {
64
- "__main__",
65
- "_main",
66
- "_boot",
67
- "webrepl",
68
- "_webrepl",
69
- "port_diag",
70
- "example_sub_led",
71
- "example_pub_button",
72
- "upip",
73
- "upip_utarfile",
74
- "upysh",
75
- "uasyncio",
76
- "builtins",
77
- "re",
78
- }
79
- log.info("Update the module list in createstubs.py")
80
- all_modules = read_modules()
81
- modules_to_stub = sorted(all_modules - set(mods_excluded | mods_problematic))
82
-
83
- # remove pycom MQTT* from defaults
84
- modules_to_stub = sorted({m for m in modules_to_stub if not m.startswith("MQTT")})
85
-
86
- log.info(f"modules to stub : {len(modules_to_stub)}")
87
- # log.debug(wrapped(modules_to_stub))
88
- lines: List[str] = []
89
- # update modules.txt
90
- modules_txt = Path("src/stubber/board/modulelist.txt")
91
- if modules_txt.exists():
92
- with open(modules_txt) as f:
93
- lines = f.readlines()
94
- # only keep comment lines
95
- lines = [l for l in lines if l[0] == "#"]
96
- else:
97
- lines = ["# list of modules to stub."]
98
- lines += [m + "\n" for m in modules_to_stub]
99
- log.info(f"writing module list to {modules_txt}")
100
- with open(modules_txt, "w") as f:
101
- f.writelines(lines)
102
-
103
- # update createstubs.py
104
- createstubs = Path("src/stubber/board/createstubs.py")
105
- if createstubs.exists():
106
- with open(createstubs) as f:
107
- lines = f.readlines()
108
-
109
- l_start = lines.index(" stubber.modules = [\n")
110
- assert l_start
111
- l_end = lines.index(" ] # spell-checker: enable\n", l_start)
112
- assert l_end
113
-
114
- # Plug in the new list of modules
115
- lines = lines[: l_start + 1] + [f' "{m}",\n' for m in modules_to_stub] + lines[l_end:]
116
-
117
- log.info(f"writing updated module list to {createstubs}")
118
- with open(createstubs, "w") as f:
119
- f.writelines(lines)
120
-
121
-
122
- if __name__ == "__main__":
123
- update_module_list()
1
+ # pragma: no cover
2
+ """
3
+ generate the list of modules that should be attempted to stub
4
+ for this :
5
+ - combine the modules from the different texts files
6
+ - split the lines into individual module names
7
+ - combine them in one set
8
+ - remove the ones than cannot be stubbed
9
+ - remove test modules, ending in `_test`
10
+ - write updates to:
11
+ - board/modulelist.txt
12
+ - board/createstubs.py
13
+ """
14
+
15
+ from pathlib import Path
16
+ from typing import List, Optional, Set
17
+
18
+ from mpflash.logger import log
19
+
20
+
21
+ def read_modules(path: Optional[Path] = None) -> Set[str]:
22
+ """
23
+ read text files with modules per firmware.
24
+ each contains the output of help("modules")
25
+ - lines starting with # are comments.
26
+ - split the other lines at whitespace seperator,
27
+ - and add each module to a set
28
+ """
29
+ path = Path(path or "./data")
30
+ assert path
31
+ all_modules: Set[str] = set()
32
+ for file in path.glob("*.txt"):
33
+ log.debug(f"processing: {file.name}")
34
+ with file.open("r") as f:
35
+ while line := f.readline():
36
+ if len(line) > 1 and line[0] != "#":
37
+ file_mods = line.split()
38
+ log.trace(line[:-1])
39
+ log.trace(set(file_mods))
40
+ # remove modules ending in _test
41
+ file_mods = [m for m in file_mods if not m.endswith("_test")]
42
+ all_modules = set(all_modules | set(file_mods))
43
+ log.trace(">" * 40)
44
+ return {m.replace(".", "/") for m in all_modules}
45
+
46
+
47
+ def update_module_list():
48
+ """
49
+ helper script
50
+ generate a few lines of code with all modules to be stubbed by createstubs
51
+ """
52
+ #######################################################################
53
+ # the exceptions
54
+ #######################################################################
55
+ mods_problematic = {
56
+ "upysh",
57
+ "webrepl_setup",
58
+ "http_client",
59
+ "http_client_ssl",
60
+ "http_server",
61
+ "http_server_ssl",
62
+ }
63
+ mods_excluded = {
64
+ "__main__",
65
+ "_main",
66
+ "_boot",
67
+ "webrepl",
68
+ "_webrepl",
69
+ "port_diag",
70
+ "example_sub_led",
71
+ "example_pub_button",
72
+ "upip",
73
+ "upip_utarfile",
74
+ "upysh",
75
+ "uasyncio",
76
+ "builtins",
77
+ "re",
78
+ }
79
+ log.info("Update the module list in createstubs.py")
80
+ all_modules = read_modules()
81
+ modules_to_stub = sorted(all_modules - set(mods_excluded | mods_problematic))
82
+
83
+ # remove pycom MQTT* from defaults
84
+ modules_to_stub = sorted({m for m in modules_to_stub if not m.startswith("MQTT")})
85
+
86
+ log.info(f"modules to stub : {len(modules_to_stub)}")
87
+ # log.debug(wrapped(modules_to_stub))
88
+ lines: List[str] = []
89
+ # update modules.txt
90
+ modules_txt = Path("src/stubber/board/modulelist.txt")
91
+ if modules_txt.exists():
92
+ with open(modules_txt) as f:
93
+ lines = f.readlines()
94
+ # only keep comment lines
95
+ lines = [l for l in lines if l[0] == "#"]
96
+ else:
97
+ lines = ["# list of modules to stub."]
98
+ lines += [m + "\n" for m in modules_to_stub]
99
+ log.info(f"writing module list to {modules_txt}")
100
+ with open(modules_txt, "w") as f:
101
+ f.writelines(lines)
102
+
103
+ # update createstubs.py
104
+ createstubs = Path("src/stubber/board/createstubs.py")
105
+ if createstubs.exists():
106
+ with open(createstubs) as f:
107
+ lines = f.readlines()
108
+
109
+ l_start = lines.index(" stubber.modules = [\n")
110
+ assert l_start
111
+ l_end = lines.index(" ] # spell-checker: enable\n", l_start)
112
+ assert l_end
113
+
114
+ # Plug in the new list of modules
115
+ lines = lines[: l_start + 1] + [f' "{m}",\n' for m in modules_to_stub] + lines[l_end:]
116
+
117
+ log.info(f"writing updated module list to {createstubs}")
118
+ with open(createstubs, "w") as f:
119
+ f.writelines(lines)
120
+
121
+
122
+ if __name__ == "__main__":
123
+ update_module_list()
stubber/utils/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
- # type: ignore
2
- from mpflash.versions import checkedout_version, clean_version
3
-
4
- from .manifest import make_manifest, manifest
5
- from .post import do_post_processing
6
- from .stubmaker import generate_pyi_files, generate_pyi_from_file
1
+ # type: ignore
2
+ from mpflash.versions import checkedout_version, clean_version
3
+
4
+ from .manifest import make_manifest, manifest
5
+ from .post import do_post_processing
6
+ from .stubmaker import generate_pyi_files, generate_pyi_from_file