piccolbo/Pweave

import sys
from optparse import OptionParser
import pweave
import logging


logger = logging.getLogger()
logger.setLevel(logging.INFO)
logging.getLogger("ipykernel").propagate = False

logger_handler = logging.StreamHandler()
logger.addHandler(logger_handler)


def weave():
    if len(sys.argv) == 1:
        print("This is pweave %s, enter Pweave -h for help" % pweave.__version__)
        sys.exit()

    # Command line options
    parser = OptionParser(
        usage="pweave [options] sourcefile", version="Pweave " + pweave.__version__
    )
    parser.add_option(
        "-f",
        "--format",
        dest="doctype",
        default=None,
        help="The output format. Available formats: "
        + pweave.PwebFormats.shortformats()
        + " Use Pweave -l to list descriptions or see http://mpastell.com/pweave/formats.html",
    )
    parser.add_option(
        "-i",
        "--input-format",
        dest="informat",
        default=None,
        help="Input format: noweb, markdown, notebook or script",
    )
    parser.add_option(
        "-k",
        "--kernel",
        dest="kernel",
        default="python3",
        help="Jupyter kernel used to run code: default is python3",
    )
    parser.add_option(
        "-o", "--output", dest="output", default=None, help="Name of the output file"
    )
    parser.add_option(
        "-l",
        "--list-formats",
        dest="listformats",
        action="store_true",
        default=False,
        help="List output formats",
    )
    parser.add_option(
        "-m",
        "--matplotlib",
        dest="plot",
        default=True,
        action="store_false",
        help="Disable matplotlib",
    )
    parser.add_option(
        "-d",
        "--documentation-mode",
        dest="docmode",
        action="store_true",
        default=False,
        help="Use documentation mode, chunk code and results will be loaded from cache and inline code will be hidden",
    )
    parser.add_option(
        "-c",
        "--cache-results",
        dest="cache",
        action="store_true",
        default=False,
        help="Cache results to disk for documentation mode",
    )
    parser.add_option(
        "-F",
        "--figure-directory",
        dest="figdir",
        default="figures",
        help="Directory path for matplolib graphics: Default 'figures'",
    )
    parser.add_option(
        "--cache-directory",
        dest="cachedir",
        default="cache",
        help="Directory path for cached results used in documentation mode: Default 'cache'",
    )
    parser.add_option(
        "-g",
        "--figure-format",
        dest="figformat",
        default=None,
        help="Figure format for matplotlib graphics: Defaults to 'png' for rst and Sphinx html documents and 'pdf' for tex",
    )
    parser.add_option(
        "-t",
        "--mimetype",
        dest="mimetype",
        default=None,
        help="Source document's text mimetype. This is used to set cell "
        + "type in Jupyter notebooks",
    )
    parser.add_option(
        "-v",
        "--verbose",
        dest="verbosity",
        action="store_true",
        default=False,
        help="Increase verbosity of logging",
    )

    (options, args) = parser.parse_args()

    try:
        infile = args[0]
    except IndexError:
        infile = ""

    opts_dict = vars(options)
    if options.figformat is not None:
        opts_dict["figformat"] = ".%s" % options.figformat

    if opts_dict.pop("verbosity"):
        logging.getLogger("ipykernel").propagate = True
        ipykernel_handler = logging.StreamHandler()
        logging.getLogger("ipykernel").addHandler(ipykernel_handler)
        ipykernel_handler.setFormatter(
            logging.Formatter("%(levelname)s:%(name)s:%(message)s")
        )

    pweave.weave(infile, **opts_dict)


def publish():
    if len(sys.argv) == 1:
        print(
            "Publish a python script. Part of Pweave %s, use -h for help"
            % pweave.__version__
        )
        sys.exit()

    parser = OptionParser(
        usage="pypublish [options] sourcefile",
        version="Part of Pweave " + pweave.__version__,
    )
    parser.add_option(
        "-f",
        "--format",
        dest="format",
        default="html",
        help="Output format html or pdf, pdf output requires pandoc and pdflatex",
    )
    parser.add_option(
        "-e",
        "--latex_engine",
        dest="latex_engine",
        default="pdflatex",
        help="The command for running latex.",
    )
    parser.add_option(
        "-t", "--theme", dest="theme", default="skeleton", help="Theme for HTML output"
    )
    parser.add_option(
        "-o", "--output", dest="output", default=None, help="Name of the output file"
    )

    (options, args) = parser.parse_args()

    try:
        infile = args[0]
    except IndexError:
        infile = ""

    pweave.publish(
        infile, options.format, options.theme, options.latex_engine, options.output
    )


def tangle():
    if len(sys.argv) == 1:
        print("This is ptangle %s, enter ptangle -h for help" % pweave.__version__)
        sys.exit()

    parser = OptionParser(
        usage="ptangle sourcefile", version="Pweave " + pweave.__version__
    )
    parser.add_option(
        "-i",
        "--input-format",
        dest="informat",
        default=None,
        help="Input format: noweb, markdown, notebook or script",
    )

    (options, args) = parser.parse_args()

    try:
        infile = args[0]
    except IndexError:
        infile = ""

    pweave.tangle(infile, options.informat)


def convert():
    if len(sys.argv) == 1:
        print(
            "This is Pweave document converter %s. Enter pweave-convert -h for help "
            % pweave.__version__
        )
        sys.exit()

    parser = OptionParser(
        usage="pweave-convert [options] sourcefile",
        version="Part of Pweave " + pweave.__version__,
    )
    parser.add_option(
        "-i",
        "--input-format",
        dest="informat",
        default="noweb",
        help="Input format: noweb, notebook or script",
    )
    parser.add_option(
        "-f",
        "--output-format",
        dest="outformat",
        default="html",
        help="Output format script, noweb or notebook",
    )
    parser.add_option(
        "-l",
        "--list-formats",
        dest="listformats",
        action="store_true",
        default=False,
        help="List input / output formats",
    )
    parser.add_option(
        "-p",
        "--pandoc",
        dest="pandoc_args",
        default=None,
        help="passed to pandoc for converting doc chunks",
    )

    (options, args) = parser.parse_args()

    try:
        infile = args[0]
    except IndexError:
        infile = ""

    pweave.convert(
        file=infile,
        informat=options.informat,
        outformat=options.outformat,
        pandoc_args=options.pandoc_args,
        listformats=options.listformats,
    )