Merge pull request #2 from tannewt/improve_verification

Better handle //| and do __init__.c first.
This commit is contained in:
dherrada 2020-04-29 13:30:40 -04:00 committed by GitHub
commit 8cdc67a157
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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.startswith("//|"):
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)