主页 / 问题 / 1330610
Accepted
Konstantin
Asked:2022-09-21 17:17:25 +0800 CST2022-09-21 17:17:25 +0800 CST

过滤多个表 QSqlTableModel

  • 772

有две桌子category和equipment。
我正在尝试为他们制作一个过滤器，但事实证明equipment它可以在其中工作，
但category不能在...... =（

最小的例子：

import re
import sys
from PyQt5 import QtSql
from PyQt5.Qt import *

class Example(QMainWindow):
    def __init__(self):
        super().__init__()

        self.createConnection()
        self.fillTable()  # !!!
        self.createModel()
        self.initUI()

        self.centralWidget = QWidget()
        self.setCentralWidget(self.centralWidget)
        self.table_box = QComboBox()
        self.table_box.addItems(["Оборудование", "Категории"])
        self.table_box.activated[str].connect(self.table_select)
        search_str = QLineEdit()
        search_str.setPlaceholderText("Введите строку поиска")
        search_str.textChanged.connect(self.update_filter)

        layout = QVBoxLayout(self.centralWidget)
        layout.addWidget(self.view)
        layout.addWidget(self.table_box)
        layout.addWidget(search_str)

    def createConnection(self):
        self.db = QtSql.QSqlDatabase.addDatabase("QSQLITE")
        self.db.setDatabaseName("test_1318914.db")  # !!! .db
        if not self.db.open():
            print("Cannot establish a database connection")
            return False

    def fillTable(self):
        self.db.transaction()
        q = QtSql.QSqlQuery()
        #                             vvvvvvvv
        q.exec_("DROP TABLE IF EXISTS category;")
        q.exec_("CREATE TABLE category (id INT PRIMARY KEY, catname TEXT);")
        q.exec_("INSERT INTO category VALUES (1, 'Расходники');")
        q.exec_("INSERT INTO category VALUES (2, 'Носители');")

        #                             vvvv
        q.exec_("DROP TABLE IF EXISTS equipment;")
        q.exec_("CREATE TABLE equipment (Name TEXT, Quantity INT, Category INT);")
        q.exec_("INSERT INTO equipment VALUES ('Барабан для принтера', 8, 1);")
        q.exec_("INSERT INTO equipment VALUES ('Бумага для принтера', 3, 1);")
        q.exec_("INSERT INTO equipment VALUES ('Дискета', 10, 2);")
        self.db.commit()

    def createModel(self):
        self.model = QtSql.QSqlRelationalTableModel()
        self.model.setTable("equipment")
        self.model.select()

    def initUI(self):
        self.view = QTableView()
        self.view.setModel(self.model)
        self.view.setColumnWidth(0, 150)
        mode = QAbstractItemView.SingleSelection
        self.view.setSelectionMode(mode)


    def update_filter(self, text):
        if self.table_box.currentText() == 'Категории':
            s = re.sub("[\W_]+", "", text)
            filter_str = 'FROM category WHERE catname LIKE "%{}%"'.format(s)
            self.model.setFilter(filter_str)
        elif self.table_box.currentText() == 'Оборудование':
            s = re.sub("[\W_]+", "", text)
            filter_str = 'Name LIKE "%{}%"'.format(s)
            self.model.setFilter(filter_str)

    def table_select(self, text):
        if text == 'Категории':
            self.model.setTable("category")
            self.model.select()
        elif text == 'Оборудование':
            self.model.setTable("equipment")
            self.model.select()

    def closeEvent(self, event):
        if (self.db.open()):
            self.db.close()



if __name__ == '__main__':
    app = QApplication(sys.argv)
    w = Example()
    w.setWindowTitle("Search")
    w.resize(430, 250)
    w.show()
    sys.exit(app.exec_())

Действия:

  1. 我运行示例，出现一个视图窗口：

在此处输入图像描述

  1. 我输入QLineEdit例如“Ba”，他选择了正确的：

在此处输入图像描述

  1. 切换到“类别”：

在此处输入图像描述

  1. 当您尝试在“类别”表中搜索时，它不会搜索 =(

在此处输入图像描述

python

2 个回答

  1. Best Answer

    作为一种选择：

    import re
import sys
from PyQt5 import QtSql
from PyQt5.Qt import *


class Example(QMainWindow):
    def __init__(self):
        super().__init__()

        self.createConnection()
        self.fillTable()  # !!!
        self.createModel()
        self.initUI()
        
# +++ vvvv       
        self.model_category  = self.model.relationModel(2)   # QtSql.QSqlTableModel
        #print(f'm_cat tableName == {self.model_category.tableName()}')
        #print(f'm_cat rowCount  == {self.model_category.rowCount()}')
        self.model_category.setHeaderData(0, Qt.Horizontal, "ID")
        self.model_category.setHeaderData(1, Qt.Horizontal, "Категория")
# +++ ^^^^

        self.centralWidget = QWidget()
        self.setCentralWidget(self.centralWidget)
        self.table_box = QComboBox()
        self.table_box.addItems(["Оборудование", "Категории"])
        self.table_box.activated[str].connect(self.table_select)
        self.search_str = QLineEdit()                               # +++ self.
        self.search_str.setPlaceholderText("Введите строку поиска")
        self.search_str.textChanged.connect(self.update_filter)

        layout = QVBoxLayout(self.centralWidget)
        layout.addWidget(self.view)
        layout.addWidget(self.table_box)
        layout.addWidget(self.search_str)

    def createConnection(self):
        self.db = QtSql.QSqlDatabase.addDatabase("QSQLITE")
        self.db.setDatabaseName("test_1318914.db")  # !!! .db
        if not self.db.open():
            print("Cannot establish a database connection")
            return False

    def fillTable(self):
        self.db.transaction()
        q = QtSql.QSqlQuery()
        
        q.exec_("DROP TABLE IF EXISTS category;")
        q.exec_("CREATE TABLE category (id INT PRIMARY KEY, catname TEXT);")
        q.exec_("INSERT INTO category VALUES (1, 'Расходники');")
        q.exec_("INSERT INTO category VALUES (2, 'Носители');")

        q.exec_("DROP TABLE IF EXISTS equipment;")
        q.exec_("CREATE TABLE equipment (Name TEXT, Quantity INT, Category INT);")
        q.exec_("INSERT INTO equipment VALUES ('Барабан для принтера', 8, 1);")
        q.exec_("INSERT INTO equipment VALUES ('Бумага для принтера', 3, 1);")
        q.exec_("INSERT INTO equipment VALUES ('Дискета', 10, 2);")
        self.db.commit()

    def createModel(self):
        self.model = QtSql.QSqlRelationalTableModel()
        self.model.setTable("equipment")
# +++ vvvv
        self.model.setHeaderData(0, Qt.Horizontal, "Название")
        self.model.setHeaderData(1, Qt.Horizontal, "Кол-во")
        self.model.setHeaderData(2, Qt.Horizontal, "Категория")
        self.set_relation()
# +++ ^^^^       
        self.model.select()
        
# +++ vvvv
    def set_relation(self):
        self.model.setRelation(2, QtSql.QSqlRelation(
            "category",
            "id",
            "catname"
        ))        
# +++ ^^^^

    def initUI(self):
        self.view = QTableView()
        self.view.setModel(self.model)
        self.view.setColumnWidth(0, 150)
        mode = QAbstractItemView.SingleSelection
        self.view.setSelectionMode(mode)

    def update_filter(self, text):
        if self.table_box.currentText() == 'Категории':
            s = re.sub("[\W_]+", "", text)
#            filter_str = 'FROM category WHERE catname LIKE "%{}%"'.format(s)
            filter_str = 'catname LIKE "%{}%"'.format(s)
            self.model_category.setFilter(filter_str)                  # +++
        elif self.table_box.currentText() == 'Оборудование':
            s = re.sub("[\W_]+", "", text)
            filter_str = 'Name LIKE "%{}%"'.format(s)
            self.model.setFilter(filter_str)

    def table_select(self, text):
        self.search_str.clear()                        # +++
        if text == 'Категории':
#            self.model.setTable("category")
#            self.model.select()
            self.view.setModel(self.model_category)     # +++   
        elif text == 'Оборудование':
#            self.model.setTable("equipment")
#            self.model.select()
            self.view.setModel(self.model)              # +++

    def closeEvent(self, event):
        if (self.db.open()):
            self.db.close()


if __name__ == '__main__':
    app = QApplication(sys.argv)
    w = Example()
    w.setWindowTitle("Search")
    w.resize(430, 250)
    w.show()
    sys.exit(app.exec_())

    在此处输入图像描述

    在此处输入图像描述

    • 1

  2. https://doc.qt.io/qtforpython/PySide6/QtSql/QSqlTableModel.html?highlight=setfilter#PySide6.QtSql.PySide6.QtSql.QSqlTableModel.setFilter

    根据文档，总的来说，我做得有点过头了：过滤器是没有关键字的WHERE句子。简单来说，不需要对整个SQL查询进行fence，我们只要引用当前表的字段就可以了。SQLWHERE

    例如，而不是：

    filter_str = 'FROM category WHERE catname LIKE "%{}%"'.format(s)

    我们写：

    filter_str = 'catname LIKE "%{}%"'.format(s)
    • 0

相关问题