智能 Uber 发票 PDF 合并工具

news2024/9/20 1:13:10

在现代商务出行中，尤其是在跨国出差中，处理和整合大量 Uber 发票已成为一项不小的挑战。手动整理和合并这些发票不仅耗时，还容易出错。作为开发者，为什么不开发一个自动化工具，将这些任务交给代码来完成呢？在这篇博客中，我将带你一步步构建一个结合 PyQt5、pdfplumber 和 PyPDF2 的智能 Uber 发票合并工具，不仅能自动提取数据，还能动态显示进度条，给用户带来极佳的使用体验。

项目亮点：

PyQt5 GUI 界面：基于 PyQt5 实现的可视化界面，简洁大方。
自动提取发票数据：利用 pdfplumber 自动提取 Uber 发票中的日期、金额、地点等信息，支持多语言（中文、英文、西班牙语）。
PDF 合并功能：通过 PyPDF2 实现多份 PDF 发票的自动合并，并且生成一份总结页，显示所有行程的详细信息。
动态进度条：实时显示合并进度，让用户一目了然处理状态。

1. 项目简介

本项目的主要目的是通过图形化用户界面（GUI）和后端的 PDF 处理技术，自动处理 Uber 发票，自动从 PDF 中提取关键信息，并合并成一个包含详细摘要的 PDF 文件。这样的工具对于经常出差、需要整理大量发票的用户来说，是一个非常实用的助手。

2. 使用技术栈

在开发过程中，我们使用了以下的技术栈：

PyQt5：用于创建用户界面，让用户可以轻松选择文件夹、选择语言以及合并 PDF 文件。
pdfplumber：用于从 Uber 发票 PDF 中提取文本和行程信息，支持多语言。
PyPDF2：用于将多个 PDF 文件合并成一个，同时在合并前生成一份包含所有行程数据的总结页。
ReportLab：用于生成总结页的 PDF 文件，方便将表格数据导出。

3. 代码实现

3.1 主窗口的设计与初始化

我们通过 QMainWindow 创建了主窗口，并初始化了必要的组件，比如上传文件按钮、合并按钮、进度条等。我们还为窗口添加了一个版权信息，并通过 setWindowIcon 方法设置了应用的图标。

class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.ui = Ui_MainWindow()  # 创建UI对象
        self.ui.setupUi(self)  # 调用setupUi方法，构建界面

        # 初始化进度条
        self.ui.progressBar.setValue(0)  # 初始化进度条值为 0

        # 设置窗口的图标（logo）
        self.setWindowIcon(QIcon('logo.png'))  # 替换为你的 logo 文件路径

        # 设置窗口标题
        self.setWindowTitle("Uber Pdf Merge")  # 设置自定义窗口标题

        # 添加版权信息
        self.add_copyright_label()

        # 其他初始化代码...

3.2 自动提取 Uber 发票中的行程数据

在这一部分中，pdfplumber 被用于提取每一张发票中的行程信息，包括日期、金额、起始地点和目的地。我们通过不同的正则表达式来处理不同语言的发票数据（支持中文、英文和西班牙语）。

    def merge_pdfs(self):
        if self.selected_folder:
            pdf_files = [os.path.join(self.selected_folder, f) for f in os.listdir(self.selected_folder) if f.endswith('.pdf')]

            total_files = len(pdf_files)
            if total_files:
                all_trips = []
                self.ui.progressBar.setValue(0)

                for i, pdf_file in enumerate(pdf_files):
                    # 提取PDF中的行程数据
                    trips = self.extract_trip_data_from_pdf(pdf_file)
                    all_trips.extend(trips)

                    # 更新进度条
                    progress = int(((i + 1) / total_files) * 100)
                    self.ui.progressBar.setValue(progress)

                # 生成总结页PDF
                summary_pdf_path = "summary.pdf"
                self.generate_summary_page(summary_pdf_path, all_trips)

                # 合并PDF
                save_path, _ = QFileDialog.getSaveFileName(self, "Save Merged PDF", "", "PDF Files (*.pdf)")
                if save_path:
                    merger = PdfMerger()
                    merger.append(summary_pdf_path)
                    for pdf in pdf_files:
                        merger.append(pdf)
                    merger.write(save_path)
                    merger.close()

3.4 生成包含行程信息的总结页

为了方便整理行程信息，我们在合并多个发票之前，生成了一份总结页，并将其合并到最终的 PDF 文件中。总结页显示了所有行程的详细信息，并通过 ReportLab 将其以表格形式展示。

def generate_summary_page(self, output_path, trips_data):
    pdfmetrics.registerFont(UnicodeCIDFont('STSong-Light'))  # 使用 STSong-Light 字体
    doc = SimpleDocTemplate(output_path, pagesize=A4)

    table_data = [["日期", "起始时间", "启程地", "结束时间", "目的地", "金额", "单位"]]
    # 循环将行程数据添加到表格
    for trip in trips_data:
        table_data.append([trip["日期"], trip["起始时间"], trip["启程地"], trip["结束时间"], trip["目的地"], trip["金额"], trip["单位"]])

    table = Table(table_data, colWidths=[3 * cm, 2 * cm, 5 * cm, 2 * cm, 5 * cm, 2 * cm, 2 * cm])
    style = TableStyle([...])
    table.setStyle(style)
    elements = [table]
    doc.build(elements)