X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/ae38fc513217c9ef8d6d92111679246572b84b4a..09b0a045683b6a965e83c1512f0cc99907f2c65c:/librarian/epub.py
diff --git a/librarian/epub.py b/librarian/epub.py
index cb2f2f4..397dc51 100644
--- a/librarian/epub.py
+++ b/librarian/epub.py
@@ -19,7 +19,7 @@ from tempfile import mkdtemp, NamedTemporaryFile
from shutil import rmtree
from librarian import RDFNS, WLNS, NCXNS, OPFNS, XHTMLNS, DCNS, OutputFile
-from librarian.cover import DefaultEbookCover
+from librarian.cover import make_cover
from librarian import functions, get_resource
@@ -113,11 +113,13 @@ def node_name(node):
return tempnode.text
-def xslt(xml, sheet):
+def xslt(xml, sheet, **kwargs):
if isinstance(xml, etree._Element):
xml = etree.ElementTree(xml)
with open(sheet) as xsltf:
- return xml.xslt(etree.parse(xsltf))
+ transform = etree.XSLT(etree.parse(xsltf))
+ params = dict((key, transform.strparam(value)) for key, value in kwargs.iteritems())
+ return transform(xml, **params)
def replace_characters(node):
@@ -199,6 +201,8 @@ class Stanza(object):
if not text:
return
for i, verse_text in enumerate(re.split(r"/\s*\n", text)):
+ if not verse_text.strip():
+ continue
if i:
self.open_normal_verse()
verse = self.get_open_verse()
@@ -409,7 +413,7 @@ def transform_chunk(chunk_xml, chunk_no, annotations, empty=False, _empty_html_s
def transform(wldoc, verbose=False, style=None, html_toc=False,
- sample=None, cover=None, flags=None, hyphenate=False, ilustr_path=''):
+ sample=None, cover=None, flags=None, hyphenate=False, ilustr_path='', output_type='epub'):
""" produces a EPUB file
sample=n: generate sample e-book (with at least n paragraphs)
@@ -431,7 +435,7 @@ def transform(wldoc, verbose=False, style=None, html_toc=False,
chars = set()
if first:
# write book title page
- html_tree = xslt(wldoc.edoc, get_resource('epub/xsltTitle.xsl'))
+ html_tree = xslt(wldoc.edoc, get_resource('epub/xsltTitle.xsl'), outputtype=output_type)
chars = used_chars(html_tree.getroot())
zip.writestr(
'OPS/title.html',
@@ -561,7 +565,7 @@ def transform(wldoc, verbose=False, style=None, html_toc=False,
if cover:
if cover is True:
- cover = DefaultEbookCover
+ cover = make_cover
cover_file = StringIO()
bound_cover = cover(document.book_info)
@@ -648,7 +652,7 @@ def transform(wldoc, verbose=False, style=None, html_toc=False,
' '))
spine.append(etree.fromstring(
''))
- html_tree = xslt(document.edoc, get_resource('epub/xsltLast.xsl'))
+ html_tree = xslt(document.edoc, get_resource('epub/xsltLast.xsl'), outputtype=output_type)
chars.update(used_chars(html_tree.getroot()))
zip.writestr('OPS/last.html', etree.tostring(
html_tree, pretty_print=True, xml_declaration=True,
@@ -675,7 +679,8 @@ def transform(wldoc, verbose=False, style=None, html_toc=False,
print "Running font-optimizer"
subprocess.check_call(optimizer_call)
else:
- subprocess.check_call(optimizer_call, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+ dev_null = open(os.devnull, 'w')
+ subprocess.check_call(optimizer_call, stdout=dev_null, stderr=dev_null)
zip.write(os.path.join(tmpdir, fname), os.path.join('OPS', fname))
manifest.append(etree.fromstring(
' ' % (fname, fname)))