PyQt5-Qtablewidget
创建表格
from PyQt5.QtWidgets import QTableView,QStyledItemDelegate, QComboBox,QTextEdit from PyQt5.QtCore import QAbstractTableModel,Qt, QModelIndex,QEvent,QRect,QSize,QVariant from PyQt5.QtGui import QFontMetrics class TaskTable(QTableWidget): def __init__(self,row_count,col_count): super().__init__(row_count,col_count) self.setHorizontalHeaderLabels(['Level', 'To do List','Progress','Sub Task','Duration','Comment','Category']) # self.verticalHeader().setSectionResizeMode(QHeaderView.ResizeToContents) self.horizontalHeader().setSectionResizeMode(QHeaderView.Interactive) self.setColumnWidth(0,100) self.setColumnWidth(1,340) self.setColumnWidth(2,100) self.setColumnWidth(3,400) self.setColumnWidth(4,100) self.setColumnWidth(5,200) self.setColumnWidth(6,100) self.verticalHeader().hide() self.horizontalHeader().setStyleSheet("QHeaderView::section { background-color: lightblue; }") for row in range(row_count): item = QTableWidgetItem() item.setFlags(item.flags()|Qt.ItemIsUserCheckable) item.setCheckState(Qt.Unchecked) self.setItem(row, 3, item) items_in_progress = ['Not start', 'In progress', 'Done','Pending others','Done but keep'] items_in_cagetory = ['MOMA', 'SIPG', 'APILearning','Others','WorkRule'] combox_progress = self.addcomBox(items_in_progress,0) self.setCellWidget(row, 2, combox_progress) combox_progress = self.addcomBox(items_in_cagetory,3) self.setCellWidget(row, 6, combox_progress) def addcomBox(self,items,default_index): comboBox1 = QComboBox() comboBox1.addItems(items) comboBox1.setCurrentIndex(default_index) return comboBox1 # 设置默认值，例如 'Option A2' def getvalue(self,row,columnindex): cell = self.cellWidget(row, columnindex) if columnindex in [2, 6] else self.item(row, columnindex) if cell and isinstance(cell, QComboBox): return cell.currentText() elif cell: value = cell.text() try: return int(value) except ValueError: return value else: return ''
显示窗口
import sys from PyQt5.QtWidgets import * # QApplication,QMainWindow,QWidget,QLabel,QVBoxLayout from PyQt5.QtGui import QIcon from PyQt5.QtCore import Qt from GUI.task import TaskTable from GUI.date import DateLabel from GUI.control import controls from Back.db import initDB iconpath = 'refer\MainLogo.png' class MainWindow(QMainWindow): def __init__(self): try: super().__init__() self.initUI() except Exception as e: print(e) def initUI(self): try: self.setWindowTitle("JXY_Note") self.setWindowIcon(QIcon(iconpath)) central_widget = QWidget(self) # self.central_widget.setSizePolicy(QSizePolicy.Expanding,QSizePolicy.Expanding) self.setCentralWidget(central_widget) # print(self.centralWidget().size()) layout = QVBoxLayout(central_widget) self.contrals = controls() self.table_date = DateLabel() self.taskdate = self.table_date.text() self.data = self.init_data(self.taskdate) self.table_view = TaskTable(self.data) layout.addWidget(self.table_date) layout.addWidget(self.table_view) print("add table") self.save_button = QPushButton('Save') self.save_button.clicked.connect(self.save_data) self.retrieve_button = QPushButton('Retrieve') self.retrieve_button.clicked.connect(self.retrieve_data) self.clear_button = QPushButton('Clear') self.clear_button.clicked.connect(self.clear_data) layout.addWidget(self.save_button) layout.addWidget(self.retrieve_button) layout.addWidget(self.clear_button) self.showMaximized() except Exception as e: print(e) def save_data(self): data = [] for row in range(self.table_view.rowCount()): row_data = [] if self.table_view.item(row,1): # state = self.table_view.item(row,3).checkState() # checkStatus = "1" if state == Qt.Checked else "0" checkStatus = "0" taskdate = self.table_date.text() for col in range(self.table_view.columnCount()): cell = self.table_view.getvalue(row,col) row_data.append(cell) row_data.insert(4,checkStatus) row_data.insert(9,taskdate) data.append(tuple(row_data)) print(data) self.contrals.save_data_task(data) QMessageBox.information(self, "Success","Data saved successfully!") def retrieve_data(self): taskdate = self.table_date.text() print(taskdate) rows = self.contrals.retrieve_data_task(taskdate) print(rows) self.initUI() for row_idx, (ids,level,to_do,progress,subtask,subtask_status,duration,comment,category,date,dependance) in enumerate(rows): self.table_view.setItem(row_idx,0,QTableWidgetItem(str(level))) self.table_view.setItem(row_idx,1,QTableWidgetItem(str(to_do))) progress_cellcombox = self.table_view.cellWidget(row_idx,2) progress_index = progress_cellcombox.findText(str(progress)) if progress_index >= 0: progress_cellcombox.setCurrentIndex(progress_index) else: progress_cellcombox.setCurrentIndex(0) check = QTableWidgetItem(subtask) if subtask_status == '0': check.setCheckState(Qt.Unchecked) self.table_view.setItem(row_idx, 3, check) if subtask_status == '1': check.setCheckState(Qt.Checked) self.table_view.setItem(row_idx, 3, check) self.table_view.setItem(row_idx,4,QTableWidgetItem(str(duration))) self.table_view.setItem(row_idx,5,QTableWidgetItem(str(comment))) category_combox = self.table_view.cellWidget(row_idx,6) category_index = category_combox.findText(str(category)) if category_index >= 0: category_combox.setCurrentIndex(category_index) else: category_combox.setCurrentIndex(0) def init_data(self,taskdate): rows = self.contrals.retrieve_data_task(taskdate) data = [] for row_idx, (ids,level,to_do,progress,subtask,subtask_status,duration,comment,category,date,dependance) in enumerate(rows): row = [level,to_do,progress,subtask,duration,comment,category] data.append(row) return data def clear_data(self): self.db = initDB() self.db.delete_data("tasks") self.retrieve_data() if __name__ == '__main__': app = QApplication(sys.argv) ex = MainWindow() ex.show() sys.exit(app.exec_())
代码不全，仅做练习使用
