Python实现工资汇总和发放

修改代码，实现两个核心的颜色标注功能：

1. QT表格中：新增的5列（每天工资、工资、银行卡号、是否审核、最终工资）的单元格背景色分别设置为不同颜色，便于视觉区分；
2. 导出的Excel文件中：新增列的表头和数据行也会标注不同背景色，保持格式统一。

完整修改后的代码

import sys
import random
import pandas as pd
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,
                             QPushButton, QTableWidget, QTableWidgetItem, QFileDialog,
                             QMessageBox, QHeaderView, QLabel, QSplitter)
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor, QBrush
from openpyxl import Workbook
from openpyxl.styles import PatternFill  # 用于Excel单元格着色
from openpyxl.utils import get_column_letter

# 解决pandas读取xls文件的依赖问题（需提前安装：pip install xlrd==1.2.0）
try:
import xlrd
except ImportError:
    print("请安装xlrd 1.2.0版本：pip install xlrd==1.2.0")

# 定义新增列的颜色配置（QT表格 + Excel导出共用）
# 格式：列名: (QT背景色, Excel背景色)
COLOR_CONFIG = {
"每天工资": (QColor(255, 240, 240), PatternFill(start_color="FFE0E0", end_color="FFE0E0", fill_type="solid")),
"工资": (QColor(240, 255, 240), PatternFill(start_color="E0FFE0", end_color="E0FFE0", fill_type="solid")),
"银行卡号": (QColor(240, 240, 255), PatternFill(start_color="E0E0FF", end_color="E0E0FF", fill_type="solid")),
"是否审核": (QColor(255, 255, 240), PatternFill(start_color="FFFFE0", end_color="FFFFE0", fill_type="solid")),
"最终工资": (QColor(255, 240, 255), PatternFill(start_color="FFE0FF", end_color="FFE0FF", fill_type="solid"))
}


classSalaryManagementSystem(QMainWindow):
def__init__(self):
        super().__init__()
        self.df = None# 存储原始数据的DataFrame
        self.init_ui()

definit_ui(self):
# 窗口基本设置
        self.setWindowTitle("工资表管理系统")
        self.setGeometry(100, 100, 1200, 800)

# 中心部件
        central_widget = QWidget()
        self.setCentralWidget(central_widget)

# 整体布局
        main_layout = QVBoxLayout(central_widget)

# 按钮布局（顶部）
        btn_layout = QHBoxLayout()

# 导入文件按钮
        self.import_btn = QPushButton("导入CSV/XLS文件")
        self.import_btn.clicked.connect(self.import_file)
        btn_layout.addWidget(self.import_btn)

# 生成工资表按钮
        self.gen_salary_btn = QPushButton("生成工资表")
        self.gen_salary_btn.clicked.connect(self.generate_salary_table)
        self.gen_salary_btn.setEnabled(False)  # 初始禁用
        btn_layout.addWidget(self.gen_salary_btn)

# 导出工资表按钮
        self.export_salary_btn = QPushButton("导出工资表")
        self.export_salary_btn.clicked.connect(self.export_salary_table)
        self.export_salary_btn.setEnabled(False)
        btn_layout.addWidget(self.export_salary_btn)

# 生成工资条按钮
        self.gen_salary_slip_btn = QPushButton("生成工资条")
        self.gen_salary_slip_btn.clicked.connect(self.generate_salary_slips)
        self.gen_salary_slip_btn.setEnabled(False)
        btn_layout.addWidget(self.gen_salary_slip_btn)

        main_layout.addLayout(btn_layout)

# 数据展示表格
        self.table = QTableWidget()
        self.table.setEditTriggers(QTableWidget.NoEditTriggers)  # 禁止编辑
        self.table.horizontalHeader().setStretchLastSection(True)
        self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        main_layout.addWidget(self.table)

defimport_file(self):
"""导入CSV/XLS/XLSX文件并展示数据"""
# 选择文件
        file_path, _ = QFileDialog.getOpenFileName(
            self, "选择文件", "",
"Excel文件 (*.xls *.xlsx);;CSV文件 (*.csv);;所有文件 (*.*)"
        )
ifnot file_path:
return

try:
# 读取文件
if file_path.endswith(".csv"):
                self.df = pd.read_csv(file_path, encoding="utf-8")
elif file_path.endswith(".xls"):
                self.df = pd.read_excel(file_path, engine="xlrd")
else:  # .xlsx
                self.df = pd.read_excel(file_path, engine="openpyxl")

# 自动计算实际考勤天数（核心逻辑：处理可能的考勤列）
            self.calc_actual_attendance()

# 展示数据到表格
            self.show_data_in_table(self.df)

# 启用功能按钮
            self.gen_salary_btn.setEnabled(True)
            QMessageBox.information(self, "成功", "文件导入成功！已自动计算实际考勤天数")

except Exception as e:
            QMessageBox.critical(self, "错误", f"文件导入失败：{str(e)}")

defcalc_actual_attendance(self):
"""自动计算实际考勤天数（兼容不同列名）"""
# 定义考勤相关列名（兼容常见命名）
        actual_cols = ["实际出勤", "实际考勤", "出勤天数"]
        should_cols = ["应出勤", "应考勤", "应到天数"]
        leave_cols = ["事假", "病假", "调休", "年假"]

# 找到实际考勤列（如果不存在则自动计算）
        actual_col = None
for col in actual_cols:
if col in self.df.columns:
                actual_col = col
break

if actual_col isNone:
# 自动计算：实际考勤 = 应出勤 - 各类请假天数
            should_col = None
for col in should_cols:
if col in self.df.columns:
                    should_col = col
break
if should_col isNone:
                QMessageBox.warning(self, "提示", "未找到应出勤列，无法自动计算实际考勤天数")
return

# 计算总请假天数
            leave_days = pd.Series([0.0] * len(self.df))
for col in leave_cols:
if col in self.df.columns:
                    leave_days += self.df[col].fillna(0)

# 新增实际出勤列
            self.df["实际出勤"] = self.df[should_col] - leave_days
# 确保天数非负
            self.df["实际出勤"] = self.df["实际出勤"].apply(lambda x: max(x, 0))

defshow_data_in_table(self, df):
"""将DataFrame数据展示到QTableWidget，并为新增列着色"""
# 设置行列数
        self.table.setRowCount(len(df))
        self.table.setColumnCount(len(df.columns))
# 设置列名
        self.table.setHorizontalHeaderLabels(df.columns.tolist())

# 填充数据并为新增列着色
for row in range(len(df)):
for col in range(len(df.columns)):
                col_name = df.columns[col]
                value = df.iloc[row, col]

# 处理空值
if pd.isna(value):
                    value = ""
# 数值格式化（保留2位小数）
elif isinstance(value, (int, float)):
                    value = round(value, 2)

# 创建表格项
                item = QTableWidgetItem(str(value))
                item.setTextAlignment(Qt.AlignCenter)  # 居中显示

# 如果是新增列，设置背景色
if col_name in COLOR_CONFIG:
                    item.setBackground(QBrush(COLOR_CONFIG[col_name][0]))

                self.table.setItem(row, col, item)

defgenerate_salary_table(self):
"""生成工资表：新增每天工资、工资、银行卡号、是否审核、最终工资列"""
if self.df isNone:
            QMessageBox.warning(self, "提示", "请先导入文件！")
return

try:
# 1. 每天工资：随机生成200-500元（可自定义范围）
if"每天工资"notin self.df.columns:
                self.df["每天工资"] = [round(random.uniform(200, 500), 2) for _ in range(len(self.df))]

# 2. 工资 = 实际出勤 × 每天工资
if"工资"notin self.df.columns:
                self.df["工资"] = self.df["实际出勤"] * self.df["每天工资"]
                self.df["工资"] = self.df["工资"].round(2)  # 保留2位小数

# 3. 银行卡号：模拟真实卡号格式（62开头）
if"银行卡号"notin self.df.columns:
                self.df["银行卡号"] = [
f"6222{random.randint(100000000000, 999999999999)}"
for _ in range(len(self.df))
                ]

# 4. 是否审核：随机生成（可后续手动修改）
if"是否审核"notin self.df.columns:
                self.df["是否审核"] = [random.choice(["是", "否"]) for _ in range(len(self.df))]

# 5. 最终工资：默认等于工资（可扩展社保/个税扣除逻辑）
if"最终工资"notin self.df.columns:
                self.df["最终工资"] = self.df["工资"].round(2)

# 刷新表格展示（会自动着色）
            self.show_data_in_table(self.df)

# 启用导出按钮
            self.export_salary_btn.setEnabled(True)
            self.gen_salary_slip_btn.setEnabled(True)

            QMessageBox.information(self, "成功", "工资表生成成功！新增列已标注不同颜色")

except Exception as e:
            QMessageBox.critical(self, "错误", f"生成工资表失败：{str(e)}")

defexport_salary_table(self):
"""导出完整工资表到Excel文件，新增列标注颜色"""
if self.df isNone:
            QMessageBox.warning(self, "提示", "暂无工资表数据！")
return

# 选择保存路径
        save_path, _ = QFileDialog.getSaveFileName(
            self, "保存工资表", "工资表.xlsx",
"Excel文件 (*.xlsx);;CSV文件 (*.csv);;所有文件 (*.*)"
        )
ifnot save_path:
return

try:
if save_path.endswith(".csv"):
                self.df.to_csv(save_path, index=False, encoding="utf-8-sig")
else:
# 使用openpyxl保存xlsx，支持格式优化和颜色标注
                wb = Workbook()
                ws = wb.active
                ws.title = "工资表"

# 写入表头
                headers = self.df.columns.tolist()
                ws.append(headers)
# 为表头的新增列着色
for col_idx, col_name in enumerate(headers, 1):
if col_name in COLOR_CONFIG:
                        ws.cell(row=1, column=col_idx).fill = COLOR_CONFIG[col_name][1]

# 写入数据行
for row_idx, (_, row) in enumerate(self.df.iterrows(), 2):
                    row_data = row.tolist()
                    ws.append(row_data)
# 为数据行的新增列着色
for col_idx, col_name in enumerate(headers, 1):
if col_name in COLOR_CONFIG:
                            ws.cell(row=row_idx, column=col_idx).fill = COLOR_CONFIG[col_name][1]

# 调整列宽
for col in ws.columns:
                    max_length = 0
                    column = col[0].column_letter
for cell in col:
try:
if len(str(cell.value)) > max_length:
                                max_length = len(str(cell.value))
except:
pass
                    adjusted_width = min(max_length + 2, 50)
                    ws.column_dimensions[column].width = adjusted_width

                wb.save(save_path)

            QMessageBox.information(self, "成功", f"工资表已导出至：{save_path}，新增列已标注不同颜色")

except Exception as e:
            QMessageBox.critical(self, "错误", f"导出失败：{str(e)}")

defgenerate_salary_slips(self):
"""生成每个人的工资条并导出，新增列标注颜色"""
if self.df isNone:
            QMessageBox.warning(self, "提示", "暂无工资表数据！")
return

# 选择保存路径
        save_path, _ = QFileDialog.getSaveFileName(
            self, "保存工资条", "工资条.xlsx",
"Excel文件 (*.xlsx);;所有文件 (*.*)"
        )
ifnot save_path:
return

try:
# 构建工资条数据（每一行数据+表头，便于打印）
            slip_data = []
            headers = self.df.columns.tolist()
for _, row in self.df.iterrows():
                slip_data.append(headers)  # 表头
                slip_data.append(row.tolist())  # 数据行
                slip_data.append([""] * len(headers))  # 空行分隔

# 保存到Excel并着色
            wb = Workbook()
            ws = wb.active
            ws.title = "工资条"

# 写入数据并着色
            current_row = 1
for data_row in slip_data:
                ws.append(data_row)
# 着色逻辑：表头行和数据行的新增列都要着色
for col_idx, col_name in enumerate(headers, 1):
if col_name in COLOR_CONFIG:
                        ws.cell(row=current_row, column=col_idx).fill = COLOR_CONFIG[col_name][1]
                current_row += 1

# 调整列宽
for col in ws.columns:
                max_length = 0
                column = col[0].column_letter
for cell in col:
try:
if len(str(cell.value)) > max_length:
                            max_length = len(str(cell.value))
except:
pass
                adjusted_width = min(max_length + 2, 50)
                ws.column_dimensions[column].width = adjusted_width

            wb.save(save_path)

            QMessageBox.information(self, "成功", f"工资条已导出至：{save_path}，新增列已标注不同颜色")

except Exception as e:
            QMessageBox.critical(self, "错误", f"生成工资条失败：{str(e)}")


if __name__ == "__main__":
# 解决中文显示问题
    QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
    app = QApplication(sys.argv)
    window = SalaryManagementSystem()
    window.show()
    sys.exit(app.exec_())

核心修改说明

1. 颜色配置定义

新增了 COLOR_CONFIG 字典，为每个新增列指定两种颜色：

• QT表格的背景色（QColor 格式）
• Excel导出的背景色（PatternFill 格式）

COLOR_CONFIG = {
"每天工资": (浅红色, Excel浅红),
"工资": (浅绿色, Excel浅绿),
"银行卡号": (浅蓝色, Excel浅蓝),
"是否审核": (浅黄色, Excel浅黄),
"最终工资": (浅紫色, Excel浅紫)
}

2. QT表格着色（show_data_in_table 方法）

• 遍历每个单元格时，判断列名是否在颜色配置中
• 如果是新增列，通过 setBackground() 设置单元格背景色
• 同时添加了 setTextAlignment(Qt.AlignCenter) 让内容居中显示，更美观

3. Excel导出着色（export_salary_table/generate_salary_slips 方法）

• 废弃了兼容问题的 dataframe_to_rows，改用逐行写入
• 表头行：遍历表头列，为新增列设置背景色
• 数据行：遍历每一行数据，为新增列设置背景色
• 工资条导出也同步实现了颜色标注

4. 交互优化

• 生成工资表/导出成功后，提示信息新增“新增列已标注不同颜色”，提升用户体验

颜色对应表

总结

1. 新增列在QT表格中会显示不同的浅色系背景色，视觉区分清晰且不影响文字阅读；
2. 导出的Excel文件中，新增列的表头和所有数据行都会标注对应的背景色，格式统一；
3. 修复了原代码中 dataframe_to_rows 的兼容问题，同时保留了所有原有功能；
4. 颜色配置集中管理，你可以通过修改 COLOR_CONFIG 轻松调整各列的颜色。

如果需要调整颜色（比如换成深色、自定义RGB值），只需修改 COLOR_CONFIG 中的颜色值即可。

基于现有代码扩展完整的工资发放功能，包括：新增公司账号信息页面、批量发送按钮、全选勾选框、发放确认弹窗、逐行显示发放信息等核心功能。以下是完整的改造后代码：

import sys
import random
import pandas as pd
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,
                             QPushButton, QTableWidget, QTableWidgetItem, QFileDialog,
                             QMessageBox, QHeaderView, QLabel, QSplitter, QLineEdit,
                             QGroupBox, QGridLayout, QCheckBox, QDialog, QDialogButtonBox,
                             QTextEdit, QScrollArea)
from PyQt5.QtCore import Qt, QThread, pyqtSignal
from PyQt5.QtGui import QColor, QBrush, QFont
from openpyxl import Workbook
from openpyxl.styles import PatternFill, Font  # 用于Excel单元格着色和字体
from openpyxl.utils import get_column_letter

# 解决pandas读取xls文件的依赖问题（需提前安装：pip install xlrd==1.2.0）
try:
import xlrd
except ImportError:
    print("请安装xlrd 1.2.0版本：pip install xlrd==1.2.0")

# 定义新增列的颜色配置（QT表格 + Excel导出共用）
# 格式：列名: (QT背景色, Excel背景色)
COLOR_CONFIG = {
"每天工资": (QColor(255, 240, 240), PatternFill(start_color="FFE0E0", end_color="FFE0E0", fill_type="solid")),
"工资": (QColor(240, 255, 240), PatternFill(start_color="E0FFE0", end_color="E0FFE0", fill_type="solid")),
"银行卡号": (QColor(240, 240, 255), PatternFill(start_color="E0E0FF", end_color="E0E0FF", fill_type="solid")),
"是否审核": (QColor(255, 255, 240), PatternFill(start_color="FFFFE0", end_color="FFFFE0", fill_type="solid")),
"最终工资": (QColor(255, 240, 255), PatternFill(start_color="FFE0FF", end_color="FFE0FF", fill_type="solid")),
"发放状态": (QColor(245, 245, 245), PatternFill(start_color="F5F5F5", end_color="F5F5F5", fill_type="solid"))
}

# 汇总行样式配置
SUMMARY_STYLE = {
"qt_bg_color": QColor(200, 200, 200),  # QT表格汇总行背景色（灰色）
"excel_fill": PatternFill(start_color="D3D3D3", end_color="D3D3D3", fill_type="solid"),  # Excel汇总行背景色
"excel_font": Font(bold=True)  # Excel汇总行字体加粗
}

# 发放状态颜色配置
PAYMENT_STATUS_COLOR = {
"未发放": QColor(255, 240, 240),
"发放中": QColor(255, 255, 204),
"发放成功": QColor(240, 255, 240),
"发放失败": QColor(255, 220, 220)
}

# 模拟发放线程
classPaymentThread(QThread):
"""工资发放模拟线程"""
    progress_signal = pyqtSignal(str)  # 发放进度信号
    finish_signal = pyqtSignal(bool, str)  # 发放完成信号 (是否成功, 消息)

def__init__(self, company_account, selected_data):
        super().__init__()
        self.company_account = company_account
        self.selected_data = selected_data

defrun(self):
"""模拟工资发放过程"""
        success_count = 0
        fail_count = 0

try:
for idx, data in enumerate(self.selected_data, 1):
                name = data.get("姓名", f"员工{idx}")
                card_num = data.get("银行卡号", "")
                amount = data.get("最终工资", 0)

# 发送进度信息
                self.progress_signal.emit(f"正在发放给 {name} | 卡号: {card_num} | 金额: {amount} 元")

# 模拟网络延迟
                self.msleep(500)

# 模拟发放结果（90%成功率）
if random.random() < 0.9:
                    self.progress_signal.emit(f"✅ {name} 工资发放成功！")
                    success_count += 1
else:
                    self.progress_signal.emit(f"❌ {name} 工资发放失败！（模拟银行卡异常）")
                    fail_count += 1

# 发放完成
            total = len(self.selected_data)
            msg = f"发放完成！总计 {total} 人，成功 {success_count} 人，失败 {fail_count} 人"
            self.finish_signal.emit(fail_count == 0, msg)

except Exception as e:
            self.finish_signal.emit(False, f"发放过程出错：{str(e)}")

# 发放确认弹窗
classPaymentConfirmDialog(QDialog):
"""工资发放确认弹窗"""
def__init__(self, company_account, selected_data, parent=None):
        super().__init__(parent)
        self.setWindowTitle("工资发放确认")
        self.setModal(True)
        self.setMinimumSize(800, 600)

# 主布局
        main_layout = QVBoxLayout(self)

# 公司账号信息
        company_group = QGroupBox("发放方信息")
        company_layout = QGridLayout(company_group)
        company_layout.addWidget(QLabel("公司账号："), 0, 0)
        company_layout.addWidget(QLabel(company_account), 0, 1)
        company_layout.addWidget(QLabel("发放总金额："), 1, 0)

# 计算总金额
        total_amount = sum([float(data.get("最终工资", 0)) for data in selected_data])
        company_layout.addWidget(QLabel(f"{total_amount:.2f} 元"), 1, 1)
        company_layout.addWidget(QLabel(f"发放人数：{len(selected_data)} 人"), 2, 0, 1, 2)
        main_layout.addWidget(company_group)

# 发放列表
        list_group = QGroupBox("发放列表")
        scroll_area = QScrollArea()
        scroll_area.setWidgetResizable(True)
        list_widget = QWidget()
        list_layout = QVBoxLayout(list_widget)

# 逐行显示发放信息
for idx, data in enumerate(selected_data, 1):
            name = data.get("姓名", f"员工{idx}")
            card_num = data.get("银行卡号", "")
            amount = data.get("最终工资", 0)
            status = data.get("是否审核", "未审核")

# 隐藏完整卡号，只显示后4位
            show_card = f"**** **** **** {card_num[-4:]}"if len(card_num) >= 4else card_num

            item_label = QLabel(f"{idx}. {name} | 卡号：{show_card} | 金额：{amount:.2f} 元 | 审核状态：{status}")
            item_label.setFont(QFont("Microsoft YaHei", 9))
            list_layout.addWidget(item_label)

        list_layout.addStretch()
        scroll_area.setWidget(list_widget)
        list_group.setLayout(QVBoxLayout())
        list_group.layout().addWidget(scroll_area)
        main_layout.addWidget(list_group)

# 发放日志
        log_group = QGroupBox("发放进度")
        self.log_text = QTextEdit()
        self.log_text.setReadOnly(True)
        self.log_text.setMinimumHeight(150)
        log_group.setLayout(QVBoxLayout())
        log_group.layout().addWidget(self.log_text)
        main_layout.addWidget(log_group)

# 按钮
        button_box = QDialogButtonBox()
        self.start_btn = QPushButton("开始发放")
        self.cancel_btn = QPushButton("取消")
        button_box.addButton(self.start_btn, QDialogButtonBox.AcceptRole)
        button_box.addButton(self.cancel_btn, QDialogButtonBox.RejectRole)

        main_layout.addWidget(button_box)

# 信号连接
        self.start_btn.clicked.connect(self.start_payment)
        self.cancel_btn.clicked.connect(self.reject)

# 发放线程
        self.payment_thread = None

defstart_payment(self):
"""开始工资发放"""
        self.start_btn.setEnabled(False)
        self.log_text.clear()

# 创建发放线程
        company_account = self.parent().company_account_edit.text().strip()
        self.payment_thread = PaymentThread(company_account, self.parent().get_selected_data())
        self.payment_thread.progress_signal.connect(self.update_log)
        self.payment_thread.finish_signal.connect(self.payment_finished)
        self.payment_thread.start()

defupdate_log(self, message):
"""更新发放日志"""
        self.log_text.append(message)
# 自动滚动到底部
        cursor = self.log_text.textCursor()
        cursor.movePosition(cursor.End)
        self.log_text.setTextCursor(cursor)

defpayment_finished(self, success, message):
"""发放完成处理"""
        self.update_log(f"\n{message}")
        self.start_btn.setEnabled(True)

if success:
            QMessageBox.information(self, "成功", message)
            self.accept()
else:
            QMessageBox.warning(self, "提示", message)

# 主窗口
classSalaryManagementSystem(QMainWindow):
def__init__(self):
        super().__init__()
        self.df = None# 存储原始数据的DataFrame
        self.summary_data = None# 存储汇总数据
        self.init_ui()

definit_ui(self):
# 窗口基本设置
        self.setWindowTitle("工资表管理系统 - 工资发放版")
        self.setGeometry(100, 100, 1400, 900)

# 中心部件
        central_widget = QWidget()
        self.setCentralWidget(central_widget)

# 整体布局
        main_layout = QVBoxLayout(central_widget)

# 顶部布局：公司账号信息 + 功能按钮
        top_layout = QHBoxLayout()

# 公司账号信息区域
        company_group = QGroupBox("公司账号信息")
        company_layout = QGridLayout(company_group)

        company_layout.addWidget(QLabel("公司名称："), 0, 0)
        self.company_name_edit = QLineEdit()
        self.company_name_edit.setPlaceholderText("请输入公司名称")
        self.company_name_edit.setText("XX科技有限公司")
        company_layout.addWidget(self.company_name_edit, 0, 1)

        company_layout.addWidget(QLabel("公司账号："), 1, 0)
        self.company_account_edit = QLineEdit()
        self.company_account_edit.setPlaceholderText("请输入公司银行账号")
        self.company_account_edit.setText("6222081234567890")
        company_layout.addWidget(self.company_account_edit, 1, 1)

        company_layout.addWidget(QLabel("开户行："), 2, 0)
        self.bank_name_edit = QLineEdit()
        self.bank_name_edit.setPlaceholderText("请输入开户银行")
        self.bank_name_edit.setText("中国工商银行XX支行")
        company_layout.addWidget(self.bank_name_edit, 2, 1)

        top_layout.addWidget(company_group)

# 功能按钮区域
        btn_layout = QVBoxLayout()

# 导入文件按钮
        self.import_btn = QPushButton("导入CSV/XLS文件")
        self.import_btn.clicked.connect(self.import_file)
        btn_layout.addWidget(self.import_btn)

# 生成工资表按钮
        self.gen_salary_btn = QPushButton("生成工资表")
        self.gen_salary_btn.clicked.connect(self.generate_salary_table)
        self.gen_salary_btn.setEnabled(False)  # 初始禁用
        btn_layout.addWidget(self.gen_salary_btn)

# 导出工资表按钮
        self.export_salary_btn = QPushButton("导出工资表")
        self.export_salary_btn.clicked.connect(self.export_salary_table)
        self.export_salary_btn.setEnabled(False)
        btn_layout.addWidget(self.export_salary_btn)

        top_layout.addLayout(btn_layout)
        main_layout.addLayout(top_layout)

# 表格操作布局
        table_op_layout = QHBoxLayout()

# 全选复选框
        self.select_all_checkbox = QCheckBox("全选")
        self.select_all_checkbox.stateChanged.connect(self.toggle_select_all)
        table_op_layout.addWidget(self.select_all_checkbox)

# 批量发送按钮
        self.batch_pay_btn = QPushButton("批量发放工资")
        self.batch_pay_btn.clicked.connect(self.batch_payment)
        self.batch_pay_btn.setEnabled(False)
        table_op_layout.addWidget(self.batch_pay_btn)

        table_op_layout.addStretch()
        main_layout.addLayout(table_op_layout)

# 数据展示表格
        self.table = QTableWidget()
        self.table.setEditTriggers(QTableWidget.NoEditTriggers)  # 禁止编辑
        self.table.horizontalHeader().setStretchLastSection(True)
        self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        main_layout.addWidget(self.table)

defimport_file(self):
"""导入CSV/XLS/XLSX文件并展示数据"""
# 选择文件
        file_path, _ = QFileDialog.getOpenFileName(
            self, "选择文件", "",
"Excel文件 (*.xls *.xlsx);;CSV文件 (*.csv);;所有文件 (*.*)"
        )
ifnot file_path:
return

try:
# 读取文件
if file_path.endswith(".csv"):
                self.df = pd.read_csv(file_path, encoding="utf-8")
elif file_path.endswith(".xls"):
                self.df = pd.read_excel(file_path, engine="xlrd")
else:  # .xlsx
                self.df = pd.read_excel(file_path, engine="openpyxl")

# 自动计算实际考勤天数（核心逻辑：处理可能的考勤列）
            self.calc_actual_attendance()

# 展示数据到表格（无汇总行，仅原始数据）
            self.show_data_in_table(self.df, show_summary=False)

# 启用功能按钮
            self.gen_salary_btn.setEnabled(True)
            QMessageBox.information(self, "成功", "文件导入成功！已自动计算实际考勤天数")

except Exception as e:
            QMessageBox.critical(self, "错误", f"文件导入失败：{str(e)}")

defcalc_actual_attendance(self):
"""自动计算实际考勤天数（兼容不同列名）"""
# 定义考勤相关列名（兼容常见命名）
        actual_cols = ["实际出勤", "实际考勤", "出勤天数"]
        should_cols = ["应出勤", "应考勤", "应到天数"]
        leave_cols = ["事假", "病假", "调休", "年假"]

# 找到实际考勤列（如果不存在则自动计算）
        actual_col = None
for col in actual_cols:
if col in self.df.columns:
                actual_col = col
break

if actual_col isNone:
# 自动计算：实际考勤 = 应出勤 - 各类请假天数
            should_col = None
for col in should_cols:
if col in self.df.columns:
                    should_col = col
break
if should_col isNone:
                QMessageBox.warning(self, "提示", "未找到应出勤列，无法自动计算实际考勤天数")
return

# 计算总请假天数
            leave_days = pd.Series([0.0] * len(self.df))
for col in leave_cols:
if col in self.df.columns:
                    leave_days += self.df[col].fillna(0)

# 新增实际出勤列
            self.df["实际出勤"] = self.df[should_col] - leave_days
# 确保天数非负
            self.df["实际出勤"] = self.df["实际出勤"].apply(lambda x: max(x, 0))

defcalculate_summary(self):
"""计算汇总数据"""
if self.df isNone:
returnNone

        summary = {}
# 遍历所有列，计算对应汇总值
for col in self.df.columns:
if col in ["工资", "最终工资", "每天工资", "实际出勤"]:
# 数值列：求和
                summary[col] = round(self.df[col].sum(), 2)
elif col in ["是否审核"]:
# 统计审核状态："已审核: X 未审核: Y"
                audit_count = self.df[col].value_counts()
                yes_count = audit_count.get("是", 0)
                no_count = audit_count.get("否", 0)
                summary[col] = f"已审核: {yes_count} 未审核: {no_count}"
elif col in ["银行卡号"]:
# 卡号列：统计人数
                summary[col] = f"总人数: {len(self.df)}"
elif col == "发放状态":
                summary[col] = "汇总"
else:
# 其他列：标注"汇总"
                summary[col] = "汇总"

        self.summary_data = summary
return summary

defshow_data_in_table(self, df, show_summary=True):
"""将DataFrame数据展示到QTableWidget，并为新增列着色，可选显示汇总行"""
# 计算总行数（数据行 + 汇总行）
        total_rows = len(df)
if show_summary and self.summary_data:
            total_rows += 1

# 插入选择列（第一列）
        columns = ["选择"] + df.columns.tolist()

# 设置行列数
        self.table.setRowCount(total_rows)
        self.table.setColumnCount(len(columns))
# 设置列名
        self.table.setHorizontalHeaderLabels(columns)

# 填充数据行
for row in range(len(df)):
# 第一列：复选框
            check_item = QTableWidgetItem()
            check_item.setCheckState(Qt.Unchecked)
            check_item.setTextAlignment(Qt.AlignCenter)
            self.table.setItem(row, 0, check_item)

# 其他列：数据
for col in range(len(df.columns)):
                col_idx = col + 1
                col_name = df.columns[col]
                value = df.iloc[row, col]

# 处理空值
if pd.isna(value):
                    value = ""
# 数值格式化（保留2位小数）
elif isinstance(value, (int, float)):
                    value = round(value, 2)

# 创建表格项
                item = QTableWidgetItem(str(value))
                item.setTextAlignment(Qt.AlignCenter)  # 居中显示

# 如果是新增列，设置背景色
if col_name in COLOR_CONFIG:
                    item.setBackground(QBrush(COLOR_CONFIG[col_name][0]))

# 发放状态列特殊处理
if col_name == "发放状态":
                    item.setBackground(QBrush(PAYMENT_STATUS_COLOR.get(str(value), QColor(245, 245, 245))))

                self.table.setItem(row, col_idx, item)

# 添加汇总行
if show_summary and self.summary_data:
            summary_row = len(df)  # 汇总行在最后一行

# 汇总行第一列：显示"汇总"
            summary_item = QTableWidgetItem("汇总")
            summary_item.setTextAlignment(Qt.AlignCenter)
            summary_item.setBackground(QBrush(SUMMARY_STYLE["qt_bg_color"]))
            font = summary_item.font()
            font.setBold(True)
            summary_item.setFont(font)
            self.table.setItem(summary_row, 0, summary_item)

# 汇总行其他列
for col in range(len(df.columns)):
                col_idx = col + 1
                col_name = df.columns[col]
                value = self.summary_data.get(col_name, "汇总")

# 创建汇总行表格项
                item = QTableWidgetItem(str(value))
                item.setTextAlignment(Qt.AlignCenter)
# 设置汇总行样式
                item.setBackground(QBrush(SUMMARY_STYLE["qt_bg_color"]))
# 字体加粗
                font = item.font()
                font.setBold(True)
                item.setFont(font)

                self.table.setItem(summary_row, col_idx, item)

deftoggle_select_all(self, state):
"""全选/取消全选"""
if self.df isNone:
return

# 遍历所有数据行（排除汇总行）
for row in range(len(self.df)):
            check_item = self.table.item(row, 0)
if check_item:
                check_item.setCheckState(state)

defget_selected_data(self):
"""获取选中的数据"""
        selected_data = []

if self.df isNone:
return selected_data

# 遍历所有行
for row in range(len(self.df)):
            check_item = self.table.item(row, 0)
if check_item and check_item.checkState() == Qt.Checked:
# 获取该行数据
                row_data = {}
for col in range(len(self.df.columns)):
                    col_name = self.df.columns[col]
                    cell_item = self.table.item(row, col + 1)
                    row_data[col_name] = cell_item.text() if cell_item else""
                selected_data.append(row_data)

return selected_data

defbatch_payment(self):
"""批量发放工资"""
# 获取选中的数据
        selected_data = self.get_selected_data()

ifnot selected_data:
            QMessageBox.warning(self, "提示", "请先选择要发放工资的员工！")
return

# 检查公司账号信息
        company_account = self.company_account_edit.text().strip()
        company_name = self.company_name_edit.text().strip()
        bank_name = self.bank_name_edit.text().strip()

ifnot company_account ornot company_name ornot bank_name:
            QMessageBox.warning(self, "提示", "请完善公司账号信息！")
return

# 打开发放确认弹窗
        dialog = PaymentConfirmDialog(company_account, selected_data, self)
if dialog.exec_() == QDialog.Accepted:
# 更新发放状态
            self.update_payment_status(selected_data)

defupdate_payment_status(self, selected_data):
"""更新发放状态"""
if self.df isNone:
return

# 确保发放状态列存在
if"发放状态"notin self.df.columns:
            self.df["发放状态"] = "未发放"

# 更新选中行的发放状态
for data in selected_data:
# 根据姓名匹配行（实际项目中建议用唯一标识）
            name = data.get("姓名", "")
if name and name in self.df["姓名"].values:
                self.df.loc[self.df["姓名"] == name, "发放状态"] = "发放成功"

# 重新计算汇总数据
        self.calculate_summary()

# 刷新表格
        self.show_data_in_table(self.df, show_summary=True)

        QMessageBox.information(self, "成功", "工资发放状态已更新！")

defgenerate_salary_table(self):
"""生成工资表：新增每天工资、工资、银行卡号、是否审核、最终工资列，并计算汇总数据"""
if self.df isNone:
            QMessageBox.warning(self, "提示", "请先导入文件！")
return

try:
# 1. 每天工资：随机生成200-500元（可自定义范围）
if"每天工资"notin self.df.columns:
                self.df["每天工资"] = [round(random.uniform(200, 500), 2) for _ in range(len(self.df))]

# 2. 工资 = 实际出勤 × 每天工资
if"工资"notin self.df.columns:
                self.df["工资"] = self.df["实际出勤"] * self.df["每天工资"]
                self.df["工资"] = self.df["工资"].round(2)  # 保留2位小数

# 3. 银行卡号：模拟真实卡号格式（62开头）
if"银行卡号"notin self.df.columns:
                self.df["银行卡号"] = [
f"6222{random.randint(100000000000, 999999999999)}"
for _ in range(len(self.df))
                ]

# 4. 是否审核：随机生成（可后续手动修改）
if"是否审核"notin self.df.columns:
                self.df["是否审核"] = [random.choice(["是", "否"]) for _ in range(len(self.df))]

# 5. 最终工资：默认等于工资（可扩展社保/个税扣除逻辑）
if"最终工资"notin self.df.columns:
                self.df["最终工资"] = self.df["工资"].round(2)

# 6. 发放状态列
if"发放状态"notin self.df.columns:
                self.df["发放状态"] = "未发放"

# 计算汇总数据
            self.calculate_summary()

# 刷新表格展示（包含汇总行）
            self.show_data_in_table(self.df, show_summary=True)

# 启用导出按钮和发放按钮
            self.export_salary_btn.setEnabled(True)
            self.batch_pay_btn.setEnabled(True)

            QMessageBox.information(self, "成功", "工资表生成成功！新增列已标注不同颜色，最后一行已添加汇总数据")

except Exception as e:
            QMessageBox.critical(self, "错误", f"生成工资表失败：{str(e)}")

defexport_salary_table(self):
"""导出完整工资表到Excel文件，新增列标注颜色，包含汇总行"""
if self.df isNone:
            QMessageBox.warning(self, "提示", "暂无工资表数据！")
return

# 选择保存路径
        save_path, _ = QFileDialog.getSaveFileName(
            self, "保存工资表", "工资表.xlsx",
"Excel文件 (*.xlsx);;CSV文件 (*.csv);;所有文件 (*.*)"
        )
ifnot save_path:
return

try:
if save_path.endswith(".csv"):
# CSV格式：先复制df，添加汇总行后导出
                df_export = self.df.copy()
if self.summary_data:
# 将汇总数据转为Series并添加到最后一行
                    summary_series = pd.Series(self.summary_data)
                    df_export = pd.concat([df_export, summary_series.to_frame().T], ignore_index=True)
                df_export.to_csv(save_path, index=False, encoding="utf-8-sig")
else:
# 使用openpyxl保存xlsx，支持格式优化、颜色标注和汇总行
                wb = Workbook()
                ws = wb.active
                ws.title = "工资表"

# 写入表头
                headers = self.df.columns.tolist()
                ws.append(headers)
# 为表头的新增列着色
for col_idx, col_name in enumerate(headers, 1):
if col_name in COLOR_CONFIG:
                        ws.cell(row=1, column=col_idx).fill = COLOR_CONFIG[col_name][1]

# 写入数据行
for row_idx, (_, row) in enumerate(self.df.iterrows(), 2):
                    row_data = row.tolist()
                    ws.append(row_data)
# 为数据行的新增列着色
for col_idx, col_name in enumerate(headers, 1):
if col_name in COLOR_CONFIG:
                            ws.cell(row=row_idx, column=col_idx).fill = COLOR_CONFIG[col_name][1]

# 发放状态列颜色
                    status_col_idx = headers.index("发放状态") + 1if"发放状态"in headers else-1
if status_col_idx > 0:
                        status_value = row.get("发放状态", "")
if status_value in ["发放成功", "发放失败", "发放中", "未发放"]:
                            fill_color = {
"未发放": "FFE0E0",
"发放中": "FFFFCC",
"发放成功": "E0FFE0",
"发放失败": "FFCCCC"
                            }.get(status_value, "F5F5F5")
                            ws.cell(row=row_idx, column=status_col_idx).fill = PatternFill(
                                start_color=fill_color, end_color=fill_color, fill_type="solid"
                            )

# 写入汇总行
if self.summary_data:
                    summary_row_idx = len(self.df) + 2# 汇总行在数据行之后
# 准备汇总行数据
                    summary_row_data = [self.summary_data.get(col, "汇总") for col in headers]
                    ws.append(summary_row_data)
# 设置汇总行样式
for col_idx in range(1, len(headers) + 1):
                        cell = ws.cell(row=summary_row_idx, column=col_idx)
                        cell.fill = SUMMARY_STYLE["excel_fill"]
                        cell.font = SUMMARY_STYLE["excel_font"]
                        cell.alignment = cell.alignment.copy(horizontal='center')  # 居中对齐

# 调整列宽
for col in ws.columns:
                    max_length = 0
                    column = col[0].column_letter
for cell in col:
try:
if len(str(cell.value)) > max_length:
                                max_length = len(str(cell.value))
except:
pass
                    adjusted_width = min(max_length + 2, 50)
                    ws.column_dimensions[column].width = adjusted_width

                wb.save(save_path)

            QMessageBox.information(self, "成功", f"工资表已导出至：{save_path}，包含汇总行数据")

except Exception as e:
            QMessageBox.critical(self, "错误", f"导出失败：{str(e)}")


if __name__ == "__main__":
# 解决中文显示问题
    QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
    app = QApplication(sys.argv)
    window = SalaryManagementSystem()
    window.show()
    sys.exit(app.exec_())

核心功能说明

1. 公司账号信息页面

• 在主窗口顶部新增公司账号信息分组框，包含：
• 公司名称输入框（默认值：XX科技有限公司）
• 公司银行账号输入框（默认值：6222081234567890）
• 开户行输入框（默认值：中国工商银行XX支行）
• 发放工资前会校验这些信息是否完整

2. 批量发送功能

• 新增批量发放工资按钮（生成工资表后启用）
• 点击按钮会弹出确认弹窗，展示发放信息并执行模拟发放
• 使用QThread实现异步发放，避免界面卡顿

3. 列表勾选功能

• 表格第一列新增选择列，包含复选框
• 顶部新增全选复选框，支持一键全选/取消全选
• get_selected_data()方法获取所有选中的员工数据

4. 发放确认弹窗

• 显示发放方信息（公司账号、总金额、发放人数）
• 逐行显示选中员工的发放信息（姓名、隐藏后的卡号、金额、审核状态）
• 发放进度日志区域，实时显示发放过程
• 模拟发放过程（500ms/人，90%成功率）
• 发放完成后显示统计结果

5. 发放状态管理

• 新增发放状态列（未发放/发放中/发放成功/发放失败）
• 不同状态使用不同颜色区分
• 发放完成后自动更新表格中的发放状态
• 导出Excel时保留发放状态颜色样式

6. 其他优化

• 修复了汇总行字体加粗的BUG
• 银行卡号显示优化（只显示后4位，保护隐私）
• 发放线程异步执行，界面无卡顿
• 完善的异常处理和用户提示

总结

1. 新增了完整的公司账号信息管理模块，支持发放方信息配置和校验。
2. 实现了列表勾选（全选/单选）+ 批量发放功能，发放过程异步执行且有进度展示。
3. 发放确认弹窗逐行显示发放信息，模拟真实的工资发放确认流程，发放完成后自动更新状态。
4. 所有功能与原有工资表生成、导出功能完全兼容，样式统一且交互友好。

你可以直接运行这段代码，体验完整的工资发放流程：导入文件 → 生成工资表 → 勾选员工 → 点击批量发放 → 确认发放信息 → 查看发放进度 → 完成发放并更新状态。