diff options
author | Zach Smith <zd@zdsmith.com> | 2018-04-25 22:55:52 -0400 |
---|---|---|
committer | Zach Smith <zd@zdsmith.com> | 2018-04-25 22:55:52 -0400 |
commit | 101b234b180a04a859ebd80db69221ace7de1021 (patch) | |
tree | e243e58b6ca35509a5e45a1a30cf0ca414474695 | |
parent | 185f999fad86967f98926b97a1ef1e220fbce39b (diff) | |
download | pycco-101b234b180a04a859ebd80db69221ace7de1021.tar.gz |
migrate optparse -> argparse
-rw-r--r-- | pycco/main.py | 40 |
1 files changed, 21 insertions, 19 deletions
diff --git a/pycco/main.py b/pycco/main.py index c062fbd..63b6e99 100644 --- a/pycco/main.py +++ b/pycco/main.py @@ -1,18 +1,18 @@ #!/usr/bin/env python -from __future__ import print_function, absolute_import +from __future__ import absolute_import, print_function # Import our external dependencies. -import optparse +import argparse import os import re import sys import time from os import path -from pycco.generate_index import generate_index -from pycco.languages import supported_languages import pygments from markdown import markdown +from pycco.generate_index import generate_index +from pycco.languages import supported_languages from pycco_resources import css as pycco_css # This module contains all of our static resources. from pycco_resources import pycco_template @@ -577,49 +577,51 @@ def main(): Hook spot for the console script. """ - parser = optparse.OptionParser() - parser.add_option('-p', '--paths', action='store_true', + parser = argparse.ArgumentParser() + parser.add_argument('-p', '--paths', action='store_true', help='Preserve path structure of original files') - parser.add_option('-d', '--directory', action='store', type='string', + parser.add_argument('-d', '--directory', action='store', type=str, dest='outdir', default='docs', help='The output directory that the rendered files should go to.') - parser.add_option('-w', '--watch', action='store_true', + parser.add_argument('-w', '--watch', action='store_true', help='Watch original files and re-generate documentation on changes') - parser.add_option('-l', '--force-language', action='store', type='string', + parser.add_argument('-l', '--force-language', action='store', type=str, dest='language', default=None, help='Force the language for the given files') - parser.add_option('-i', '--generate_index', action='store_true', + parser.add_argument('-i', '--generate_index', action='store_true', help='Generate an index.html document with sitemap content') - parser.add_option('-s', '--skip-bad-files', action='store_true', + parser.add_argument('-s', '--skip-bad-files', action='store_true', dest='skip_bad_files', help='Continue processing after hitting a bad file') - opts, sources = parser.parse_args() - if opts.outdir == '': + parser.add_argument('sources', nargs='*') + + args = parser.parse_args() + if args.outdir == '': outdir = '.' else: - outdir = opts.outdir + outdir = args.outdir - process(sources, outdir=outdir, preserve_paths=opts.paths, - language=opts.language, index=opts.generate_index, - skip=opts.skip_bad_files) + process(args.sources, outdir=outdir, preserve_paths=args.paths, + language=args.language, index=args.generate_index, + skip=args.skip_bad_files) # If the -w / \-\-watch option was present, monitor the source directories # for changes and re-generate documentation for source files whenever they # are modified. - if opts.watch: + if args.watch: try: import watchdog.events import watchdog.observers # noqa except ImportError: sys.exit('The -w/--watch option requires the watchdog package.') - monitor(sources, opts) + monitor(args.sources, args) # Run the script. |