Data for #3396: picture style, full creation date.

[librarian.git] / librarian / mobi.py

diff --git a/librarian/mobi.py b/librarian/mobi.py
index 82ff343..a0e463a 100644 (file)
--- a/librarian/mobi.py
+++ b/librarian/mobi.py
@@ -3,17 +3,16 @@
 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #

+

 from copy import deepcopy
 import os
 import subprocess
 from tempfile import NamedTemporaryFile
 
 from librarian import OutputFile
 from copy import deepcopy
 import os
 import subprocess
 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):
     """ produces a MOBI file
 
               use_kindlegen=False, flags=None):
     """ produces a MOBI file
 


@@ -25,34 +24,26 @@ 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)

     if verbose:
         kwargs = {}
     else:
         devnull = open("/dev/null", 'w')
         kwargs = {"stdout": devnull, "stderr": devnull}
 
     if verbose:
         kwargs = {}
     else:
         devnull = open("/dev/null", 'w')
         kwargs = {"stdout": devnull, "stderr": devnull}
 

-    output_file = NamedTemporaryFile(prefix='librarian', suffix='.mobi', delete=False)
+    output_file = NamedTemporaryFile(prefix='librarian', suffix='.mobi',
+                                     delete=False)

     output_file.close()
 
     if use_kindlegen:
         output_file_basename = os.path.basename(output_file.name)
     output_file.close()
 
     if use_kindlegen:
         output_file_basename = os.path.basename(output_file.name)

-        subprocess.check_call(
-            ['kindlegen', '-c2', epub.get_filename(), '-o', output_file_basename],
-            **kwargs
-        )
+        subprocess.check_call(['kindlegen', '-c2', epub.get_filename(),
+                              '-o', output_file_basename], **kwargs)

     else:
     else:

-        subprocess.check_call(
-            ['ebook-convert', epub.get_filename(), output_file.name,
-            '--no-inline-toc', 
-            '--mobi-file-type=both', 
-            '--mobi-ignore-margins'], **kwargs
-        )
+        subprocess.check_call(['ebook-convert', epub.get_filename(),
+                               output_file.name, '--no-inline-toc',
+                               '--mobi-file-type=both',
+                               '--mobi-ignore-margins'], **kwargs)

     return OutputFile.from_filename(output_file.name)
     return OutputFile.from_filename(output_file.name)