PDF工具应用程序是一个使用wxPython和PyMuPDF库编写的简单工具,用于合并和分拆PDF文件。它提供了一个用户友好的图形界面,允许用户选择源文件夹和目标文件夹,并对PDF文件进行操作。
C:\pythoncode\blog\pdfmergandsplit.py
功能特点
- 选择文件夹:用户可以通过应用程序界面轻松选择源文件夹和目标文件夹。
- 合并PDF文件:应用程序允许用户选择要合并的PDF文件,并将它们合并成一个单独的PDF文件。
- 分拆PDF文件:用户可以选择一个PDF文件,将其分拆成多个单独的PDF文件,每个文件包含一个页面。
程序实现
该应用程序使用了以下库和模块:
- wxPython:用于创建应用程序的图形用户界面。
- PyMuPDF:用于处理PDF文件的库。
应用程序的主要部分是一个继承自wxPython的wx.Frame
类的主窗口。窗口包含以下组件:
- 选择源文件夹和目标文件夹的按钮。
- PDF文件列表框,显示源文件夹中的PDF文件。
- 合并和分拆按钮,用于执行相应的操作。
当用户点击选择源文件夹按钮时,应用程序显示一个文件夹选择对话框,用户可以选择源文件夹。选择后,应用程序获取文件夹路径,并列出文件夹中的PDF文件。然后,合并和分拆按钮变为可用状态,用户可以执行相应的操作。
合并按钮的点击事件会弹出一个文本输入对话框,要求用户输入合并后的文件名。用户输入后,应用程序调用PyMuPDF库合并选定的PDF文件,并将合并后的PDF文件保存到目标文件夹中。
分拆按钮的点击事件会遍历选定的PDF文件,并使用PyMuPDF库将每个页面保存为单独的PDF文件。
代码示例
以下是应用程序的代码示例:
import os
import wx
import fitz
class PDFToolApp(wx.Frame):
def __init__(self, parent, title):
super(PDFToolApp, self).__init__(parent, title=title, size=(400, 400))
self.panel = wx.Panel(self)
self.source_folder_btn = wx.Button(self.panel, -1, "选择源文件夹")
self.target_folder_btn = wx.Button(self.panel, -1, "选择目标文件夹")
self.pdf_list = wx.CheckListBox(self.panel, -1, choices=[], style=wx.LB_MULTIPLE)
self.merge_btn = wx.Button(self.panel, -1, "合并")
self.split_btn = wx.Button(self.panel, -1, "分拆")
self.merge_btn.Disable()
self.split_btn.Disable()
self.source_folder_btn.Bind(wx.EVT_BUTTON, self.on_select_source_folder)
self.target_folder_btn.Bind(wx.EVT_BUTTON, self.on_select_target_folder)
self.merge_btn.Bind(wx.EVT_BUTTON, self.on_merge)
self.split_btn.Bind(wx.EVT_BUTTON, self.on_split)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(self.source_folder_btn, 0, wx.ALIGN_CENTER | wx.ALL, 10)
sizer.Add(self.target_folder_btn, 0, wx.ALIGN_CENTER | wx.ALL, 10)
sizer.Add(self.pdf_list, 1, wx.EXPAND | wx.ALL, 10)
sizer.Add(self.merge_btn, 0, wx.ALIGN_CENTER | wx.ALL, 10)
sizer.Add(self.split_btn, 0, wx.ALIGN_CENTER | wx.ALL, 10)
self.panel.SetSizer(sizer)
self.Show()
def on_select_source_folder(self, event):
dlg = wx.DirDialog(self, "选择源文件夹", style=wx.DD_DEFAULT_STYLE)
if dlg.ShowModal() == wx.ID_OK:
source_folder_path = dlg.GetPath()
self.pdf_list.Set(self.get_pdf_files(source_folder_path))
self.merge_btn.Enable()
self.split_btn.Enable()
dlg.Destroy()
def on_select_target_folder(self, event):
dlg = wx.DirDialog(self, "选择目标文件夹", style=wx.DD_DEFAULT_STYLE)
if dlg.ShowModal() == wx.ID_OK:
self.target_folder_path = dlg.GetPath()
dlg.Destroy()
def on_merge(self, event):
selected_items = self.pdf_list.GetCheckedItems()
if len(selected_items) > 0:
dlg = wx.TextEntryDialog(self, "请输入合并后的文件名(不带扩展名):", "合并文件")
if dlg.ShowModal() == wx.ID_OK:
output_filename = dlg.GetValue()
output_filepath = os.path.join(self.target_folder_path, output_filename + ".pdf")
self.merge_pdfs(selected_items, output_filepath)
wx.MessageBox("PDF文件合并完成!", "完成", wx.OK | wx.ICON_INFORMATION)
dlg.Destroy()
def on_split(self, event):
selected_items = self.pdf_list.GetCheckedItems()
if len(selected_items) > 0:
for index in selected_items:
pdf_filename = self.pdf_list.GetString(index)
pdf_filepath = os.path.join(self.target_folder_path, pdf_filename)
self.split_pdf(pdf_filename, pdf_filepath)
wx.MessageBox("PDF文件分拆完成!", "完成", wx.OK | wx.ICON_INFORMATION)
def merge_pdfs(self, selected_items, output_filepath):
pdf_merger = fitz.open()
for index in selected_items:
pdf_filename = self.pdf_list.GetString(index)
pdf_filepath = os.path.join(self.target_folder_path, pdf_filename)
pdf = fitz.open(pdf_filepath)
pdf_merger.insert_pdf(pdf)
pdf_merger.save(output_filepath)
pdf_merger.close()
def split_pdf(self, pdf_filename, pdf_filepath):
pdf = fitz.open(pdf_filepath)
num_pages = pdf.page_count
for i in range(num_pages):
output_filename = f"{pdf_filename[:-4]}_{i+1}.pdf"
output_filepath = os.path.join(self.target_folder_path, os.path.basename(output_filename) )
print("output_filename:"+output_filename)
page = pdf[i]
new_pdf = fitz.open()
new_pdf.insert_pdf(pdf, from_page=i, to_page=i)
new_pdf.save(output_filepath)
print("self.target_folder_path:"+self.target_folder_path)
print("output_filepath:"+output_filepath)
new_pdf.close()
pdf.close()
def get_pdf_files(self, folder_path):
pdf_files = []
for filename in os.listdir(folder_path):
if filename.endswith(".pdf"):
pdf_files.append(folder_path+'/'+filename)
return pdf_files
if __name__ == "__main__":
app = wx.App()
PDFToolApp(None, "PDF工具")
app.MainLoop()
总结
PDF工具应用程序是一个使用wxPython和PyMuPDF库编写的简单实用工具,可用于合并和分拆PDF文件。它提供了一个直观的图形界面,使用户能够轻松选择文件夹和执行操作。无论是处理大量PDF文件还是简单的分拆操作,该应用程序都能满足用户的需求。