fix chapter titles in toc

[librarian.git] / librarian / mobi.py

diff --git a/librarian/mobi.py b/librarian/mobi.py
index b30c1fe..c3c8f28 100644 (file)
--- a/librarian/mobi.py
+++ b/librarian/mobi.py
@@ -10,12 +10,10 @@ import subprocess
 from tempfile import NamedTemporaryFile
 
 from librarian import OutputFile
 from tempfile import NamedTemporaryFile
 
 from librarian import OutputFile

-from librarian.cover import DefaultEbookCover
-from librarian import get_resource

 
 
 def transform(wldoc, verbose=False, sample=None, cover=None,
 
 
 def transform(wldoc, verbose=False, sample=None, cover=None,

-              use_kindlegen=False, flags=None):
+              use_kindlegen=False, flags=None, hyphenate=True, ilustr_path=''):

     """ produces a MOBI file
 
     wldoc: a WLDocument
     """ produces a MOBI file
 
     wldoc: a WLDocument


@@ -26,14 +24,10 @@ def transform(wldoc, verbose=False, sample=None, cover=None,
 
     document = deepcopy(wldoc)
     del wldoc
 
     document = deepcopy(wldoc)
     del wldoc

-    book_info = document.book_info
-
-    if not flags:
-        flags = []
-    flags = list(flags)

 
     epub = document.as_epub(verbose=verbose, sample=sample,
 
     epub = document.as_epub(verbose=verbose, sample=sample,

-                            html_toc=True, cover=True, flags=flags)
+                            html_toc=True, cover=cover or True, flags=flags,
+                            hyphenate=hyphenate, ilustr_path=ilustr_path, output_type='mobi')

     if verbose:
         kwargs = {}
     else:
     if verbose:
         kwargs = {}
     else: