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
@@ -1,307 +1,308 @@
1
- x='No report file'
2
- w='Failed to create the report.'
3
- v='{}/{}'
4
- u='method'
5
- t='function'
6
- s='bool'
7
- r='str'
8
- q='float'
9
- p='int'
10
- o='stubber'
11
- n=Exception
12
- m=KeyError
13
- l=sorted
14
- k=NotImplementedError
15
- f=',\n'
16
- e='dict'
17
- d='list'
18
- c='tuple'
19
- b='micropython'
20
- a=TypeError
21
- Z=repr
22
- W='-preview'
23
- V='-'
24
- U='board'
25
- T=IndexError
26
- S=print
27
- R=True
28
- Q='family'
29
- P=len
30
- O=open
31
- N=ImportError
32
- M=dir
33
- K='port'
34
- J='.'
35
- I=AttributeError
36
- H=False
37
- G='/'
38
- F=OSError
39
- E=None
40
- C='version'
41
- B=''
42
- import gc as D,os,sys
43
- from time import sleep
44
- try:from ujson import dumps
45
- except:from json import dumps
46
- try:from machine import reset
47
- except N:pass
48
- try:from collections import OrderedDict as g
49
- except N:from ucollections import OrderedDict as g
50
- __version__='v1.23.1'
51
- y=2
52
- z=2
53
- A0=['lib','/lib','/sd/lib','/flash/lib',J]
54
- class L:
55
- INFO=20;WARNING=30;ERROR=40;level=INFO;prnt=S
56
- @staticmethod
57
- def getLogger(name):return L()
58
- @classmethod
59
- def basicConfig(A,level):A.level=level
60
- def info(A,msg):
61
- if A.level<=L.INFO:A.prnt('INFO :',msg)
62
- def warning(A,msg):
63
- if A.level<=L.WARNING:A.prnt('WARN :',msg)
64
- def error(A,msg):
65
- if A.level<=L.ERROR:A.prnt('ERROR :',msg)
66
- A=L.getLogger(o)
67
- L.basicConfig(level=L.INFO)
68
- class Stubber:
69
- def __init__(B,path=E,firmware_id=E):
70
- C=firmware_id
71
- try:
72
- if os.uname().release=='1.13.0'and os.uname().version<'v1.13-103':raise k('MicroPython 1.13.0 cannot be stubbed')
73
- except I:pass
74
- B.info=_info();A.info('Port: {}'.format(B.info[K]));A.info('Board: {}'.format(B.info[U]));D.collect()
75
- if C:B._fwid=C.lower()
76
- elif B.info[Q]==b:B._fwid='{family}-v{version}-{port}-{board}'.format(**B.info).rstrip(V)
77
- else:B._fwid='{family}-v{version}-{port}'.format(**B.info)
78
- B._start_free=D.mem_free()
79
- if path:
80
- if path.endswith(G):path=path[:-1]
81
- else:path=get_root()
82
- B.path='{}/stubs/{}'.format(path,B.flat_fwid).replace('//',G)
83
- try:X(path+G)
84
- except F:A.error('error creating stub folder {}'.format(path))
85
- B.problematic=['upip','upysh','webrepl_setup','http_client','http_client_ssl','http_server','http_server_ssl'];B.excluded=['webrepl','_webrepl','port_diag','example_sub_led.py','example_pub_button.py'];B.modules=[];B._json_name=E;B._json_first=H
86
- def get_obj_attributes(L,item_instance):
87
- H=item_instance;C=[];K=[]
88
- for A in M(H):
89
- if A.startswith('__')and not A in L.modules:continue
90
- try:
91
- E=getattr(H,A)
92
- try:F=Z(type(E)).split("'")[1]
93
- except T:F=B
94
- if F in{p,q,r,s,c,d,e}:G=1
95
- elif F in{t,u}:G=2
96
- elif F in'class':G=3
97
- else:G=4
98
- C.append((A,Z(E),Z(type(E)),E,G))
99
- except I as J:K.append("Couldn't get attribute '{}' from object '{}', Err: {}".format(A,H,J))
100
- except MemoryError as J:S('MemoryError: {}'.format(J));sleep(1);reset()
101
- C=l([A for A in C if not A[0].startswith('__')],key=lambda x:x[4]);D.collect();return C,K
102
- def add_modules(A,modules):A.modules=l(set(A.modules)|set(modules))
103
- def create_all_stubs(B):
104
- A.info('Start micropython-stubber {} on {}'.format(__version__,B._fwid));B.report_start();D.collect()
105
- for C in B.modules:B.create_one_stub(C)
106
- B.report_end();A.info('Finally done')
107
- def create_one_stub(C,module_name):
108
- B=module_name
109
- if B in C.problematic:A.warning('Skip module: {:<25} : Known problematic'.format(B));return H
110
- if B in C.excluded:A.warning('Skip module: {:<25} : Excluded'.format(B));return H
111
- I='{}/{}.pyi'.format(C.path,B.replace(J,G));D.collect();E=H
112
- try:E=C.create_module_stub(B,I)
113
- except F:return H
114
- D.collect();return E
115
- def create_module_stub(K,module_name,file_name=E):
116
- I=file_name;C=module_name
117
- if I is E:L=C.replace(J,'_')+'.pyi';I=K.path+G+L
118
- else:L=I.split(G)[-1]
119
- if G in C:C=C.replace(G,J)
120
- M=E
121
- try:M=__import__(C,E,E,'*');Q=D.mem_free();A.info('Stub module: {:<25} to file: {:<70} mem:{:>5}'.format(C,L,Q))
122
- except N:return H
123
- X(I)
124
- with O(I,'w')as P:S=str(K.info).replace('OrderedDict(',B).replace('})','}');T='"""\nModule: \'{0}\' on {1}\n"""\n# MCU: {2}\n# Stubber: {3}\n'.format(C,K._fwid,S,__version__);P.write(T);P.write('from __future__ import annotations\nfrom typing import Any, Generator\nfrom _typeshed import Incomplete\n\n');K.write_object_stub(P,M,C,B)
125
- K.report_add(C,I)
126
- if C not in{'os','sys','logging','gc'}:
127
- try:del M
128
- except(F,m):A.warning('could not del new_module')
129
- D.collect();return R
130
- def write_object_stub(L,fp,object_expr,obj_name,indent,in_class=0):
131
- Z=' at ...>';Y='generator';X='{0}{1}: {3} = {2}\n';W='bound_method';V='Incomplete';O=in_class;N='Exception';M=object_expr;K=' at ';J=fp;E=indent;D.collect()
132
- if M in L.problematic:A.warning('SKIPPING problematic module:{}'.format(M));return
133
- a,Q=L.get_obj_attributes(M)
134
- if Q:A.error(Q)
135
- for(F,H,I,b,g)in a:
136
- if F in['classmethod','staticmethod','BaseException',N]:continue
137
- if F[0].isdigit():A.warning('NameError: invalid name {}'.format(F));continue
138
- if I=="<class 'type'>"and P(E)<=z*4:
139
- R=B;S=F.endswith(N)or F.endswith('Error')or F in['KeyboardInterrupt','StopIteration','SystemExit']
140
- if S:R=N
141
- C='\n{}class {}({}):\n'.format(E,F,R)
142
- if S:C+=E+' ...\n';J.write(C);continue
143
- J.write(C);L.write_object_stub(J,b,'{0}.{1}'.format(obj_name,F),E+' ',O+1);C=E+' def __init__(self, *argv, **kwargs) -> None:\n';C+=E+' ...\n\n';J.write(C)
144
- elif any(A in I for A in[u,t,'closure']):
145
- T=V;U=B
146
- if O>0:U='self, '
147
- if W in I or W in H:C='{}@classmethod\n'.format(E)+'{}def {}(cls, *args, **kwargs) -> {}:\n'.format(E,F,T)
148
- else:C='{}def {}({}*args, **kwargs) -> {}:\n'.format(E,F,U,T)
149
- C+=E+' ...\n\n';J.write(C)
150
- elif I=="<class 'module'>":0
151
- elif I.startswith("<class '"):
152
- G=I[8:-2];C=B
153
- if G in(r,p,q,s,'bytearray','bytes'):C=X.format(E,F,H,G)
154
- elif G in(e,d,c):f={e:'{}',d:'[]',c:'()'};C=X.format(E,F,f[G],G)
155
- elif G in('object','set','frozenset','Pin',Y):
156
- if G==Y:G='Generator'
157
- C='{0}{1}: {2} ## = {4}\n'.format(E,F,G,I,H)
158
- else:
159
- G=V
160
- if K in H:H=H.split(K)[0]+Z
161
- if K in H:H=H.split(K)[0]+Z
162
- C='{0}{1}: {2} ## {3} = {4}\n'.format(E,F,G,I,H)
163
- J.write(C)
164
- else:J.write("# all other, type = '{0}'\n".format(I));J.write(E+F+' # type: Incomplete\n')
165
- @property
166
- def flat_fwid(self):
167
- A=self._fwid;B=' .()/\\:$'
168
- for C in B:A=A.replace(C,'_')
169
- return A
170
- def clean(C,path=E):
171
- if path is E:path=C.path
172
- A.info('Clean/remove files in folder: {}'.format(path))
173
- try:os.stat(path);D=os.listdir(path)
174
- except(F,I):return
175
- for G in D:
176
- B=v.format(path,G)
177
- try:os.remove(B)
178
- except F:
179
- try:C.clean(B);os.rmdir(B)
180
- except F:pass
181
- def report_start(B,filename='modules.json'):
182
- H='firmware';B._json_name=v.format(B.path,filename);B._json_first=R;X(B._json_name);A.info('Report file: {}'.format(B._json_name));D.collect()
183
- try:
184
- with O(B._json_name,'w')as G:G.write('{');G.write(dumps({H:B.info})[1:-1]);G.write(f);G.write(dumps({o:{C:__version__},'stubtype':H})[1:-1]);G.write(f);G.write('"modules" :[\n')
185
- except F as I:A.error(w);B._json_name=E;raise I
186
- def report_add(B,module_name,stub_file):
187
- if not B._json_name:raise n(x)
188
- try:
189
- with O(B._json_name,'a')as C:
190
- if not B._json_first:C.write(f)
191
- else:B._json_first=H
192
- D='{{"module": "{}", "file": "{}"}}'.format(module_name,stub_file.replace('\\',G));C.write(D)
193
- except F:A.error(w)
194
- def report_end(B):
195
- if not B._json_name:raise n(x)
196
- with O(B._json_name,'a')as C:C.write('\n]}')
197
- A.info('Path: {}'.format(B.path))
198
- def X(path):
199
- B=D=0
200
- while B!=-1:
201
- B=path.find(G,D)
202
- if B!=-1:
203
- C=path[0]if B==0 else path[:B]
204
- try:I=os.stat(C)
205
- except F as E:
206
- if E.args[0]==y:
207
- try:os.mkdir(C)
208
- except F as H:A.error('failed to create folder {}'.format(C));raise H
209
- D=B+1
210
- def Y(s):
211
- C=' on '
212
- if not s:return B
213
- s=s.split(C,1)[0]if C in s else s
214
- if s.startswith('v'):
215
- if not V in s:return B
216
- A=s.split(V)[1];return A
217
- if not W in s:return B
218
- A=s.split(W)[1].split(J)[1];return A
219
- def _info():
220
- c='ev3-pybricks';Z='pycom';X='pycopy';V='unix';S='win32';R='arch';P='cpu';O='ver';F='mpy';D='build'
221
- try:H=sys.implementation[0]
222
- except a:H=sys.implementation.name
223
- A=g({Q:H,C:B,D:B,O:B,K:sys.platform,U:'UNKNOWN',P:B,F:B,R:B})
224
- if A[K].startswith('pyb'):A[K]='stm32'
225
- elif A[K]==S:A[K]='windows'
226
- elif A[K]=='linux':A[K]=V
227
- try:A[C]=A1(sys.implementation.version)
228
- except I:pass
229
- try:J=sys.implementation._machine if'_machine'in M(sys.implementation)else os.uname().machine;A[U]=J;A[P]=J.split('with')[-1].strip();A[F]=sys.implementation._mpy if'_mpy'in M(sys.implementation)else sys.implementation.mpy if F in M(sys.implementation)else B
230
- except(I,T):pass
231
- A[U]=A2()
232
- try:
233
- if'uname'in M(os):
234
- A[D]=Y(os.uname()[3])
235
- if not A[D]:A[D]=Y(os.uname()[2])
236
- elif C in M(sys):A[D]=Y(sys.version)
237
- except(I,T,a):pass
238
- if A[C]==B and sys.platform not in(V,S):
239
- try:d=os.uname();A[C]=d.release
240
- except(T,I,a):pass
241
- for(e,f,h)in[(X,X,'const'),(Z,Z,'FAT'),(c,'pybricks.hubs','EV3Brick')]:
242
- try:i=__import__(f,E,E,h);A[Q]=e;del i;break
243
- except(N,m):pass
244
- if A[Q]==c:A['release']='2.0.0'
245
- if A[Q]==b:
246
- A[C]
247
- if A[C]and A[C].endswith('.0')and A[C]>='1.10.0'and A[C]<='1.19.9':A[C]=A[C][:-2]
248
- if F in A and A[F]:
249
- G=int(A[F]);L=[E,'x86','x64','armv6','armv6m','armv7m','armv7em','armv7emsp','armv7emdp','xtensa','xtensawin'][G>>10]
250
- if L:A[R]=L
251
- A[F]='v{}.{}'.format(G&255,G>>8&3)
252
- if A[D]and not A[C].endswith(W):A[C]=A[C]+W
253
- A[O]=f"{A[C]}-{A[D]}"if A[D]else f"{A[C]}";return A
254
- def A1(version):
255
- A=version;B=J.join([str(A)for A in A[:3]])
256
- if P(A)>3 and A[3]:B+=V+A[3]
257
- return B
258
- def A2():
259
- try:from boardname import BOARDNAME as C;A.info('Found BOARDNAME: {}'.format(C))
260
- except N:A.warning('BOARDNAME not found');C=B
261
- return C
262
- def get_root():
263
- try:A=os.getcwd()
264
- except(F,I):A=J
265
- B=A
266
- for B in[A,'/sd','/flash',G,J]:
267
- try:C=os.stat(B);break
268
- except F:continue
269
- return B
270
- def h(filename):
271
- try:
272
- if os.stat(filename)[0]>>14:return R
273
- return H
274
- except F:return H
275
- def i():S("-p, --path path to store the stubs in, defaults to '.'");sys.exit(1)
276
- def read_path():
277
- path=B
278
- if P(sys.argv)==3:
279
- A=sys.argv[1].lower()
280
- if A in('--path','-p'):path=sys.argv[2]
281
- else:i()
282
- elif P(sys.argv)==2:i()
283
- return path
284
- def j():
285
- try:A=bytes('abc',encoding='utf8');B=j.__module__;return H
286
- except(k,I):return R
287
- def main():
288
- stubber=Stubber(path=read_path());stubber.clean()
289
- def A(stubber):
290
- D.collect();stubber.modules=[]
291
- for C in A0:
292
- B=C+'/modulelist.txt'
293
- if not h(B):continue
294
- with O(B)as E:
295
- while R:
296
- A=E.readline().strip()
297
- if not A:break
298
- if P(A)>0 and A[0]!='#':stubber.modules.append(A)
299
- D.collect();S('BREAK');break
300
- if not stubber.modules:stubber.modules=[b]
301
- D.collect()
302
- stubber.modules=[];A(stubber);D.collect();stubber.create_all_stubs()
303
- if __name__=='__main__'or j():
304
- if not h('no_auto_stubber.txt'):
305
- try:D.threshold(4*1024);D.enable()
306
- except BaseException:pass
1
+ y='No report file'
2
+ x='Failed to create the report.'
3
+ w='{}/{}'
4
+ v='method'
5
+ u='function'
6
+ t='bool'
7
+ s='str'
8
+ r='float'
9
+ q='int'
10
+ p='stubber'
11
+ o=Exception
12
+ n=KeyError
13
+ m=sorted
14
+ l=NotImplementedError
15
+ g=',\n'
16
+ f='dict'
17
+ e='list'
18
+ d='tuple'
19
+ c='micropython'
20
+ b=TypeError
21
+ a=repr
22
+ X='-preview'
23
+ W='-'
24
+ V='board'
25
+ U=IndexError
26
+ T=print
27
+ R=True
28
+ Q='family'
29
+ P=len
30
+ O=open
31
+ N=ImportError
32
+ M=dir
33
+ K='port'
34
+ J='.'
35
+ I=AttributeError
36
+ H=False
37
+ G='/'
38
+ F=OSError
39
+ E=None
40
+ C='version'
41
+ B=''
42
+ import gc as D,os,sys
43
+ from time import sleep
44
+ try:from ujson import dumps
45
+ except:from json import dumps
46
+ try:from machine import reset
47
+ except N:pass
48
+ try:from collections import OrderedDict as h
49
+ except N:from ucollections import OrderedDict as h
50
+ __version__='v1.23.2'
51
+ S=__version__.rsplit(J,1)[0]
52
+ z=2
53
+ A0=2
54
+ A1=['lib','/lib','/sd/lib','/flash/lib',J]
55
+ class L:
56
+ INFO=20;WARNING=30;ERROR=40;level=INFO;prnt=T
57
+ @staticmethod
58
+ def getLogger(name):return L()
59
+ @classmethod
60
+ def basicConfig(A,level):A.level=level
61
+ def info(A,msg):
62
+ if A.level<=L.INFO:A.prnt('INFO :',msg)
63
+ def warning(A,msg):
64
+ if A.level<=L.WARNING:A.prnt('WARN :',msg)
65
+ def error(A,msg):
66
+ if A.level<=L.ERROR:A.prnt('ERROR :',msg)
67
+ A=L.getLogger(p)
68
+ L.basicConfig(level=L.INFO)
69
+ class Stubber:
70
+ def __init__(B,path=E,firmware_id=E):
71
+ C=firmware_id
72
+ try:
73
+ if os.uname().release=='1.13.0'and os.uname().version<'v1.13-103':raise l('MicroPython 1.13.0 cannot be stubbed')
74
+ except I:pass
75
+ B.info=_info();A.info('Port: {}'.format(B.info[K]));A.info('Board: {}'.format(B.info[V]));D.collect()
76
+ if C:B._fwid=C.lower()
77
+ elif B.info[Q]==c:B._fwid='{family}-v{version}-{port}-{board}'.format(**B.info).rstrip(W)
78
+ else:B._fwid='{family}-v{version}-{port}'.format(**B.info)
79
+ B._start_free=D.mem_free()
80
+ if path:
81
+ if path.endswith(G):path=path[:-1]
82
+ else:path=get_root()
83
+ B.path='{}/stubs/{}'.format(path,B.flat_fwid).replace('//',G)
84
+ try:Y(path+G)
85
+ except F:A.error('error creating stub folder {}'.format(path))
86
+ B.problematic=['upip','upysh','webrepl_setup','http_client','http_client_ssl','http_server','http_server_ssl'];B.excluded=['webrepl','_webrepl','port_diag','example_sub_led.py','example_pub_button.py'];B.modules=[];B._json_name=E;B._json_first=H
87
+ def get_obj_attributes(L,item_instance):
88
+ H=item_instance;C=[];K=[]
89
+ for A in M(H):
90
+ if A.startswith('__')and not A in L.modules:continue
91
+ try:
92
+ E=getattr(H,A)
93
+ try:F=a(type(E)).split("'")[1]
94
+ except U:F=B
95
+ if F in{q,r,s,t,d,e,f}:G=1
96
+ elif F in{u,v}:G=2
97
+ elif F in'class':G=3
98
+ else:G=4
99
+ C.append((A,a(E),a(type(E)),E,G))
100
+ except I as J:K.append("Couldn't get attribute '{}' from object '{}', Err: {}".format(A,H,J))
101
+ except MemoryError as J:T('MemoryError: {}'.format(J));sleep(1);reset()
102
+ C=m([A for A in C if not A[0].startswith('__')],key=lambda x:x[4]);D.collect();return C,K
103
+ def add_modules(A,modules):A.modules=m(set(A.modules)|set(modules))
104
+ def create_all_stubs(B):
105
+ A.info('Start micropython-stubber {} on {}'.format(__version__,B._fwid));B.report_start();D.collect()
106
+ for C in B.modules:B.create_one_stub(C)
107
+ B.report_end();A.info('Finally done')
108
+ def create_one_stub(C,module_name):
109
+ B=module_name
110
+ if B in C.problematic:A.warning('Skip module: {:<25} : Known problematic'.format(B));return H
111
+ if B in C.excluded:A.warning('Skip module: {:<25} : Excluded'.format(B));return H
112
+ I='{}/{}.pyi'.format(C.path,B.replace(J,G));D.collect();E=H
113
+ try:E=C.create_module_stub(B,I)
114
+ except F:return H
115
+ D.collect();return E
116
+ def create_module_stub(K,module_name,file_name=E):
117
+ I=file_name;C=module_name
118
+ if I is E:L=C.replace(J,'_')+'.pyi';I=K.path+G+L
119
+ else:L=I.split(G)[-1]
120
+ if G in C:C=C.replace(G,J)
121
+ M=E
122
+ try:M=__import__(C,E,E,'*');Q=D.mem_free();A.info('Stub module: {:<25} to file: {:<70} mem:{:>5}'.format(C,L,Q))
123
+ except N:return H
124
+ Y(I)
125
+ with O(I,'w')as P:T=str(K.info).replace('OrderedDict(',B).replace('})','}');U='"""\nModule: \'{0}\' on {1}\n"""\n# MCU: {2}\n# Stubber: {3}\n'.format(C,K._fwid,T,S);P.write(U);P.write('from __future__ import annotations\nfrom typing import Any, Generator\nfrom _typeshed import Incomplete\n\n');K.write_object_stub(P,M,C,B)
126
+ K.report_add(C,I)
127
+ if C not in{'os','sys','logging','gc'}:
128
+ try:del M
129
+ except(F,n):A.warning('could not del new_module')
130
+ D.collect();return R
131
+ def write_object_stub(L,fp,object_expr,obj_name,indent,in_class=0):
132
+ Z=' at ...>';Y='generator';X='{0}{1}: {3} = {2}\n';W='bound_method';V='Incomplete';O=in_class;N='Exception';M=object_expr;K=' at ';J=fp;E=indent;D.collect()
133
+ if M in L.problematic:A.warning('SKIPPING problematic module:{}'.format(M));return
134
+ a,Q=L.get_obj_attributes(M)
135
+ if Q:A.error(Q)
136
+ for(F,H,I,b,g)in a:
137
+ if F in['classmethod','staticmethod','BaseException',N]:continue
138
+ if F[0].isdigit():A.warning('NameError: invalid name {}'.format(F));continue
139
+ if I=="<class 'type'>"and P(E)<=A0*4:
140
+ R=B;S=F.endswith(N)or F.endswith('Error')or F in['KeyboardInterrupt','StopIteration','SystemExit']
141
+ if S:R=N
142
+ C='\n{}class {}({}):\n'.format(E,F,R)
143
+ if S:C+=E+' ...\n';J.write(C);continue
144
+ J.write(C);L.write_object_stub(J,b,'{0}.{1}'.format(obj_name,F),E+' ',O+1);C=E+' def __init__(self, *argv, **kwargs) -> None:\n';C+=E+' ...\n\n';J.write(C)
145
+ elif any(A in I for A in[v,u,'closure']):
146
+ T=V;U=B
147
+ if O>0:U='self, '
148
+ if W in I or W in H:C='{}@classmethod\n'.format(E)+'{}def {}(cls, *args, **kwargs) -> {}:\n'.format(E,F,T)
149
+ else:C='{}def {}({}*args, **kwargs) -> {}:\n'.format(E,F,U,T)
150
+ C+=E+' ...\n\n';J.write(C)
151
+ elif I=="<class 'module'>":0
152
+ elif I.startswith("<class '"):
153
+ G=I[8:-2];C=B
154
+ if G in(s,q,r,t,'bytearray','bytes'):C=X.format(E,F,H,G)
155
+ elif G in(f,e,d):c={f:'{}',e:'[]',d:'()'};C=X.format(E,F,c[G],G)
156
+ elif G in('object','set','frozenset','Pin',Y):
157
+ if G==Y:G='Generator'
158
+ C='{0}{1}: {2} ## = {4}\n'.format(E,F,G,I,H)
159
+ else:
160
+ G=V
161
+ if K in H:H=H.split(K)[0]+Z
162
+ if K in H:H=H.split(K)[0]+Z
163
+ C='{0}{1}: {2} ## {3} = {4}\n'.format(E,F,G,I,H)
164
+ J.write(C)
165
+ else:J.write("# all other, type = '{0}'\n".format(I));J.write(E+F+' # type: Incomplete\n')
166
+ @property
167
+ def flat_fwid(self):
168
+ A=self._fwid;B=' .()/\\:$'
169
+ for C in B:A=A.replace(C,'_')
170
+ return A
171
+ def clean(C,path=E):
172
+ if path is E:path=C.path
173
+ A.info('Clean/remove files in folder: {}'.format(path))
174
+ try:os.stat(path);D=os.listdir(path)
175
+ except(F,I):return
176
+ for G in D:
177
+ B=w.format(path,G)
178
+ try:os.remove(B)
179
+ except F:
180
+ try:C.clean(B);os.rmdir(B)
181
+ except F:pass
182
+ def report_start(B,filename='modules.json'):
183
+ H='firmware';B._json_name=w.format(B.path,filename);B._json_first=R;Y(B._json_name);A.info('Report file: {}'.format(B._json_name));D.collect()
184
+ try:
185
+ with O(B._json_name,'w')as G:G.write('{');G.write(dumps({H:B.info})[1:-1]);G.write(g);G.write(dumps({p:{C:S},'stubtype':H})[1:-1]);G.write(g);G.write('"modules" :[\n')
186
+ except F as I:A.error(x);B._json_name=E;raise I
187
+ def report_add(B,module_name,stub_file):
188
+ if not B._json_name:raise o(y)
189
+ try:
190
+ with O(B._json_name,'a')as C:
191
+ if not B._json_first:C.write(g)
192
+ else:B._json_first=H
193
+ D='{{"module": "{}", "file": "{}"}}'.format(module_name,stub_file.replace('\\',G));C.write(D)
194
+ except F:A.error(x)
195
+ def report_end(B):
196
+ if not B._json_name:raise o(y)
197
+ with O(B._json_name,'a')as C:C.write('\n]}')
198
+ A.info('Path: {}'.format(B.path))
199
+ def Y(path):
200
+ B=D=0
201
+ while B!=-1:
202
+ B=path.find(G,D)
203
+ if B!=-1:
204
+ C=path[0]if B==0 else path[:B]
205
+ try:I=os.stat(C)
206
+ except F as E:
207
+ if E.args[0]==z:
208
+ try:os.mkdir(C)
209
+ except F as H:A.error('failed to create folder {}'.format(C));raise H
210
+ D=B+1
211
+ def Z(s):
212
+ C=' on '
213
+ if not s:return B
214
+ s=s.split(C,1)[0]if C in s else s
215
+ if s.startswith('v'):
216
+ if not W in s:return B
217
+ A=s.split(W)[1];return A
218
+ if not X in s:return B
219
+ A=s.split(X)[1].split(J)[1];return A
220
+ def _info():
221
+ d='ev3-pybricks';a='pycom';Y='pycopy';W='unix';T='win32';R='arch';P='cpu';O='ver';F='mpy';D='build'
222
+ try:H=sys.implementation[0]
223
+ except b:H=sys.implementation.name
224
+ A=h({Q:H,C:B,D:B,O:B,K:sys.platform,V:'UNKNOWN',P:B,F:B,R:B})
225
+ if A[K].startswith('pyb'):A[K]='stm32'
226
+ elif A[K]==T:A[K]='windows'
227
+ elif A[K]=='linux':A[K]=W
228
+ try:A[C]=S(sys.implementation.version)
229
+ except I:pass
230
+ try:J=sys.implementation._machine if'_machine'in M(sys.implementation)else os.uname().machine;A[V]=J;A[P]=J.split('with')[-1].strip();A[F]=sys.implementation._mpy if'_mpy'in M(sys.implementation)else sys.implementation.mpy if F in M(sys.implementation)else B
231
+ except(I,U):pass
232
+ A[V]=A2()
233
+ try:
234
+ if'uname'in M(os):
235
+ A[D]=Z(os.uname()[3])
236
+ if not A[D]:A[D]=Z(os.uname()[2])
237
+ elif C in M(sys):A[D]=Z(sys.version)
238
+ except(I,U,b):pass
239
+ if A[C]==B and sys.platform not in(W,T):
240
+ try:e=os.uname();A[C]=e.release
241
+ except(U,I,b):pass
242
+ for(f,g,i)in[(Y,Y,'const'),(a,a,'FAT'),(d,'pybricks.hubs','EV3Brick')]:
243
+ try:j=__import__(g,E,E,i);A[Q]=f;del j;break
244
+ except(N,n):pass
245
+ if A[Q]==d:A['release']='2.0.0'
246
+ if A[Q]==c:
247
+ A[C]
248
+ if A[C]and A[C].endswith('.0')and A[C]>='1.10.0'and A[C]<='1.19.9':A[C]=A[C][:-2]
249
+ if F in A and A[F]:
250
+ G=int(A[F]);L=[E,'x86','x64','armv6','armv6m','armv7m','armv7em','armv7emsp','armv7emdp','xtensa','xtensawin'][G>>10]
251
+ if L:A[R]=L
252
+ A[F]='v{}.{}'.format(G&255,G>>8&3)
253
+ if A[D]and not A[C].endswith(X):A[C]=A[C]+X
254
+ A[O]=f"{A[C]}-{A[D]}"if A[D]else f"{A[C]}";return A
255
+ def S(version):
256
+ A=version;B=J.join([str(A)for A in A[:3]])
257
+ if P(A)>3 and A[3]:B+=W+A[3]
258
+ return B
259
+ def A2():
260
+ try:from boardname import BOARDNAME as C;A.info('Found BOARDNAME: {}'.format(C))
261
+ except N:A.warning('BOARDNAME not found');C=B
262
+ return C
263
+ def get_root():
264
+ try:A=os.getcwd()
265
+ except(F,I):A=J
266
+ B=A
267
+ for B in[A,'/sd','/flash',G,J]:
268
+ try:C=os.stat(B);break
269
+ except F:continue
270
+ return B
271
+ def i(filename):
272
+ try:
273
+ if os.stat(filename)[0]>>14:return R
274
+ return H
275
+ except F:return H
276
+ def j():T("-p, --path path to store the stubs in, defaults to '.'");sys.exit(1)
277
+ def read_path():
278
+ path=B
279
+ if P(sys.argv)==3:
280
+ A=sys.argv[1].lower()
281
+ if A in('--path','-p'):path=sys.argv[2]
282
+ else:j()
283
+ elif P(sys.argv)==2:j()
284
+ return path
285
+ def k():
286
+ try:A=bytes('abc',encoding='utf8');B=k.__module__;return H
287
+ except(l,I):return R
288
+ def main():
289
+ stubber=Stubber(path=read_path());stubber.clean()
290
+ def A(stubber):
291
+ D.collect();stubber.modules=[]
292
+ for C in A1:
293
+ B=C+'/modulelist.txt'
294
+ if not i(B):continue
295
+ with O(B)as E:
296
+ while R:
297
+ A=E.readline().strip()
298
+ if not A:break
299
+ if P(A)>0 and A[0]!='#':stubber.modules.append(A)
300
+ D.collect();T('BREAK');break
301
+ if not stubber.modules:stubber.modules=[c]
302
+ D.collect()
303
+ stubber.modules=[];A(stubber);D.collect();stubber.create_all_stubs()
304
+ if __name__=='__main__'or k():
305
+ if not i('no_auto_stubber.txt'):
306
+ try:D.threshold(4*1024);D.enable()
307
+ except BaseException:pass
307
308
  main()
Binary file