#!/usr/bin/env python ############################################################################# ## ## Copyright (C) 2010 Hans-Peter Jansen <hpj@urpla.net>. ## Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies). ## All rights reserved. ## ## This file is part of the examples of PyQt. ## ## $QT_BEGIN_LICENSE:BSD$ ## You may use this file under the terms of the BSD license as follows: ## ## "Redistribution and use in source and binary forms, with or without ## modification, are permitted provided that the following conditions are ## met: ## * Redistributions of source code must retain the above copyright ## notice, this list of conditions and the following disclaimer. ## * Redistributions in binary form must reproduce the above copyright ## notice, this list of conditions and the following disclaimer in ## the documentation and/or other materials provided with the ## distribution. ## * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor ## the names of its contributors may be used to endorse or promote ## products derived from this software without specific prior written ## permission. ## ## THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ## "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT ## LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR ## A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT ## OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, ## SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT ## LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, ## DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY ## THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT ## (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE ## OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." ## $QT_END_LICENSE$ ## ############################################################################# # This is only needed for Python v2 but is harmless for Python v3. import sip sip.setapi('QVariant', 2) from PyQt4 import QtCore, QtGui class MySortFilterProxyModel(QtGui.QSortFilterProxyModel): def __init__(self, parent=None): super(MySortFilterProxyModel, self).__init__(parent) self.minDate = QtCore.QDate() self.maxDate = QtCore.QDate() def setFilterMinimumDate(self, date): self.minDate = date self.invalidateFilter() def filterMinimumDate(self): return self.minDate def setFilterMaximumDate(self, date): self.maxDate = date self.invalidateFilter() def filterMaximumDate(self): return self.maxDate def filterAcceptsRow(self, sourceRow, sourceParent): index0 = self.sourceModel().index(sourceRow, 0, sourceParent) index1 = self.sourceModel().index(sourceRow, 1, sourceParent) index2 = self.sourceModel().index(sourceRow, 2, sourceParent) return ( (self.filterRegExp().indexIn(self.sourceModel().data(index0)) >= 0 or self.filterRegExp().indexIn(self.sourceModel().data(index1)) >= 0) and self.dateInRange(self.sourceModel().data(index2))) def lessThan(self, left, right): leftData = self.sourceModel().data(left) rightData = self.sourceModel().data(right) if not isinstance(leftData, QtCore.QDate): emailPattern = QtCore.QRegExp("([\\w\\.]*@[\\w\\.]*)") if left.column() == 1 and emailPattern.indexIn(leftData) != -1: leftData = emailPattern.cap(1) if right.column() == 1 and emailPattern.indexIn(rightData) != -1: rightData = emailPattern.cap(1) return leftData < rightData def dateInRange(self, date): if isinstance(date, QtCore.QDateTime): date = date.date() return ( (not self.minDate.isValid() or date >= self.minDate) and (not self.maxDate.isValid() or date <= self.maxDate)) class Window(QtGui.QWidget): def __init__(self): super(Window, self).__init__() self.proxyModel = MySortFilterProxyModel(self) self.proxyModel.setDynamicSortFilter(True) self.sourceView = QtGui.QTreeView() self.sourceView.setRootIsDecorated(False) self.sourceView.setAlternatingRowColors(True) sourceLayout = QtGui.QHBoxLayout() sourceLayout.addWidget(self.sourceView) sourceGroupBox = QtGui.QGroupBox("Original Model") sourceGroupBox.setLayout(sourceLayout) self.filterCaseSensitivityCheckBox = QtGui.QCheckBox("Case sensitive filter") self.filterCaseSensitivityCheckBox.setChecked(True) self.filterPatternLineEdit = QtGui.QLineEdit() self.filterPatternLineEdit.setText("Grace|Sports") filterPatternLabel = QtGui.QLabel("&Filter pattern:") filterPatternLabel.setBuddy(self.filterPatternLineEdit) self.filterSyntaxComboBox = QtGui.QComboBox() self.filterSyntaxComboBox.addItem("Regular expression", QtCore.QRegExp.RegExp) self.filterSyntaxComboBox.addItem("Wildcard", QtCore.QRegExp.Wildcard) self.filterSyntaxComboBox.addItem("Fixed string", QtCore.QRegExp.FixedString) self.fromDateEdit = QtGui.QDateEdit() self.fromDateEdit.setDate(QtCore.QDate(2006, 12, 22)) self.fromDateEdit.setCalendarPopup(True) fromLabel = QtGui.QLabel("F&rom:") fromLabel.setBuddy(self.fromDateEdit) self.toDateEdit = QtGui.QDateEdit() self.toDateEdit.setDate(QtCore.QDate(2007, 1, 5)) self.toDateEdit.setCalendarPopup(True) toLabel = QtGui.QLabel("&To:") toLabel.setBuddy(self.toDateEdit) self.filterPatternLineEdit.textChanged.connect(self.textFilterChanged) self.filterSyntaxComboBox.currentIndexChanged.connect(self.textFilterChanged) self.filterCaseSensitivityCheckBox.toggled.connect(self.textFilterChanged) self.fromDateEdit.dateChanged.connect(self.dateFilterChanged) self.toDateEdit.dateChanged.connect(self.dateFilterChanged) self.proxyView = QtGui.QTreeView() self.proxyView.setRootIsDecorated(False) self.proxyView.setAlternatingRowColors(True) self.proxyView.setModel(self.proxyModel) self.proxyView.setSortingEnabled(True) self.proxyView.sortByColumn(1, QtCore.Qt.AscendingOrder) self.textFilterChanged() self.dateFilterChanged() proxyLayout = QtGui.QGridLayout() proxyLayout.addWidget(self.proxyView, 0, 0, 1, 3) proxyLayout.addWidget(filterPatternLabel, 1, 0) proxyLayout.addWidget(self.filterPatternLineEdit, 1, 1) proxyLayout.addWidget(self.filterSyntaxComboBox, 1, 2) proxyLayout.addWidget(self.filterCaseSensitivityCheckBox, 2, 0, 1, 3) proxyLayout.addWidget(fromLabel, 3, 0) proxyLayout.addWidget(self.fromDateEdit, 3, 1, 1, 2) proxyLayout.addWidget(toLabel, 4, 0) proxyLayout.addWidget(self.toDateEdit, 4, 1, 1, 2) proxyGroupBox = QtGui.QGroupBox("Sorted/Filtered Model") proxyGroupBox.setLayout(proxyLayout) mainLayout = QtGui.QVBoxLayout() mainLayout.addWidget(sourceGroupBox) mainLayout.addWidget(proxyGroupBox) self.setLayout(mainLayout) self.setWindowTitle("Custom Sort/Filter Model") self.resize(500, 450) def setSourceModel(self, model): self.proxyModel.setSourceModel(model) self.sourceView.setModel(model) def textFilterChanged(self): syntax = QtCore.QRegExp.PatternSyntax( self.filterSyntaxComboBox.itemData( self.filterSyntaxComboBox.currentIndex())) caseSensitivity = ( self.filterCaseSensitivityCheckBox.isChecked() and QtCore.Qt.CaseSensitive or QtCore.Qt.CaseInsensitive) regExp = QtCore.QRegExp(self.filterPatternLineEdit.text(), caseSensitivity, syntax) self.proxyModel.setFilterRegExp(regExp) def dateFilterChanged(self): self.proxyModel.setFilterMinimumDate(self.fromDateEdit.date()) self.proxyModel.setFilterMaximumDate(self.toDateEdit.date()) def addMail(model, subject, sender, date): model.insertRow(0) model.setData(model.index(0, 0), subject) model.setData(model.index(0, 1), sender) model.setData(model.index(0, 2), date) def createMailModel(parent): model = QtGui.QStandardItemModel(0, 3, parent) model.setHeaderData(0, QtCore.Qt.Horizontal, "Subject") model.setHeaderData(1, QtCore.Qt.Horizontal, "Sender") model.setHeaderData(2, QtCore.Qt.Horizontal, "Date") addMail(model, "Happy New Year!", "Grace K. <grace@software-inc.com>", QtCore.QDateTime(QtCore.QDate(2006, 12, 31), QtCore.QTime(17, 3))) addMail(model, "Radically new concept", "Grace K. <grace@software-inc.com>", QtCore.QDateTime(QtCore.QDate(2006, 12, 22), QtCore.QTime(9, 44))) addMail(model, "Accounts", "pascale@nospam.com", QtCore.QDateTime(QtCore.QDate(2006, 12, 31), QtCore.QTime(12, 50))) addMail(model, "Expenses", "Joe Bloggs <joe@bloggs.com>", QtCore.QDateTime(QtCore.QDate(2006, 12, 25), QtCore.QTime(11, 39))) addMail(model, "Re: Expenses", "Andy <andy@nospam.com>", QtCore.QDateTime(QtCore.QDate(2007, 1, 2), QtCore.QTime(16, 5))) addMail(model, "Re: Accounts", "Joe Bloggs <joe@bloggs.com>", QtCore.QDateTime(QtCore.QDate(2007, 1, 3), QtCore.QTime(14, 18))) addMail(model, "Re: Accounts", "Andy <andy@nospam.com>", QtCore.QDateTime(QtCore.QDate(2007, 1, 3), QtCore.QTime(14, 26))) addMail(model, "Sports", "Linda Smith <linda.smith@nospam.com>", QtCore.QDateTime(QtCore.QDate(2007, 1, 5), QtCore.QTime(11, 33))) addMail(model, "AW: Sports", "Rolf Newschweinstein <rolfn@nospam.com>", QtCore.QDateTime(QtCore.QDate(2007, 1, 5), QtCore.QTime(12, 0))) addMail(model, "RE: Sports", "Petra Schmidt <petras@nospam.com>", QtCore.QDateTime(QtCore.QDate(2007, 1, 5), QtCore.QTime(12, 1))) return model if __name__ == "__main__": import sys app = QtGui.QApplication(sys.argv) window = Window() window.setSourceModel(createMailModel(window)) window.show() sys.exit(app.exec_())