结果效果

1、将pdf文件按指定页数进行分割
2、将pdf文件按页导出jpg文件

image.png

源码

import tkinter as tk
from tkinter import filedialog, messagebox, scrolledtext
import os

import fitz #python -m pip install PyMuPDF==1.18.7 -i https://pypi.tuna.tsinghua.edu.cn/simple
import PyPDF2
import datetime

import threading
import queue

class PDFInputApp:
    def __init__(self, root):
        self.root = root
        self.root.title("PDF-处理工具")
        self.root.geometry("500x330")

        # 创建标签和输入框
        self.create_widgets()

        # 创建日志队列和日志线程
        self.log_queue = queue.Queue()
        self.log_thread = threading.Thread(target=self.process_log_queue, daemon=True)
        self.log_thread.start()


    def create_widgets(self):
        # 设置列权重，使输入框自动扩展
        self.root.grid_columnconfigure(1, weight=1)
        self.root.grid_columnconfigure(3, weight=1)  # 使第 3 列也具有扩展性，确保按钮居中

        # 选择 PDF 文件
        self.pdf_label = tk.Label(self.root, text="选择 PDF 文件:")
        self.pdf_label.grid(row=0, column=0, padx=10, pady=5, sticky=tk.E)

        self.pdf_path_var = tk.StringVar()
        self.pdf_entry = tk.Entry(self.root, textvariable=self.pdf_path_var, width=50)
        self.pdf_entry.grid(row=0, column=1, padx=10, pady=5, sticky=tk.W+tk.E)

        self.browse_button = tk.Button(self.root, text="浏览", command=self.browse_pdf)
        self.browse_button.grid(row=0, column=2, padx=10, pady=5, sticky=tk.W)

        # 输入数字
        self.number_label = tk.Label(self.root, text="每份页数:")
        self.number_label.grid(row=1, column=0, padx=10, pady=5, sticky=tk.E)

        self.number_var = tk.IntVar()
        self.number_var.set(1)
        self.number_entry = tk.Entry(self.root, textvariable=self.number_var, width=50)
        self.number_entry.grid(row=1, column=1, padx=10, pady=5, sticky=tk.W+tk.E)

        # 显示所选文件的路径
        self.path_label = tk.Label(self.root, text="结果文件路径:")
        self.path_label.grid(row=2, column=0, padx=10, pady=5, sticky=tk.E)

        self.out_path_var = tk.StringVar()
        self.path_entry = tk.Entry(self.root, textvariable=self.out_path_var, width=50, state='readonly')
        self.path_entry.grid(row=2, column=1, padx=10, pady=5, sticky=tk.W+tk.E)

        # 提交和导出按钮放在同一行，居中显示，中间有 10px 间隔
        self.submit_button = tk.Button(self.root, text="切割PDF", command=self.submit,width=15)
        self.export_button = tk.Button(self.root, text="导出图片", command=self.export,width=15)

        # 使用空列 (column 3) 来创建 10px 的间隔
        self.root.grid_columnconfigure(3, minsize=5)  # 设置第 3 列的最小宽度为 10px

        # 将按钮放置在第 4 行，跨越 3 列，居中对齐
        self.submit_button.grid(row=3, column=1, padx=10, pady=15, sticky=tk.W)
        self.export_button.grid(row=3, column=1, padx=10, pady=15, sticky=tk.E)

        # 添加日志文本框
        self.log_label = tk.Label(self.root, text="处理日志:")
        self.log_label.grid(row=4, column=0, padx=10, pady=5, sticky=tk.NE)

        # 使用 ScrolledText 控件创建带滚动条的多行文本框
        self.log_text = scrolledtext.ScrolledText(self.root, width=70, height=10, state='disabled')
        self.log_text.grid(row=4, column=1, columnspan=3, padx=10, pady=5, sticky=tk.W+tk.E)

    def browse_pdf(self):
        # 打开文件对话框，限制只能选择 PDF 文件
        file_path = filedialog.askopenfilename(
            filetypes=[("PDF 文件", "*.pdf")],
            title="选择 PDF 文件"
        )
        if file_path:
            self.pdf_path_var.set(file_path)
            self.out_path_var.set(os.path.dirname(file_path))

    def submit(self):
        # 获取输入的 PDF 文件路径和数字
        pdf_path = self.pdf_path_var.get()
        number = self.number_var.get()
        out_path = self.out_path_var.get()

        # 验证输入
        if not pdf_path:
            messagebox.showwarning("警告", "请选择一个 PDF 文件")
            return

        # if not number.i():
        #     messagebox.showwarning("警告", "请输入有效的数字")
        #     return
        self.log(f"准备开始分隔文件{pdf_path}")
        self.split_pdf(pdf_path,out_path,number)
        # 处理逻辑（这里可以根据需要添加更多处理）
        messagebox.showinfo("分割处理成功!", f"PDF 文件: {pdf_path}\n每页: {number}完成分隔!")
        self.log(f"完成文件{pdf_path}分隔")

    def export(self):
         # 获取输入的 PDF 文件路径和数字
        pdf_path = self.pdf_path_var.get()
        number = self.number_var.get()
        out_path = self.out_path_var.get()

        # 验证输入
        if not pdf_path:
            messagebox.showwarning("警告", "请选择一个 PDF 文件")
            return
        self.log(f"准备开始分隔文件{pdf_path}")
        self.pdf_image(pdf_path,out_path,1.0,1.0,0)
        # 导出功能的逻辑（根据需要实现）
        messagebox.showinfo("导出成功", f"PDF 文件: {pdf_path}\n导出jpg成功!")
        self.log(f"完成文件{pdf_path}处理")

    #分割pdf
    def split_pdf(self,input_pdf, output_dir, num_pages_per_file):
        with open(input_pdf, "rb") as file:
            reader = PyPDF2.PdfReader(file)
            total_pages = len(reader.pages)
            
            for i in range(0, total_pages, num_pages_per_file):
                start_page = i
                end_page = min(i + num_pages_per_file, total_pages)
                self.log(f"开始处理第{start_page}至｛end_page｝页内容")
                output_pdf = f"{output_dir}/split_{start_page+1}-{end_page}.pdf"
                writer = PyPDF2.PdfWriter()
                
                for page_num in range(start_page, end_page):
                    page = reader.pages[page_num]
                    writer.add_page(page)
                
                with open(output_pdf, "wb") as out_file:
                    writer.write(out_file)
                    self.log(f"Saved {output_pdf}")
    '''
    # 将PDF转化为图片
    pdfPath pdf文件的路径
    imgPath 图像要保存的文件夹
    zoom_x x方向的缩放系数 - 值越大生成的文件越大
    zoom_y y方向的缩放系数 - 值越大生成的文件越大
    rotation_angle 旋转角度
    '''
    def pdf_image(self,pdfPath,imgPath,zoom_x,zoom_y,rotation_angle):
        # 打开PDF文件
        pdf = fitz.open(pdfPath)
        # 逐页读取PDF
        for pg in range(0, pdf.pageCount):
            self.log(f"开始处理第{pg}页")
            page = pdf[pg]
            # 设置缩放和旋转系数
            trans = fitz.Matrix(zoom_x, zoom_y).preRotate(rotation_angle)
            pm = page.getPixmap(matrix=trans, alpha=False)
            # 开始写图像
            pm.writePNG(imgPath+"\\image-"+str(pg)+".jpg")
            self.log(f"生成图片 image-"+str(pg)+".jpg")
        pdf.close()


    def log(self, message):
        """将消息放入日志队列中，由日志线程处理"""
        current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        log_message = f"[{current_time}] {message}\n"
        self.log_queue.put(log_message)

    def process_log_queue(self):
        """从日志队列中取出消息并更新日志文本框"""
        while True:
            try:
                log_message = self.log_queue.get(timeout=0.1)
                self.update_log_text(log_message)
            except queue.Empty:
                continue

    def update_log_text(self, log_message):
        """更新日志文本框的内容"""
        # 确保在主线程中更新 GUI
        self.root.after(0, self._update_log_text, log_message)

    def _update_log_text(self, log_message):
        """实际更新日志文本框的方法"""
        # 启用文本框编辑模式，插入日志，然后禁用编辑
        self.log_text.config(state='normal')
        self.log_text.insert(tk.END, log_message)
        self.log_text.config(state='disabled')

        # 滚动到最新日志
        self.log_text.yview_moveto(1.0)

if __name__ == "__main__":
    root = tk.Tk()
    app = PDFInputApp(root)
    root.mainloop()