micropython-stubber 1.23.1.post1__py3-none-any.whl → 1.23.3__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 (149) hide show
  1. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.3.dist-info}/LICENSE +30 -30
  2. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.3.dist-info}/METADATA +5 -5
  3. micropython_stubber-1.23.3.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 +304 -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 +207 -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 +48 -43
  44. mpflash/mpflash/mpremoteboard/__init__.py +266 -266
  45. mpflash/mpflash/mpremoteboard/mpy_fw_info.py +152 -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 +1687 -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 +986 -986
  59. stubber/board/createstubs_db.py +825 -825
  60. stubber/board/createstubs_db_min.py +331 -331
  61. stubber/board/createstubs_lvgl.py +741 -741
  62. stubber/board/createstubs_lvgl_min.py +741 -741
  63. stubber/board/createstubs_mem.py +766 -766
  64. stubber/board/createstubs_mem_min.py +306 -306
  65. stubber/board/createstubs_min.py +294 -294
  66. stubber/board/fw_info.py +141 -141
  67. stubber/board/info.py +183 -183
  68. stubber/board/main.py +19 -19
  69. stubber/board/modulelist.txt +247 -247
  70. stubber/board/pyrightconfig.json +34 -34
  71. stubber/bulk/mcu_stubber.py +437 -437
  72. stubber/codemod/_partials/__init__.py +48 -48
  73. stubber/codemod/_partials/db_main.py +147 -147
  74. stubber/codemod/_partials/lvgl_main.py +77 -77
  75. stubber/codemod/_partials/modules_reader.py +80 -80
  76. stubber/codemod/add_comment.py +53 -53
  77. stubber/codemod/add_method.py +65 -65
  78. stubber/codemod/board.py +317 -317
  79. stubber/codemod/enrich.py +145 -145
  80. stubber/codemod/merge_docstub.py +284 -284
  81. stubber/codemod/modify_list.py +54 -54
  82. stubber/codemod/utils.py +56 -56
  83. stubber/commands/build_cmd.py +94 -94
  84. stubber/commands/cli.py +49 -49
  85. stubber/commands/clone_cmd.py +78 -78
  86. stubber/commands/config_cmd.py +29 -29
  87. stubber/commands/enrich_folder_cmd.py +71 -71
  88. stubber/commands/get_core_cmd.py +71 -71
  89. stubber/commands/get_docstubs_cmd.py +92 -92
  90. stubber/commands/get_frozen_cmd.py +117 -117
  91. stubber/commands/get_mcu_cmd.py +102 -102
  92. stubber/commands/merge_cmd.py +66 -66
  93. stubber/commands/publish_cmd.py +118 -118
  94. stubber/commands/stub_cmd.py +31 -31
  95. stubber/commands/switch_cmd.py +62 -62
  96. stubber/commands/variants_cmd.py +48 -48
  97. stubber/cst_transformer.py +178 -178
  98. stubber/data/board_info.csv +193 -193
  99. stubber/data/board_info.json +1729 -1729
  100. stubber/data/micropython_tags.csv +15 -15
  101. stubber/data/requirements-core-micropython.txt +38 -38
  102. stubber/data/requirements-core-pycopy.txt +39 -39
  103. stubber/downloader.py +37 -37
  104. stubber/freeze/common.py +72 -72
  105. stubber/freeze/freeze_folder.py +69 -69
  106. stubber/freeze/freeze_manifest_2.py +126 -126
  107. stubber/freeze/get_frozen.py +131 -131
  108. stubber/get_cpython.py +112 -112
  109. stubber/get_lobo.py +59 -59
  110. stubber/minify.py +423 -423
  111. stubber/publish/bump.py +86 -86
  112. stubber/publish/candidates.py +275 -275
  113. stubber/publish/database.py +18 -18
  114. stubber/publish/defaults.py +40 -40
  115. stubber/publish/enums.py +24 -24
  116. stubber/publish/helpers.py +29 -29
  117. stubber/publish/merge_docstubs.py +132 -132
  118. stubber/publish/missing_class_methods.py +51 -51
  119. stubber/publish/package.py +150 -150
  120. stubber/publish/pathnames.py +51 -51
  121. stubber/publish/publish.py +120 -120
  122. stubber/publish/pypi.py +42 -42
  123. stubber/publish/stubpackage.py +1051 -1051
  124. stubber/rst/__init__.py +9 -9
  125. stubber/rst/classsort.py +78 -78
  126. stubber/rst/lookup.py +531 -531
  127. stubber/rst/output_dict.py +401 -401
  128. stubber/rst/reader.py +814 -814
  129. stubber/rst/report_return.py +77 -77
  130. stubber/rst/rst_utils.py +541 -541
  131. stubber/stubber.py +38 -38
  132. stubber/stubs_from_docs.py +90 -90
  133. stubber/tools/manifestfile.py +654 -654
  134. stubber/tools/readme.md +6 -6
  135. stubber/update_fallback.py +117 -117
  136. stubber/update_module_list.py +123 -123
  137. stubber/utils/__init__.py +6 -6
  138. stubber/utils/config.py +137 -137
  139. stubber/utils/makeversionhdr.py +54 -54
  140. stubber/utils/manifest.py +90 -90
  141. stubber/utils/post.py +80 -80
  142. stubber/utils/repos.py +156 -156
  143. stubber/utils/stubmaker.py +139 -139
  144. stubber/utils/typed_config_toml.py +80 -80
  145. stubber/variants.py +106 -106
  146. micropython_stubber-1.23.1.post1.dist-info/RECORD +0 -159
  147. mpflash/basicgit.py +0 -288
  148. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.3.dist-info}/WHEEL +0 -0
  149. {micropython_stubber-1.23.1.post1.dist-info → micropython_stubber-1.23.3.dist-info}/entry_points.txt +0 -0
@@ -1,29 +1,29 @@
1
- """Show the current configuration"""
2
-
3
- # pragma: no cover
4
-
5
- from mpflash.logger import log
6
-
7
- from stubber.utils.config import CONFIG
8
-
9
- from .cli import stubber_cli
10
-
11
-
12
- @stubber_cli.command(name="show-config")
13
- def cli_config():
14
- """
15
- Show the current configuration
16
- """
17
- try:
18
- # todo: sources[1] is depedent on the order the providers are added in
19
- log.info(f"config file {CONFIG._provider._config_sources[1].filename}") # type: ignore
20
- except Exception:
21
- pass
22
- log.info(f"CONFIG.repo_path {CONFIG.repo_path}")
23
- log.info(f"CONFIG.mpy_path {CONFIG.mpy_path}")
24
- log.info(f"CONFIG.mpy_lib_path {CONFIG.mpy_lib_path}")
25
- log.info(f"CONFIG.mpy_stubs_path {CONFIG.mpy_stubs_path}")
26
-
27
- log.info(f"CONFIG.stub_path {CONFIG.stub_path}")
28
- log.info(f"CONFIG.publish_path {CONFIG.publish_path}")
29
- log.info(f"CONFIG.template_path {CONFIG.template_path}")
1
+ """Show the current configuration"""
2
+
3
+ # pragma: no cover
4
+
5
+ from mpflash.logger import log
6
+
7
+ from stubber.utils.config import CONFIG
8
+
9
+ from .cli import stubber_cli
10
+
11
+
12
+ @stubber_cli.command(name="show-config")
13
+ def cli_config():
14
+ """
15
+ Show the current configuration
16
+ """
17
+ try:
18
+ # todo: sources[1] is depedent on the order the providers are added in
19
+ log.info(f"config file {CONFIG._provider._config_sources[1].filename}") # type: ignore
20
+ except Exception:
21
+ pass
22
+ log.info(f"CONFIG.repo_path {CONFIG.repo_path}")
23
+ log.info(f"CONFIG.mpy_path {CONFIG.mpy_path}")
24
+ log.info(f"CONFIG.mpy_lib_path {CONFIG.mpy_lib_path}")
25
+ log.info(f"CONFIG.mpy_stubs_path {CONFIG.mpy_stubs_path}")
26
+
27
+ log.info(f"CONFIG.stub_path {CONFIG.stub_path}")
28
+ log.info(f"CONFIG.publish_path {CONFIG.publish_path}")
29
+ log.info(f"CONFIG.template_path {CONFIG.template_path}")
@@ -1,71 +1,71 @@
1
- """
2
- enrich machinestubs with docstubs
3
- """
4
-
5
- from pathlib import Path
6
- from typing import Union
7
-
8
- import rich_click as click
9
- from mpflash.logger import log
10
-
11
- from stubber.codemod.enrich import enrich_folder
12
- from stubber.utils.config import CONFIG
13
-
14
- from .cli import stubber_cli
15
-
16
-
17
- @stubber_cli.command(name="enrich")
18
- @click.option(
19
- "--stubs",
20
- "-s",
21
- "stubs_folder",
22
- default=CONFIG.stub_path.as_posix(),
23
- type=click.Path(exists=True, file_okay=True, dir_okay=True),
24
- help="File or folder containing the MCU stubs to be updated",
25
- show_default=True,
26
- )
27
- @click.option(
28
- "--docstubs",
29
- "-ds",
30
- "docstubs_folder",
31
- default=CONFIG.stub_path.as_posix(),
32
- type=click.Path(exists=True, file_okay=True, dir_okay=True),
33
- help="File or folder containing the docstubs to be applied",
34
- show_default=True,
35
- )
36
- @click.option("--diff", default=False, help="Show diff", show_default=True, is_flag=True)
37
- @click.option(
38
- "--dry-run",
39
- default=False,
40
- help="Dry run does not write the files back",
41
- show_default=True,
42
- is_flag=True,
43
- )
44
- @click.option(
45
- "--package-name",
46
- "-p",
47
- "package_name",
48
- default="",
49
- help="Package name to be enriched (Optional)",
50
- show_default=True,
51
- )
52
- def cli_enrich_folder(
53
- stubs_folder: Union[str, Path],
54
- docstubs_folder: Union[str, Path],
55
- diff: bool = False,
56
- dry_run: bool = False,
57
- package_name: str = "",
58
- ):
59
- """
60
- Enrich the stubs in stub_folder with the docstubs in docstubs_folder.
61
- """
62
- write_back = not dry_run
63
- log.info(f"Enriching {stubs_folder} with {docstubs_folder}")
64
- _ = enrich_folder(
65
- Path(stubs_folder),
66
- Path(docstubs_folder),
67
- show_diff=diff,
68
- write_back=write_back,
69
- require_docstub=False,
70
- package_name=package_name,
71
- )
1
+ """
2
+ enrich machinestubs with docstubs
3
+ """
4
+
5
+ from pathlib import Path
6
+ from typing import Union
7
+
8
+ import rich_click as click
9
+ from mpflash.logger import log
10
+
11
+ from stubber.codemod.enrich import enrich_folder
12
+ from stubber.utils.config import CONFIG
13
+
14
+ from .cli import stubber_cli
15
+
16
+
17
+ @stubber_cli.command(name="enrich")
18
+ @click.option(
19
+ "--stubs",
20
+ "-s",
21
+ "stubs_folder",
22
+ default=CONFIG.stub_path.as_posix(),
23
+ type=click.Path(exists=True, file_okay=True, dir_okay=True),
24
+ help="File or folder containing the MCU stubs to be updated",
25
+ show_default=True,
26
+ )
27
+ @click.option(
28
+ "--docstubs",
29
+ "-ds",
30
+ "docstubs_folder",
31
+ default=CONFIG.stub_path.as_posix(),
32
+ type=click.Path(exists=True, file_okay=True, dir_okay=True),
33
+ help="File or folder containing the docstubs to be applied",
34
+ show_default=True,
35
+ )
36
+ @click.option("--diff", default=False, help="Show diff", show_default=True, is_flag=True)
37
+ @click.option(
38
+ "--dry-run",
39
+ default=False,
40
+ help="Dry run does not write the files back",
41
+ show_default=True,
42
+ is_flag=True,
43
+ )
44
+ @click.option(
45
+ "--package-name",
46
+ "-p",
47
+ "package_name",
48
+ default="",
49
+ help="Package name to be enriched (Optional)",
50
+ show_default=True,
51
+ )
52
+ def cli_enrich_folder(
53
+ stubs_folder: Union[str, Path],
54
+ docstubs_folder: Union[str, Path],
55
+ diff: bool = False,
56
+ dry_run: bool = False,
57
+ package_name: str = "",
58
+ ):
59
+ """
60
+ Enrich the stubs in stub_folder with the docstubs in docstubs_folder.
61
+ """
62
+ write_back = not dry_run
63
+ log.info(f"Enriching {stubs_folder} with {docstubs_folder}")
64
+ _ = enrich_folder(
65
+ Path(stubs_folder),
66
+ Path(docstubs_folder),
67
+ show_diff=diff,
68
+ write_back=write_back,
69
+ require_docstub=False,
70
+ package_name=package_name,
71
+ )
@@ -1,71 +1,71 @@
1
- """Get core CPython stubs from PyPi."""
2
-
3
- ##########################################################################################
4
- # core
5
- ##########################################################################################
6
-
7
- from pathlib import Path
8
- from typing import List
9
-
10
- import rich_click as click
11
- from mpflash.logger import log
12
-
13
- import stubber.get_cpython as get_cpython
14
- import stubber.utils as utils
15
- from stubber.utils.config import CONFIG
16
-
17
- from .cli import stubber_cli
18
-
19
- ##########################################################################################
20
- # log = logging.getLogger("stubber")
21
- #########################################################################################
22
-
23
-
24
- @stubber_cli.command(name="get-core")
25
- @click.option(
26
- "--stub-folder",
27
- "-stubs",
28
- default=CONFIG.stub_path.as_posix(),
29
- type=click.Path(exists=True, file_okay=False, dir_okay=True),
30
- show_default=True,
31
- )
32
- @click.option(
33
- "--stubgen/--no-stubgen",
34
- default=True,
35
- help="run stubgen to create .pyi files for the (new) frozen modules",
36
- show_default=True,
37
- )
38
- @click.option(
39
- "--black/--no-black",
40
- default=True,
41
- help="Run black on the (new) frozen modules",
42
- show_default=True,
43
- )
44
- def cli_get_core(
45
- stub_folder: str = "",
46
- # core_type: str = "pycopy", # pycopy or Micropython CPython stubs
47
- stubgen: bool = True,
48
- black: bool = True,
49
- ):
50
- """
51
- Download core CPython stubs from PyPi.
52
-
53
- Get the core (CPython compat) modules for both MicroPython and Pycopy.
54
- """
55
- # default parameter values
56
- stub_folder = stub_folder or CONFIG.stub_path.as_posix()
57
-
58
- stub_paths: List[Path] = []
59
- for core_type in ["pycopy", "micropython"]:
60
- log.info(f"::group:: Get Cpython core :{core_type}")
61
- req_filename = f"requirements-core-{core_type}.txt"
62
- stub_path = Path(stub_folder) / f"cpython_core-{core_type}"
63
-
64
- get_cpython.get_core(
65
- stub_path=stub_path.as_posix(), requirements=req_filename, family=core_type
66
- )
67
- stub_paths.append(stub_path)
68
-
69
- log.info("::group:: start post processing of retrieved stubs")
70
- utils.do_post_processing(stub_paths, stubgen=stubgen, black=black, autoflake=True)
71
- log.info("::group:: Done")
1
+ """Get core CPython stubs from PyPi."""
2
+
3
+ ##########################################################################################
4
+ # core
5
+ ##########################################################################################
6
+
7
+ from pathlib import Path
8
+ from typing import List
9
+
10
+ import rich_click as click
11
+ from mpflash.logger import log
12
+
13
+ import stubber.get_cpython as get_cpython
14
+ import stubber.utils as utils
15
+ from stubber.utils.config import CONFIG
16
+
17
+ from .cli import stubber_cli
18
+
19
+ ##########################################################################################
20
+ # log = logging.getLogger("stubber")
21
+ #########################################################################################
22
+
23
+
24
+ @stubber_cli.command(name="get-core")
25
+ @click.option(
26
+ "--stub-folder",
27
+ "-stubs",
28
+ default=CONFIG.stub_path.as_posix(),
29
+ type=click.Path(exists=True, file_okay=False, dir_okay=True),
30
+ show_default=True,
31
+ )
32
+ @click.option(
33
+ "--stubgen/--no-stubgen",
34
+ default=True,
35
+ help="run stubgen to create .pyi files for the (new) frozen modules",
36
+ show_default=True,
37
+ )
38
+ @click.option(
39
+ "--black/--no-black",
40
+ default=True,
41
+ help="Run black on the (new) frozen modules",
42
+ show_default=True,
43
+ )
44
+ def cli_get_core(
45
+ stub_folder: str = "",
46
+ # core_type: str = "pycopy", # pycopy or Micropython CPython stubs
47
+ stubgen: bool = True,
48
+ black: bool = True,
49
+ ):
50
+ """
51
+ Download core CPython stubs from PyPi.
52
+
53
+ Get the core (CPython compat) modules for both MicroPython and Pycopy.
54
+ """
55
+ # default parameter values
56
+ stub_folder = stub_folder or CONFIG.stub_path.as_posix()
57
+
58
+ stub_paths: List[Path] = []
59
+ for core_type in ["pycopy", "micropython"]:
60
+ log.info(f"::group:: Get Cpython core :{core_type}")
61
+ req_filename = f"requirements-core-{core_type}.txt"
62
+ stub_path = Path(stub_folder) / f"cpython_core-{core_type}"
63
+
64
+ get_cpython.get_core(
65
+ stub_path=stub_path.as_posix(), requirements=req_filename, family=core_type
66
+ )
67
+ stub_paths.append(stub_path)
68
+
69
+ log.info("::group:: start post processing of retrieved stubs")
70
+ utils.do_post_processing(stub_paths, stubgen=stubgen, black=black, autoflake=True)
71
+ log.info("::group:: Done")
@@ -1,92 +1,92 @@
1
- """
2
- get-docstubs
3
-
4
- """
5
-
6
- from pathlib import Path
7
- from typing import Optional
8
-
9
- import rich_click as click
10
- from mpflash.logger import log
11
-
12
- import mpflash.basicgit as git
13
- import stubber.utils as utils
14
- from stubber.stubs_from_docs import generate_from_rst
15
- from stubber.utils.config import CONFIG
16
- from stubber.utils.repos import fetch_repos
17
-
18
- from .cli import stubber_cli
19
-
20
- ##########################################################################################
21
- # log = logging.getLogger("stubber")
22
- #########################################################################################
23
-
24
-
25
- @stubber_cli.command(
26
- name="get-docstubs",
27
- aliases=["get-doc-stubs", "docstubs"],
28
- )
29
- @click.option(
30
- "--path",
31
- "-p",
32
- default=CONFIG.repo_path.as_posix(),
33
- type=click.Path(file_okay=False, dir_okay=True),
34
- show_default=True,
35
- )
36
- @click.option(
37
- "--stub-path",
38
- "--stub-folder",
39
- "target",
40
- default=CONFIG.stub_path.as_posix(),
41
- type=click.Path(exists=True, file_okay=False, dir_okay=True),
42
- help="Destination of the files to be generated.",
43
- show_default=True,
44
- )
45
- # @click.option("--family", "-f", "basename", default="micropython", help="Micropython family.", show_default=True)
46
- @click.option(
47
- "--version", "--tag", default="", type=str, help="Version number to use. [default: Git tag]"
48
- )
49
- @click.option("--black/--no-black", "-b/-nb", default=True, help="Run black", show_default=True)
50
- @click.pass_context
51
- def cli_docstubs(
52
- ctx: click.Context,
53
- path: Optional[str] = None,
54
- target: Optional[str] = None,
55
- black: bool = True,
56
- basename: Optional[str] = None,
57
- version: str = "",
58
- ):
59
- """
60
- Build stubs from documentation.
61
-
62
- Read the Micropython library documentation files and use them to build stubs that can be used for static typechecking.
63
- """
64
- # default parameter values
65
- path = path or CONFIG.repo_path.as_posix()
66
- target = target or CONFIG.stub_path.as_posix()
67
- basename = basename or "micropython"
68
-
69
- if path == CONFIG.repo_path.as_posix():
70
- # default
71
- rst_path = CONFIG.mpy_path / "docs" / "library"
72
- elif Path(path).stem == "micropython":
73
- # path to a micropython repo
74
- rst_path = Path(path) / "docs" / "library"
75
- else:
76
- rst_path = Path(path) # or specify full path
77
-
78
- if version:
79
- version = utils.clean_version(version, drop_v=False)
80
- result = fetch_repos(version, CONFIG.mpy_path, CONFIG.mpy_lib_path)
81
- if not result:
82
- return -1
83
- # get the current checked out version
84
- version = utils.checkedout_version(CONFIG.mpy_path)
85
-
86
- release = git.get_local_tag(rst_path.as_posix(), abbreviate=False) or ""
87
-
88
- dst_path = Path(target) / f"{basename}-{utils.clean_version(version, flat=True)}-docstubs"
89
-
90
- log.info(f"Get docstubs for MicroPython {utils.clean_version(version, drop_v=False)}")
91
- generate_from_rst(rst_path, dst_path, version, release=release, suffix=".pyi", black=black)
92
- log.info("::group:: Done")
1
+ """
2
+ get-docstubs
3
+
4
+ """
5
+
6
+ from pathlib import Path
7
+ from typing import Optional
8
+
9
+ import rich_click as click
10
+ from mpflash.logger import log
11
+
12
+ import mpflash.basicgit as git
13
+ import stubber.utils as utils
14
+ from stubber.stubs_from_docs import generate_from_rst
15
+ from stubber.utils.config import CONFIG
16
+ from stubber.utils.repos import fetch_repos
17
+
18
+ from .cli import stubber_cli
19
+
20
+ ##########################################################################################
21
+ # log = logging.getLogger("stubber")
22
+ #########################################################################################
23
+
24
+
25
+ @stubber_cli.command(
26
+ name="get-docstubs",
27
+ aliases=["get-doc-stubs", "docstubs"],
28
+ )
29
+ @click.option(
30
+ "--path",
31
+ "-p",
32
+ default=CONFIG.repo_path.as_posix(),
33
+ type=click.Path(file_okay=False, dir_okay=True),
34
+ show_default=True,
35
+ )
36
+ @click.option(
37
+ "--stub-path",
38
+ "--stub-folder",
39
+ "target",
40
+ default=CONFIG.stub_path.as_posix(),
41
+ type=click.Path(exists=True, file_okay=False, dir_okay=True),
42
+ help="Destination of the files to be generated.",
43
+ show_default=True,
44
+ )
45
+ # @click.option("--family", "-f", "basename", default="micropython", help="Micropython family.", show_default=True)
46
+ @click.option(
47
+ "--version", "--tag", default="", type=str, help="Version number to use. [default: Git tag]"
48
+ )
49
+ @click.option("--black/--no-black", "-b/-nb", default=True, help="Run black", show_default=True)
50
+ @click.pass_context
51
+ def cli_docstubs(
52
+ ctx: click.Context,
53
+ path: Optional[str] = None,
54
+ target: Optional[str] = None,
55
+ black: bool = True,
56
+ basename: Optional[str] = None,
57
+ version: str = "",
58
+ ):
59
+ """
60
+ Build stubs from documentation.
61
+
62
+ Read the Micropython library documentation files and use them to build stubs that can be used for static typechecking.
63
+ """
64
+ # default parameter values
65
+ path = path or CONFIG.repo_path.as_posix()
66
+ target = target or CONFIG.stub_path.as_posix()
67
+ basename = basename or "micropython"
68
+
69
+ if path == CONFIG.repo_path.as_posix():
70
+ # default
71
+ rst_path = CONFIG.mpy_path / "docs" / "library"
72
+ elif Path(path).stem == "micropython":
73
+ # path to a micropython repo
74
+ rst_path = Path(path) / "docs" / "library"
75
+ else:
76
+ rst_path = Path(path) # or specify full path
77
+
78
+ if version:
79
+ version = utils.clean_version(version, drop_v=False)
80
+ result = fetch_repos(version, CONFIG.mpy_path, CONFIG.mpy_lib_path)
81
+ if not result:
82
+ return -1
83
+ # get the current checked out version
84
+ version = utils.checkedout_version(CONFIG.mpy_path)
85
+
86
+ release = git.get_local_tag(rst_path.as_posix(), abbreviate=False) or ""
87
+
88
+ dst_path = Path(target) / f"{basename}-{utils.clean_version(version, flat=True)}-docstubs"
89
+
90
+ log.info(f"Get docstubs for MicroPython {utils.clean_version(version, drop_v=False)}")
91
+ generate_from_rst(rst_path, dst_path, version, release=release, suffix=".pyi", black=black)
92
+ log.info("::group:: Done")