X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/143b2e4e4ebbcc4e3de1b23f56df933317b269ff..ef7911fba9c330552599bc6eb9dc22606246dd7e:/librarian/epub.py?ds=sidebyside
diff --git a/librarian/epub.py b/librarian/epub.py
index 92df2d4..84a745f 100644
--- a/librarian/epub.py
+++ b/librarian/epub.py
@@ -7,41 +7,15 @@ from __future__ import with_statement
import os
import os.path
-import shutil
-import sys
from copy import deepcopy
from lxml import etree
import zipfile
-from librarian import XMLNamespace, RDFNS, DCNS, WLNS, XHTMLNS, NoDublinCore
-from librarian.parser import WLDocument
-
-#TODO: shouldn't be repeated here
-NCXNS = XMLNamespace("http://www.daisy.org/z3986/2005/ncx/")
-OPFNS = XMLNamespace("http://www.idpf.org/2007/opf")
-
-
-class DocProvider(object):
- class DoesNotExist(Exception):
- pass
-
- def by_slug(self, slug):
- raise NotImplemented
-
- def __getitem__(self, slug):
- return self.by_slug(slug)
-
- def by_uri(self, uri):
- return self.by_slug(uri.rsplit('/', 1)[1])
-
-
-class DirDocProvider(DocProvider):
- def __init__(self, dir):
- self.dir = dir
- self.files = {}
+import sys
+sys.path.append('..') # for running from working copy
- def by_slug(self, slug):
- return open(os.path.join(self.dir, '%s.xml' % slug))
+from librarian import XMLNamespace, RDFNS, DCNS, WLNS, NCXNS, OPFNS, NoDublinCore
+from librarian.dcparser import BookInfo
def inner_xml(node):
@@ -63,7 +37,6 @@ def set_inner_xml(node, text):
xyz
"""
-
p = etree.fromstring('%s' % text)
node.text = p.text
node[:] = p[:]
@@ -273,11 +246,12 @@ def transform_chunk(chunk_xml, chunk_no, annotations):
return output_html, toc
-def transform(provider, slug, output_file):
+def transform(provider, slug, output_file=None, output_dir=None):
""" produces an epub
provider is a DocProvider
- output_file should be filelike object
+ either output_file (a file-like object) or output_dir (path to file/dir) should be specified
+ if output_dir is specified, file will be written to //.epub
"""
def transform_file(input_xml, chunk_counter=1, first=True):
@@ -312,7 +286,7 @@ def transform(provider, slug, output_file):
if main_text is not None:
replace_characters(main_text)
- for chunk_no, chunk_xml in enumerate(chop(main_text), chunk_counter):
+ for chunk_xml in chop(main_text):
chunk_html, chunk_toc = transform_chunk(chunk_xml, chunk_counter, annotations)
toc.extend(chunk_toc)
zip.writestr('OPS/part%d.html' % chunk_counter, chunk_html)
@@ -328,6 +302,24 @@ def transform(provider, slug, output_file):
return toc, chunk_counter
+ # read metadata from the first file
+ input_xml = etree.parse(provider[slug])
+ metadata = input_xml.find('.//'+RDFNS('Description'))
+ if metadata is None:
+ raise NoDublinCore('Document has no DublinCore - which is required.')
+ book_info = BookInfo.from_element(input_xml)
+ metadata = etree.ElementTree(metadata)
+
+ # if output to dir, create the file
+ if output_dir is not None:
+ author = unicode(book_info.author)
+ author_dir = os.path.join(output_dir, author)
+ try:
+ os.makedirs(author_dir)
+ except OSError:
+ pass
+ output_file = open(os.path.join(author_dir, '%s.epub' % slug), 'w')
+
zip = zipfile.ZipFile(output_file, 'w', zipfile.ZIP_DEFLATED)
@@ -345,12 +337,6 @@ def transform(provider, slug, output_file):
for fname in 'style.css', 'logo_wolnelektury.png':
zip.write(res(fname), os.path.join('OPS', fname))
- # metadata from first file
- input_xml = etree.parse(provider[slug])
- metadata = input_xml.find('.//'+RDFNS('Description'))
- if metadata is None:
- raise NoDublinCore('Document has no DublinCore - which is required.')
- metadata = etree.ElementTree(metadata)
opf = xslt(metadata, res('xsltContent.xsl'))
manifest = opf.find('.//' + OPFNS('manifest'))
spine = opf.find('.//' + OPFNS('spine'))
@@ -363,13 +349,13 @@ def transform(provider, slug, output_file):
'version="2005-1">' \
'' \
'Strona tytuÅowa' \
- '' \
- 'PoczÄ
tek utworu' \
'')
nav_map = toc_file[-1]
toc, chunk_counter = transform_file(input_xml)
- toc_counter = toc.write_to_xml(nav_map, 3) # we already have 2 navpoints
+ if not toc.children:
+ toc.add(u"PoczÄ
tek utworu", 1)
+ toc_counter = toc.write_to_xml(nav_map, 2)
# Last modifications in container files and EPUB creation
if len(annotations) > 0:
@@ -401,6 +387,8 @@ def transform(provider, slug, output_file):
if __name__ == '__main__':
+ from librarian import DirDocProvider
+
if len(sys.argv) < 2:
print >> sys.stderr, 'Usage: python epub.py '
sys.exit(1)
@@ -408,7 +396,6 @@ if __name__ == '__main__':
main_input = sys.argv[1]
basepath, ext = os.path.splitext(main_input)
path, slug = os.path.realpath(basepath).rsplit('/', 1)
- output = basepath + '.epub'
provider = DirDocProvider(path)
- transform(provider, slug, open(output, 'w'))
+ transform(provider, slug, output_dir=path)