dump-ast.py

#!/usr/bin/env python3
"""
Parse source files and print the abstract syntax trees.
"""

from __future__ importannotations

importargparse
importsys

frommypyimportdefaults
frommypy.errorsimportCompileError, Errors
frommypy.optionsimportOptions
frommypy.parseimportparse


defdump(fname: str, python_version: tuple[int, int], quiet: bool=False) ->None:
options=Options()
options.python_version=python_version
withopen(fname, "rb") asf:
s=f.read()
tree=parse(s, fname, None, errors=Errors(options), options=options)
ifnotquiet:
print(tree)


defmain() ->None:
# Parse a file and dump the AST (or display errors).
parser=argparse.ArgumentParser(
description="Parse source files and print the abstract syntax tree (AST)."
 )
parser.add_argument("--quiet", action="store_true", help="do not print AST")
parser.add_argument("FILE", nargs="*", help="files to parse")
args=parser.parse_args()

status=0
forfnameinargs.FILE:
try:
dump(fname, defaults.PYTHON3_VERSION, args.quiet)
exceptCompileErrorase:
formsgine.messages:
sys.stderr.write("%s\n"%msg)
status=1
sys.exit(status)


if__name__=="__main__":
main()