Mercurial > hg > MakeItSo
diff makeitso/mkpydir.py @ 189:7d8f3660fabb
fix syntax error and use argparse
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Fri, 02 May 2014 10:38:12 -0700 |
parents | f6474c7dfb39 |
children |
line wrap: on
line diff
--- a/makeitso/mkpydir.py Fri Apr 11 18:50:40 2014 -0700 +++ b/makeitso/mkpydir.py Fri May 02 10:38:12 2014 -0700 @@ -4,22 +4,20 @@ make a python module directory with an __init__.py """ -import optparse +import argparse import os import sys def main(args=sys.argv[1:]): + """CLI""" - usage = '%prog [options] directory_name' - parser = optparse.OptionParser(usage=usage, description=__doc__) - options, args = parser.parse_args(args) - if len(args) != 1: - parser.print_usage() - parser.error() + parser = argparse.ArgumentParser(description=__doc__) + parser.add_argument('directory') + options = parser.parse_args(args) - os.makedirs(args[0]) - init = os.path.join(args[0], '__init__.py') - with f as open(init, 'w'): + os.makedirs(options.directory) + init = os.path.join(options.directory, '__init__.py') + with open(init, 'w') as f: f.write('#\n') if __name__ == '__main__':