diff options
author | Arturs Artamonovs <dos21h@gmail.com> | 2023-01-29 10:30:54 +0000 |
---|---|---|
committer | Arturs Artamonovs <dos21h@gmail.com> | 2023-01-29 10:30:54 +0000 |
commit | 66fa71a8f11b6ce5e8471b533f67cc3a1fdb85a8 (patch) | |
tree | 7aed7f385826a3bd88c76a373e28c6cfae4f396e /src/mistune/__main__.py | |
parent | 129c1201ea5c4418f0f89ad932633c7cea2439b7 (diff) | |
download | md-site-66fa71a8f11b6ce5e8471b533f67cc3a1fdb85a8.tar.gz md-site-66fa71a8f11b6ce5e8471b533f67cc3a1fdb85a8.zip |
Update to new mistune, removed old mistune, rewrite to python3
Diffstat (limited to 'src/mistune/__main__.py')
-rw-r--r-- | src/mistune/__main__.py | 124 |
1 files changed, 124 insertions, 0 deletions
diff --git a/src/mistune/__main__.py b/src/mistune/__main__.py new file mode 100644 index 0000000..053a379 --- /dev/null +++ b/src/mistune/__main__.py @@ -0,0 +1,124 @@ +import sys +import argparse +from .renderers.rst import RSTRenderer +from .renderers.markdown import MarkdownRenderer +from . import ( + create_markdown, + __version__ as version +) + + +def _md(args): + if args.plugin: + plugins = args.plugin + else: + # default plugins + plugins = ['strikethrough', 'footnotes', 'table', 'speedup'] + + if args.renderer == 'rst': + renderer = RSTRenderer() + elif args.renderer == 'markdown': + renderer = MarkdownRenderer() + else: + renderer = args.renderer + return create_markdown( + escape=args.escape, + hard_wrap=args.hardwrap, + renderer=renderer, + plugins=plugins, + ) + + +def _output(text, args): + if args.output: + with open(args.output, 'w') as f: + f.write(text) + else: + print(text) + + +CMD_HELP = '''Mistune, a sane and fast python markdown parser. + +Here are some use cases of the command line tool: + + $ python -m mistune -m "Hi **Markdown**" + <p>Hi <strong>Markdown</strong></p> + + $ python -m mistune -f README.md + <p>... + + $ cat README.md | python -m mistune + <p>... +''' + + +def cli(): + parser = argparse.ArgumentParser( + prog='python -m mistune', + description=CMD_HELP, + formatter_class=argparse.RawDescriptionHelpFormatter, + ) + parser.add_argument( + '-m', '--message', + help='the markdown message to convert', + ) + parser.add_argument( + '-f', '--file', + help='the markdown file to convert', + ) + parser.add_argument( + '-p', '--plugin', + metavar='NAME', + action='extend', + nargs='+', + help='specifiy a plugin to use', + ) + parser.add_argument( + '--escape', + action='store_true', + help='turn on escape option', + ) + parser.add_argument( + '--hardwrap', + action='store_true', + help='turn on hardwrap option', + ) + parser.add_argument( + '-o', '--output', + help='write the rendered result into file', + ) + parser.add_argument( + '-r', '--renderer', + default='html', + help='specify the output renderer', + ) + parser.add_argument('--version', action='version', version='mistune ' + version) + args = parser.parse_args() + + message = args.message + if not message and not args.file: + message = read_stdin() + + if message: + md = _md(args) + text = md(message) + _output(text, args) + elif args.file: + md = _md(args) + text = md.read(args.file)[0] + _output(text, args) + else: + print('You MUST specify a message or file') + return sys.exit(1) + + +def read_stdin(): + is_stdin_pipe = not sys.stdin.isatty() + if is_stdin_pipe: + return sys.stdin.read() + else: + return None + + +if __name__ == '__main__': + cli() |