blob: 95370f7619a7f362f77f71294e7f7ada9dae1699 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import os
import sys
import astroid
import traceback
top_level = sys.argv[1].strip("/")
stub_directory = sys.argv[2]
if top_level.count("/") == 1:
top_level, module = top_level.split("/")
modules = [module]
else:
modules = os.listdir(top_level)
modules = sorted(modules)
ok = 0
total = 0
for module in modules:
module_path = os.path.join(top_level, module)
if not os.path.isdir(module_path):
continue
pyi_lines = []
classes = os.listdir(module_path)
classes = [x for x in sorted(classes) if x.endswith(".c")]
if classes and 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("//|"):
if line[3] == " ":
line = line[4:]
elif line[3] == "\n":
line = line[3:]
else:
continue
pyi_lines.append(line)
raw_stubs = [x for x in sorted(classes) if x.endswith(".pyi")]
if raw_stubs and raw_stubs[-1] == "__init__.pyi":
raw_stubs.insert(0, raw_stubs.pop())
for raw_stub in raw_stubs:
raw_stub_path = os.path.join(module_path, raw_stub)
with open(raw_stub_path, "r") as f:
pyi_lines.extend(f.readlines())
stub_filename = os.path.join(stub_directory, module + ".pyi")
print(stub_filename)
stub_contents = "".join(pyi_lines)
with open(stub_filename, "w") as f:
f.write(stub_contents)
# Validate that the module is a parseable stub.
total += 1
try:
astroid.parse(stub_contents)
ok += 1
except astroid.exceptions.AstroidSyntaxError as e:
e = e.__cause__
traceback.print_exception(type(e), e, e.__traceback__)
print()
print(f"{ok} ok out of {total}")
if ok != total:
sys.exit(total - ok)
|
