This commit is contained in:
Kovid Goyal 2011-08-29 17:00:16 -06:00
commit 52d1bac2d8
2 changed files with 9 additions and 8 deletions

View File

@ -437,7 +437,7 @@ class LRFMetadataWriter(MetadataWriterPlugin):
class MOBIMetadataWriter(MetadataWriterPlugin):
name = 'Set MOBI metadata'
file_types = set(['mobi', 'prc', 'azw'])
file_types = set(['mobi', 'prc', 'azw', 'azw4'])
description = _('Set metadata in %s files')%'MOBI'
author = 'Marshall T. Vandegrift'

View File

@ -10,7 +10,7 @@ __docformat__ = 'restructuredtext en'
from calibre.ebooks.pdb.formatreader import FormatReader
from calibre.ptempfile import TemporaryFile
from calibre.ptempfile import PersistentTemporaryFile
class Reader(FormatReader):
@ -23,10 +23,12 @@ class Reader(FormatReader):
def extract_content(self, output_dir):
self.log.info('Extracting PDF...')
with TemporaryFile() as pdf_n:
pdf = open(pdf_n, 'rwb')
pdf = PersistentTemporaryFile('.pdf')
pdf.close()
pdf = open(pdf, 'wb')
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
@ -35,5 +37,4 @@ class Reader(FormatReader):
if not hasattr(self.options, opt.option.name):
setattr(self.options, opt.option.name, opt.recommended_value)
pdf.seek(0)
return pdf_plugin.convert(pdf, self.options, 'pdf', self.log, {})
return pdf_plugin.convert(open(pdf, 'rb'), self.options, 'pdf', self.log, {})