PyQt5 QTableWidget中实现健壮的单元格合并与取消合并功能

import sys
from PyQt5.QtCore import Qt, QEvent
from PyQt5.QtWidgets import QApplication, QMainWindow, QTableWidget, QVBoxLayout, QWidget, QPushButton


class ExcelLikeTable(QMainWindow):
    def __init__(self):
        super().__init__()
        self.mergeButton = None
        self.unmergeButton = None
        self.tableWidget = QTableWidget()
        self.initUI()

    def initUI(self):
        self.setWindowTitle("PyQt5 QTableWidget 单元格合并示例")
        self.setGeometry(100, 100, 800, 600)

        self.tableWidget.setColumnCount(10)
        self.tableWidget.setHorizontalHeaderLabels([f'Column {chr(65+i)}' for i in range(10)])
        self.tableWidget.setRowCount(10) # 增加行数以便测试合并

        # 初始设置，确保选择模式正确
        self.tableWidget.clearSelection()
        self.tableWidget.setSelectionMode(QTableWidget.ContiguousSelection)
        self.tableWidget.setSelectionBehavior(QTableWidget.SelectItems)

        self.mergeButton = QPushButton("合并单元格")
        self.unmergeButton = QPushButton("取消合并单元格")
        self.mergeButton.clicked.connect(self.mergeCells)
        self.unmergeButton.clicked.connect(self.unmergeCells)

        layout = QVBoxLayout()
        layout.addWidget(self.tableWidget)
        layout.addWidget(self.mergeButton)
        layout.addWidget(self.unmergeButton)

        centralWidget = QWidget()
        centralWidget.setLayout(layout)
        self.setCentralWidget(centralWidget)

        self.tableWidget.installEventFilter(self)

    def mergeCells(self):
        # 1. 获取所有选中的单元格索引
        selection = self.tableWidget.selectedIndexes()

        if not selection:
            print("未选择任何单元格进行合并。")
            return

        if len(selection) == 1:
            print("只选择了一个单元格，无需合并。")
            return

        # 2. 清理选中区域内已存在的合并
        # 这一步非常关键，它能避免复杂的嵌套合并问题，并确保后续计算的准确性。
        # 当选中区域内包含已被合并的单元格时，先将其取消合并。
        for index in selection:
            row, column = index.row(), index.column()
            # 检查当前单元格是否是某个合并区域的起始点，或者是否属于某个合并区域
            # 如果是，则将其取消合并，恢复为1x1的跨度
            if (self.tableWidget.rowSpan(row, column) > 1 or
                self.tableWidget.columnSpan(row, column) > 1):
                self.tableWidget.setSpan(row, column, 1, 1)

        # 3. 重新获取并排序选择，以确保获取到最准确的边界
        # 清理合并操作可能会影响当前的选中状态，因此需要重新获取。
        # 排序是为了确保能够正确地找到最左上角和最右下角的单元格。
        selection = sorted(self.tableWidget.selectedIndexes())

        # 4. 计算合并区域的边界
        # 最小行、最小列、最大行、最大列
        topRow = selection[0].row()
        leftColumn = selection[0].column()
        bottomRow = selection[-1].row()
        rightColumn = selection[-1].column()

        # 计算合并所需的行数和列数
        rowCount = bottomRow - topRow + 1
        columnCount = rightColumn - leftColumn + 1

        print(
            f"选择范围 - 起始行: {topRow}, 起始列: {leftColumn}, 行数: {rowCount}, 列数: {columnCount}")

        # 5. 应用单元格合并
        self.tableWidget.setSpan(topRow, leftColumn, rowCount, columnCount)
        print(
            f"合并完成，从单元格 {chr(65 + leftColumn)}{topRow + 1} 到 {chr(65 + rightColumn)}{bottomRow + 1}")

    # 辅助方法，用于添加行（与合并功能无关，但保留以保持完整示例）
    def addRow(self):
        rowCount = self.tableWidget.rowCount()
        self.tableWidget.insertRow(rowCount)

    # 事件过滤器，用于处理回车键添加新行（与合并功能无关，但保留以保持完整示例）
    def eventFilter(self, source, event):
        if event.type() == QEvent.KeyPress and event.key() == Qt.Key_Return:
            currentRow = self.tableWidget.currentRow()
            currentColumn = self.tableWidget.currentColumn()
            if currentRow == self.tableWidget.rowCount() - 1:
                self.addRow()
            self.tableWidget.setCurrentCell(currentRow + 1, currentColumn)
            return True
        return super(ExcelLikeTable, self).eventFilter(