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

Better handle //| and do __init__.c first.

Browse Source
This commit is contained in:
Scott Shawcroft 2020-04-29 10:29:12 -07:00
parent c7b721f4f1
commit 8c77252adf
No known key found for this signature in database
GPG Key ID: 9349BC7E64B1921E
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tools/extract_pyi.py
View File

@ -20,12 +20,22 @@ for module in modules:
if not os.path.isdir(module_path):
continue
pyi_lines = []
for class_file in os.listdir(module_path):
classes = os.listdir(module_path)
classes = [x for x in sorted(classes) if x.endswith(".c")]
if classes[-1] == "__init__.c":
classes.insert(0, classes.pop())
for class_file in classes:
class_path = os.path.join(module_path, class_file)
with open(class_path, "r") as f:
for line in f:
if line.startswith("//|"):
pyi_lines.append(line[4:])
if line[3] == " ":
line = line[4:]
elif line[3] == "\n":
line = line[3:]
else:
continue
pyi_lines.append(line)
stub_filename = os.path.join(stub_directory, module + ".pyi")
print(stub_filename)