Python PDF转JPG

今天一个搞档案数字化的客户给我打电话，需要一个PDF转JPG的工具。答应下来之后开始找方案，在网上以“Python PDF转JPG”关键词搜索了一些文章，进入一看都是转的PNG的，用不成，有个别文章里确实是赚的JPG，但是需要安装这个库那个库的，啰里啰嗦的写好多代码，感觉好麻烦。心想：PDF转JPG应该是很简单的事情啊，怎么会这么麻烦？是不是大家都走弯路了？找了一篇文章看了下，参考：https://www.jianshu.com/p/83d9d1e05eb2，大家注意：这篇文件的标题声称“python实现pdf转jpg”，但是实际上只是生成了PNG而已，不过这篇文正的代码够简洁的，如果能在此基础上生成JPG岂不是很好？经过阅读fitz的源代码，发现实际上是可以生成JPG的，也就在原文基础上改一行代码而已，为了更多的人不走弯路，特写此文章，希望对大家有点帮助。好了，闲话不说了，直接上代码吧。

# -*- coding: utf-8 -*-

"""

1、安装库 pip install pymupdf
2、安装库 pip install pillow
3、直接运行

"""
import os
import fitz

pdf_dir = []


def get_file():
    docunames = os.listdir()
    for docuname in docunames:
        if os.path.splitext(docuname)[1] == '.pdf':  # 目录下包含.pdf的文件
            pdf_dir.append(docuname)


def conver_img():
    for pdf in pdf_dir:
        doc = fitz.open(pdf)
        pdf_name = os.path.splitext(pdf)[0]
        for pg in range(doc.pageCount):
            page = doc[pg]
            rotate = int(0)
            # 每个尺寸的缩放系数为2，这将为我们生成分辨率提高四倍的图像。
            zoom_x = 2.0
            zoom_y = 2.0
            trans = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)
            pm = page.getPixmap(matrix=trans, alpha=False)
            # 注意下边的一行，这是本的重点。原文是生成的PNG，我给改成了JPG
            pm.pillowWrite('%s.jpg' % pg, quality = 95.0)

if __name__ == '__main__':
    get_file()
    conver_img()

在生成JPG之前还可以设置分辨率等等，我本着示例代码尽量少写的原则，没有写这些，大家可以自己查看一下fitz的源代码。