QT 表格操作

from PyQt5 import QtGui
from PyQt5.QtCore import QSize, Qt, QRect
from PyQt5.QtGui import QPalette, QColor, QBrush
from PyQt5.QtWidgets import QWidget, QTableWidget, QTableWidgetItem, QApplication, QHeaderView, QGridLayout, QPushButton
import sys


Datas = []


class Table(QWidget):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.h_list = ["公司名称", "企业名称", "任务类型", "报表名称", "任务结果", "地区版本", "地区", "账套日期", "任务创建时间", "任务id"]
        if not len(Datas):
            self.v_list = ["1"]
        else:
            self.v_list = [str(i + 1) for i in range(len(Datas))]
        self.current_hover = [0, 0]
        self.resize(900, 800)
        self.grid = QGridLayout()

        self.add_btn()
        self.cancel_btn()
        self.cancel_select_btn()
        self.insert_up_btn()
        self.insert_down_btn()

        self.setObjectName('表格')
        self.table = QTableWidget(self)
        self.set_col_width()  # 自动设置表格列宽
        self.set_col_color()
        self.table.setMouseTracking(True)  # 设置鼠标追踪
        self.table.cellEntered.connect(self.set_col_hover)  # 鼠标划过变色
        self.table_main()  # 设置表格表头
        self.load_data()
        self.grid.addWidget(self.table, 1, 0, 1, 5, Qt.AlignCenter)
        self.setLayout(self.grid)

    def load_data(self):
        """
        加载数据
        Returns:

        """
        for key, data in enumerate(Datas):
            self.table.setItem(key, 0, QTableWidgetItem(data['gsName']))
            self.table.setItem(key, 1, QTableWidgetItem(data['qyName']))
            self.table.setItem(key, 2, QTableWidgetItem(data['rwlxText']))
            self.table.setItem(key, 3, QTableWidgetItem(""))
            self.table.setItem(key, 4, QTableWidgetItem(data['statusErrMsg']))
            self.table.setItem(key, 5, QTableWidgetItem(data['version']))
            self.table.setItem(key, 6, QTableWidgetItem(data['xzqh']))
            self.table.setItem(key, 7, QTableWidgetItem(data['createTime'][:7]))
            self.table.setItem(key, 8, QTableWidgetItem(data['createTime']))
            self.table.setItem(key, 9, QTableWidgetItem(data['strTaskId']))

    def add_btn(self):
        """
        添加行 按钮
        Returns:

        """
        btn = QPushButton("添加行")
        btn.clicked.connect(self.add_row)  # 设置表格行数
        self.grid.addWidget(btn, 0, 0)

    def cancel_btn(self):
        """
        删除行 按钮
        Returns:

        """
        btn = QPushButton("删除行")
        btn.clicked.connect(self.cancel_row)  # 设置表格行数
        self.grid.addWidget(btn, 0, 1)

    def cancel_select_btn(self):
        """
        删除行 按钮
        Returns:

        """
        btn = QPushButton("删除选中行")
        btn.clicked.connect(self.cancel_select_row)  # 设置表格行数
        self.grid.addWidget(btn, 0, 2)

    def add_row(self):
        """
        在最后一行添加
        Returns:

        """
        row_count = self.table.rowCount()
        if not row_count:
            row_count = 0
        self.table.setRowCount(row_count+1)

    def cancel_row(self):
        """
        删除最后一行
        Returns:

        """
        row_count = self.table.rowCount()
        if not row_count:
            row_count = 0
        self.table.removeRow(row_count-1)

    def cancel_select_row(self):
        """
        删除选中行
        Returns:

        """
        row_select_row = self.table.currentIndex()
        if not row_select_row:
            return
        self.table.removeRow(row_select_row.row())
        self.v_list.pop(-1)
        self.table.setVerticalHeaderLabels(self.v_list)  # 重新刷新当前列名

    def insert_up_btn(self):
        """
        向上插入 按钮
        Returns:

        """
        btn = QPushButton("向上插入")
        btn.clicked.connect(lambda: self.insert_row(e=True))  # 向上插入一行
        self.grid.addWidget(btn, 0, 3)

    def insert_down_btn(self):
        """
        向下插入 按钮
        Returns:

        """
        btn = QPushButton("向下插入")
        btn.clicked.connect(self.insert_row)  # 向下插入一行
        self.grid.addWidget(btn, 0, 4)

    def insert_row(self, e=False):
        """
        向 上/下 插入一行
        Args:
            e: 是否向上插入 默认 False 向下插入

        Returns:

        """
        max_row = self.table.rowCount()  # 获取总共有多少列
        row_select_row = self.table.currentIndex()  # 当前列的索引
        if not row_select_row:
            return
        if e:
            self.table.insertRow(row_select_row.row())  # 向上插入 使用当前索引
        else:
            self.table.insertRow(row_select_row.row() + 1)  # 向下插入需要索引加 1 向上插入 使用当前索引
        self.v_list.append(str(max_row + 1))
        self.table.setVerticalHeaderLabels(self.v_list)  # 重新刷新当前列名

    def table_main(self):
        """
        设置表格表头
        Returns:

        """
        self.table.setObjectName('ha')
        self.table.setHorizontalHeaderLabels(self.h_list)
        self.table.setVerticalHeaderLabels(self.v_list)

    def set_col_width(self):
        """
        自动设置表格列宽  程序当前宽度 / 总列数
        Returns:

        """
        self.table.setSortingEnabled(True)  # 默认为False
        self.table.setColumnCount(len(self.h_list))  # 设置表格列数
        self.table.setRowCount(len(self.v_list))  # 设置表格行数
        print(111)
        self.table.setFixedSize(QSize(self.size().width(), self.size().height() - 50))  # 设置表格宽高
        # 方法一:  手动设置
        # for i in range(len(self.h_list)):
        #     self.table.setColumnWidth(i, self.size().width() / len(self.h_list))
        # 方法二:  自动设置
        # self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)  # 列自适应，(水平方向占满窗体)
        # # self.table.verticalHeader().setSectionResizeMode(QHeaderView.Stretch)  # 行自适应，(垂直方向占满窗体)

        # self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        # self.table.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeToContents)

        self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.table.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeToContents)

    def set_col_color(self):
        """
        设置隔行变色
        Returns:

        """
        # self.table.setPalette(QPalette(QColor(102, 220, 121)))  # 自定义设置隔行显示的颜色  默认为灰色
        self.table.setAlternatingRowColors(True)  # 隔行变色设置

    def set_col_hover(self, row, column):
        """
        在有数据的情况下  鼠标悬停后整行变色
        Args:
            row: 悬停行
            column: 悬停列

        Returns:

        """
        max_col = self.table.columnCount()  # 获取总共有多少列
        for i in range(max_col):
            item = self.table.item(row, i)
            if not item:  # 检测到如果没有数据情况下 赋予空值
                self.table.setItem(row, i, QTableWidgetItem(''))
            old_item = self.table.item(self.current_hover[0], i)
            if self.current_hover[0] != row:
                if old_item:
                    old_item.setBackground(QBrush(QColor('white')))  # 悬停前一行设置白色
                if item:
                    item.setBackground(QBrush(QColor('yellow')))  # 悬停行设置黄色
        self.current_hover = [row, column]

    def set_cell_hover(self, row, column):
        """
        在有数据的情况下  鼠标悬停后单个单元格变色
        Args:
            row: 悬停行
            column: 悬停列

        Returns:

        """
        item = self.table.item(row, column)
        old_item = self.table.item(self.current_hover[0], self.current_hover[1])
        if self.current_hover != [row, column]:
            old_item.setBackground(QBrush(QColor('white')))
            item.setBackground(QBrush(QColor('yellow')))
        self.current_hover = [row, column]

    def set_cross_hover(self, row, column):
        """
        在有数据的情况下  鼠标悬停后当前单元格行列变色
        Args:
            row: 悬停行
            column: 悬停列

        Returns:

        """
        max_row = self.table.rowCount()  # 获取总共有多少行
        max_col = self.table.columnCount()  # 获取总共有多少列
        for i in range(max_col):
            if i == self.current_hover[1]:
                continue
            item = self.table.item(row, i)
            old_item = self.table.item(self.current_hover[0], i)
            if self.current_hover[0] != row:
                if old_item:
                    old_item.setBackground(QBrush(QColor('white')))  # 悬停前一行设置白色
                if item:
                    item.setBackground(QBrush(QColor('yellow')))  # 悬停行设置黄色

        for i in range(max_row):
            if i == self.current_hover[0]:
                continue
            item = self.table.item(i, column)
            old_item = self.table.item(i, self.current_hover[1])
            if self.current_hover != [row, column]:
                if old_item:
                    old_item.setBackground(QBrush(QColor('white')))  # 悬停前一行设置白色
                if item:
                    item.setBackground(QBrush(QColor('yellow')))  # 悬停行设置黄色

        self.current_hover = [row, column]


if __name__ == '__main__':
    app = QApplication(sys.argv)
    table = Table()
    table.show()
    sys.exit(app.exec())