Merge pull request #1 from mgorny/master

[librarian.git] / librarian / epub.py

diff --git a/librarian/epub.py b/librarian/epub.py
index b063380..5481516 100644 (file)
--- a/librarian/epub.py
+++ b/librarian/epub.py
@@ -16,6 +16,7 @@ from tempfile import mkdtemp, NamedTemporaryFile
 from shutil import rmtree
 
 from librarian import RDFNS, WLNS, NCXNS, OPFNS, XHTMLNS, OutputFile
 from shutil import rmtree
 
 from librarian import RDFNS, WLNS, NCXNS, OPFNS, XHTMLNS, OutputFile

+from librarian.cover import WLCover

 
 from librarian import functions, get_resource
 
 
 from librarian import functions, get_resource
 


@@ -290,8 +291,8 @@ def transform(wldoc, verbose=False,
     """ produces a EPUB file
 
     sample=n: generate sample e-book (with at least n paragraphs)
     """ produces a EPUB file
 
     sample=n: generate sample e-book (with at least n paragraphs)

-    cover: a cover.Cover object
-    flags: less-advertising, without-fonts
+    cover: a cover.Cover object or True for default
+    flags: less-advertising, without-fonts, working-copy

     """
 
     def transform_file(wldoc, chunk_counter=1, first=True, sample=None):
     """
 
     def transform_file(wldoc, chunk_counter=1, first=True, sample=None):


@@ -392,10 +393,17 @@ def transform(wldoc, verbose=False,
         style = get_resource('epub/style.css')
     zip.write(style, os.path.join('OPS', 'style.css'))
 
         style = get_resource('epub/style.css')
     zip.write(style, os.path.join('OPS', 'style.css'))
 

-

     if cover:
     if cover:

+        if cover is True:
+            cover = WLCover
+        if cover.uses_dc_cover:
+            if document.book_info.cover_by:
+                document.edoc.getroot().set('data-cover-by', document.book_info.cover_by)
+            if document.book_info.cover_source:
+                document.edoc.getroot().set('data-cover-source', document.book_info.cover_source)
+

         cover_file = StringIO()
         cover_file = StringIO()

-        c = cover(document.book_info.author.readable(), document.book_info.title)
+        c = cover(document.book_info)

         c.save(cover_file)
         c_name = 'cover.%s' % c.ext()
         zip.writestr(os.path.join('OPS', c_name), cover_file.getvalue())
         c.save(cover_file)
         c_name = 'cover.%s' % c.ext()
         zip.writestr(os.path.join('OPS', c_name), cover_file.getvalue())