Mercurial > hg > MakeItSo
comparison 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 |
comparison
equal
deleted
inserted
replaced
| 188:1bcf60bcc455 | 189:7d8f3660fabb |
|---|---|
| 2 | 2 |
| 3 """ | 3 """ |
| 4 make a python module directory with an __init__.py | 4 make a python module directory with an __init__.py |
| 5 """ | 5 """ |
| 6 | 6 |
| 7 import optparse | 7 import argparse |
| 8 import os | 8 import os |
| 9 import sys | 9 import sys |
| 10 | 10 |
| 11 def main(args=sys.argv[1:]): | 11 def main(args=sys.argv[1:]): |
| 12 """CLI""" | |
| 12 | 13 |
| 13 usage = '%prog [options] directory_name' | 14 parser = argparse.ArgumentParser(description=__doc__) |
| 14 parser = optparse.OptionParser(usage=usage, description=__doc__) | 15 parser.add_argument('directory') |
| 15 options, args = parser.parse_args(args) | 16 options = parser.parse_args(args) |
| 16 if len(args) != 1: | |
| 17 parser.print_usage() | |
| 18 parser.error() | |
| 19 | 17 |
| 20 os.makedirs(args[0]) | 18 os.makedirs(options.directory) |
| 21 init = os.path.join(args[0], '__init__.py') | 19 init = os.path.join(options.directory, '__init__.py') |
| 22 with f as open(init, 'w'): | 20 with open(init, 'w') as f: |
| 23 f.write('#\n') | 21 f.write('#\n') |
| 24 | 22 |
| 25 if __name__ == '__main__': | 23 if __name__ == '__main__': |
| 26 main() | 24 main() |
