Python 创建一个能够筛选文件的PDF合并工具

winfredzhang2025-03-02 2:02

合并多个 PDF 文件。这款 PDF 合并工具允许用户浏览文件夹、选择 PDF 文件，并将其合并为一个新的 PDF 文件。我们将详细分析代码结构和如何一步步实现每个功能。

C:\pythoncode\new\PDFFileInFolderMergeToNewPDFFile.py

全部代码

python 复制代码 
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()
背景

wxPython 是一个基于 wxWidgets 的 Python GUI 库，它提供了创建跨平台桌面应用程序的强大工具。PyPDF2 则是一个 Python 库，用于处理 PDF 文件的操作，如合并、拆分、提取文本等。

本示例程序结合了 wxPython 用于构建图形界面和 PyPDF2 用于处理 PDF 合并的功能。

主要功能
  1. 浏览文件夹：用户可以选择一个文件夹，程序会扫描该文件夹下所有的 PDF 文件。
  2. 选择文件：用户可以从文件夹中选择需要合并的 PDF 文件。
  3. 合并文件：用户选择多个 PDF 文件后，程序会将其合并成一个新的 PDF 文件。

接下来，让我们详细分析代码。

代码解析

1. 初始化 wx.Frame 窗口
python 复制代码 
class PDFMergerFrame(wx.Frame):
    def __init__(self, parent, title):
        super(PDFMergerFrame, self).__init__(parent, title=title, size=(900, 600))

我们创建了一个继承自 wx.Frame 的类 PDFMergerFrame，这是应用程序的主窗口。窗口的大小为 900x600。

2. 创建工具栏
python 复制代码 
# 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 创建了主界面布局，并将不同的控件添加到面板中。

python 复制代码 
# Create main sizer
main_sizer = wx.BoxSizer(wx.HORIZONTAL)

main_sizer 是一个水平的 BoxSizer，用于将控件横向排列。接下来，我们定义了四个面板，分别用于显示：

  • 文件夹选择：显示文件夹列表。
  • 文件选择：显示当前文件夹中的 PDF 文件。
  • 按钮面板：显示添加和移除文件的按钮。
  • 已选文件：显示用户选择的文件列表。

每个面板使用 wx.Panel 创建，并且通过 wx.ListBox 控件展示文件夹或文件列表。

4. 文件夹扫描功能
python 复制代码 
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. 选择文件夹后的操作
python 复制代码 
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. 添加和移除文件
python 复制代码 
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文件
python 复制代码 
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 文件合并并保存。

运行结果

相关推荐
Tttian6221 小时前
Python办公自动化（3）对Excel的操作
开发语言·python·excel
蹦蹦跳跳真可爱5892 小时前
Python----机器学习（KNN：使用数学方法实现KNN）
人工智能·python·机器学习
独好紫罗兰2 小时前
洛谷题单2-P5713 【深基3.例5】洛谷团队系统-python-流程图重构
开发语言·python·算法
DREAM.ZL3 小时前
基于python的电影数据分析及可视化系统
开发语言·python·数据分析
Uncertainty!!4 小时前
python函数装饰器
开发语言·python·装饰器
ZhangChuChu_92484 小时前
Word在生成PDF后，PDF左侧导航书签目录错误显示的解决方法
pdf·word
吾日三省吾码5 小时前
Python 脚本：自动化你的日常任务
数据库·python·自动化
snowfoootball5 小时前
基于 Ollama DeepSeek、Dify RAG 和 Fay 框架的高考咨询 AI 交互系统项目方案
前端·人工智能·后端·python·深度学习·高考
橙色小博6 小时前
长短期记忆神经网络（LSTM）基础学习与实例：预测序列的未来
人工智能·python·深度学习·神经网络·lstm
SsummerC6 小时前
【leetcode100】每日温度
数据结构·python·leetcode
热门推荐
01我决定放弃搞 Java 了02DeepSeek各版本说明与优缺点分析03如何在WPS和Word/Excel中直接使用DeepSeek功能04RAG 实践- Ollama+RagFlow 部署本地知识库05本地化部署AI知识库：基于Ollama+DeepSeek+AnythingLLM保姆级教程06从零安装 LLaMA-Factory 微调 Qwen 大模型成功及所有的坑07如何本地部署AI智能体平台，带你手搓一个AI Agent08苍穹外卖面试总结09DeepSeek RAGFlow构建本地知识库系统10【芯片封测学习专栏 -- D2D 和 C2C 之间的区别】