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
|
# -*- coding: utf-8 -*-
#
# This file is part of Linux Show Player
#
# Copyright 2012-2016 Francesco Ceruti <ceppofrancy@gmail.com>
#
# Linux Show Player is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Linux Show Player is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Linux Show Player. If not, see <http://www.gnu.org/licenses/>.
from PyQt5.QtCore import QAbstractTableModel, QModelIndex, Qt
from lisp.cues.cue import Cue
# Application defended data-roles
CueClassRole = Qt.UserRole + 1
class SimpleTableModel(QAbstractTableModel):
""" Simple implementation of a QAbstractTableModel
.. warning::
Currently this is a partial implementation of a complete R/W Qt model
intended for a simple and straightforward usage.
"""
def __init__(self, columns):
super().__init__()
self.columns = columns
self.rows = []
def appendRow(self, *values):
row = len(self.rows) - 1
self.beginInsertRows(QModelIndex(), row, row)
self.rows.append(list(values))
self.endInsertRows()
def removeRow(self, row, parent=QModelIndex()):
if -1 < row < len(self.rows):
self.beginRemoveRows(parent, row, row)
self.rows.pop(row)
self.endRemoveRows()
return True
return False
def rowCount(self, parent=QModelIndex()):
return len(self.rows)
def columnCount(self, parent=QModelIndex()):
return len(self.columns)
def headerData(self, section, orientation, role=Qt.DisplayRole):
if role == Qt.DisplayRole and orientation == Qt.Horizontal:
if section < len(self.columns):
return self.columns[section]
else:
return section + 1
if role == Qt.SizeHintRole and orientation == Qt.Vertical:
return 0
def data(self, index, role=Qt.DisplayRole):
if index.isValid():
if role == Qt.DisplayRole or role == Qt.EditRole:
return self.rows[index.row()][index.column()]
elif role == Qt.TextAlignmentRole:
return Qt.AlignCenter
def setData(self, index, value, role=Qt.DisplayRole):
if index.isValid():
if role == Qt.DisplayRole or role == Qt.EditRole:
self.rows[index.row()][index.column()] = value
self.dataChanged.emit(self.index(index.row(), 0),
self.index(index.row(), index.column()),
[Qt.DisplayRole, Qt.EditRole])
return True
return False
def flags(self, index):
return Qt.ItemIsEditable | Qt.ItemIsEnabled | Qt.ItemIsSelectable
class SimpleCueListModel(SimpleTableModel):
"""Extension of SimpleTableModel supporting the CueClassRole for rows."""
def __init__(self, columns):
super().__init__(columns)
self.rows_cc = []
def appendRow(self, cue_class, *values):
self.rows_cc.append(cue_class)
super().appendRow(*values)
def removeRow(self, row, parent=None):
if super().removeRow(row):
self.rows_cc.pop(row)
def data(self, index, role=Qt.DisplayRole):
if role == CueClassRole and index.isValid:
return self.rows_cc[index.row()]
return super().data(index, role)
def setData(self, index, value, role=Qt.DisplayRole):
if role == CueClassRole and index.isValid():
if issubclass(value, Cue):
self.rows_cc[index.row()] = value
return True
return False
return super().setData(index, value, role)
|
