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

16
tools/extract_pyi.py
View File

@ -20,12 +20,22 @@ for module in modules:
if not os.path.isdir(module_path): if not os.path.isdir(module_path):
continue continue
pyi_lines = [] 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) class_path = os.path.join(module_path, class_file)
with open(class_path, "r") as f: with open(class_path, "r") as f:
for line in f: for line in f:
if line.startswith("//| "): 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") stub_filename = os.path.join(stub_directory, module + ".pyi")
print(stub_filename) print(stub_filename)