foc-specific changes

[librarian.git] / scripts / book2epub

diff --git a/scripts/book2epub b/scripts/book2epub
index 9af3692..e60b932 100755 (executable)
--- a/scripts/book2epub
+++ b/scripts/book2epub
@@ -20,6 +20,10 @@ if __name__ == '__main__':
 
     parser.add_option('-v', '--verbose', action='store_true', dest='verbose', default=False,
         help='print status messages to stdout')
 
     parser.add_option('-v', '--verbose', action='store_true', dest='verbose', default=False,
         help='print status messages to stdout')

+    parser.add_option('-c', '--with-cover', action='store_true', dest='with_cover', default=False,
+                      help='create default cover')
+    parser.add_option('-w', '--working-copy', action='store_true', dest='working_copy', default=False,
+                      help='specifies the directory for output')

     parser.add_option('-d', '--make-dir', action='store_true', dest='make_dir', default=False,
                       help='create a directory for author and put the PDF in it')
     parser.add_option('-o', '--output-file', dest='output_file', metavar='FILE',
     parser.add_option('-d', '--make-dir', action='store_true', dest='make_dir', default=False,
                       help='create a directory for author and put the PDF in it')
     parser.add_option('-o', '--output-file', dest='output_file', metavar='FILE',


@@ -33,6 +37,10 @@ if __name__ == '__main__':
         parser.print_help()
         exit(1)
 
         parser.print_help()
         exit(1)
 

+    flags = []
+    if options.working_copy:
+        flags.append('working-copy')
+

     # Do some real work
     try:
         for main_input in input_filenames:
     # Do some real work
     try:
         for main_input in input_filenames:


@@ -47,7 +55,7 @@ if __name__ == '__main__':
                 output_file = None
 
             doc = WLDocument.from_file(main_input, provider=provider)
                 output_file = None
 
             doc = WLDocument.from_file(main_input, provider=provider)

-            epub = doc.as_epub()
+            epub = doc.as_epub(cover=options.with_cover, flags=flags)

 
             doc.save_output_file(epub,
                 output_file, options.output_dir, options.make_dir, 'epub')
 
             doc.save_output_file(epub,
                 output_file, options.output_dir, options.make_dir, 'epub')