fix
[librarian.git] / src / librarian / command_line.py
1 import argparse
2 import os.path
3 from .builders import builders
4 from .document import WLDocument
5
6
7 def main(*args, **kwargs):
8     parser = argparse.ArgumentParser(description="PARSER DESCRIPTION")
9
10     parser.add_argument(
11         'builder',
12         choices=builders.keys(),
13         help="Builder"
14     )
15     parser.add_argument('input_file')
16     parser.add_argument(
17         '-o', '--output-file', metavar='FILE',
18         help='specifies the output file'
19     )
20     parser.add_argument(
21         '-O', '--output-dir', metavar='DIR',
22         help='specifies the directory for output'
23     )
24
25     # Specific 
26     parser.add_argument(
27         '-b', '--base-url', metavar="URL",
28         help="Base for relative URLs in documents (like image sources)"
29     )
30
31     parser.add_argument(
32         '--mp3',
33         metavar="FILE",
34         nargs="*",
35         help='specifies an MP3 file, if needed'
36     )
37
38     args = parser.parse_args()
39     builder = builders[args.builder]
40
41     if args.output_file:
42         output_file_path = args.output_file
43     else:
44         output_file_path = '.'.join((
45             os.path.splitext(args.input_file)[0],
46             builder.file_extension
47         ))
48         if args.output_dir:
49             output_file_path = '/'.join((
50                 args.output_dir,
51                 output_file_path.rsplit('/', 1)[-1]
52             ))
53
54     document = WLDocument(filename=args.input_file)
55
56     builder = builders[args.builder]
57     kwargs = {
58         "mp3": args.mp3,
59     }
60
61     output = document.build(builder, base_url=args.base_url, **kwargs)
62     with open(output_file_path, 'wb') as f:
63         f.write(output.get_bytes())