告别报销噩梦！用Python自动化处理Uber发票，一键生成完美PDF

本文链接：https://blog.csdn.net/qq_36224726/article/details/142283340

开篇

还记得你最后一次面对一堆Uber发票手动整理时的崩溃瞬间吗？每次出差回来，你都要把那些发票一张张摊开，像福尔摩斯一样解析日期、金额、起点和终点，还得手动输入表格。是时候改变这种繁琐的手动操作了！为什么不让Python来帮你做这些无聊的事呢？今天，我将带你实现一个能够自动解析Uber发票、生成总结页，并合并所有发票为一个整洁PDF的超级工具，从此报销不再让你抓狂！

技术工具准备

在开始之前，以下工具将成为我们的"秘密武器"：

PyQt5：构建一个简单的图形用户界面（GUI），让工具使用起来像Windows程序一样方便，而不是依赖命令行。
pdfplumber：从PDF中挖掘Uber发票的详细信息，比如时间、地点、金额等。
PyPDF2：合并多个PDF，生成一个统一的文档。
reportlab：生成包含行程摘要的PDF表格。

下面是具体的代码实现。

代码实现：从文件选择到PDF合并

1. 创建主界面

首先，我们用PyQt5来创建一个简单的图形用户界面，允许用户选择包含Uber发票的文件夹，并提供一个按钮来执行PDF合并操作。下面是主界面的代码部分：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog, QMessageBox
from Uber_MAINWINDOW import Ui_MainWindow  # 导入生成的UI类

class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.ui = Ui_MainWindow()  # UI实例化
        self.ui.setupUi(self)  # 设置UI
        self.selected_folder = None

        # 连接按钮点击事件
        self.ui.pushButto_upLoadFile.clicked.connect(self.start_folder_selection)
        self.ui.pushButto_Merge_PDF.clicked.connect(self.merge_pdfs)

    # 启动文件夹选择
    def start_folder_selection(self):
        folder = QFileDialog.getExistingDirectory(self, "选择文件夹")
        if folder:
            self.selected_folder = folder  # 保存选择的文件夹路径
            QMessageBox.information(self, "文件夹已选择", f"已选择文件夹: {folder}")

解释：这个界面非常简洁，用户可以选择包含Uber发票的文件夹，并点击“合并PDF”按钮执行操作。我们通过PyQt5的QFileDialog让用户选择文件夹路径。

2. PDF数据提取

接下来，我们使用pdfplumber来解析Uber发票PDF，提取关键信息，如行程日期、时间、起点、终点和金额。我们定义了一个函数来处理这些发票。

import pdfplumber
import re

# 提取PDF中的行程信息
def extract_trip_data_from_pdf(pdf_path):
    trips = []
    with pdfplumber.open(pdf_path) as pdf:
        for page in pdf.pages:
            text = page.extract_text()

            # 使用正则表达式提取日期、时间、地点和金额
            date_match = re.search(r'(\d{1,2}/\d{1,2}/\d{2,4})', text)
            time_matches = re.findall(r'(\d{1,2}:\d{2})', text)
            locations = re.findall(r'(\d{1,2}:\d{2}\s\|\s.+)', text)
            amount_match = re.search(r'Total\s([\d,.]+)\s([A-Z]{2,3})', text)

            if date_match and len(time_matches) >= 2 and len(locations) >= 2 and amount_match:
                trips.append({
                    "日期": date_match.group(1),
                    "起始时间": time_matches[0],
                    "启程地": locations[0].split('|')[1].strip(),
                    "结束时间": time_matches[1],
                    "目的地": locations[1].split('|')[1].strip(),
                    "金额": float(amount_match.group(1).replace(',', '')),
                    "单位": amount_match.group(2)
                })
    return trips

解释：在这个函数中，我们利用正则表达式从PDF文本中提取出行程的日期、时间、出发地、目的地和金额。每个发票页都包含这些信息，我们将其存储在一个列表中，方便后续操作。

3. 生成总结页

接下来是生成PDF总结页的功能，它将所有行程信息汇总为一个表格，方便后续审核。

from reportlab.lib.pagesizes import A4
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle
from reportlab.lib import colors

# 生成总结页
def generate_summary_page(output_path, trips_data):
    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)
    table.setStyle(TableStyle([('BACKGROUND', (0, 0), (-1, 0), colors.grey),
                               ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
                               ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
                               ('GRID', (0, 0), (-1, -1), 0.5, colors.black)]))
    
    elements = [table]
    doc.build(elements)

解释：这里我们使用reportlab创建一个PDF表格，总结所有提取的行程数据。该表格会成为我们合并PDF时的第一页。

4. 合并PDF文件

最后一步是将所有的发票PDF和总结页合并在一起，生成一个最终的文件。

from PyPDF2 import PdfMerger

def merge_pdfs(selected_folder, summary_pdf_path, save_path):
    merger = PdfMerger()
    merger.append(summary_pdf_path)  # 添加总结页

    pdf_files = [os.path.join(selected_folder, f) for f in os.listdir(selected_folder) if f.endswith('.pdf')]
    for pdf in pdf_files:
        merger.append(pdf)  # 合并所有发票

    merger.write(save_path)
    merger.close()

解释：这个函数会将生成的总结页与所选文件夹中的所有发票PDF合并，最后保存为一个新的PDF文件。