Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fixes for PDB-PDF input.

Browse Source
This commit is contained in:
John Schember 2011-08-28 12:27:10 -04:00
parent 472e8d1743
commit 7b5cecfc2e
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/calibre/ebooks/pdb/pdf/reader.py
View File

@ -10,7 +10,7 @@ __docformat__ = 'restructuredtext en'
from calibre.ebooks.pdb.formatreader import FormatReader from calibre.ebooks.pdb.formatreader import FormatReader
from calibre.ptempfile import TemporaryFile from calibre.ptempfile import PersistentTemporaryFile
class Reader(FormatReader): class Reader(FormatReader):
@ -23,10 +23,12 @@ class Reader(FormatReader):
def extract_content(self, output_dir): def extract_content(self, output_dir):
self.log.info('Extracting PDF...') self.log.info('Extracting PDF...')
with TemporaryFile() as pdf_n: pdf = PersistentTemporaryFile('.pdf')
pdf = open(pdf_n, 'rwb') pdf.close()
for x in xrange(self.header.section_count()): pdf = open(pdf, 'wb')
pdf.write(self.header.section_data(x)) for x in xrange(self.header.section_count()):
pdf.write(self.header.section_data(x))
pdf.close()
from calibre.customize.ui import plugin_for_input_format from calibre.customize.ui import plugin_for_input_format
@ -35,5 +37,4 @@ class Reader(FormatReader):
if not hasattr(self.options, opt.option.name): if not hasattr(self.options, opt.option.name):
setattr(self.options, opt.option.name, opt.recommended_value) setattr(self.options, opt.option.name, opt.recommended_value)
pdf.seek(0) return pdf_plugin.convert(open(pdf, 'rb'), self.options, 'pdf', self.log, {})
return pdf_plugin.convert(pdf, self.options, 'pdf', self.log, {})