42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from argparse import ArgumentParser
|
|
|
|
from chordpro.base import Song
|
|
from chordpro.renderers.html import render as html_render, get_options as html_get_options
|
|
from chordpro.renderers.text import render as text_render
|
|
|
|
|
|
def get_args():
|
|
"""Parse the command line arguments"""
|
|
parser = ArgumentParser()
|
|
parser.add_argument('input', metavar='FILENAME', help='Input ChordPro file')
|
|
parser.add_argument('-o', '--output', metavar='FILENAME', help='Output to a file')
|
|
parser.add_argument('-f', '--format', metavar='FORMAT', choices=['html', 'text'], default='html',
|
|
help='Output format')
|
|
parser.add_argument('-p', '--param', metavar='KEY=VALUE', action='append', help='Parameter to send to renderer')
|
|
return parser.parse_args()
|
|
|
|
|
|
def main():
|
|
"""The command line entrypoint"""
|
|
args = get_args()
|
|
render_params = {kv.split('=')[0]: kv.split('=')[-1] for kv in args.param} if args.param else {}
|
|
# Prepare the params
|
|
for key, value in render_params.items():
|
|
if value.lower() == "true":
|
|
render_params[key] = True
|
|
elif value.lower() == "false":
|
|
render_params[key] = False
|
|
elif value.isdigit():
|
|
render_params[key] = int(value)
|
|
|
|
song = Song(args.input)
|
|
if args.format == 'text':
|
|
output = text_render(song, **render_params)
|
|
else:
|
|
output = html_render(song, **render_params)
|
|
if args.output:
|
|
with open(args.output, 'w') as html_file:
|
|
html_file.write(output)
|
|
else:
|
|
print(output)
|