Merge pull request #2 from mgorny/fb2

[librarian.git] / scripts / book2epub

diff --git a/scripts/book2epub b/scripts/book2epub
index bdb5ac6..04e4515 100755 (executable)
--- a/scripts/book2epub
+++ b/scripts/book2epub
@@ -22,6 +22,8 @@ if __name__ == '__main__':
         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='mark the output as a working copy')
     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',
@@ -35,6 +37,10 @@ if __name__ == '__main__':
         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:
@@ -49,7 +55,7 @@ if __name__ == '__main__':
                 output_file = None
 
             doc = WLDocument.from_file(main_input, provider=provider)
-            epub = doc.as_epub(cover=options.with_cover)
+            epub = doc.as_epub(cover=options.with_cover, flags=flags)
 
             doc.save_output_file(epub,
                 output_file, options.output_dir, options.make_dir, 'epub')