1
0
mirror of https://github.com/gryf/ebook-converter.git synced 2026-01-30 10:15:45 +01:00
Files
ebook-converter/ebook_converter/ebooks/pdb/pdf/reader.py
gryf 48fedea799 Removing unneeded from __future__ import statements.
Since we are on Python 3.6 and up, we don't need those anymore.
2020-04-19 17:39:02 +02:00

40 lines
1.2 KiB
Python

"""
Read content from palmdoc pdb file.
"""
from ebook_converter.ebooks.pdb.formatreader import FormatReader
from ebook_converter.ptempfile import PersistentTemporaryFile
from ebook_converter.polyglot.builtins import range
__license__ = 'GPL v3'
__copyright__ = '2010, John Schember <john@nachtimwald.com>'
__docformat__ = 'restructuredtext en'
class Reader(FormatReader):
def __init__(self, header, stream, log, options):
self.header = header
self.stream = stream
self.log = log
self.options = options
def extract_content(self, output_dir):
self.log.info('Extracting PDF...')
pdf = PersistentTemporaryFile('.pdf')
pdf.close()
pdf = open(pdf, 'wb')
for x in range(self.header.section_count()):
pdf.write(self.header.section_data(x))
pdf.close()
from ebook_converter.customize.ui import plugin_for_input_format
pdf_plugin = plugin_for_input_format('pdf')
for opt in pdf_plugin.options:
if not hasattr(self.options, opt.option.name):
setattr(self.options, opt.option.name, opt.recommended_value)
return pdf_plugin.convert(open(pdf, 'rb'), self.options, 'pdf', self.log, {})