基于PyQt5如何完成PDF拆分功能

引言

PDF（Portable Document Format）是一种广泛使用的文档格式，因其跨平台、易于共享和保持格式一致等优点而备受青睐。然而，在实际应用中，我们经常需要对PDF文件进行拆分，例如将一个大文件拆分成多个小文件，或者提取其中的某些页面。本文将介绍如何使用Python的PyQt5库来实现一个简单的PDF拆分工具。

环境准备

在开始之前，我们需要确保已经安装了以下Python库：

• PyQt5：用于创建图形用户界面（GUI）。
• PyPDF2：用于处理PDF文件。

可以通过以下命令安装这些库：

pip install PyQt5 PyPDF2

创建PyQt5应用程序

首先，我们需要创建一个基本的PyQt5应用程序框架。以下是一个简单的示例：

import sys
from PyQt5.QtWidgets import QApplication, QWidget, QVBoxLayout, QPushButton, QFileDialog, QLabel

class PDFSplitterApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setWindowTitle('PDF Splitter')
        self.setGeometry(100, 100, 400, 200)

        layout = QVBoxLayout()

        self.label = QLabel('Select a PDF file to split', self)
        layout.addWidget(self.label)

        self.btn_select = QPushButton('Select PDF', self)
        self.btn_select.clicked.connect(self.select_pdf)
        layout.addWidget(self.btn_select)

        self.btn_split = QPushButton('Split PDF', self)
        self.btn_split.clicked.connect(self.split_pdf)
        layout.addWidget(self.btn_split)

        self.setLayout(layout)

    def select_pdf(self):
        options = QFileDialog.Options()
        self.file_name, _ = QFileDialog.getOpenFileName(self, "Select PDF File", "", "PDF Files (*.pdf)", options=options)
        if self.file_name:
            self.label.setText(f'Selected PDF: {self.file_name}')

    def split_pdf(self):
        if not hasattr(self, 'file_name'):
            self.label.setText('Please select a PDF file first.')
            return

        # TODO: Implement PDF splitting logic here
        self.label.setText('PDF splitting not implemented yet.')

if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = PDFSplitterApp()
    ex.show()
    sys.exit(app.exec_())

在这个示例中，我们创建了一个简单的窗口，包含两个按钮：一个用于选择PDF文件，另一个用于执行拆分操作。选择文件后，文件名会显示在标签中。

实现PDF拆分功能

接下来，我们需要实现PDF拆分的核心逻辑。我们将使用PyPDF2库来读取和拆分PDF文件。

1. 读取PDF文件

首先，我们需要读取用户选择的PDF文件。可以使用PyPDF2.PdfFileReader来读取PDF文件：

from PyPDF2 import PdfFileReader, PdfFileWriter

def split_pdf(self):
    if not hasattr(self, 'file_name'):
        self.label.setText('Please select a PDF file first.')
        return

    try:
        reader = PdfFileReader(self.file_name)
        num_pages = reader.getNumPages()
        self.label.setText(f'PDF has {num_pages} pages.')
    except Exception as e:
        self.label.setText(f'Error reading PDF: {str(e)}')

2. 拆分PDF文件

假设我们希望将PDF文件拆分为每页一个文件。我们可以遍历每一页，并将其保存为一个单独的PDF文件：

import os

def split_pdf(self):
    if not hasattr(self, 'file_name'):
        self.label.setText('Please select a PDF file first.')
        return

    try:
        reader = PdfFileReader(self.file_name)
        num_pages = reader.getNumPages()

        output_dir = os.path.dirname(self.file_name)
        base_name = os.path.splitext(os.path.basename(self.file_name))[0]

        for i in range(num_pages):
            writer = PdfFileWriter()
            writer.addPage(reader.getPage(i))

            output_file = os.path.join(output_dir, f'{base_name}_page_{i+1}.pdf')
            with open(output_file, 'wb') as output_pdf:
                writer.write(output_pdf)

        self.label.setText(f'PDF split into {num_pages} files.')
    except Exception as e:
        self.label.setText(f'Error splitting PDF: {str(e)}')

在这个示例中，我们遍历PDF文件的每一页，将其添加到新的PdfFileWriter对象中，并保存为一个单独的PDF文件。文件名基于原始文件名和页码生成。

3. 添加用户输入

为了增加灵活性，我们可以允许用户指定拆分的页码范围。例如，用户可以选择从第几页到第几页进行拆分。我们可以通过添加输入框来实现这一点：

from PyQt5.QtWidgets import QLineEdit

class PDFSplitterApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setWindowTitle('PDF Splitter')
        self.setGeometry(100, 100, 400, 200)

        layout = QVBoxLayout()

        self.label = QLabel('Select a PDF file to split', self)
        layout.addWidget(self.label)

        self.btn_select = QPushButton('Select PDF', self)
        self.btn_select.clicked.connect(self.select_pdf)
        layout.addWidget(self.btn_select)

        self.start_page = QLineEdit(self)
        self.start_page.setPlaceholderText('Start Page')
        layout.addWidget(self.start_page)

        self.end_page = QLineEdit(self)
        self.end_page.setPlaceholderText('End Page')
        layout.addWidget(self.end_page)

        self.btn_split = QPushButton('Split PDF', self)
        self.btn_split.clicked.connect(self.split_pdf)
        layout.addWidget(self.btn_split)

        self.setLayout(layout)

    def select_pdf(self):
        options = QFileDialog.Options()
        self.file_name, _ = QFileDialog.getOpenFileName(self, "Select PDF File", "", "PDF Files (*.pdf)", options=options)
        if self.file_name:
            self.label.setText(f'Selected PDF: {self.file_name}')

    def split_pdf(self):
        if not hasattr(self, 'file_name'):
            self.label.setText('Please select a PDF file first.')
            return

        try:
            reader = PdfFileReader(self.file_name)
            num_pages = reader.getNumPages()

            start_page = int(self.start_page.text()) if self.start_page.text() else 1
            end_page = int(self.end_page.text()) if self.end_page.text() else num_pages

            if start_page < 1 or end_page > num_pages or start_page > end_page:
                self.label.setText('Invalid page range.')
                return

            output_dir = os.path.dirname(self.file_name)
            base_name = os.path.splitext(os.path.basename(self.file_name))[0]

            for i in range(start_page - 1, end_page):
                writer = PdfFileWriter()
                writer.addPage(reader.getPage(i))

                output_file = os.path.join(output_dir, f'{base_name}_page_{i+1}.pdf')
                with open(output_file, 'wb') as output_pdf:
                    writer.write(output_pdf)

            self.label.setText(f'PDF split from page {start_page} to {end_page}.')
        except Exception as e:
            self.label.setText(f'Error splitting PDF: {str(e)}')

在这个版本中，我们添加了两个QLineEdit输入框，允许用户指定拆分的页码范围。如果用户没有输入页码范围，则默认拆分整个PDF文件。

结论

通过本文的介绍，我们学习了如何使用PyQt5和PyPDF2库创建一个简单的PDF拆分工具。这个工具允许用户选择PDF文件，并指定拆分的页码范围，然后将选定的页面保存为单独的PDF文件。这个工具可以进一步扩展，例如添加更多的拆分选项、支持批量处理等。

希望本文对你有所帮助，祝你在使用PyQt5开发GUI应用程序时取得成功！