博客主页：https://tomcat.blog.csdn.net
博主昵称：农民工老王
主要领域：Java、Linux、K8S
期待大家的关注点赞收藏留言

背景

我们部门偶尔加班，领导让我统计同事们的加班情况，并处理加班餐的报销工作，其中比较麻烦的一个事情是整理发票，我们公司对电子发票的要求是：

1. 电子发票的文件名修改为日期+发票代码+发票号码。
2. 电子发票需要打印出来，并在背面写上报销金额，消费时间，加班参与人员等报销信息。

每月有20多张报销发票，平均每天加班有10多名同事，光誊写这些名字就挺耗时。如果手动操作文件的重命名，也很容易出错。我就想到用代码来解决这个问题。Java在处理Excel表格，PDF文件稍显麻烦，我就用Python作为编程语言，写了个自动化工具，要求1是文件的重命名，实现起来就比较简单，可以复制原有的发票文件，存储为所要求的文件名即可；要求2的实现是通过在电子发票PDF文件中再加一页，并在其中写入报销信息，然后就可以直接批量双面打印。

实现

项目结构

我在PyCharm中创建了项目。其中outputForPrint为双面打印的输出文件夹，outputForRename为重命名后的输出文件夹，receiptOriginal为原始发票文件夹，msyh.ttc为字体文件，receipt.xlsx为存储报销信息的表格，receiptTool.py为python脚本。

表格内容

receipt.xlsx仅有sheet1工作表，内容如下表所示，备注中的金额与人员用英文分号进行分割。

日期	备注
4月01日	金额-50;人员-老王,老罗
4月06日	金额-300;人员-老王,肖美女,兵哥,阿杜,老李,老胡,小毛,老荣,老罗,涛哥,杰哥,核姐
4月07日	金额-275;人员-肖美女,兵哥,阿杜,老李,老胡,小毛,老荣,老罗,涛哥,杰哥,核姐
4月08日	金额-150;人员-老王,老荣,老罗,涛哥,杰哥,核姐
4月09日	金额-250;人员-老王,肖美女,兵哥,阿杜,老李,老胡,小毛,老荣,涛哥,核姐
4月11日	金额-250;人员-老王,肖美女,兵哥,老李,老荣,老罗,涛哥,杰哥,核姐,老姚
4月12日	金额-250;人员-老王,肖美女,兵哥,阿杜,老荣,老罗,涛哥,杰哥,核姐,老姚
4月13日	金额-250;人员-老王,肖美女,兵哥,阿杜,老荣,老罗,涛哥,杰哥,核姐,老姚
4月14日	金额-250;人员-老王,肖美女,兵哥,阿杜,老荣,老罗,涛哥,杰哥,核姐,老姚
4月16日	金额-175;人员-老王,阿杜,老荣,老罗,杰哥,核姐,老姚
4月18日	金额-225;人员-肖美女,兵哥,阿杜,老李,老荣,老罗,涛哥,杰哥,核姐
4月19日	金额-275;人员-老王,肖美女,兵哥,阿杜,老李,老荣,老罗,涛哥,杰哥,核姐,老姚
4月20日	金额-275;人员-老王,肖美女,兵哥,阿杜,老李,老荣,老罗,涛哥,杰哥,核姐,老姚
4月21日	金额-275;人员-老王,肖美女,兵哥,阿杜,老李,老荣,老罗,涛哥,杰哥,核姐,老姚
4月24日	金额-225;人员-老王,肖美女,阿杜,老荣,老罗,涛哥,杰哥,核姐,老姚
4月25日	金额-300;人员-老王,肖美女,兵哥,阿杜,老李,老荣,老罗,涛哥,杰哥,核姐,老姚,追风少年
4月26日	金额-275;人员-肖美女,兵哥,阿杜,老李,老荣,老罗,涛哥,杰哥,核姐,老姚,追风少年
4月27日	金额-275;人员-老王,肖美女,兵哥,老李,老荣,老罗,涛哥,杰哥,核姐,老姚,追风少年
4月28日	金额-275;人员-老王,肖美女,兵哥,阿杜,老李,老荣,涛哥,杰哥,核姐,老姚,追风少年

python代码

# ------------------------------------------
# Filename    : receiptTool.py
# Version     : 1.1
# Date        : 2022-7-15 19:50:15
# Author      : 农民工老王@CSDN
# Email       : [email protected]
# Website     : https://blog.csdn.net/monarch91
# Description : 用于处理电子发票
# ------------------------------------------

import pandas
import PyPDF2
import os
import pathlib
import pdfplumber
import re
from shutil import copyfile
from PyPDF2 import PdfFileReader
from reportlab.lib.pagesizes import A4
from reportlab.pdfgen.canvas import Canvas
from reportlab.pdfbase import pdfmetrics, ttfonts

outputDir = "outputForRename"
printDir = "outputForPrint"
if not pathlib.Path(outputDir).is_dir():
    os.makedirs(outputDir)

if not pathlib.Path(printDir).is_dir():
    os.makedirs(printDir)

pdfmetrics.registerFont(ttfonts.TTFont('msyh', 'msyh.ttc'))
receiptData = pandas.read_excel("receipt.xlsx", sheet_name="Sheet1", index_col="日期")

for index, row in receiptData.iterrows():
    pdfPath = "receiptOriginal" + os.sep + str(index) + ".pdf"
    if type(index) == str and pathlib.Path(pdfPath).is_file():
        print("检测到%s有对应发票，正在处理。" % (index))

        pdfStr = ""
        with pdfplumber.open(pdfPath) as p:
            page = p.pages[0]
            pdfStr = page.extract_text()
        fpdm = re.search(r"发票代码[：:]\s?(\d+)", pdfStr).group(1)
        fphm = re.search(r"发票号码[：:]\s?(\d+)", pdfStr).group(1)

        receiptPdf = PdfFileReader(open(pdfPath, 'rb'))
        pdfSize = receiptPdf.getPage(0).mediaBox
        pdfH = float(pdfSize[2])
        pdfV = float(pdfSize[3])
        canvas = Canvas("temp.pdf", pagesize=(pdfH, pdfV))
        fontSize = 18 * pdfV / A4[1]
        canvas.setFont("msyh", fontSize)
        canvas.drawString(pdfH / 20, pdfV / 2 + fontSize * 1.5, index)
        for i, item in enumerate(row["备注"].split(";")):
            canvas.drawString(pdfH / 20, pdfV / 2 - fontSize * i * 1.5, item)
        canvas.save()

        pdf_merger = PyPDF2.PdfFileMerger()
        pdf_merger.append(pdfPath)
        pdf_merger.append("temp.pdf")

        pdfOutput = open(printDir + os.sep + index + "-print.pdf", 'wb')
        copyfile(pdfPath, outputDir + os.sep + index + "-" + fpdm + "-" + fphm + ".pdf")
        pdf_merger.write(pdfOutput)
        pdf_merger.close()

os.remove("temp.pdf")

运行效果

运行后，在控制台有以下输出：

顺利实现文件重命名：
顺利生成用于双面打印的文件：
用于双面打印的文件：

---

如需转载，请注明本文的出处：农民工老王的程序员宅基地https://blog.csdn.net/monarch91 。