20多行 Python 代码优雅搞定 PDF 转换成图片

 1# -*- coding: utf-8 -*-
 2
 3import io
 4from wand.image import Image
 5from wand.color import Color
 6from PyPDF2 import PdfFileReader, PdfFileWriter
 7memo = {}
 8'''
 9需要注意的是一般PDF文件较大，如果一次性转换
10整个PDF文件需要小心内存溢出的问题，我们这里
11将第一次载入的整个PDF文件保存到内存，避免每
12次读取的时候都重新载入。
13'''
14def getPdfReader(filename):
15   reader = memo.get(filename, None)
16   if reader is None:
17       reader = PdfFileReader(filename, strict=False)
18       memo[filename] = reader
19   return reader
20
21def _run_convert(filename, page, res=120):
22   idx = page + 1
23   pdfile = getPdfReader(filename)
24   pageObj = pdfile.getPage(page)
25   dst_pdf = PdfFileWriter()
26   dst_pdf.addPage(pageObj)
27   pdf_bytes = io.BytesIO()
28   dst_pdf.write(pdf_bytes)
29   pdf_bytes.seek(0)
30   img = Image(file=pdf_bytes, resolution=res)
31   img.format =  png 
32   img.compression_quality = 90
33   img.background_color = Color("white")
34   img_path =  %s%d.png  % (filename[:filename.rindex( . )], idx)
35   img.save(filename=img_path)
36   img.destroy()