X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/f318053fb3349c5364cfb866b2a3d33c2423e12a..6ef490dd10bfed98771bcae22bcdeb3fe6ee8830:/librarian/epub.py diff --git a/librarian/epub.py b/librarian/epub.py index 223bde9..8dc11c7 100644 --- a/librarian/epub.py +++ b/librarian/epub.py @@ -15,10 +15,11 @@ 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.latex import LatexFragment from librarian import functions, get_resource functions.reg_person_name() @@ -329,9 +330,27 @@ def transform_chunk(chunk_xml, chunk_no, annotations, empty=False, _empty_html_s return output_html, toc, chars +def render_latex(wldoc, prefix="latex"): + """ + Renders CODE as images and returns + (changed_wldoc, [ (epub_filepath1, latexfragment_object1), ... ] +""" + root = wldoc.edoc.getroot() + latex_nodes = root.findall(".//latex") + images = [] + for ln in latex_nodes: + fragment = LatexFragment(ln.text, resize=40) + images.append((os.path.join(prefix, fragment.filename), fragment)) + ln.tag = "img" + ln.text = os.path.join(prefix, fragment.filename) + + return wldoc, images + + 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 +374,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 +455,34 @@ 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')) + document, latex_images = render_latex(document) + for image in latex_images: + zip.write(image[1].path, os.path.join('OPS', image[0])) + image[1].remove() + + 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 +492,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 +505,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 +521,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: @@ -506,16 +565,16 @@ def transform(wldoc, verbose=False, zip.writestr('OPS/annotations.html', etree.tostring( html_tree, method="html", pretty_print=True)) - toc.add("Weprzyj Wolne Lektury", "support.html") - manifest.append(etree.fromstring( - '')) - spine.append(etree.fromstring( - '')) - html_string = open(get_resource('epub/support.html')).read() - chars.update(used_chars(etree.fromstring(html_string))) - zip.writestr('OPS/support.html', html_string) + # toc.add("Weprzyj Wolne Lektury", "support.html") + # manifest.append(etree.fromstring( + # '')) + # spine.append(etree.fromstring( + # '')) + # html_string = open(get_resource('epub/support.html')).read() + # 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(