summaryrefslogtreecommitdiffstats
path: root/Tools/peg_generator/scripts/ast_timings.py
blob: 7ebd46fdac6851589c53c094c1703b4d5449102c (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
import ast
import sys
import time
import token
import tokenize

from pegen.testutil import print_memstats


def main() -> None:
    t0 = time.time()
    for filename in sys.argv[1:]:
        print(filename, end="\r")
        try:
            with open(filename) as file:
                source = file.read()
            tree = ast.parse(source, filename)
        except Exception as err:
            print(f"{filename}: {err.__class__.__name__}: {err}", file=sys.stderr)
    tok = None
    t1 = time.time()
    dt = t1 - t0
    print(f"Parsed in {dt:.3f} secs", file=sys.stderr)
    print_memstats()


if __name__ == "__main__":
    main()