X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/d04e61819290fc8d6d71b1932c55a774014c1f05..4b0833f222da430d31356b6f065e56073b849d0f:/src/librarian/command_line.py diff --git a/src/librarian/command_line.py b/src/librarian/command_line.py index 3c17daf..08c769f 100644 --- a/src/librarian/command_line.py +++ b/src/librarian/command_line.py @@ -1,3 +1,6 @@ +# This file is part of Librarian, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Wolne Lektury. See NOTICE for more information. +# import argparse import os.path from .builders import builders @@ -22,6 +25,19 @@ def main(*args, **kwargs): help='specifies the directory for output' ) + # Specific + parser.add_argument( + '-b', '--base-url', metavar="URL", + help="Base for relative URLs in documents (like image sources)" + ) + + parser.add_argument( + '--mp3', + metavar="FILE", + nargs="*", + help='specifies an MP3 file, if needed' + ) + args = parser.parse_args() builder = builders[args.builder] @@ -39,6 +55,12 @@ def main(*args, **kwargs): )) document = WLDocument(filename=args.input_file) - output = document.build(args.builder) + + builder = builders[args.builder] + kwargs = { + "mp3": args.mp3, + } + + output = document.build(builder, base_url=args.base_url, **kwargs) with open(output_file_path, 'wb') as f: f.write(output.get_bytes())