使用Python创建一个能够筛选文件的PDF合并工具
更新时间：2025年02月28日 14:23:37 作者：winfredzhang
这篇文章主要为大家详细介绍了如何使用Python创建一个能够筛选文件的PDF合并工具,文中的示例代码讲解详细,感兴趣的小伙伴可以了解下
背景
wxPython 是一个基于 wxWidgets 的 Python GUI 库，它提供了创建跨平台桌面应用程序的强大工具。PyPDF2 则是一个 Python 库，用于处理 PDF 文件的操作，如合并、拆分、提取文本等。
本示例程序结合了 wxPython 用于构建图形界面和 PyPDF2 用于处理 PDF 合并的功能。
这款 PDF 合并工具允许用户浏览文件夹、选择 PDF 文件，并将其合并为一个新的 PDF 文件。我们将详细分析代码结构和如何一步步实现每个功能。
主要功能
浏览文件夹：用户可以选择一个文件夹，程序会扫描该文件夹下所有的 PDF 文件。
选择文件：用户可以从文件夹中选择需要合并的 PDF 文件。
合并文件：用户选择多个 PDF 文件后，程序会将其合并成一个新的 PDF 文件。
全部代码
import os
import wx
import PyPDF2
class PDFMergerFrame(wx.Frame):
def __init__(self, parent, title):
super(PDFMergerFrame, self).__init__(parent, title=title, size=(900, 600))
# Create main panel
self.panel = wx.Panel(self)
# Create toolbar
self.toolbar = self.CreateToolBar()
# Add toolbar buttons
browse_tool = self.toolbar.AddTool(wx.ID_ANY, "浏览", wx.ArtProvider.GetBitmap(wx.ART_FOLDER_OPEN), "浏览文件夹")
merge_tool = self.toolbar.AddTool(wx.ID_ANY, "合成", wx.ArtProvider.GetBitmap(wx.ART_NEW), "合成PDF文件")
self.toolbar.Realize()
# Bind toolbar events
self.Bind(wx.EVT_TOOL, self.on_browse, browse_tool)
self.Bind(wx.EVT_TOOL, self.on_merge, merge_tool)
# Create main sizer
main_sizer = wx.BoxSizer(wx.HORIZONTAL)
# Create left panel for folders
left_panel = wx.Panel(self.panel)
left_sizer = wx.BoxSizer(wx.VERTICAL)
folder_label = wx.StaticText(left_panel, label="文件夹:")
self.folder_listbox = wx.ListBox(left_panel, size=(200, -1), style=wx.LB_SINGLE)
left_sizer.Add(folder_label, 0, wx.ALL, 5)
left_sizer.Add(self.folder_listbox, 1, wx.EXPAND | wx.ALL, 5)
left_panel.SetSizer(left_sizer)
# Create middle panel for files
middle_panel = wx.Panel(self.panel)
middle_sizer = wx.BoxSizer(wx.VERTICAL)
file_label = wx.StaticText(middle_panel, label="子文件:")
self.file_listbox = wx.ListBox(middle_panel, size=(200, -1), style=wx.LB_SINGLE)
middle_sizer.Add(file_label, 0, wx.ALL, 5)
middle_sizer.Add(self.file_listbox, 1, wx.EXPAND | wx.ALL, 5)
middle_panel.SetSizer(middle_sizer)
# Create button panel
button_panel = wx.Panel(self.panel)
button_sizer = wx.BoxSizer(wx.VERTICAL)
add_button = wx.Button(button_panel, label=">")
remove_button = wx.Button(button_panel, label="<")
button_sizer.Add(add_button, 0, wx.ALL, 5)
button_sizer.Add(remove_button, 0, wx.ALL, 5)
button_panel.SetSizer(button_sizer)
# Create right panel for selected files
right_panel = wx.Panel(self.panel)
right_sizer = wx.BoxSizer(wx.VERTICAL)
selected_label = wx.StaticText(right_panel, label="已选文件:")
self.selected_listbox = wx.ListBox(right_panel, size=(200, -1), style=wx.LB_SINGLE)
right_sizer.Add(selected_label, 0, wx.ALL, 5)
right_sizer.Add(self.selected_listbox, 1, wx.EXPAND | wx.ALL, 5)
right_panel.SetSizer(right_sizer)
# Add panels to main sizer
main_sizer.Add(left_panel, 1, wx.EXPAND | wx.ALL, 5)
main_sizer.Add(middle_panel, 1, wx.EXPAND | wx.ALL, 5)
main_sizer.Add(button_panel, 0, wx.EXPAND | wx.ALL, 5)
main_sizer.Add(right_panel, 1, wx.EXPAND | wx.ALL, 5)
# Set main sizer
self.panel.SetSizerAndFit(main_sizer)
# # Set main sizer
# self.panel.SetSizerAndFit(main_sizer)
# Bind events
self.folder_listbox.Bind(wx.EVT_LISTBOX, self.on_folder_select)
add_button.Bind(wx.EVT_BUTTON, self.on_add_file)
remove_button.Bind(wx.EVT_BUTTON, self.on_remove_file)
# Initialize variables
self.root_path = ""
self.folders_with_files = []
# Center and show frame
self.Centre()
self.Show()
def on_browse(self, event):
"""Handle browse button click"""
dlg = wx.DirDialog(self, "选择文件夹", style=wx.DD_DEFAULT_STYLE)
if dlg.ShowModal() == wx.ID_OK:
self.root_path = dlg.GetPath()
self.scan_folders()
dlg.Destroy()
def scan_folders(self):
"""Scan folders and list those with PDF files"""
if not self.root_path:
return
self.folders_with_files = []
self.folder_listbox.Clear()
for root, dirs, files in os.walk(self.root_path):
has_pdf = False
for file in files:
if file.lower().endswith('.pdf'):
has_pdf = True
break
if has_pdf:
self.folders_with_files.append(root)
folder_name = os.path.basename(root) or root
self.folder_listbox.Append(folder_name)
def on_folder_select(self, event):
"""Handle folder selection"""
selected_index = self.folder_listbox.GetSelection()
if selected_index == wx.NOT_FOUND:
return
selected_folder = self.folders_with_files[selected_index]
self.file_listbox.Clear()
for file in os.listdir(selected_folder):
if file.lower().endswith('.pdf'):
self.file_listbox.Append(file)
def on_add_file(self, event):
"""Add selected file to the list of files to merge"""
selected_file_index = self.file_listbox.GetSelection()
selected_folder_index = self.folder_listbox.GetSelection()
if selected_file_index == wx.NOT_FOUND or selected_folder_index == wx.NOT_FOUND:
return
selected_file = self.file_listbox.GetString(selected_file_index)
selected_folder = self.folders_with_files[selected_folder_index]
full_path = os.path.join(selected_folder, selected_file)
# Check if already in the list
for i in range(self.selected_listbox.GetCount()):
if self.selected_listbox.GetClientData(i) == full_path:
return
self.selected_listbox.Append(selected_file, full_path)
def on_remove_file(self, event):
"""Remove selected file from the list of files to merge"""
selected_index = self.selected_listbox.GetSelection()
if selected_index != wx.NOT_FOUND:
self.selected_listbox.Delete(selected_index)
def on_merge(self, event):
"""Merge selected PDF files"""
if self.selected_listbox.GetCount() == 0:
wx.MessageBox("请选择至少一个PDF文件", "警告", wx.OK | wx.ICON_WARNING)
return
# Ask for output file
with wx.FileDialog(self, "保存合并的PDF", wildcard="PDF files (*.pdf)|*.pdf",
style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT) as fileDialog:
if fileDialog.ShowModal() == wx.ID_CANCEL:
return
output_path = fileDialog.GetPath()
try:
merger = PyPDF2.PdfMerger()
# Add all selected files
for i in range(self.selected_listbox.GetCount()):
file_path = self.selected_listbox.GetClientData(i)
merger.append(file_path)
# Write to output file
merger.write(output_path)
merger.close()
wx.MessageBox(f"PDF文件已成功合并为: {output_path}", "成功", wx.OK | wx.ICON_INFORMATION)
except Exception as e:
wx.MessageBox(f"合并PDF时出错: {str(e)}", "错误", wx.OK | wx.ICON_ERROR)
class PDFMergerApp(wx.App):
def OnInit(self):
frame = PDFMergerFrame(None, "PDF合并工具")
self.SetTopWindow(frame)
return True
if __name__ == "__main__":
app = PDFMergerApp()
app.MainLoop()
接下来，让我们详细分析代码。
代码解析
1. 初始化 wx.Frame 窗口
class PDFMergerFrame(wx.Frame):
def __init__(self, parent, title):
super(PDFMergerFrame, self).__init__(parent, title=title, size=(900, 600))
我们创建了一个继承自 wx.Frame 的类 PDFMergerFrame，这是应用程序的主窗口。窗口的大小为 900x600。
2. 创建工具栏
# Create toolbar self.toolbar = self.CreateToolBar() # Add toolbar buttons browse_tool = self.toolbar.AddTool(wx.ID_ANY, "浏览", wx.ArtProvider.GetBitmap(wx.ART_FOLDER_OPEN), "浏览文件夹") merge_tool = self.toolbar.AddTool(wx.ID_ANY, "合成", wx.ArtProvider.GetBitmap(wx.ART_NEW), "合成PDF文件") self.toolbar.Realize()
在窗口中添加了一个工具栏，并且为工具栏添加了两个按钮：
- 浏览：用来选择文件夹，浏览文件夹并列出包含 PDF 文件的文件夹。
- 合成：用来合并用户选择的 PDF 文件。
按钮的图标通过 wx.ArtProvider.GetBitmap 方法加载。
3. 创建布局和界面控件
我们使用 wx.BoxSizer 创建了主界面布局，并将不同的控件添加到面板中。
# Create main sizer main_sizer = wx.BoxSizer(wx.HORIZONTAL)
main_sizer 是一个水平的 BoxSizer，用于将控件横向排列。接下来，我们定义了四个面板，分别用于显示：
- 文件夹选择：显示文件夹列表。
- 文件选择：显示当前文件夹中的 PDF 文件。
- 按钮面板：显示添加和移除文件的按钮。
- 已选文件：显示用户选择的文件列表。
每个面板使用 wx.Panel 创建，并且通过 wx.ListBox 控件展示文件夹或文件列表。
4. 文件夹扫描功能
def scan_folders(self):
"""Scan folders and list those with PDF files"""
if not self.root_path:
return
self.folders_with_files = []
self.folder_listbox.Clear()
for root, dirs, files in os.walk(self.root_path):
has_pdf = False
for file in files:
if file.lower().endswith('.pdf'):
has_pdf = True
break
if has_pdf:
self.folders_with_files.append(root)
folder_name = os.path.basename(root) or root
self.folder_listbox.Append(folder_name)
scan_folders 函数扫描用户选择的文件夹并列出其中包含 PDF 文件的子文件夹。使用 os.walk 递归遍历文件夹，并筛选出后缀为 .pdf 的文件。如果文件夹中包含 PDF 文件，则将文件夹添加到 folder_listbox。
5. 选择文件夹后的操作
def on_folder_select(self, event):
"""Handle folder selection"""
selected_index = self.folder_listbox.GetSelection()
if selected_index == wx.NOT_FOUND:
return
selected_folder = self.folders_with_files[selected_index]
self.file_listbox.Clear()
for file in os.listdir(selected_folder):
if file.lower().endswith('.pdf'):
self.file_listbox.Append(file)
用户在 folder_listbox 中选择一个文件夹后，on_folder_select 方法会被触发。此时，程序会列出该文件夹中的所有 PDF 文件，并显示在 file_listbox 中供用户选择。
6. 添加和移除文件
def on_add_file(self, event):
"""Add selected file to the list of files to merge"""
selected_file_index = self.file_listbox.GetSelection()
selected_folder_index = self.folder_listbox.GetSelection()
if selected_file_index == wx.NOT_FOUND or selected_folder_index == wx.NOT_FOUND:
return
selected_file = self.file_listbox.GetString(selected_file_index)
selected_folder = self.folders_with_files[selected_folder_index]
full_path = os.path.join(selected_folder, selected_file)
# Check if already in the list
for i in range(self.selected_listbox.GetCount()):
if self.selected_listbox.GetClientData(i) == full_path:
return
self.selected_listbox.Append(selected_file, full_path)
当用户选择文件并点击 > 按钮时，选中的文件将被添加到 selected_listbox 中。在添加之前，程序会检查该文件是否已经在列表中，防止重复添加。
移除文件的操作也是类似的，用户可以通过 < 按钮从 selected_listbox 中删除已选文件。
7. 合并PDF文件
def on_merge(self, event):
"""Merge selected PDF files"""
if self.selected_listbox.GetCount() == 0:
wx.MessageBox("请选择至少一个PDF文件", "警告", wx.OK | wx.ICON_WARNING)
return
# Ask for output file
with wx.FileDialog(self, "保存合并的PDF", wildcard="PDF files (*.pdf)|*.pdf",
style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT) as fileDialog:
if fileDialog.ShowModal() == wx.ID_CANCEL:
return
output_path = fileDialog.GetPath()
try:
merger = PyPDF2.PdfMerger()
# Add all selected files
for i in range(self.selected_listbox.GetCount()):
file_path = self.selected_listbox.GetClientData(i)
merger.append(file_path)
# Write to output file
merger.write(output_path)
merger.close()
wx.MessageBox(f"PDF文件已成功合并为: {output_path}", "成功", wx.OK | wx.ICON_INFORMATION)
except Exception as e:
wx.MessageBox(f"合并PDF时出错: {str(e)}", "错误", wx.OK | wx.ICON_ERROR)
最后，合并 PDF 的操作由 on_merge 方法实现。用户选择了要合并的 PDF 文件后，程序会弹出文件保存对话框，询问用户保存合并后文件的位置。使用 PyPDF2.PdfMerger 将所有选择的 PDF 文件合并并保存。
运行结果
以上就是使用Python创建一个能够筛选文件的PDF合并工具的详细内容，更多关于Python PDF合并的资料请关注脚本之家其它相关文章！
相关文章
如果你刚刚开始学习 PyTorch,并想学习如何做一些基本的图像分类,可以看看本文,这篇文章主要给大家介绍了关于如何使用Pytorch完成图像分类任务的相关资料,需要的朋友可以参考下2022-08-08
Python提供了丰富的字符串处理方法，可以方便地对字符串进行操作、处理和转换。在本文中，我们将介绍Python中常用的字符串方法，需要的可以参考一下2023-04-04
view函数是一个极为重要的张量操作函数,本文主要介绍了Pytorch中view()函数的实现示例,具有一定的参考价值,感兴趣的可以了解一下2025-05-05
这篇文章主要介绍了用python查找统一局域网下ip对应的mac地址的示例代码，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下2021-01-01
python单向链表的基本实现与使用方法【定义、遍历、添加、删除、查找等】
这篇文章主要介绍了python单向链表的基本实现与使用方法,结合实例形式分析了Python单向链表的定义、遍历、添加、删除、查找等相关操作技巧,需要的朋友可以参考下2019-10-10
本文是参考《python数据分析》的附录对生成器和文件系统结合案例的一个简单回顾,文中对python生成器与文件系统作了非常详细的介绍,对正在学习python的小伙伴们有很好地帮助,需要的朋友可以参考下2021-05-05
这篇文章主要介绍了python利用dlib获取人脸的68个landmark，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-11-11
这篇文章主要介绍了Python实现仿射密码的思路详解,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-04-04
这篇文章主要介绍了python根据经纬度计算距离示例, 计算两点之间距离,需要的朋友可以参考下2014-02-02
在本篇内容里小编给大家整理的是一篇关于Python3爬虫中pyspider的安装步骤，需要的朋友们可以学习参考下。2020-07-07
最新评论