49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
![]() |
from __future__ import print_function
|
||
|
import sys
|
||
|
import os
|
||
|
import argparse
|
||
|
from .unparser import roundtrip
|
||
|
from . import dump
|
||
|
|
||
|
|
||
|
def roundtrip_recursive(target, dump_tree=False):
|
||
|
if os.path.isfile(target):
|
||
|
print(target)
|
||
|
print("=" * len(target))
|
||
|
if dump_tree:
|
||
|
dump(target)
|
||
|
else:
|
||
|
roundtrip(target)
|
||
|
print()
|
||
|
elif os.path.isdir(target):
|
||
|
for item in os.listdir(target):
|
||
|
if item.endswith(".py"):
|
||
|
roundtrip_recursive(os.path.join(target, item), dump_tree)
|
||
|
else:
|
||
|
print(
|
||
|
"WARNING: skipping '%s', not a file or directory" % target,
|
||
|
file=sys.stderr
|
||
|
)
|
||
|
|
||
|
|
||
|
def main(args):
|
||
|
parser = argparse.ArgumentParser(prog="astunparse")
|
||
|
parser.add_argument(
|
||
|
'target',
|
||
|
nargs='+',
|
||
|
help="Files or directories to show roundtripped source for"
|
||
|
)
|
||
|
parser.add_argument(
|
||
|
'--dump',
|
||
|
type=bool,
|
||
|
help="Show a pretty-printed AST instead of the source"
|
||
|
)
|
||
|
|
||
|
arguments = parser.parse_args(args)
|
||
|
for target in arguments.target:
|
||
|
roundtrip_recursive(target, dump_tree=arguments.dump)
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main(sys.argv[1:])
|