mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-01-05 18:54:11 +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.
65 lines
2.3 KiB
Python
65 lines
2.3 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
|
|
|
|
from ebook_converter.customize.conversion import OutputFormatPlugin, \
|
|
OptionRecommendation
|
|
from ebook_converter.ebooks.pdb import PDBError, get_writer, ALL_FORMAT_WRITERS
|
|
|
|
|
|
class PDBOutput(OutputFormatPlugin):
|
|
|
|
name = 'PDB Output'
|
|
author = 'John Schember'
|
|
file_type = 'pdb'
|
|
commit_name = 'pdb_output'
|
|
ui_data = {'formats': tuple(ALL_FORMAT_WRITERS)}
|
|
|
|
options = {
|
|
OptionRecommendation(name='format', recommended_value='doc',
|
|
level=OptionRecommendation.LOW,
|
|
short_switch='f', choices=list(ALL_FORMAT_WRITERS),
|
|
help=(_('Format to use inside the pdb container. Choices are:') + ' %s' % sorted(ALL_FORMAT_WRITERS))),
|
|
OptionRecommendation(name='pdb_output_encoding', recommended_value='cp1252',
|
|
level=OptionRecommendation.LOW,
|
|
help=_('Specify the character encoding of the output document. '
|
|
'The default is cp1252. Note: This option is not honored by all '
|
|
'formats.')),
|
|
OptionRecommendation(name='inline_toc',
|
|
recommended_value=False, level=OptionRecommendation.LOW,
|
|
help=_('Add Table of Contents to beginning of the book.')),
|
|
}
|
|
|
|
def convert(self, oeb_book, output_path, input_plugin, opts, log):
|
|
close = False
|
|
if not hasattr(output_path, 'write'):
|
|
close = True
|
|
if not os.path.exists(os.path.dirname(output_path)) and os.path.dirname(output_path):
|
|
os.makedirs(os.path.dirname(output_path))
|
|
out_stream = lopen(output_path, 'wb')
|
|
else:
|
|
out_stream = output_path
|
|
|
|
Writer = get_writer(opts.format)
|
|
|
|
if Writer is None:
|
|
raise PDBError('No writer available for format %s.' % format)
|
|
|
|
setattr(opts, 'max_line_length', 0)
|
|
setattr(opts, 'force_max_line_length', False)
|
|
|
|
writer = Writer(opts, log)
|
|
|
|
out_stream.seek(0)
|
|
out_stream.truncate()
|
|
|
|
writer.write_content(oeb_book, out_stream, oeb_book.metadata)
|
|
|
|
if close:
|
|
out_stream.close()
|