mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-09 05:04:12 +01:00
gryf
0f9792df36
Here is the first batch of modules, which are needed for converting several formats to LRF. Some of the logic has been change, more cleanups will follow.
78 lines
3.1 KiB
Python
78 lines
3.1 KiB
Python
# -*- 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 ebook_converter.customize.conversion import (OutputFormatPlugin,
|
|
OptionRecommendation)
|
|
from ebook_converter.ptempfile import TemporaryDirectory
|
|
from ebook_converter.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 ebook_converter.ebooks.pml.pmlml import PMLMLizer
|
|
from ebook_converter.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 ebook_converter.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)
|