Сортировка без учета регистра в QAbstractItemModel - PullRequest
Новая
       55

Сортировка без учета регистра в QAbstractItemModel

0 голосов
/ 14 мая 2018

У меня проблемы с попыткой создать собственную функцию сортировки с помощью QAbstractItemModel. Это работает, но без учета регистра. Я пытался использовать QSortFilterProxyModel, но никакого успеха. Моя функция сортировки: 

def sort(self, col, order):
    self.emit(SIGNAL("layoutAboutToBeChanged()"))
    self.tableData = sorted(self.tableData, key=operator.itemgetter(col))       
    if order == Qt.AscendingOrder:
        self.tableData.reverse()
    self.emit(SIGNAL("layoutChanged()"))

Я использую QTableView. Как я могу сделать это без учета регистра?

Полный пример: 

from PyQt4.QtCore import *
from PyQt4.QtGui import *
import operator
import sys

class Window(QWidget):
    def __init__(self):
        super(Window, self).__init__()

        header = ["one", "two"]
        tableDict = [["abcdef", "tuvwx"], ["aBcde", "TUvWx"], ["acdef","tUvWX"], ["Acdef", "TUVwx"], ["ACdef", "TUVwx"]]
        self.myTable = newTableModel(header, tableDict)

        mainLayout = QHBoxLayout()
        mainLayout.addWidget(self.myTable.tableView)
        self.setLayout(mainLayout)
        self.setWindowTitle("Test table")

class newTableModel(QAbstractTableModel): 
    def __init__(self, header, data, parent=None, *args):
        super(newTableModel, self).__init__(parent)
        self.tableView = QTableView()
        self.tableData = data
        self.header = header

        self.tableView.setShowGrid(True)
        self.tableView.setFrameStyle( QFrame.NoFrame )
        self.tableView.setFocusPolicy( Qt.NoFocus )
        self.tableView.setSelectionMode( QAbstractItemView.NoSelection )

        vHeader = self.tableView.verticalHeader()
        vHeader.setVisible(False)
        vHeader.setStretchLastSection(False)
        hHeader = self.tableView.horizontalHeader()
        hHeader.setVisible(True)
        hHeader.setStretchLastSection(False)
        self.tableView.setSortingEnabled(True)

        self.tableView.setModel(self)
        self.tableView.resizeRowsToContents()
        self.tableView.resizeColumnsToContents()
        vHeader.setResizeMode(QHeaderView.ResizeToContents)

        self.tableView.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded)
        self.tableView.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)

    def rowCount(self, parent): 
        return len(self.tableData) 

    def columnCount(self, parent): 
        return len(self.tableData[0]) 

    def data(self, index, role=Qt.DisplayRole):
        row = index.row()
        col = index.column()
        if role == Qt.DisplayRole:
            return "{0}".format(self.tableData[row][col])
        return None

    def setData(self, index, value, role):
        if index.isValid():
             return True
        return False

    def flags(self, index):
        fl = QAbstractTableModel.flags(self, index)
        if index.column() == 0:
            fl |= Qt.ItemIsUserCheckable
        return fl

    def headerData(self, col, orientation, role):
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return self.header[col]

    def sort(self, col, order):
        self.emit(SIGNAL("layoutAboutToBeChanged()"))
        self.tableData = sorted(self.tableData, key=operator.itemgetter(col))      
        if order == Qt.AscendingOrder:
            self.tableData.reverse()
        self.emit(SIGNAL("layoutChanged()"))

if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = Window()
    window.show()
    sys.exit(app.exec_())

Эти данные таблицы в примере показывают сортировку без учета регистра.

1 Ответ

0 голосов
/ 14 мая 2018

Вам просто нужно преобразовать значения в нижний (или верхний) регистр в ключе сортировки, который передается в функцию sorted.Для повышения эффективности вы также можете использовать аргумент reverse , чтобы избежать этого на отдельном шаге: 

def sort(self, col, order):
    self.layoutAboutToBeChanged.emit()
    self.tableData = sorted(
        self.tableData,
        key=lambda row: row[col].lower(),
        reverse=(order != Qt.AscendingOrder),
        )
    self.layoutChanged.emit()

Обратите внимание, что sorted делает стабильную сортировку , поэтому равные значения (после того, как ключ был применен), сохранят свои исходные места.Таким образом, второй столбец в вашем примере не будет показывать никаких изменений при сортировке, поскольку все значения одинаковы (если вы игнорируете регистр).

UPDATE :

Вот решение, которое будет работать для строк и чисел.Предполагается, что столбцы не являются смесью двух типов: 

def sort(self, col, order):
    if self.tableData and self.tableData[0]:
        self.layoutAboutToBeChanged.emit()
        if isinstance(self.tableData[0][col], str):
            sortkey = lambda row: row[col].lower()
        else:
            sortkey = operator.itemgetter(col)
        self.tableData = sorted(
            self.tableData, key=sortkey,
            reverse=(order != Qt.AscendingOrder))
        self.layoutChanged.emit()
...