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
| 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()]
# 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')
print(model.columnCount(), model.rowCount())
QAbstractItemModelTester(model)
view = QTableView()
view.setModel(model)
view.resizeColumnsToContents()
layout.addWidget(view)
if __name__ == '__main__':
app = QApplication(sys.argv)
main_window = Window()
main_window.show()
sys.exit(app.exec())
|