···

       
1
       
1
       
 
       
import ast

     

       
2
       
2
       
 
       
import sys

     

       
3
       
3
       
+
       
from pathlib import Path

     

       
3
       
4
       
 
       


     

       
4
       
5
       
 
       
"""

     

       
5
       
6
       
 
       
This program takes all the Machine class methods and prints its methods in

     
···

       
40
       
41
       
 
       


     

       
41
       
42
       
 
       
"""

     

       
42
       
43
       
 
       


     

       
43
       
43
       
-
       
assert len(sys.argv) == 2

     

       
44
       
44
       
+
       


     

       
45
       
45
       
+
       
def main() -> None:

     

       
46
       
46
       
+
       
    if len(sys.argv) != 2:

     

       
47
       
47
       
+
       
        print(f"Usage: {sys.argv[0]} <path-to-test-driver>")

     

       
48
       
48
       
+
       
        sys.exit(1)

     

       
49
       
49
       
+
       


     

       
50
       
50
       
+
       
    module = ast.parse(Path(sys.argv[1]).read_text())

     

       
51
       
51
       
+
       


     

       
52
       
52
       
+
       
    class_definitions = (node for node in module.body if isinstance(node, ast.ClassDef))

     

       
44
       
53
       
 
       


     

       
45
       
45
       
-
       
with open(sys.argv[1], "r") as f:

     

       
46
       
46
       
-
       
    module = ast.parse(f.read())

     

       
54
       
54
       
+
       
    machine_class = next(filter(lambda x: x.name == "Machine", class_definitions))

     

       
55
       
55
       
+
       
    assert machine_class is not None

     

       
47
       
56
       
 
       


     

       
48
       
48
       
-
       
class_definitions = (node for node in module.body if isinstance(node, ast.ClassDef))

     

       
57
       
57
       
+
       
    function_definitions = [

     

       
58
       
58
       
+
       
        node for node in machine_class.body if isinstance(node, ast.FunctionDef)

     

       
59
       
59
       
+
       
    ]

     

       
60
       
60
       
+
       
    function_definitions.sort(key=lambda x: x.name)

     

       
49
       
61
       
 
       


     

       
50
       
50
       
-
       
machine_class = next(filter(lambda x: x.name == "Machine", class_definitions))

     

       
51
       
51
       
-
       
assert machine_class is not None

     

       
62
       
62
       
+
       
    for f in function_definitions:

     

       
63
       
63
       
+
       
        docstr = ast.get_docstring(f)

     

       
64
       
64
       
+
       
        if docstr is not None:

     

       
65
       
65
       
+
       
            args = ", ".join(a.arg for a in f.args.args[1:])

     

       
66
       
66
       
+
       
            args = f"({args})"

     

       
52
       
67
       
 
       


     

       
53
       
53
       
-
       
function_definitions = [

     

       
54
       
54
       
-
       
    node for node in machine_class.body if isinstance(node, ast.FunctionDef)

     

       
55
       
55
       
-
       
]

     

       
56
       
56
       
-
       
function_definitions.sort(key=lambda x: x.name)

     

       
68
       
68
       
+
       
            docstr = "\n".join(f"    {l}" for l in docstr.strip().splitlines())

     

       
57
       
69
       
 
       


     

       
58
       
58
       
-
       
for f in function_definitions:

     

       
59
       
59
       
-
       
    docstr = ast.get_docstring(f)

     

       
60
       
60
       
-
       
    if docstr is not None:

     

       
61
       
61
       
-
       
        args = ", ".join((a.arg for a in f.args.args[1:]))

     

       
62
       
62
       
-
       
        args = f"({args})"

     

       
70
       
70
       
+
       
            print(f"{f.name}{args}\n\n:{docstr[1:]}\n")

     

       
63
       
71
       
 
       


     

       
64
       
64
       
-
       
        docstr = "\n".join((f"    {l}" for l in docstr.strip().splitlines()))

     

       
65
       
72
       
 
       


     

       
66
       
66
       
-
       
        print(f"{f.name}{args}\n\n:{docstr[1:]}\n")

     

       
73
       
73
       
+
       
if __name__ == "__main__":

     

       
74
       
74
       
+
       
    main()