复杂SQL查询的可编辑QTableView

Posted 2023-02-19

【中文标题】复杂SQL查询的可编辑QTableView

【英文标题】：Editable QTableView of complex SQL query

【发布时间】：2018-09-19 23:59:47

【问题描述】：

如何制作可编辑的 QTableView，显示来自复杂 SQLite 查询的数据？

我需要用来自多个 SQLite 表的数据填充 QTableView。这需要用户可编辑。

由于查询有点复杂（包括 JOIN 和 CASE WHEN 等），我通过 QSqlTableModel 和 QSqlQuery 执行此操作。然而，有人告诉我，这不是 QSqlTableModels 应该使用的方式。那么，有人可以告诉我如何通过正确的方式获得像这里显示的结果吗？

另外，虽然我的 QTableViews 是可编辑的，但结果似乎并没有存储在 SQLite 数据库中。 （当我注释掉fill_tables时，重新启动GUI后仍然得到原始结果。将EditStrategy更改为OnFieldChange没有帮助。）是因为我处理QSqlTableModel错误吗？

#!/usr/bin/python3

from PyQt5.QtSql import (QSqlDatabase, QSqlQuery, QSqlTableModel, 
                         QSqlRelationalTableModel, QSqlRelation)
from PyQt5.QtWidgets import QTableView, QApplication
from PyQt5.Qt import QSortFilterProxyModel
import sys

db_file = "test.db"

def create_connection(db_file):
    db = QSqlDatabase.addDatabase("QSQLITE")
    db.setDatabaseName(db_file)
    if not db.open():
        print("Cannot establish a database connection to !".format(db_file))
        return False
    return db


def fill_tables():
    q = QSqlQuery()
    q.exec_("DROP TABLE IF EXISTS Manufacturers;")
    q.exec_("CREATE TABLE Manufacturers (Company TEXT, Country TEXT);")
    q.exec_("INSERT INTO Manufacturers VALUES ('VW', 'Germany');")
    q.exec_("INSERT INTO Manufacturers VALUES ('Honda' , 'Japan');")

    q.exec_("DROP TABLE IF EXISTS Cars;")
    q.exec_("CREATE TABLE Cars (Company TEXT, Model TEXT, Year INT);")
    q.exec_("INSERT INTO Cars VALUES ('Honda', 'Civic', 2009);")
    q.exec_("INSERT INTO Cars VALUES ('VW', 'Golf', 2013);")
    q.exec_("INSERT INTO Cars VALUES ('VW', 'Polo', 1999);")


class CarTable(QTableView):
    def __init__(self):
        super().__init__()
        self.init_UI()
        self.create_model()

    def create_model(self):
        query = """
        SELECT (comp.company || " " || cars.model) as Car,
                comp.Country,
                (CASE WHEN cars.Year > 2000 THEN 'yes' ELSE 'no' END) as this_century
        from manufacturers comp left join cars
            on comp.company = cars.company
        """
        raw_model = QSqlTableModel()
        q = QSqlQuery()
        q.exec_(query)
        self.check_error(q)
        raw_model.setQuery(q)

        self.model = QSortFilterProxyModel()
        self.model.setSourceModel(raw_model)
        self.setModel(self.model)

        # filtering:
        self.model.setFilterKeyColumn(0)
        self.model.setFilterFixedString('VW')

    def init_UI(self):
        self.resize(500,300)

    def check_error(self, q):
        lasterr = q.lastError()
        if lasterr.isValid():
            print(lasterr.text())
            exit(1)


def main():
    mydb = create_connection(db_file)
    if not mydb:
        sys.exit(-1)
    fill_tables()
    app = QApplication(sys.argv)
    ex = CarTable()
    ex.show()
    result = app.exec_()

    if (mydb.open()):
        mydb.close()

    sys.exit(result)


if __name__ == '__main__':
    main()

我已尝试改用 QSqlRelationalTableModel，但我无法完成同样复杂的查询，而且它也不会保存更改，就像上面的代码一样。这是我的尝试：

     def create_model_alternative(self):
        self.model = QSqlRelationalTableModel()
        self.model.setTable("Cars")
        self.model.setRelation(0, QSqlRelation("Manufacturers", "Company",
                                               "Company, Country"))  
        self.setModel(self.model)
        self.model.select()

        # filtering:
        self.model.setFilter("cars.Company = 'VW'")

---

回答传入的问题：

可编辑性：

在这个例子中，唯一明确需要可编辑的列（以这种方式使更改到达数据库）是国家列（并且那里的更改应该影响所有其他行共享相同的内容；例如，如果您将任一大众汽车的“德国”更改为“法国”，则两者都应将“法国”列为国家）。

如果您知道如何让第一个也可编辑，以便更新数据库中的相应列，那将是非常漂亮的，但这不是必需的。 （在我的真实表格中，我将此类“列连接”用于不可编辑的字段。） 在这种特定情况下，我希望将“VW Polo”更改为“Marco Polo”也会将“VW Golf”更新为“Marco Golf”，因为列连接中使用的列是制造商.公司而不是汽车。公司。 （实际上，可能会使用cars.company 进行连接，在这种情况下，'VW Golf' 将保持不变。但让我们假设上面给出的查询。）

第三列是计算统计结果的示例，这些通常仅供阅读（编辑它们没有意义）。

列顺序：

我非常感谢能够选择列的显示顺序，即使跨连接表（就像我可以使用查询一样）。

【问题讨论】：

我要问你很多问题以正确理解你的要求，推进解决方案将非常特别，我认为几乎不可能概括它并使其可移植。首先我看到第一列是 Cars 表的第一列和第二列的并集，用空格分隔，所以你想在编辑一个时，更改相应列中的数据吗？

第二列比较好理解，第三列是涉及一年的运算结果只会被读取，对吗？

@eyllanesc 我更新了我的帖子来回答你的问题。感谢您尝试解决此问题。

最优方案是使用QSqlQueryModel，我只会编辑，可以吗？

最后一个问题，如果您更改了 VW Gold 行中的 Country，因此也更改了 VW Polo 中的 Country，这是您所期望的吗？

【参考方案1】：

QSqlTableModel是继承自QSqlQueryModel的类，所以可以说QSqlTableModel是专门编辑表格的QSqlQueryModel，所以可以限制或超大。

对于这种特殊情况，我建议使用QSqlQueryModel 可编辑，为此我进行了以下更改：

我已为第二列启用标志Qt.ItemIsEditable。

我已经覆盖了setData() 方法来更新制造商表。

我添加了一个代表公司的列，这将被隐藏，但它有助于获取之前更改中必须更改的行。

我已经实现了setFilter()方法来制作过滤器。

---

import sys

from PyQt5.QtCore import Qt
from PyQt5.QtSql import QSqlDatabase, QSqlQuery, QSqlQueryModel
from PyQt5.QtWidgets import QApplication, QTableView

db_file = "test.db"


def create_connection(file_path):
    db = QSqlDatabase.addDatabase("QSQLITE")
    db.setDatabaseName(file_path)
    if not db.open():
        print("Cannot establish a database connection to !".format(file_path))
        return False
    return True


def fill_tables():
    q = QSqlQuery()
    q.exec_("DROP TABLE IF EXISTS Manufacturers;")
    q.exec_("CREATE TABLE Manufacturers (Company TEXT, Country TEXT);")
    q.exec_("INSERT INTO Manufacturers VALUES ('VW', 'Germany');")
    q.exec_("INSERT INTO Manufacturers VALUES ('Honda' , 'Japan');")

    q.exec_("DROP TABLE IF EXISTS Cars;")
    q.exec_("CREATE TABLE Cars (Company TEXT, Model TEXT, Year INT);")
    q.exec_("INSERT INTO Cars VALUES ('Honda', 'Civic', 2009);")
    q.exec_("INSERT INTO Cars VALUES ('VW', 'Golf', 2013);")
    q.exec_("INSERT INTO Cars VALUES ('VW', 'Polo', 1999);")


class SqlQueryModel(QSqlQueryModel):
    def flags(self, index):
        fl = QSqlQueryModel.flags(self, index)
        if index.column() == 1:
            fl |= Qt.ItemIsEditable
        return fl

    def setData(self, index, value, role=Qt.EditRole):
        if index.column() == 1:
            company = self.index(index.row(), 2).data()
            q = QSqlQuery("UPDATE Manufacturers SET Country = '' WHERE Company =  ''".format(value, company))
            result = q.exec_()
            if result:
                self.query().exec_()
            else:
                print(self.query().lastError().text())
            return result
        return QSqlQueryModel.setData(self, index, value, role)

    def setFilter(self, filter):
        text = (self.query().lastQuery() + " WHERE " + filter)
        self.setQuery(text)


query = '''
        SELECT (comp.company || " " || cars.model) as Car,
                comp.Country,
                cars.company,
                (CASE WHEN cars.Year > 2000 THEN 'yes' ELSE 'no' END) as this_century
        from manufacturers comp left join cars
            on comp.company = cars.company
        '''

if __name__ == '__main__':
    app = QApplication(sys.argv)
    if not create_connection(db_file):
        sys.exit(-1)

    fill_tables()

    view = QTableView()

    model = SqlQueryModel()
    q = QSqlQuery(query)
    model.setQuery(q)
    model.setFilter("cars.Company = 'VW'")
    view.setModel(model)
    view.hideColumn(2)
    view.show()
    sys.exit(app.exec_())

【讨论】：

非常感谢，这很好用！我喜欢只使必要的列特别可编辑的想法。我根据您的代码制作了一个通用版本，我将其作为单独的答案发布，因为它可能对其他有类似问题的人有所帮助。但是如果没有你的帮助，我是做不到的（或者至少不会很快），所以你得到了赏金。 :)

@CodingCat 如果我不了解所有具体要求，概括是非常困难的，所以我只是尽可能具体地进行，然后我希望您的反馈可以微调它，但是如果您已经完成了看起来很棒。我在等待赏金：D

我必须再等 4 个小时才能获得赏金。 :D 是的，越具体越好，因为通常更容易看到正在发生的事情。

如何在这个模型中实现 model.setEditStrategy(QSqlTableModel.OnManualSubmit) ？ QSqlQueryModel 似乎没有这个（不足为奇，因为它不是可编辑的）。你能帮忙吗？ （或者我应该提出一个新问题？）

我在这里发了一个后续问题：***.com/questions/52166429/…

【参考方案2】：

基于@eyllanesc 的出色特定解决方案，我制作了一个通用版本的 QSqlQueryModel，可以在其中指定哪些列应该是可编辑的。它可能需要针对其他人的查询进行调整，但我希望它对那些在类似问题中苦苦挣扎的人有所帮助：

import sys

from PyQt5.QtCore import Qt
from PyQt5.QtSql import QSqlDatabase, QSqlQuery, QSqlQueryModel
from PyQt5.QtWidgets import QApplication, QTableView

db_file = "test.db"


def create_connection(file_path):
    db = QSqlDatabase.addDatabase("QSQLITE")
    db.setDatabaseName(file_path)
    if not db.open():
        print("Cannot establish a database connection to !".format(file_path))
        return False
    return True


def fill_tables():
    q = QSqlQuery()
    q.exec_("DROP TABLE IF EXISTS Manufacturers;")
    q.exec_("CREATE TABLE Manufacturers (Company TEXT, Country TEXT);")
    q.exec_("INSERT INTO Manufacturers VALUES ('VW', 'Germany');")
    q.exec_("INSERT INTO Manufacturers VALUES ('Honda' , 'Japan');")

    q.exec_("DROP TABLE IF EXISTS Cars;")
    q.exec_("CREATE TABLE Cars (Company TEXT, Model TEXT, Year INT);")
    q.exec_("INSERT INTO Cars VALUES ('Honda', 'Civic', 2009);")
    q.exec_("INSERT INTO Cars VALUES ('VW', 'Golf', 2013);")
    q.exec_("INSERT INTO Cars VALUES ('VW', 'Polo', 1999);")


class SqlQueryModel_editable(QSqlQueryModel):
    """a subclass of QSqlQueryModel where individual columns can be defined as editable
    """
    def __init__(self, editables):
        """editables should be a dict of format: 
        INT editable_column_nr : (STR update query to be performed when changes are made on this column
                                   INT model's column number for the filter-column (used in the where-clause),
                                   ) 
        """
        super().__init__()
        self.editables = editables

    def flags(self, index):
        fl = QSqlQueryModel.flags(self, index)
        if index.column() in self.editables:
            fl |= Qt.ItemIsEditable
        return fl

    def setData(self, index, value, role=Qt.EditRole):
        if role == Qt.EditRole:
            mycolumn = index.column()
            if mycolumn in self.editables:
                (query, filter_col) = self.editables[mycolumn]
                filter_value = self.index(index.row(), filter_col).data()
                q = QSqlQuery(query.format(value, filter_value))
                result = q.exec_()
                if result:
                    self.query().exec_()
                else:
                    print(self.query().lastError().text())
                return result
        return QSqlQueryModel.setData(self, index, value, role)

    def setFilter(self, myfilter):
        text = (self.query().lastQuery() + " WHERE " + myfilter)
        self.setQuery(text)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    if not create_connection(db_file):
        sys.exit(-1)

    fill_tables()

    view = QTableView()

    editables = 1 : ("UPDATE Manufacturers SET Country = '' WHERE Company = ''", 2)
    model = SqlQueryModel_editable(editables)
    query = '''
        SELECT (comp.company || " " || cars.model) as Car,
                comp.Country,
                cars.company,
                (CASE WHEN cars.Year > 2000 THEN 'yes' ELSE 'no' END) as this_century
        from manufacturers comp left join cars
            on comp.company = cars.company
        '''
    q = QSqlQuery(query)
    model.setQuery(q)
    model.setFilter("cars.Company = 'VW'")
    view.setModel(model)
    view.hideColumn(2)
    view.show()
    sys.exit(app.exec_())

要使连接的列可编辑，需要更多的工作和不同的可编辑格式，但这应该适用于任何不包含连接/计算/聚合数据的列（如本例中的“国家”）。

【讨论】：

好的，这就是你所说的泛化，我认为它是正确的：D