1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
| import sys
import csv
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt
from PySide6.QtWidgets import (QApplication,
QWidget, QTableView, QVBoxLayout)
from PySide6.QtTest import QAbstractItemModelTester
class CsvModel(QAbstractTableModel):
def __init__(self, source, parent=None):
super().__init__(parent)
self.csv_data = []
with open(source) as csv_file:
reader = csv.reader(csv_file)
self.header = next(reader)
for row in reader:
self.csv_data.append(row)
def rowCount(self, parent=QModelIndex()):
return len(self.csv_data)
def columnCount(self, parent=QModelIndex()):
return 4
def data(self, index, role):
if role == Qt.ItemDataRole.DisplayRole:
return self.csv_data[index.row()][index.column()]
# Editable models implement setData() and flags()
def setData(self, index, value, role):
if role == Qt.ItemDataRole.EditRole:
if self.csv_data[index.row()][index.column()] != value:
self.csv_data[index.row()][index.column()] = value
self.dataChanged.emit(index, index)
return True
return False
return False
def flags(self, index):
flags = Qt.ItemFlags.ItemIsSelectable | \
Qt.ItemFlags.ItemIsEnabled | \
Qt.ItemFlags.ItemIsEditable
return flags
# QTableView can have a header
def headerData(self, section, orientation, role):
if orientation == Qt.Orientation.Horizontal:
if role == Qt.ItemDataRole.DisplayRole:
return self.header[section]
class Window(QWidget):
def __init__(self):
super().__init__()
layout = QVBoxLayout()
self.setLayout(layout)
model = CsvModel('data.csv')
QAbstractItemModelTester(model)
view = QTableView()
view.setModel(model)
view.resizeColumnsToContents()
layout.addWidget(view)
model.dataChanged.connect(self.on_data_changed)
def on_data_changed(self, topLeft, bottomRight, roles):
print(f'Model changed, r: {topLeft.row()}, c: {topLeft.column()}')
data = topLeft.model().data(topLeft, Qt.ItemDataRole.DisplayRole)
print(f'Data: {data}')
if __name__ == '__main__':
app = QApplication(sys.argv)
main_window = Window()
main_window.show()
sys.exit(app.exec())
|