Mercurial > hg > decoupage
diff decoupage/index.py @ 107:450aff4c97e3
py35 compat
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Fri, 31 Mar 2017 17:06:59 -0700 |
parents | 4a9c5cf9fec9 |
children |
line wrap: on
line diff
--- a/decoupage/index.py Fri Mar 31 16:38:02 2017 -0700 +++ b/decoupage/index.py Fri Mar 31 17:06:59 2017 -0700 @@ -7,19 +7,25 @@ import argparse import os -import subprocess import sys here = os.path.dirname(os.path.realpath(__file__)) -string = (str, unicode) def index(directory): + """ + returns string representation of directory + """ + return '\n'.join(['{name} = {name}'.format(name=name) - for name in sorted(os.listdir(directory), key=lambda name: name.lower()) + for name in sorted(os.listdir(directory), + key=lambda name: name.lower()) if not name.startswith('.')]) + def main(args=sys.argv[1:]): + """CLI""" + # parse command line parser = argparse.ArgumentParser(description=__doc__) parser.add_argument('directory', help='directory') parser.add_argument('-o', '--output', dest='output', @@ -27,5 +33,13 @@ help='output') options = parser.parse_args(args) + # sanity + if not os.path.isdir(options.directory): + parser.error("Not a directory: '{}'".format(options.directory)) + + # output + options.output.write(index(options.directory)) + + if __name__ == '__main__': main()