cli.py

import argparse
import sys

from . import document
import logging

def do_main(paths, output_dir, use_ascii: bool, escape_html: bool, single_file: bool):
    try:
        document(paths, output_dir, use_ascii, escape_html, single_file)
    except IOError as e:
        logging.error('Unable to open file: %s', e)
        return 1
    except Exception as e:
        logging.error('An error occurred: %s', e)
        return 1
    else:
        return 0

def main():
    arg_parser = argparse.ArgumentParser(prog='dycco', description='Literate-style documentation generator.')
    arg_parser.add_argument('source_file', nargs='+', default=sys.stdin, help='Source files to document')
    arg_parser.add_argument('-o', '--output-dir', default='docs', help='Output directory (will be created if necessary)')
    arg_parser.add_argument('-a', '--asciidoc3', action='store_true', default=False, dest='use_ascii',
                            help='Process with asciidoc3 instead of markdown (you will have to install asciidoc3, of course)')
    arg_parser.add_argument('-e', '--escape-html', action='store_true', default=False, dest='escape_html',
                            help='Run the documentation through html.escape() before markdown or asciidoc3')
    arg_parser.add_argument('-f', '--single-file', action='store_true', default=False, dest='single_file',
                            help='Just produce a .md or .adoc file in single-column to be processed externally')

    args = arg_parser.parse_args()

    return do_main(args.source_file, args.output_dir, args.use_ascii, args.escape_html, args.single_file)