pyside6 控件三：自定义表格控件

朱雀 2024-03-23 14:46 69阅读 0赞

**目录**

一、自定义控件（读者可直接Copy）

二、使用示例

--------------------

#### 一、自定义控件（读者可直接Copy） ####

import os
    import platform
    from dataclasses import dataclass, field
    
    from PySide6 import QtWidgets
    from PySide6.QtCore import QObject, Qt, Signal
    from PySide6.QtGui import QIcon
    from PySide6.QtWidgets import QWidget, QVBoxLayout, QTableWidget, QApplication, QPushButton, QLineEdit, \
        QTableWidgetItem, QHeaderView
    
    
    # 定义模板子类
    class TableWidgetItem:
        def __init__(self, name, prop, callback, resize_model=QHeaderView.ResizeMode.Stretch):
            # 展示名
            self.name: str = name
            # 属性名
            self.prop: str = prop
            # 回调函数，返回控件实例
            self.callback = callback
            # 表头尺寸模式
            self.resize_model = resize_model
    
    
    # 定义模板基类
    class TableWidgetTemplateBase:
        def __init__(self):
            self.models: list[TableWidgetItem] = []
    
        def handle_cell_changed(self, tableWidgetItem: QTableWidgetItem):
            pass
    
    
    """
    初始化CSS文件位置
    """
    CSS_PATH = None
    if str(platform.system().lower()) == 'windows':
        path = __file__.replace(fr"\{os.path.basename(__file__)}", "").replace("\\\\", "\\")
        CSS_PATH = fr'{path}\el_table.py'
    elif str(platform.system().lower()) == 'linux':
        path = __file__.replace(fr"/{os.path.basename(__file__)}", "").replace("//", "/")
        CSS_PATH = fr'{path}/el_table.py'
    else:
        print(f"未知系统：{platform.system().lower()}")
    
    
    class ElTableUi(object):
        """
            User Interface
            """
    
        def setupUi(self, window):
            with open(CSS_PATH, "r", encoding="UTF-8") as f:
                window.setStyleSheet(f.read())
            window.setWindowTitle("ElTable")
            window.setWindowIcon(QIcon(':/icons/add.png'))
            window.resize(40, 30)
    
            # 设置整个窗体的透明度
            window.setWindowOpacity(0.90)  # 设置窗体透明度
            # 设置隐藏背景
            window.setAttribute(Qt.WA_TranslucentBackground)
            # 垂直布局
            self.layout = QVBoxLayout()
            # 设置边距为0
            self.layout.setContentsMargins(0, 0, 0, 0)
            window.setLayout(self.layout)
            self._table = QTableWidget()
    
            self._table.setSizePolicy(QtWidgets.QSizePolicy.Policy.Expanding, QtWidgets.QSizePolicy.Policy.Expanding)
            self.layout.addWidget(self._table)
    
    
    class ElTable(QWidget, ElTableUi):
        cellChanged: Signal = Signal(QLineEdit)
    
        def __init__(self, parent=None):
            super(ElTable, self).__init__(parent=parent)
            self.setupUi(self)
            self.tableTemplate: TableWidgetTemplateBase
            self.tableData: list[object]
            self.tableDataFunction: any
            self.__bind()
    
        def setTableTemplate(self, templateBase: TableWidgetTemplateBase):
            """
            设置模型
            """
            self.tableTemplate: TableWidgetTemplateBase = templateBase
            self.cellChanged.connect(self.tableTemplate.handle_cell_changed)
    
        def render_table(self, table_data: list[object]):
            self.tableData = table_data
    
            self._table.clear()
            self._table.setRowCount(len(self.tableData))
            self._table.setColumnCount(len(self.tableTemplate.models))
            self._table.setHorizontalHeaderLabels(map(lambda x: x.name, self.tableTemplate.models))
            self._table.horizontalHeader().setSectionResizeMode(QHeaderView.ResizeMode.Stretch)
            for i_index, i in enumerate(self.tableTemplate.models):
                self._table.horizontalHeader().setSectionResizeMode(i_index, i.resize_model)
    
            for i_index, i in enumerate(self.tableData):
                for j_index, j in enumerate(self.tableTemplate.models):
                    callback = j.callback(i, self)
                    if isinstance(callback, QTableWidgetItem):
                        self._table.setItem(i_index, j_index, callback)
                    else:
                        self._table.setCellWidget(i_index, j_index, callback)
            self._table.cellChanged.connect(lambda row, column: self.cellChanged.emit(self._table.item(row, column)))
    
        def __bind(self):
            """
            数据绑定到控件属性
            """
            pass

#### 二、使用示例 ####

if __name__ == '__main__':
        # 模拟数据
        @dataclass
        class Account:
            username: str = field(default=None)
            password: str = field(default=None)
            age: int = field(default=None)
    
    
        # 定义模型
        class TableWidgetTemplate(TableWidgetTemplateBase):
    
            def __init__(self):
                super(TableWidgetTemplate, self).__init__()
                self.models = [
                    TableWidgetItem(name="姓名", prop="username", callback=self.username_function),
                    TableWidgetItem(name="密码", prop="password", callback=self.password_function),
                    TableWidgetItem(name="年龄", prop="age", callback=self.age_function),
                ]
    
            def handle_cell_changed(self, tableWidgetItem: QTableWidgetItem):
                print("相关控件：", tableWidgetItem)
                print("相关扩展：", tableWidgetItem.data(Qt.UserRole + 1))
                print("相关数据：", tableWidgetItem.text())
    
            @staticmethod
            def username_function(item, el: ElTable):
                tableWidgetItem = QTableWidgetItem()
                tableWidgetItem.setText(item.username)
                tableWidgetItem.setData(Qt.UserRole + 1, {'item': item, 'prop': 'username'})
                return tableWidgetItem
    
            @staticmethod
            def password_function(item, el: ElTable):
                button = QPushButton()
                button.setText(item.password)
                return button
    
            @staticmethod
            def age_function(item, el: ElTable):
                edit = QLineEdit()
                edit.setText(str(item.age))
    
                return edit
    
    
        """
        Test Demo
        """
        # 引入资源（这一步不能少！）
        from images.resources_rc import *
    
        app = QApplication([])
        table = ElTable()
        table.setTableTemplate(TableWidgetTemplate())
        table.render_table([
            Account("张三", "123", 19),
            Account("李四", "456", 20),
        ])
        table.show()
        app.exec()

![01c68114354a4202a282f89bd29d091d.png][]

[01c68114354a4202a282f89bd29d091d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/23/655cd60a36bd4c2e94ff4b7c8e5aad7b.png