PyQt: QTableView + QSqlTableModel -将所有选定的行或列复制并粘贴到记事本或QSqlTableModel中

from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QAbstractItemView, QMenu, QAction, QFileDialog
from PyQt5.QtGui import QClipboard
from PyQt5.QtCore import Qt, QModelIndex
from PyQt5.QtSql import QSqlDatabase, QSqlTableModel
import sys

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

        self.table_view = QTableView(self)
        self.setCentralWidget(self.table_view)

        # 创建数据库连接
        self.db = QSqlDatabase.addDatabase("QSQLITE")
        self.db.setDatabaseName("database.db")
        self.db.open()

        # 创建模型
        self.model = QSqlTableModel(self, self.db)
        self.model.setTable("table_name")
        self.model.select()

        # 设置表格视图
        self.table_view.setModel(self.model)
        self.table_view.setSelectionMode(QAbstractItemView.ExtendedSelection)
        self.table_view.setContextMenuPolicy(Qt.CustomContextMenu)
        self.table_view.customContextMenuRequested.connect(self.show_context_menu)

    def show_context_menu(self, pos):
        menu = QMenu(self.table_view)
        copy_action = QAction("复制", self)
        copy_action.triggered.connect(self.copy_selected_rows_or_columns)
        menu.addAction(copy_action)
        menu.exec_(self.table_view.viewport().mapToGlobal(pos))

    def copy_selected_rows_or_columns(self):
        selection_model = self.table_view.selectionModel()
        indexes = selection_model.selectedIndexes()

        if len(indexes) == 0:
            return

        # 获取选定的行或列
        rows = set()
        columns = set()
        for index in indexes:
            rows.add(index.row())
            columns.add(index.column())

        # 复制选定的行或列到剪贴板
        clipboard = QApplication.clipboard()
        mime_data = []
        for row in rows:
            row_data = []
            for column in columns:
                index = self.model.index(row, column)
                row_data.append(index.data())
            mime_data.append("\t".join(row_data))
        clipboard.setText("\n".join(mime_data))

        # 粘贴到记事本或QSqlTableModel中
        file_dialog = QFileDialog(self)
        file_dialog.setAcceptMode(QFileDialog.AcceptSave)
        file_dialog.setDefaultSuffix("txt")
        file_dialog.setOption(QFileDialog.DontUseNativeDialog, True)
        file_dialog.setFileMode(QFileDialog.AnyFile)
        if file_dialog.exec_() == QFileDialog.Accepted:
            file_path = file_dialog.selectedFiles()[0]
            with open(file_path, "w") as file:
                file.write("\n".join(mime_data))

            # 或者将数据插入到QSqlTableModel中
            for row_data in mime_data:
                self.model.insertRows(self.model.rowCount(), 1)
                for column, data in enumerate(row_data.split("\t")):
                    index = self.model.index(self.model.rowCount() - 1, column)
                    self.model.setData(index, data)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())