Initial import

ebook_converter/ebooks/conversion/plugins/pml_output.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+__license__ = 'GPL 3'
+__copyright__ = '2009, John Schember <john@nachtimwald.com>'
+__docformat__ = 'restructuredtext en'
+
+import os, io
+
+from calibre.customize.conversion import (OutputFormatPlugin,
+        OptionRecommendation)
+from calibre.ptempfile import TemporaryDirectory
+from polyglot.builtins import unicode_type
+
+
+class PMLOutput(OutputFormatPlugin):
+
+    name = 'PML Output'
+    author = 'John Schember'
+    file_type = 'pmlz'
+    commit_name = 'pml_output'
+
+    options = {
+        OptionRecommendation(name='pml_output_encoding', recommended_value='cp1252',
+            level=OptionRecommendation.LOW,
+            help=_('Specify the character encoding of the output document. '
+            'The default is cp1252.')),
+        OptionRecommendation(name='inline_toc',
+            recommended_value=False, level=OptionRecommendation.LOW,
+            help=_('Add Table of Contents to beginning of the book.')),
+        OptionRecommendation(name='full_image_depth',
+            recommended_value=False, level=OptionRecommendation.LOW,
+            help=_('Do not reduce the size or bit depth of images. Images '
+                   'have their size and depth reduced by default to accommodate '
+                   'applications that can not convert images on their '
+                   'own such as Dropbook.')),
+    }
+
+    def convert(self, oeb_book, output_path, input_plugin, opts, log):
+        from calibre.ebooks.pml.pmlml import PMLMLizer
+        from calibre.utils.zipfile import ZipFile
+
+        with TemporaryDirectory('_pmlz_output') as tdir:
+            pmlmlizer = PMLMLizer(log)
+            pml = unicode_type(pmlmlizer.extract_content(oeb_book, opts))
+            with lopen(os.path.join(tdir, 'index.pml'), 'wb') as out:
+                out.write(pml.encode(opts.pml_output_encoding, 'replace'))
+
+            img_path = os.path.join(tdir, 'index_img')
+            if not os.path.exists(img_path):
+                os.makedirs(img_path)
+            self.write_images(oeb_book.manifest, pmlmlizer.image_hrefs, img_path, opts)
+
+            log.debug('Compressing output...')
+            pmlz = ZipFile(output_path, 'w')
+            pmlz.add_dir(tdir)
+
+    def write_images(self, manifest, image_hrefs, out_dir, opts):
+        from PIL import Image
+
+        from calibre.ebooks.oeb.base import OEB_RASTER_IMAGES
+        for item in manifest:
+            if item.media_type in OEB_RASTER_IMAGES and item.href in image_hrefs.keys():
+                if opts.full_image_depth:
+                    im = Image.open(io.BytesIO(item.data))
+                else:
+                    im = Image.open(io.BytesIO(item.data)).convert('P')
+                    im.thumbnail((300,300), Image.ANTIALIAS)
+
+                data = io.BytesIO()
+                im.save(data, 'PNG')
+                data = data.getvalue()
+
+                path = os.path.join(out_dir, image_hrefs[item.href])
+
+                with lopen(path, 'wb') as out:
+                    out.write(data)