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())