Python 图片与pdf相互转换

谁借莪１个温暖的怀抱￠ 2023-02-18 01:47 25阅读 0赞

使用 Python 进行图片和pdf之间的相互转换  
使用到第三方库 PyMuPDF 在 python 环境下对 PDF 文件的操作。

安装方式:pip install pymupdf

PDF 转为图片  
需新建文件夹 pdf2png

import fitz
    import glob
    
    def rightinput(desc):
        flag=True
        while(flag):
            instr = input(desc)
            try:
                intnum = eval(instr)
                if type(intnum)==int:
                    flag = False
            except:
                print('请输入正整数！')
                pass
        return intnum
    
    pdffile = glob.glob("*.pdf")[0]
    doc = fitz.open(pdffile)
    
    flag = rightinput("输入：1：全部页面；2：选择页面\t")
    if flag == 1:
        strat = 0
        totaling = doc.pageCount
    else:
        strat = rightinput('输入起始页面：') - 1
        totaling = rightinput('输入结束页面：')
    
    for pg in range(strat, totaling):
        page = doc[pg]
        zoom = int(100)
        rotate = int(0)
        trans = fitz.Matrix(zoom / 100.0, zoom / 100.0).preRotate(rotate)
        pm = page.getPixmap(matrix=trans, alpha=False)
        pm.writePNG('pdf2png/%s.png' % str(pg+1))

图片合并为 PDF 文件  
将所需合并的图片放置与文件夹 pic2pdf 中。

import glob
    import fitz
    import os
    
    def pic2pdf():
        doc = fitz.open()
        for img in sorted(glob.glob("pic2pdf/*")):  # 读取图片，确保按文件名排序
            print(img)
            imgdoc = fitz.open(img)                 # 打开图片
            pdfbytes = imgdoc.convertToPDF()        # 使用图片创建单页的 PDF
            imgpdf = fitz.open("pdf", pdfbytes)
            doc.insertPDF(imgpdf)                   # 将当前页插入文档
        if os.path.exists("allimages.pdf"):
            os.remove("allimages.pdf")
        doc.save("allimages.pdf")                   # 保存pdf文件
        doc.close()
    
    if __name__ == '__main__':
        pic2pdf()