pyqt5重写treewidget并加入自定义函数

好的，我可以为您提供一个重写`QTreeWidget`并加入自定义函数的示例代码。您可以将其作为参考并进行修改以满足您的需求。

from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem

class CustomTreeWidget(QTreeWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        
    def custom_function(self):
        # TODO: Add your custom function code here
        pass
        
    def add_custom_item(self, text):
        item = QTreeWidgetItem(self)
        item.setText(0, text)
        item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
        item.setCheckState(0, Qt.Unchecked)
        return item

在这个示例代码中，我们创建了一个名为`CustomTreeWidget`的子类，并重写了`QTreeWidget`的`__init__`方法。此外，我们还添加了两个自定义函数`custom_function`和`add_custom_item`。

`custom_function`是您自定义的函数，用于实现您想要的功能。在这里，我只是添加了一个`pass`语句，您需要将其替换为您的实际代码。

`add_custom_item`是用于在树状视图中添加自定义项的函数。它接受一个文本参数，并返回新的`QTreeWidgetItem`对象。在这个函数中，我们创建了一个新的项，并设置了它的文本和标志。

您可以通过以下方式使用这个自定义树状视图：

tree = CustomTreeWidget()
tree.add_custom_item('Item 1')
tree.add_custom_item('Item 2')
tree.custom_function()

在这个示例中，我们首先创建了一个新的`CustomTreeWidget`对象。然后，我们使用`add_custom_item`函数添加了两个自定义项。最后，我们调用了`custom_function`函数来执行您自定义的功能。