[epub, mobi] added optional command line switch helpful with kindlegen conversion

[librarian.git] / librarian / book2anything.py

diff --git a/librarian/book2anything.py b/librarian/book2anything.py
index 7660ec7..f1044b7 100755 (executable)
--- a/librarian/book2anything.py
+++ b/librarian/book2anything.py
@@ -10,7 +10,7 @@ import optparse
 
 from librarian import DirDocProvider, ParseError
 from librarian.parser import WLDocument
 
 from librarian import DirDocProvider, ParseError
 from librarian.parser import WLDocument

-from librarian.cover import WLCover
+from librarian.cover import DefaultEbookCover

 
 
 class Option(object):
 
 
 class Option(object):


@@ -56,6 +56,9 @@ class Book2Anything(object):
         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('-t', '--html-toc', 
+                action='store_true', dest='html_toc', default=False,
+                help='with inline html toc')

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


@@ -97,11 +100,15 @@ class Book2Anything(object):
         if transform_flags:
             transform_args['flags'] = transform_flags
         # Add cover support, if any.
         if transform_flags:
             transform_args['flags'] = transform_flags
         # Add cover support, if any.

+        if options.html_toc:
+            transform_args['html_toc'] = True

         if cls.uses_cover:
             if options.image_cache:
         if cls.uses_cover:
             if options.image_cache:

-                transform_args['cover'] = lambda x: WLCover(x, image_cache = options.image_cache)
+                def cover_class(*args, **kwargs):
+                    return DefaultEbookCover(image_cache=options.image_cache, *args, **kwargs)
+                transform_args['cover'] = cover_class

             elif not cls.cover_optional or options.with_cover:
             elif not cls.cover_optional or options.with_cover:

-                transform_args['cover'] = WLCover
+                transform_args['cover'] = DefaultEbookCover

 
 
         # Do some real work
 
 
         # Do some real work