X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/75abfdc7e8f14937b825a1502bb564fe2063ced4..637475880b35561e349a4a5df3048256cb58fb76:/librarian/epub.py?ds=inline
diff --git a/librarian/epub.py b/librarian/epub.py
index 17550b2..43eb295 100644
--- a/librarian/epub.py
+++ b/librarian/epub.py
@@ -15,9 +15,10 @@ from lxml import etree
import zipfile
from tempfile import mkdtemp, NamedTemporaryFile
from shutil import rmtree
+from mimetypes import guess_type
from librarian import RDFNS, WLNS, NCXNS, OPFNS, XHTMLNS, OutputFile
-from librarian.cover import WLCover
+from librarian.cover import WLCover, FutureOfCopyrightCover
from librarian import functions, get_resource
@@ -331,7 +332,8 @@ 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):
+ sample=None, cover=None, flags=None, resources=None,
+ intro_file=None, cover_file=None):
""" produces a EPUB file
sample=n: generate sample e-book (with at least n paragraphs)
@@ -355,7 +357,7 @@ def transform(wldoc, verbose=False,
zip.writestr('OPS/title.html',
etree.tostring(html_tree, method="html", pretty_print=True))
# add a title page TOC entry
- toc.add(u"Strona tytuÅowa", "title.html")
+ toc.add(u"TytuÅ", "title.html")
elif wldoc.book_info.parts:
# write title page for every parent
if sample is not None and sample <= 0:
@@ -436,14 +438,28 @@ def transform(wldoc, verbose=False,
'media-type="application/oebps-package+xml" />' \
'')
zip.write(get_resource('res/wl-logo-small.png'), os.path.join('OPS', 'logo_wolnelektury.png'))
+ zip.write(get_resource('res/logo.png'), os.path.join('OPS', 'logo.png'))
zip.write(get_resource('res/jedenprocent.png'), os.path.join('OPS', 'jedenprocent.png'))
if not style:
style = get_resource('epub/style.css')
zip.write(style, os.path.join('OPS', 'style.css'))
+ if resources:
+ if os.path.isdir(resources):
+ for dp, dirs, files in os.walk(resources):
+ for fname in files:
+ fpath = os.path.join(dp, fname)
+ if os.path.isfile(fpath):
+ zip.write(fpath, os.path.join('OPS', fname))
+ manifest.append(etree.fromstring(
+ ' ' % (os.path.splitext(fname)[0], fname, guess_type(fpath)[0])))
+
+ else:
+ print "resources path %s is not directory" % resources
+
if cover:
if cover is True:
- cover = WLCover
+ cover = FutureOfCopyrightCover
cover_file = StringIO()
c = cover(document.book_info)
@@ -453,11 +469,11 @@ def transform(wldoc, verbose=False,
del cover_file
cover_tree = etree.parse(get_resource('epub/cover.html'))
- cover_tree.find('//' + XHTMLNS('img')).set('src', cover_name)
+ cover_tree.find('//' + XHTMLNS('img')).set('src', c_name)
zip.writestr('OPS/cover.html', etree.tostring(
cover_tree, method="html", pretty_print=True))
- if bound_cover.uses_dc_cover:
+ if c.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:
@@ -466,11 +482,12 @@ def transform(wldoc, verbose=False,
manifest.append(etree.fromstring(
' '))
manifest.append(etree.fromstring(
- ' ' % (cover_name, bound_cover.mime_type())))
+ ' ' % (c_name, c.mime_type())))
spine.insert(0, etree.fromstring(''))
opf.getroot()[0].append(etree.fromstring(''))
guide.append(etree.fromstring(''))
-
+
+
annotations = etree.Element('annotations')
@@ -481,17 +498,36 @@ def transform(wldoc, verbose=False,
'')
nav_map = toc_file[-1]
+ manifest.append(etree.fromstring(
+ ' '))
+ spine.append(etree.fromstring(
+ ''))
+ html_tree = xslt(document.edoc, get_resource('epub/xsltFirst.xsl'))
+# chars.update(used_chars(html_tree.getroot()))
+ zip.writestr('OPS/first.html', etree.tostring(
+ html_tree, method="html", pretty_print=True))
+
+ if intro_file:
+ manifest.append(etree.fromstring(
+ ' '))
+ spine.append(etree.fromstring(
+ ''))
+ zip.writestr('OPS/intro.html', open(intro_file or get_resource('epub/intro.html')).read())
+
+
if html_toc:
manifest.append(etree.fromstring(
' '))
spine.append(etree.fromstring(
''))
- guide.append(etree.fromstring(''))
+ guide.append(etree.fromstring(''))
toc, chunk_counter, chars, sample = transform_file(document, sample=sample)
+ toc.add("Informacje redakcyjne", "first.html", index=0)
+
if len(toc.children) < 2:
- toc.add(u"PoczÄ
tek utworu", "part1.html")
+ toc.add(u"PoczÄ
tek ksiÄ
żki", "part1.html")
# Last modifications in container files and EPUB creation
if len(annotations) > 0:
@@ -515,7 +551,7 @@ def transform(wldoc, verbose=False,
# chars.update(used_chars(etree.fromstring(html_string)))
# zip.writestr('OPS/support.html', html_string)
- toc.add("Strona redakcyjna", "last.html")
+ toc.add("Informacje redakcyjne", "last.html")
manifest.append(etree.fromstring(
' '))
spine.append(etree.fromstring(