Peter_Torbenn | 2021-02-11 00:46:33 UTC | #1
I want to create a filter (search bar) for a table that receives data from Sqlite, I found something with QSortFilterProxyModel but only in C ++ is it possible to create the same function in pyqt5? if possible how it is done? all the tutorials I find are very empty and confusing I'm new to PyQt and would like to learn more
martin | 2021-02-11 10:21:44 UTC | #2
Hi @Peter_Torbenn welcome to the forum.
Here is a working example using QSortFilterProxyModel to filter a table by text input using a QLineEdit. You can adjust which columns are used to filter on, and how the filter works (plain text vs. regex, etc.)
This is using a custom model, based on a Python list but the principle is the same for the standard Qt models.
import sys
from PyQt5.QtWidgets import QApplication, QWidget, QTableView, QMainWindow, QVBoxLayout, QLineEdit
from PyQt5.QtCore import Qt, QSortFilterProxyModel, QAbstractTableModel
class TableModel(QAbstractTableModel):
def __init__(self, data):
super().__init__()
self._data = data
def data(self, index, role):
if role == Qt.DisplayRole:
# See below for the nested-list data structure.
# .row() indexes into the outer list,
# .column() indexes into the sub-list
return self._data[index.row()][index.column()]
def rowCount(self, index):
# The length of the outer list.
return len(self._data)
def columnCount(self, index):
# The following takes the first sub-list, and returns
# the length (only works if all rows are an equal length)
return len(self._data[0])
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.table = QTableView()
data = [
[4, 9, 2],
[1, "hello", 0],
[3, 5, 0],
[3, 3, "what"],
["this", 8, 9],
]
self.model = TableModel(data)
self.proxy_model = QSortFilterProxyModel()
self.proxy_model.setFilterKeyColumn(-1) # Search all columns.
self.proxy_model.setSourceModel(self.model)
self.proxy_model.sort(0, Qt.AscendingOrder)
self.table.setModel(self.proxy_model)
self.searchbar = QLineEdit()
# You can choose the type of search by connecting to a different slot here.
# see https://doc.qt.io/qt-5/qsortfilterproxymodel.html#public-slots
self.searchbar.textChanged.connect(self.proxy_model.setFilterFixedString)
layout = QVBoxLayout()
layout.addWidget(self.searchbar)
layout.addWidget(self.table)
container = QWidget()
container.setLayout(layout)
self.setCentralWidget(container)
app = QApplication(sys.argv)
window = MainWindow()
window.show()
app.exec_()
