djsundog/circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed extract_pyi script to allow NoneType

Browse Source
This commit is contained in:
dherrada 2020-07-02 10:25:20 -04:00
parent 596f2f6e7d
commit d1664bdde2
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tools/extract_pyi.py
View File

@ -1,3 +1,7 @@
# Run with 'python tools/extract_pyi.py shared-bindings/ path/to/stub/dir
# You can also test a specific library in shared-bindings by putting the path
# to that directory instead
import os
import sys
import astroid
@ -58,8 +62,10 @@ def convert_folder(top_level, stub_directory):
print(i.__dict__['name'])
for j in i.body:
if isinstance(j, astroid.scoped_nodes.FunctionDef):
if None in j.args.__dict__['annotations']:
print(f"Missing parameter type: {j.__dict__['name']} on line {j.__dict__['lineno']}\n")
for i in j.args.__dict__['annotations']:
if type(i) == astroid.node_classes.Name:
if i.name == 'Any':
print(f"Missing parameter type: {j.__dict__['name']} on line {j.__dict__['lineno']}\n")
if j.returns:
if 'Any' in j.returns.__dict__.values():
print(f"Missing return type: {j.__dict__['name']} on line {j.__dict__['lineno']}")