Fix in extract_annotations
[librarian.git] / librarian / epub.py
1 # -*- coding: utf-8 -*-
2 #
3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 #
6 from __future__ import with_statement
7
8 import os
9 import os.path
10 import re
11 import subprocess
12 from StringIO import StringIO
13 from copy import deepcopy
14 from lxml import etree
15 import zipfile
16 from tempfile import mkdtemp, NamedTemporaryFile
17 from shutil import rmtree
18
19 from librarian import RDFNS, WLNS, NCXNS, OPFNS, XHTMLNS, DCNS, OutputFile
20 from librarian.cover import DefaultEbookCover
21
22 from librarian import functions, get_resource
23
24 from librarian.hyphenator import Hyphenator
25
26 functions.reg_person_name()
27 functions.reg_lang_code_3to2()
28
29
30 def set_hyph_language(source_tree):
31     def get_short_lng_code(text):
32         result = ''
33         text = ''.join(text)
34         with open(get_resource('res/ISO-639-2_8859-1.txt'), 'rb') as f:
35             for line in f:
36                 list = line.strip().split('|')
37                 if list[0] == text:
38                     result = list[2]
39         if result == '':
40             return text
41         else:
42             return result
43     bibl_lng = etree.XPath('//dc:language//text()',
44                            namespaces={'dc': str(DCNS)})(source_tree)
45     short_lng = get_short_lng_code(bibl_lng[0])
46     try:
47         return Hyphenator(get_resource('res/hyph-dictionaries/hyph_' +
48                                        short_lng + '.dic'))
49     except:
50         pass
51
52
53 def hyphenate_and_fix_conjunctions(source_tree, hyph):
54     if hyph is not None:
55         texts = etree.XPath('/utwor/*[2]//text()')(source_tree)
56         for t in texts:
57             parent = t.getparent()
58             newt = ''
59             wlist = re.compile(r'\w+|[^\w]', re.UNICODE).findall(t)
60             for w in wlist:
61                 newt += hyph.inserted(w, u'\u00AD')
62             newt = re.sub(r'(?<=\s\w)\s+', u'\u00A0', newt)
63             if t.is_text:
64                 parent.text = newt
65             elif t.is_tail:
66                 parent.tail = newt
67
68
69 def inner_xml(node):
70     """ returns node's text and children as a string
71
72     >>> print inner_xml(etree.fromstring('<a>x<b>y</b>z</a>'))
73     x<b>y</b>z
74     """
75
76     nt = node.text if node.text is not None else ''
77     return ''.join([nt] + [etree.tostring(child) for child in node])
78
79
80 def set_inner_xml(node, text):
81     """ sets node's text and children from a string
82
83     >>> e = etree.fromstring('<a>b<b>x</b>x</a>')
84     >>> set_inner_xml(e, 'x<b>y</b>z')
85     >>> print etree.tostring(e)
86     <a>x<b>y</b>z</a>
87     """
88
89     p = etree.fromstring('<x>%s</x>' % text)
90     node.text = p.text
91     node[:] = p[:]
92
93
94 def node_name(node):
95     """ Find out a node's name
96
97     >>> print node_name(etree.fromstring('<a>X<b>Y</b>Z</a>'))
98     XYZ
99     """
100
101     tempnode = deepcopy(node)
102
103     for p in ('pe', 'pa', 'pt', 'pr', 'motyw'):
104         for e in tempnode.findall('.//%s' % p):
105             t = e.tail
106             e.clear()
107             e.tail = t
108     etree.strip_tags(tempnode, '*')
109     return tempnode.text
110
111
112 def xslt(xml, sheet):
113     if isinstance(xml, etree._Element):
114         xml = etree.ElementTree(xml)
115     with open(sheet) as xsltf:
116         return xml.xslt(etree.parse(xsltf))
117
118
119 def replace_characters(node):
120     def replace_chars(text):
121         if text is None:
122             return None
123         return text.replace(u"\ufeff", u"")\
124                    .replace("---", u"\u2014")\
125                    .replace("--", u"\u2013")\
126                    .replace(",,", u"\u201E")\
127                    .replace('"', u"\u201D")\
128                    .replace("'", u"\u2019")
129     if node.tag in ('uwaga', 'extra'):
130         t = node.tail
131         node.clear()
132         node.tail = t
133     node.text = replace_chars(node.text)
134     node.tail = replace_chars(node.tail)
135     for child in node:
136         replace_characters(child)
137
138
139 def find_annotations(annotations, source, part_no):
140     for child in source:
141         if child.tag in ('pe', 'pa', 'pt', 'pr'):
142             annotation = deepcopy(child)
143             number = str(len(annotations) + 1)
144             annotation.set('number', number)
145             annotation.set('part', str(part_no))
146             annotation.tail = ''
147             annotations.append(annotation)
148             tail = child.tail
149             child.clear()
150             child.tail = tail
151             child.text = number
152         if child.tag not in ('extra', 'uwaga'):
153             find_annotations(annotations, child, part_no)
154
155
156 class Stanza(object):
157     """
158     Converts / verse endings into verse elements in a stanza.
159
160     Slashes may only occur directly in the stanza. Any slashes in subelements
161     will be ignored, and the subelements will be put inside verse elements.
162
163     >>> s = etree.fromstring("<strofa>a <b>c</b> <b>c</b>/\\nb<x>x/\\ny</x>c/ \\nd</strofa>")
164     >>> Stanza(s).versify()
165     >>> print etree.tostring(s)
166     <strofa><wers_normalny>a <b>c</b> <b>c</b></wers_normalny><wers_normalny>b<x>x/
167     y</x>c</wers_normalny><wers_normalny>d</wers_normalny></strofa>
168
169     """
170     def __init__(self, stanza_elem):
171         self.stanza = stanza_elem
172         self.verses = []
173         self.open_verse = None
174
175     def versify(self):
176         self.push_text(self.stanza.text)
177         for elem in self.stanza:
178             self.push_elem(elem)
179             self.push_text(elem.tail)
180         tail = self.stanza.tail
181         self.stanza.clear()
182         self.stanza.tail = tail
183         self.stanza.extend(self.verses)
184
185     def open_normal_verse(self):
186         self.open_verse = self.stanza.makeelement("wers_normalny")
187         self.verses.append(self.open_verse)
188
189     def get_open_verse(self):
190         if self.open_verse is None:
191             self.open_normal_verse()
192         return self.open_verse
193
194     def push_text(self, text):
195         if not text:
196             return
197         for i, verse_text in enumerate(re.split(r"/\s*\n", text)):
198             if i:
199                 self.open_normal_verse()
200             verse = self.get_open_verse()
201             if len(verse):
202                 verse[-1].tail = (verse[-1].tail or "") + verse_text
203             else:
204                 verse.text = (verse.text or "") + verse_text
205
206     def push_elem(self, elem):
207         if elem.tag.startswith("wers"):
208             verse = deepcopy(elem)
209             verse.tail = None
210             self.verses.append(verse)
211             self.open_verse = verse
212         else:
213             appended = deepcopy(elem)
214             appended.tail = None
215             self.get_open_verse().append(appended)
216
217
218 def replace_by_verse(tree):
219     """ Find stanzas and create new verses in place of a '/' character """
220
221     stanzas = tree.findall('.//' + WLNS('strofa'))
222     for stanza in stanzas:
223         Stanza(stanza).versify()
224
225
226 def add_to_manifest(manifest, partno):
227     """ Adds a node to the manifest section in content.opf file """
228
229     partstr = 'part%d' % partno
230     e = manifest.makeelement(
231         OPFNS('item'), attrib={'id': partstr, 'href': partstr + '.html',
232                                'media-type': 'application/xhtml+xml'}
233     )
234     manifest.append(e)
235
236
237 def add_to_spine(spine, partno):
238     """ Adds a node to the spine section in content.opf file """
239
240     e = spine.makeelement(OPFNS('itemref'), attrib={'idref': 'part%d' % partno})
241     spine.append(e)
242
243
244 class TOC(object):
245     def __init__(self, name=None, part_href=None):
246         self.children = []
247         self.name = name
248         self.part_href = part_href
249         self.sub_number = None
250
251     def add(self, name, part_href, level=0, is_part=True, index=None):
252         assert level == 0 or index is None
253         if level > 0 and self.children:
254             return self.children[-1].add(name, part_href, level - 1, is_part)
255         else:
256             t = TOC(name)
257             t.part_href = part_href
258             if index is not None:
259                 self.children.insert(index, t)
260             else:
261                 self.children.append(t)
262             if not is_part:
263                 t.sub_number = len(self.children) + 1
264                 return t.sub_number
265
266     def append(self, toc):
267         self.children.append(toc)
268
269     def extend(self, toc):
270         self.children.extend(toc.children)
271
272     def depth(self):
273         if self.children:
274             return max((c.depth() for c in self.children)) + 1
275         else:
276             return 0
277
278     def href(self):
279         src = self.part_href
280         if self.sub_number is not None:
281             src += '#sub%d' % self.sub_number
282         return src
283
284     def write_to_xml(self, nav_map, counter=1):
285         for child in self.children:
286             nav_point = nav_map.makeelement(NCXNS('navPoint'))
287             nav_point.set('id', 'NavPoint-%d' % counter)
288             nav_point.set('playOrder', str(counter))
289
290             nav_label = nav_map.makeelement(NCXNS('navLabel'))
291             text = nav_map.makeelement(NCXNS('text'))
292             if child.name is not None:
293                 text.text = re.sub(r'\n', ' ', child.name)
294             else:
295                 text.text = child.name
296             nav_label.append(text)
297             nav_point.append(nav_label)
298
299             content = nav_map.makeelement(NCXNS('content'))
300             content.set('src', child.href())
301             nav_point.append(content)
302             nav_map.append(nav_point)
303             counter = child.write_to_xml(nav_point, counter + 1)
304         return counter
305
306     def html_part(self, depth=0):
307         texts = []
308         for child in self.children:
309             texts.append(
310                 "<div style='margin-left:%dem;'><a href='%s'>%s</a></div>" %
311                 (depth, child.href(), child.name))
312             texts.append(child.html_part(depth + 1))
313         return "\n".join(texts)
314
315     def html(self):
316         with open(get_resource('epub/toc.html')) as f:
317             t = unicode(f.read(), 'utf-8')
318         return t % self.html_part()
319
320
321 def used_chars(element):
322     """ Lists characters used in an ETree Element """
323     chars = set((element.text or '') + (element.tail or ''))
324     for child in element:
325         chars = chars.union(used_chars(child))
326     return chars
327
328
329 def chop(main_text):
330     """ divide main content of the XML file into chunks """
331
332     # prepare a container for each chunk
333     part_xml = etree.Element('utwor')
334     etree.SubElement(part_xml, 'master')
335     main_xml_part = part_xml[0]  # master
336
337     last_node_part = False
338
339     # the below loop are workaround for a problem with epubs in drama ebooks without acts
340     is_scene = False
341     is_act = False
342     for one_part in main_text:
343         name = one_part.tag
344         if name == 'naglowek_scena':
345             is_scene = True
346         elif name == 'naglowek_akt':
347             is_act = True
348
349     for one_part in main_text:
350         name = one_part.tag
351         if is_act is False and is_scene is True:
352             if name == 'naglowek_czesc':
353                 yield part_xml
354                 last_node_part = True
355                 main_xml_part[:] = [deepcopy(one_part)]
356             elif not last_node_part and name == "naglowek_scena":
357                 yield part_xml
358                 main_xml_part[:] = [deepcopy(one_part)]
359             else:
360                 main_xml_part.append(deepcopy(one_part))
361                 last_node_part = False
362         else:
363             if name == 'naglowek_czesc':
364                 yield part_xml
365                 last_node_part = True
366                 main_xml_part[:] = [deepcopy(one_part)]
367             elif not last_node_part and name in ("naglowek_rozdzial", "naglowek_akt", "srodtytul"):
368                 yield part_xml
369                 main_xml_part[:] = [deepcopy(one_part)]
370             else:
371                 main_xml_part.append(deepcopy(one_part))
372                 last_node_part = False
373     yield part_xml
374
375
376 def transform_chunk(chunk_xml, chunk_no, annotations, empty=False, _empty_html_static=[]):
377     """ transforms one chunk, returns a HTML string, a TOC object and a set of used characters """
378
379     toc = TOC()
380     for element in chunk_xml[0]:
381         if element.tag == "naglowek_czesc":
382             toc.add(node_name(element), "part%d.html#book-text" % chunk_no)
383         elif element.tag in ("naglowek_rozdzial", "naglowek_akt", "srodtytul"):
384             toc.add(node_name(element), "part%d.html" % chunk_no)
385         elif element.tag in ('naglowek_podrozdzial', 'naglowek_scena'):
386             subnumber = toc.add(node_name(element), "part%d.html" % chunk_no, level=1, is_part=False)
387             element.set('sub', str(subnumber))
388     if empty:
389         if not _empty_html_static:
390             _empty_html_static.append(open(get_resource('epub/emptyChunk.html')).read())
391         chars = set()
392         output_html = _empty_html_static[0]
393     else:
394         find_annotations(annotations, chunk_xml, chunk_no)
395         replace_by_verse(chunk_xml)
396         html_tree = xslt(chunk_xml, get_resource('epub/xsltScheme.xsl'))
397         chars = used_chars(html_tree.getroot())
398         output_html = etree.tostring(
399             html_tree, pretty_print=True, xml_declaration=True,
400             encoding="utf-8",
401             doctype='<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" ' +
402                     '"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'
403         )
404     return output_html, toc, chars
405
406
407 def transform(wldoc, verbose=False,
408               style=None, html_toc=False,
409               sample=None, cover=None, flags=None):
410     """ produces a EPUB file
411
412     sample=n: generate sample e-book (with at least n paragraphs)
413     cover: a cover.Cover factory or True for default
414     flags: less-advertising, without-fonts, working-copy
415     """
416
417     def transform_file(wldoc, chunk_counter=1, first=True, sample=None):
418         """ processes one input file and proceeds to its children """
419
420         replace_characters(wldoc.edoc.getroot())
421
422         hyphenator = set_hyph_language(wldoc.edoc.getroot())
423         hyphenate_and_fix_conjunctions(wldoc.edoc.getroot(), hyphenator)
424
425         # every input file will have a TOC entry,
426         # pointing to starting chunk
427         toc = TOC(wldoc.book_info.title, "part%d.html" % chunk_counter)
428         chars = set()
429         if first:
430             # write book title page
431             html_tree = xslt(wldoc.edoc, get_resource('epub/xsltTitle.xsl'))
432             chars = used_chars(html_tree.getroot())
433             zip.writestr(
434                 'OPS/title.html',
435                 etree.tostring(
436                     html_tree, pretty_print=True, xml_declaration=True,
437                     encoding="utf-8",
438                     doctype='<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN"' +
439                             ' "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'
440                 )
441             )
442             # add a title page TOC entry
443             toc.add(u"Strona tytułowa", "title.html")
444         elif wldoc.book_info.parts:
445             # write title page for every parent
446             if sample is not None and sample <= 0:
447                 chars = set()
448                 html_string = open(get_resource('epub/emptyChunk.html')).read()
449             else:
450                 html_tree = xslt(wldoc.edoc, get_resource('epub/xsltChunkTitle.xsl'))
451                 chars = used_chars(html_tree.getroot())
452                 html_string = etree.tostring(
453                     html_tree, pretty_print=True, xml_declaration=True,
454                     encoding="utf-8",
455                     doctype='<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN"' +
456                             ' "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'
457                 )
458             zip.writestr('OPS/part%d.html' % chunk_counter, html_string)
459             add_to_manifest(manifest, chunk_counter)
460             add_to_spine(spine, chunk_counter)
461             chunk_counter += 1
462
463         if len(wldoc.edoc.getroot()) > 1:
464             # rdf before style master
465             main_text = wldoc.edoc.getroot()[1]
466         else:
467             # rdf in style master
468             main_text = wldoc.edoc.getroot()[0]
469             if main_text.tag == RDFNS('RDF'):
470                 main_text = None
471
472         if main_text is not None:
473             for chunk_xml in chop(main_text):
474                 empty = False
475                 if sample is not None:
476                     if sample <= 0:
477                         empty = True
478                     else:
479                         sample -= len(chunk_xml.xpath('//strofa|//akap|//akap_cd|//akap_dialog'))
480                 chunk_html, chunk_toc, chunk_chars = transform_chunk(chunk_xml, chunk_counter, annotations, empty)
481
482                 toc.extend(chunk_toc)
483                 chars = chars.union(chunk_chars)
484                 zip.writestr('OPS/part%d.html' % chunk_counter, chunk_html)
485                 add_to_manifest(manifest, chunk_counter)
486                 add_to_spine(spine, chunk_counter)
487                 chunk_counter += 1
488
489         for child in wldoc.parts():
490             child_toc, chunk_counter, chunk_chars, sample = transform_file(
491                 child, chunk_counter, first=False, sample=sample)
492             toc.append(child_toc)
493             chars = chars.union(chunk_chars)
494
495         return toc, chunk_counter, chars, sample
496
497     document = deepcopy(wldoc)
498     del wldoc
499
500     if flags:
501         for flag in flags:
502             document.edoc.getroot().set(flag, 'yes')
503
504     # add editors info
505     editors = document.editors()
506     if editors:
507         document.edoc.getroot().set('editors', u', '.join(sorted(
508             editor.readable() for editor in editors)))
509     if document.book_info.funders:
510         document.edoc.getroot().set('funders', u', '.join(
511             document.book_info.funders))
512     if document.book_info.thanks:
513         document.edoc.getroot().set('thanks', document.book_info.thanks)
514
515     opf = xslt(document.book_info.to_etree(), get_resource('epub/xsltContent.xsl'))
516     manifest = opf.find('.//' + OPFNS('manifest'))
517     guide = opf.find('.//' + OPFNS('guide'))
518     spine = opf.find('.//' + OPFNS('spine'))
519
520     output_file = NamedTemporaryFile(prefix='librarian', suffix='.epub', delete=False)
521     zip = zipfile.ZipFile(output_file, 'w', zipfile.ZIP_DEFLATED)
522
523     # write static elements
524     mime = zipfile.ZipInfo()
525     mime.filename = 'mimetype'
526     mime.compress_type = zipfile.ZIP_STORED
527     mime.extra = ''
528     zip.writestr(mime, 'application/epub+zip')
529     zip.writestr(
530         'META-INF/container.xml',
531         '<?xml version="1.0" ?>'
532         '<container version="1.0" '
533         'xmlns="urn:oasis:names:tc:opendocument:xmlns:container">'
534         '<rootfiles><rootfile full-path="OPS/content.opf" '
535         'media-type="application/oebps-package+xml" />'
536         '</rootfiles></container>'
537     )
538     zip.write(get_resource('res/wl-logo-small.png'),
539               os.path.join('OPS', 'logo_wolnelektury.png'))
540     zip.write(get_resource('res/jedenprocent.png'),
541               os.path.join('OPS', 'jedenprocent.png'))
542     if not style:
543         style = get_resource('epub/style.css')
544     zip.write(style, os.path.join('OPS', 'style.css'))
545
546     if cover:
547         if cover is True:
548             cover = DefaultEbookCover
549
550         cover_file = StringIO()
551         bound_cover = cover(document.book_info)
552         bound_cover.save(cover_file)
553         cover_name = 'cover.%s' % bound_cover.ext()
554         zip.writestr(os.path.join('OPS', cover_name), cover_file.getvalue())
555         del cover_file
556
557         cover_tree = etree.parse(get_resource('epub/cover.html'))
558         cover_tree.find('//' + XHTMLNS('img')).set('src', cover_name)
559         zip.writestr('OPS/cover.html', etree.tostring(
560             cover_tree, pretty_print=True, xml_declaration=True,
561             encoding="utf-8",
562             doctype='<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" ' +
563                     '"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'
564         ))
565
566         if bound_cover.uses_dc_cover:
567             if document.book_info.cover_by:
568                 document.edoc.getroot().set('data-cover-by', document.book_info.cover_by)
569             if document.book_info.cover_source:
570                 document.edoc.getroot().set('data-cover-source', document.book_info.cover_source)
571
572         manifest.append(etree.fromstring(
573             '<item id="cover" href="cover.html" media-type="application/xhtml+xml" />'))
574         manifest.append(etree.fromstring(
575             '<item id="cover-image" href="%s" media-type="%s" />' % (cover_name, bound_cover.mime_type())))
576         spine.insert(0, etree.fromstring('<itemref idref="cover"/>'))
577         opf.getroot()[0].append(etree.fromstring('<meta name="cover" content="cover-image"/>'))
578         guide.append(etree.fromstring('<reference href="cover.html" type="cover" title="Okładka"/>'))
579
580     annotations = etree.Element('annotations')
581
582     toc_file = etree.fromstring(
583         '<?xml version="1.0" encoding="utf-8"?><!DOCTYPE ncx PUBLIC '
584         '"-//NISO//DTD ncx 2005-1//EN" '
585         '"http://www.daisy.org/z3986/2005/ncx-2005-1.dtd">'
586         '<ncx xmlns="http://www.daisy.org/z3986/2005/ncx/" xml:lang="pl" '
587         'version="2005-1"><head></head><docTitle></docTitle><navMap>'
588         '</navMap></ncx>'
589     )
590     nav_map = toc_file[-1]
591
592     if html_toc:
593         manifest.append(etree.fromstring(
594             '<item id="html_toc" href="toc.html" media-type="application/xhtml+xml" />'))
595         spine.append(etree.fromstring(
596             '<itemref idref="html_toc" />'))
597         guide.append(etree.fromstring('<reference href="toc.html" type="toc" title="Spis treści"/>'))
598
599     toc, chunk_counter, chars, sample = transform_file(document, sample=sample)
600
601     if len(toc.children) < 2:
602         toc.add(u"Początek utworu", "part1.html")
603
604     # Last modifications in container files and EPUB creation
605     if len(annotations) > 0:
606         toc.add("Przypisy", "annotations.html")
607         manifest.append(etree.fromstring(
608             '<item id="annotations" href="annotations.html" media-type="application/xhtml+xml" />'))
609         spine.append(etree.fromstring(
610             '<itemref idref="annotations" />'))
611         replace_by_verse(annotations)
612         html_tree = xslt(annotations, get_resource('epub/xsltAnnotations.xsl'))
613         chars = chars.union(used_chars(html_tree.getroot()))
614         zip.writestr('OPS/annotations.html', etree.tostring(
615             html_tree, pretty_print=True, xml_declaration=True,
616             encoding="utf-8",
617             doctype='<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" ' +
618                     '"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'
619         ))
620
621     toc.add("Wesprzyj Wolne Lektury", "support.html")
622     manifest.append(etree.fromstring(
623         '<item id="support" href="support.html" media-type="application/xhtml+xml" />'))
624     spine.append(etree.fromstring(
625         '<itemref idref="support" />'))
626     html_string = open(get_resource('epub/support.html')).read()
627     chars.update(used_chars(etree.fromstring(html_string)))
628     zip.writestr('OPS/support.html', html_string)
629
630     toc.add("Strona redakcyjna", "last.html")
631     manifest.append(etree.fromstring(
632         '<item id="last" href="last.html" media-type="application/xhtml+xml" />'))
633     spine.append(etree.fromstring(
634         '<itemref idref="last" />'))
635     html_tree = xslt(document.edoc, get_resource('epub/xsltLast.xsl'))
636     chars.update(used_chars(html_tree.getroot()))
637     zip.writestr('OPS/last.html', etree.tostring(
638         html_tree, pretty_print=True, xml_declaration=True,
639         encoding="utf-8",
640         doctype='<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" ' +
641                 '"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">'
642     ))
643
644     if not flags or not 'without-fonts' in flags:
645         # strip fonts
646         tmpdir = mkdtemp('-librarian-epub')
647         try:
648             cwd = os.getcwd()
649         except OSError:
650             cwd = None
651
652         os.chdir(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'font-optimizer'))
653         for fname in 'DejaVuSerif.ttf', 'DejaVuSerif-Bold.ttf', 'DejaVuSerif-Italic.ttf', 'DejaVuSerif-BoldItalic.ttf':
654             optimizer_call = ['perl', 'subset.pl', '--chars',
655                               ''.join(chars).encode('utf-8'),
656                               get_resource('fonts/' + fname),
657                               os.path.join(tmpdir, fname)]
658             if verbose:
659                 print "Running font-optimizer"
660                 subprocess.check_call(optimizer_call)
661             else:
662                 subprocess.check_call(optimizer_call, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
663             zip.write(os.path.join(tmpdir, fname), os.path.join('OPS', fname))
664             manifest.append(etree.fromstring(
665                 '<item id="%s" href="%s" media-type="application/x-font-truetype" />' % (fname, fname)))
666         rmtree(tmpdir)
667         if cwd is not None:
668             os.chdir(cwd)
669     zip.writestr('OPS/content.opf', etree.tostring(opf, pretty_print=True,
670                  xml_declaration=True, encoding="utf-8"))
671     title = document.book_info.title
672     attributes = "dtb:uid", "dtb:depth", "dtb:totalPageCount", "dtb:maxPageNumber"
673     for st in attributes:
674         meta = toc_file.makeelement(NCXNS('meta'))
675         meta.set('name', st)
676         meta.set('content', '0')
677         toc_file[0].append(meta)
678     toc_file[0][0].set('content', str(document.book_info.url))
679     toc_file[0][1].set('content', str(toc.depth()))
680     set_inner_xml(toc_file[1], ''.join(('<text>', title, '</text>')))
681
682     # write TOC
683     if html_toc:
684         toc.add(u"Spis treści", "toc.html", index=1)
685         zip.writestr('OPS/toc.html', toc.html().encode('utf-8'))
686     toc.write_to_xml(nav_map)
687     zip.writestr('OPS/toc.ncx', etree.tostring(toc_file, pretty_print=True,
688                  xml_declaration=True, encoding="utf-8"))
689     zip.close()
690
691     return OutputFile.from_filename(output_file.name)