Cleanup PDF cover extraction

This commit is contained in:
Kovid Goyal 2009-04-18 01:12:56 -07:00
parent 7dd20f593b
commit 927be72be5

View File

@ -10,10 +10,11 @@ from threading import Thread
from calibre.ebooks.metadata import MetaInformation, authors_to_string from calibre.ebooks.metadata import MetaInformation, authors_to_string
from calibre.ptempfile import TemporaryDirectory from calibre.ptempfile import TemporaryDirectory
from pyPdf import PdfFileReader, PdfFileWriter from pyPdf import PdfFileReader, PdfFileWriter
import Image #import Image
try: try:
from calibre.utils.PythonMagickWand import \ from calibre.utils.PythonMagickWand import \
NewMagickWand, MagickReadImage, MagickSetImageFormat, MagickWriteImage NewMagickWand, MagickReadImage, MagickSetImageFormat, \
MagickWriteImage, ImageMagick
_imagemagick_loaded = True _imagemagick_loaded = True
except: except:
_imagemagick_loaded = False _imagemagick_loaded = False
@ -108,15 +109,15 @@ def get_cover(stream):
outputStream = file(cover_path, "wb") outputStream = file(cover_path, "wb")
output.write(outputStream) output.write(outputStream)
outputStream.close() outputStream.close()
with ImageMagick():
wand = NewMagickWand()
MagickReadImage(wand, cover_path)
MagickSetImageFormat(wand, 'JPEG')
MagickWriteImage(wand, '%s.jpg' % cover_path)
wand = NewMagickWand() #img = Image.open('%s.jpg' % cover_path)
MagickReadImage(wand, cover_path)
MagickSetImageFormat(wand, 'JPEG')
MagickWriteImage(wand, '%s.jpg' % cover_path)
img = Image.open('%s.jpg' % cover_path) #img.save(data, 'JPEG')
img.save(data, 'JPEG')
except: except:
import traceback import traceback
traceback.print_exc() traceback.print_exc()