自定义QSqlTableModel实现QTableView拖拽CSV导入失效求助

问题场景 #

在PySide6环境中，使用QTableView结合子类化的QSqlTableModel展示SQLite数据表，尝试通过拖拽CSV文件插入新记录。测试发现canDropMimeData和supportedDropActions能正常调用，但dropMimeData始终不触发，拖拽时界面显示文件不被接受；改用QStandardItemModel则功能正常。

原因分析 #

QSqlTableModel的默认flags()方法未启用Qt.ItemIsDropEnabled标志，导致模型不接受放置操作。而QStandardItemModel默认开启了该标志，所以能正常响应拖拽。此外，原代码中未调用模型的select()方法，导致初始时表格无数据显示，也会影响拖拽交互的视觉反馈。

解决方案 #

• 重写模型的flags()方法：为根索引（或目标索引）添加Qt.ItemIsDropEnabled标志，允许放置操作。
• 设置QTableView的dropMode：根据需求设置合适的放置模式，比如DropOnWidget允许将文件拖拽到表格任意区域，DropOnItem则限制在单元格上。
• 调用模型的select()方法：加载数据库表数据，确保表格正常显示。

修改后的完整代码 #

import sys

from qtpy.QtWidgets import QApplication, QTableView
from qtpy.QtCore import QModelIndex, QMimeData, Qt
from qtpy.QtSql import QSqlDatabase, QSqlTableModel, QSqlQuery
from pandas import read_csv

def create_table():
    # 创建测试表
    _query_str = """CREATE TABLE MyTable ( 
    ID     INTEGER PRIMARY KEY AUTOINCREMENT,
    Field1 INTEGER,
    Field2 TEXT);"""

    query = QSqlQuery(db=db, query=_query_str)
    query.exec_()

class MyTableModel(QSqlTableModel):
    def __init__(self, table_name, db):
        super().__init__(db=db)
        self.setTable(table_name)
        self.select()  # 加载表数据

    def flags(self, index: QModelIndex) -> Qt.ItemFlags:
        # 为所有索引添加DropEnabled标志，根索引单独处理确保表格区域可拖放
        flags = super().flags(index)
        if not index.isValid():
            flags |= Qt.ItemIsDropEnabled
        return flags

    def canDropMimeData(self, data: QMimeData, action: Qt.DropAction, row: int, column: int, parent: QModelIndex) -> bool:
        # 仅接受包含文件URL的MIME数据
        return data.hasUrls() and len(data.urls()) > 0

    def supportedDropActions(self) -> Qt.DropAction:
        return Qt.CopyAction

    def dropMimeData(self, data: QMimeData, action: Qt.DropAction, row: int, column: int, parent: QModelIndex) -> bool:
        if action != Qt.CopyAction:
            return False
        if not data.hasUrls():
            return False
        
        csv_filename = data.urls()[0].toLocalFile()
        try:
            df = read_csv(csv_filename, delimiter=',', header=0)
        except Exception as e:
            print(f"读取CSV失败: {e}")
            return False

        self.database().transaction()  # 开启事务提升性能
        for _, row_data in df.iterrows():
            record = self.record()
            record.remove(0)  # 移除自增ID字段
            record.setValue('Field1', row_data['Field1'])
            record.setValue('Field2', row_data['Field2'])
            if not self.insertRecord(-1, record):
                self.database().rollback()
                print(f"插入记录失败: {self.lastError().text()}")
                return False
        self.database().commit()
        self.select()  # 刷新模型数据
        return True

if __name__ == '__main__':
    # 创建内存SQLite数据库
    db = QSqlDatabase.addDatabase("QSQLITE")
    db.setDatabaseName(":memory:")
    if not db.open():
        raise RuntimeError(f"数据库打开失败: {db.lastError().text()}")
    create_table()
    
    app = QApplication(sys.argv)
    table = QTableView()
    table.setAcceptDrops(True)
    table.setDropMode(QTableView.DropOnWidget)  # 设置为允许拖拽到表格任意区域
    
    model = MyTableModel('MyTable', db)
    table.setModel(model)
    
    table.show()
    sys.exit(app.exec_())

关键修改点说明 #

• 重写flags()：针对无效索引（表格空白区域）添加ItemIsDropEnabled，确保拖拽到表格任意位置都能触发放置。
• 事务处理：批量插入时使用数据库事务，减少IO开销并保证数据一致性。
• 增强错误处理：添加CSV读取和记录插入的异常捕获，便于调试。
• 设置dropMode：DropOnWidget让整个表格区域都接受拖拽，提升用户体验。

内容的提问来源于stack exchange，提问作者Buzz