Added epub write support

2026-04-27 17:13:32 +02:00 · 2020-04-13 12:46:37 +02:00
parent 9f18513787
commit 79cad46732
9 changed files with 3049 additions and 0 deletions
@@ -0,0 +1,172 @@
+#!/usr/bin/env python2
+# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+__license__   = 'GPL v3'
+__copyright__ = '2010, Kovid Goyal <kovid@kovidgoyal.net>'
+__docformat__ = 'restructuredtext en'
+
+import textwrap
+
+from calibre import guess_type
+from calibre.utils.imghdr import identify
+from calibre.utils.xml_parse import safe_xml_fromstring
+from polyglot.builtins import unicode_type
+from polyglot.urllib import unquote
+
+
+class CoverManager(object):
+
+    SVG_TEMPLATE = textwrap.dedent('''\
+        <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en">
+            <head>
+                <meta http-equiv="Content-Type" content="text/html; charset=UTF-8" />
+                <meta name="calibre:cover" content="true" />
+                <title>Cover</title>
+                <style type="text/css" title="override_css">
+                    @page {padding: 0pt; margin:0pt}
+                    body { text-align: center; padding:0pt; margin: 0pt; }
+                </style>
+            </head>
+            <body>
+                <div>
+                    <svg version="1.1" xmlns="http://www.w3.org/2000/svg"
+                        xmlns:xlink="http://www.w3.org/1999/xlink"
+                        width="100%%" height="100%%" viewBox="__viewbox__"
+                        preserveAspectRatio="__ar__">
+                        <image width="__width__" height="__height__" xlink:href="%s"/>
+                    </svg>
+                </div>
+            </body>
+        </html>
+        ''')
+
+    NONSVG_TEMPLATE = textwrap.dedent('''\
+        <html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en">
+            <head>
+                <meta http-equiv="Content-Type" content="text/html; charset=UTF-8" />
+                <meta name="calibre:cover" content="true" />
+                <title>Cover</title>
+                <style type="text/css" title="override_css">
+                    @page {padding: 0pt; margin:0pt}
+                    body { text-align: center; padding:0pt; margin: 0pt }
+                    div { padding:0pt; margin: 0pt }
+                    img { padding:0pt; margin: 0pt }
+                </style>
+            </head>
+            <body>
+                <div>
+                    <img src="%s" alt="cover" __style__ />
+                </div>
+            </body>
+        </html>
+    ''')
+
+    def __init__(self, no_default_cover=False, no_svg_cover=False,
+            preserve_aspect_ratio=False, fixed_size=None):
+        self.no_default_cover = no_default_cover
+        self.no_svg_cover = no_svg_cover
+        self.preserve_aspect_ratio = preserve_aspect_ratio
+
+        ar = 'xMidYMid meet' if preserve_aspect_ratio else 'none'
+        self.svg_template = self.SVG_TEMPLATE.replace('__ar__', ar)
+
+        if fixed_size is None:
+            style = 'style="height: 100%%"'
+        else:
+            width, height = fixed_size
+            style = 'style="height: %s; width: %s"'%(height, width)
+        self.non_svg_template = self.NONSVG_TEMPLATE.replace('__style__',
+                style)
+
+    def __call__(self, oeb, opts, log):
+        self.oeb = oeb
+        self.log = log
+        self.insert_cover()
+
+    def default_cover(self):
+        '''
+        Create a generic cover for books that dont have a cover
+        '''
+        if self.no_default_cover:
+            return None
+        self.log('Generating default cover')
+        m = self.oeb.metadata
+        title = unicode_type(m.title[0])
+        authors = [unicode_type(x) for x in m.creator if x.role == 'aut']
+        try:
+            from calibre.ebooks.covers import create_cover
+            series = series_index = None
+            if m.series:
+                try:
+                    series, series_index = unicode_type(m.series[0]), m.series_index[0]
+                except IndexError:
+                    pass
+            img_data = create_cover(title, authors, series, series_index)
+            id, href = self.oeb.manifest.generate('cover',
+                    'cover_image.jpg')
+            item = self.oeb.manifest.add(id, href, guess_type('t.jpg')[0],
+                        data=img_data)
+            m.clear('cover')
+            m.add('cover', item.id)
+
+            return item.href
+        except:
+            self.log.exception('Failed to generate default cover')
+        return None
+
+    def inspect_cover(self, href):
+        from calibre.ebooks.oeb.base import urlnormalize
+        for x in self.oeb.manifest:
+            if x.href == urlnormalize(href):
+                try:
+                    raw = x.data
+                    return identify(raw)[1:]
+                except Exception:
+                    self.log.exception('Failed to read cover image dimensions')
+        return -1, -1
+
+    def insert_cover(self):
+        from calibre.ebooks.oeb.base import urldefrag
+        g, m = self.oeb.guide, self.oeb.manifest
+        item = None
+        if 'titlepage' not in g:
+            if 'cover' in g:
+                href = g['cover'].href
+            else:
+                href = self.default_cover()
+            if href is None:
+                return
+            width, height = self.inspect_cover(href)
+            if width == -1 or height == -1:
+                self.log.warning('Failed to read cover dimensions')
+                width, height = 600, 800
+            # if self.preserve_aspect_ratio:
+            #    width, height = 600, 800
+            self.svg_template = self.svg_template.replace('__viewbox__',
+                    '0 0 %d %d'%(width, height))
+            self.svg_template = self.svg_template.replace('__width__',
+                    unicode_type(width))
+            self.svg_template = self.svg_template.replace('__height__',
+                    unicode_type(height))
+
+            if href is not None:
+                templ = self.non_svg_template if self.no_svg_cover \
+                        else self.svg_template
+                tp = templ%unquote(href)
+                id, href = m.generate('titlepage', 'titlepage.xhtml')
+                item = m.add(id, href, guess_type('t.xhtml')[0],
+                        data=safe_xml_fromstring(tp))
+        else:
+            item = self.oeb.manifest.hrefs[
+                    urldefrag(self.oeb.guide['titlepage'].href)[0]]
+        if item is not None:
+            self.oeb.spine.insert(0, item, True)
+            if 'cover' not in self.oeb.guide.refs:
+                self.oeb.guide.add('cover', 'Title Page', 'a')
+            self.oeb.guide.refs['cover'].href = item.href
+            if 'titlepage' in self.oeb.guide.refs:
+                self.oeb.guide.refs['titlepage'].href = item.href
+            titem = getattr(self.oeb.toc, 'item_that_refers_to_cover', None)
+            if titem is not None:
+                titem.href = item.href