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

Fix PDF cover extraction

Browse Source
This commit is contained in:
Kovid Goyal 2009-04-18 09:29:16 -07:00
parent f969ed39fe
commit 02cfaac014
1 changed files with 11 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/calibre/ebooks/metadata/pdf.py
View File

@ -10,7 +10,6 @@ from calibre import FileWrapper
from calibre.ebooks.metadata import MetaInformation, authors_to_string
from calibre.ptempfile import TemporaryDirectory
from pyPdf import PdfFileReader, PdfFileWriter
#import Image
try:
from calibre.utils.PythonMagickWand import \
NewMagickWand, MagickReadImage, MagickSetImageFormat, \
@ -95,7 +94,6 @@ def set_metadata(stream, mi):
stream.seek(0)
def get_cover(stream):
data = cStringIO.StringIO()
try:
pdf = PdfFileReader(stream)
@ -107,21 +105,18 @@ def get_cover(stream):
with TemporaryDirectory('_pdfmeta') as tdir:
cover_path = os.path.join(tdir, 'cover.pdf')
outputStream = file(cover_path, "wb")
with open(cover_path, "wb") as outputStream:
output.write(outputStream)
outputStream.close()
with ImageMagick():
wand = NewMagickWand()
MagickReadImage(wand, cover_path)
MagickSetImageFormat(wand, 'JPEG')
MagickWriteImage(wand, '%s.jpg' % cover_path)
return open('%s.jpg' % cover_path, 'rb').read()
#img = Image.open('%s.jpg' % cover_path)
#img.save(data, 'JPEG')
except:
import traceback
traceback.print_exc()
return data.getvalue()
return ''