【python批量插入图片到一个pdf中】

批量插入图片到一个pdf中

    • 需求
    • 解决思路
    • 代码详解

需求

把大量的img图片插入到一个pdf中,一张图片为pdf的一页

解决思路

先把大量的图片转为一个个的pdf,然后合并所有的pdf

话不多说,直接上代码:

import os
from PIL import Image
from PyPDF2 import PdfFileMerger

# 防止字符串乱码
os.environ['NLS_LANG'] = 'SIMPLIFIED CHINESE_CHINA.UTF8'

class AllImagesToPdf:

    def __init__(self):
        self.imgs_path = "imgs"  # 将所有的图片放到此文件夹中
        self.pdfs_path = "pdfs"  # 将所有转换后的pdf放到此文件夹

    def imgs_to_pdfs(self):
        imgs_path = self.imgs_path
        imgs_list = os.listdir(imgs_path)
        for img_name in imgs_list:
            if "jpg" in img_name or "png" in img_name:
                read_img_path = os.path.join(self.imgs_path,img_name)
                obj_img = Image.open(read_img_path)  # 打开指定路径下的图片
                save_img_path = os.path.join(self.pdfs_path,img_name.split(".")[0]+".pdf")  # 拼接pdf的路径和名称
                obj_img.save(save_img_path)  # 保存为pdf

    def pdfs_to_pdf(self):
        pdfs_path = self.pdfs_path
        pdfs_list = os.listdir(pdfs_path)
        pdfs_path_list = []
        for pdf_name in pdfs_list:
            if pdf_name.endswith(".pdf"):
                pdfs_path_list.append(os.path.join(self.pdfs_path,pdf_name))
        # print(pdfs_path_list)
        pdf_merger = PdfFileMerger()  # 创建PdfFileMerger对象,用来合并pdf文件
        for file_path in pdfs_path_list:
            pdf_merger.append(file_path)

        pdf_merger.write("mergerd.pdf")  # 使用write方法将所有pdf文件写入到一个文件

    def run(self):

        self.imgs_to_pdfs()  # 把大量的图片一个个转为pdf
        print("批量图片转批量pdf完毕")
        self.pdfs_to_pdf()   # 把大量的pdf合并到一个pdf
        print("批量pdf转pdf完毕")


if __name__ == '__main__':
    itp = AllImagesToPdf()
    itp.run()

代码详解

1、需要事先在代码文件的同级目录下新建两个文件夹:imgspdfs。文件夹imgs内放入相关图片。
2、安装好相应的依赖包:PyPDF2
3、代码中有两个函数:

imgs_to_pdfs:将所有的图片文件转为相应的pdf
pdfs_to_pdf:将所有的pdf合并为一个pdf

4、准备好文件夹和安装完包后,直接运行代码,然后同级目录下会出现一个pdf文件:mergerd.pdf,这就是我们最终要的结果

你可能感兴趣的:(python,开发语言)