# -*- coding:utf-8 -*-

import pyperclip
from PyQt5.QtWidgets import QApplication, QWidget
from PyQt5.QtGui import QStandardItemModel, QStandardItem
from PyQt5.QtCore import Qt
from PyQt5 import QtCore, QtWidgets


# 復(fù)制選擇表格數(shù)據(jù)
def selected_tb_text(table_view):
  try:
    indexes = table_view.selectedIndexes() # 獲取表格對(duì)象中被選中的數(shù)據(jù)索引列表
    indexes_dict = {}
    for index in indexes: # 遍歷每個(gè)單元格
      row, column = index.row(), index.column() # 獲取單元格的行號(hào)，列號(hào)
      if row in indexes_dict.keys():
        indexes_dict[row].append(column)
      else:
        indexes_dict[row] = [column]

    # 將數(shù)據(jù)表數(shù)據(jù)用制表符(\t)和換行符(\n)連接，使其可以復(fù)制到excel文件中
    text = ''
    for row, columns in indexes_dict.items():
      row_data = ''
      for column in columns:
        data = table_view.model().item(row, column).text()
        if row_data:
          row_data = row_data + '\t' + data
        else:
          row_data = data

      if text:
        text = text + '\n' + row_data
      else:
        text = row_data
    return text
  except BaseException as e:
    print(e)
    return ''


class Ui_Form(object): # UI類(lèi)
  def setupUi(self, Form):
    Form.setObjectName("Form")
    self.tableView = QtWidgets.QTableView(Form)
    self.tableView.setEnabled(True)
    self.tableView.setGeometry(QtCore.QRect(5, 5, 400, 200))
    self.tableView.setObjectName("tableView")
    QtCore.QMetaObject.connectSlotsByName(Form)


# 邏輯類(lèi)
class StartRun(QWidget, Ui_Form):
  def __init__(self):
    super().__init__()
    self.init_ui() # 實(shí)例化窗體
    self.show()

  def init_ui(self):
    self.setupUi(self) # 實(shí)例化控件
    self.update_table_view()

  def update_table_view(self):
    data = [
      (1, '張三', 18),
      (2, '李四', 29),
      (3, '王五', 25),
      (4, '趙六', 26),
    ]
    columns = ['id', 'name', 'age']
    model = QStandardItemModel(len(data), len(columns)) # 設(shè)置數(shù)據(jù)層次結(jié)構(gòu)，rows行cols列
    model.setHorizontalHeaderLabels([str(i) for i in columns]) # 設(shè)置列名
    for row in range(len(data)):
      for column in range(len(data[row])):
        item = QStandardItem(str(data[row][column]))
        model.setItem(row, column, item) # 設(shè)置每個(gè)位置的文本值
    self.tableView.setModel(model) # 實(shí)例化表格視圖，設(shè)置模型為自定義的模型

  def keyPressEvent(self, event):   # 重寫(xiě)鍵盤(pán)監(jiān)聽(tīng)事件
    # 監(jiān)聽(tīng) CTRL+C 組合鍵，實(shí)現(xiàn)復(fù)制數(shù)據(jù)到粘貼板
    if (event.key() == Qt.Key_C) and QApplication.keyboardModifiers() == Qt.ControlModifier:
      text = selected_tb_text(self.tableView) # 獲取當(dāng)前表格選中的數(shù)據(jù)
      if text:
        pyperclip.copy(text) # 復(fù)制數(shù)據(jù)到粘貼板


if __name__ == '__main__':
  import sys
  app = QApplication(sys.argv)
  S = StartRun()
  sys.exit(app.exec_())