Python 图片与pdf相互转换

发布时间:2019-09-12 07:54:07编辑:auto阅读(1549)

    使用 Python 进行图片和pdf之间的相互转换

    使用到第三方库 PyMuPDF 在 python 环境下对 PDF 文件的操作。

    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()

关键字