900人的公司财务，一小时内分发完所有工资条。用Python是如何实现的

原创已于 2022-02-09 16:43:43 修改 · 524 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

于 2022-02-09 16:20:04 首次发布

本文讲述了如何使用Python的openpyxl库自动化处理Excel工资表，通过编写程序实现工资条的批量生成，从而提高工作效率，避免财务人员的重复劳动。程序涉及到复制工作表、样式拷贝、数据读取与插入以及公式修正等关键步骤，让财务工作更加便捷。

对很多工薪族来说，发工资这一天基本上是脱贫致富的重要日子，也是很多人的还款日子，交房租，还信用卡……想到接下来的日子又得靠拆东墙补西墙过。

即使发薪日就是还款日，但这个美好的日子还是值得每个职场人期待的！

在这里插入图片描述

近日，公司的财务小姐姐请我吃饭，当时我就有预感，绝对不是就吃一顿饭那么简单。
果然，在吃饭的时候，她羞涩的把脑袋凑过来悄悄跟我说，听说你Python很厉害呀。因为平时埋头写代码，很少跟女生讲话。今天小姐姐居然一开口就问我最擅长的技能，那我肯定要好好表现一番啊，
于是拍着胸脯说：“是的，我用Python上能控制火箭，下能操作坦克，你想做什么？”。
她立马一副崇拜的样子看着我说：“我每个月都要用Excel做工资表，但是每个月都要做一遍，你能不能用Python帮我写一个程序，能够自动完成这个工作。”。
我当然立马答应了啊，这么点小事，分分钟搞定！

在这里插入图片描述

需求：

在讲实现过程之前，先来给大家说说，这位财务小姐姐的需求是什么。首先他是有一个员工工资表，这个工资表是长这样的：
在这里插入图片描述
大家给自己工资条签过字的小伙伴都知道，我们需要的工资条上面都有一行标题，所以她的需求是要把工资条这样的：

在这里插入图片描述
之前她处理的方式就是一个个拷贝添加，效率实在太低。今天我给这位小姐姐做个程序，以后做工资条，一秒钟搞定！！！

在这里插入图片描述

实现逻辑分析：

Python中操作Excel的包有很多，比如有pandas、openpyxl、xlrd/xlwt等。在这里我们选择的是openpyxl，主要是他除了能处理数据外还能处理样式，功能比较完整。

首先加载excel文件，并且创建一个新的sheet来保存新生成的工资条样式。
代码如下：

wb = openpyxl.load_workbook('工资条.xlsx')
wb.copy_worksheet(wb['工资表'])
ws = wb.worksheets[-1]
ws.title = 'final_工资表'

然后再把标题部分的样式（对齐样式、边框样式、填充样式、字体样式等）拷贝一份，以后每次新增一个工资条的时候，都把标题名称和这些样式拷贝过去。这部分实现的代码如下：

def cell_style(cell):
    alignment = copy(cell.alignment)    # 对齐样式
    border = copy(cell.border)          # 边框样式
    fill = copy(cell.fill)              # 填充样式
    font = copy(cell.font)              # 字体样式
    return alignment, border, fill, font

# 获取标题
header = [cell.value for cell in cells_rows[0]]

# 获取标题行中，每个单元格中的各种样式
alignment, border, fill, font = cell_style(cell=cells_rows[0][0])

上面代码是先定义了一个拷贝样式的函数，然后下面把样式拷贝出来。

最后一部分，就是把原始的工资数据读取出来，然后按照工资条的样式重新插入数据。逻辑有点多，建议理解一下：

# 获取每一列的值，拼接在一个列表中
cells_rows = [[cell for cell in row] for row in ws.rows]

for i, _ in enumerate(cells_rows[1:]):
    if i > 0:
        index = i*3
        # 每读取一行，就在下方插入两行
        ws.insert_rows(idx=index, amount=2)
        # 写表头
        for j, v in enumerate(header):
            r, c = index+1, j+1
            cell = ws.cell(row=r, column=c)
            cell.value = v
            cell.alignment = alignment
            cell.font = font
            cell.border = border
            cell.fill = fill
            # 更新后面的公式
            if cell.coordinate[:1] in ('H', 'J'):
                cell = ws.cell(row=r+1, column=c)
                cell.value = re.sub('\d+', str(r+1), cell.value)