Mercurial > hg > config
changeset 670:93dc0507ab3b
to argparse
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Tue, 29 Apr 2014 15:01:40 -0700 |
parents | bcf5781b615e |
children | eeb38dfa17d0 |
files | python/tree.py |
diffstat | 1 files changed, 17 insertions(+), 11 deletions(-) [+] |
line wrap: on
line diff
--- a/python/tree.py Thu Apr 17 13:06:12 2014 -0700 +++ b/python/tree.py Tue Apr 29 15:01:40 2014 -0700 @@ -7,7 +7,7 @@ # TODO: script2package -import optparse +import argparse import os import sys @@ -100,6 +100,7 @@ item_marker=unicode_delimeters['item_marker'], vertical_line=unicode_delimeters['vertical_line'], last_child=unicode_delimeters['last_child'], + display_files=True, sort_key=lambda x: x.lower()): """ display tree directory structure for `directory` @@ -159,19 +160,24 @@ return '\n'.join(retval) def main(args=sys.argv[1:]): + """CLI""" # parse command line options - usage = '%prog [options]' - parser = optparse.OptionParser(usage=usage, description=__doc__) - parser.add_option('-a', '--ascii', dest='use_ascii', - action='store_true', default=False, - help="use ascii delimeters (%s)" % ascii_delimeters) - options, args = parser.parse_args(args) - if not args: - args = ['.'] + parser = argparse.ArgumentParser(description=__doc__) + parser.add_argument('-a', '--ascii', dest='use_ascii', + action='store_true', default=False, + help="use ascii delimeters ({})".format(', '.join(ascii_delimeters.values()))) + parser.add_argument('path', nargs='*', + help="paths to display the tree of") + options = parser.parse_args(args) + + # get paths to operate on + paths = options.path + if not paths: + paths = ['.'] # sanity check - not_directory = [arg for arg in args + not_directory = [arg for arg in paths if not os.path.isdir(arg)] if not_directory: parser.error("Not a directory: %s" % (', '.join(not_directory))) @@ -181,7 +187,7 @@ delimeters = ascii_delimeters # print the tree - for arg in args: + for arg in paths: print (tree(arg, **delimeters)) if __name__ == '__main__':