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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
| import sys
import csv
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt
from PySide6.QtWidgets import (QApplication, QWidget,
QTableView, QVBoxLayout, QFormLayout, QDataWidgetMapper,
QLineEdit, QPushButton)
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()]
elif role == Qt.ItemDataRole.EditRole:
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)
self.model = CsvModel('data.csv')
QAbstractItemModelTester(self.model)
self.view = QTableView()
self.view.setModel(self.model)
self.view.selectionModel().currentChanged.connect(
self.on_current_changed)
self.view.resizeColumnsToContents()
self.fname_edit = QLineEdit()
self.lname_edit = QLineEdit()
self.prof_edit = QLineEdit()
self.submit_button = QPushButton('Submit')
self.submit_button.clicked.connect(self.on_submit)
self.mapper = QDataWidgetMapper()
self.mapper.setModel(self.model)
self.mapper.setSubmitPolicy(
QDataWidgetMapper.SubmitPolicy.ManualSubmit)
self.mapper.addMapping(self.fname_edit, 1)
self.mapper.addMapping(self.lname_edit, 2)
self.mapper.addMapping(self.prof_edit, 3)
self.mapper.toFirst()
form_layout = QFormLayout()
form_layout.addWidget(self.fname_edit)
form_layout.addWidget(self.lname_edit)
form_layout.addWidget(self.prof_edit)
form_layout.addWidget(self.submit_button)
layout.addWidget(self.view)
layout.addLayout(form_layout)
self.model.dataChanged.connect(self.on_data_changed)
def on_submit(self):
self.mapper.submit()
def on_current_changed(self, current, previous):
self.mapper.setCurrentIndex(current.row())
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())
|