Фильтрация QTreeView с QSortFilterProxyModel

class FindFilterProxyModel(QtCore.QSortFilterProxyModel):
    def filterAcceptsRow(self, source_row, source_parent):
        if (self.filterAcceptsRowItself(source_row, source_parent)):
            return True
 
        if (self.hasAcceptedChildren(source_row, source_parent)):
            return True
 
        return False
 
    def filterAcceptsRowItself(self, source_row, source_parent):
        return super(FindFilterProxyModel, self).\
        filterAcceptsRow(source_row, source_parent)
 
    def hasAcceptedChildren(self, source_row, source_parent):
        model = self.sourceModel()
        sourceIndex = model.index(source_row, 0, source_parent)
        if not (sourceIndex.isValid()):
            return False
 
        childCount = model.rowCount(sourceIndex)
        if (childCount == 0):
            return False
 
        for i in range (childCount): 
            if (self.filterAcceptsRowItself(i, sourceIndex)):
                return True
            if (self.hasAcceptedChildren(i, sourceIndex)):
                return True
 
        return False

def findTextPushButton_clicked(self):
        text = self.lineEdit.text()
        self.sourceModel = self.model
        self.proxyModel = FindFilterProxyModel(self)
        self.proxyModel.setSourceModel(self.sourceModel)
        # Если раскоментить строку ниже, то поиск будет работать
        # не только по дереву параметров, но и по содержимому ячеек
        # self.proxyModel.setFilterKeyColumn(-1)
        self.proxyModel.setFilterCaseSensitivity(0)
        self.proxyModel.setFilterWildcard(text)
        self.treeView.setModel(self.proxyModel)

class DiffFilterProxyModel(QtCore.QSortFilterProxyModel):
def filterAcceptsRow(self, source_row, source_parent):
    # проверим, что моделька валидная
    # model = self.sourceModel()
    # if model is None:
    #     return False
 
    # index для первого столбца проверяемой строки
    src_index = model.index(source_row, 0, source_parent)
 
    # рекурсивно пробежимся по дереву
    for i in range (model.rowCount(src_index)):
        child_index = src_index.child(i, 0)
        c1 = child_index.sibling(child_index.row(), 1).data()
        c2 = child_index.sibling(child_index.row(), 2).data()
 
        # если есть хоть один потомок, удовлетворяющий условиям фильтра - ок!
        if (c1 != c2 or self.filterAcceptsRow(i, src_index)):
            return super(DiffFilterProxyModel, self).filterAcceptsRow(source_row, source_parent)                                                   
    # если потомки не отфильтрованы, проверим себя
    c1 = src_index.sibling(src_index.row(), 1).data()
    c2 = src_index.sibling(src_index.row(), 2).data()
 
    return c1 != c2 and super(DiffFilterProxyModel, self).filterAcceptsRow(source_row, source_parent)