def eventFilter(self, viewport, event):
# type: (QObject, QEvent) -> bool
view = viewport.parent()
if event.type() == QEvent.MouseButtonPress and \
event.button() == Qt.LeftButton:
index = view.indexAt(event.pos())
if index is not None:
self._pos = event.pos()
self._index = QPersistentModelIndex(index)
elif event.type() == QEvent.MouseMove and self._pos is not None and \
((self._pos - event.pos()).manhattanLength() >=
QApplication.startDragDistance()):
assert self._index is not None
if self._index.isValid():
# Map to a QModelIndex in the model.
index = QModelIndex(self._index)
self._pos = None
self._index = None
self.dragStarted.emit(index)
return super().eventFilter(view, event)
def add_row(self, attr=None, condition_type=None, condition_value=None):
model = self.cond_list.model()
row = model.rowCount()
model.insertRow(row)
attr_combo = gui.OrangeComboBox(
minimumContentsLength=12,
sizeAdjustPolicy=QComboBox.AdjustToMinimumContentsLengthWithIcon)
attr_combo.row = row
for var in self._visible_variables(self.data.domain):
attr_combo.addItem(*gui.attributeItem(var))
attr_combo.setCurrentIndex(attr or 0)
self.cond_list.setCellWidget(row, 0, attr_combo)
index = QPersistentModelIndex(model.index(row, 3))
temp_button = QPushButton(
'×',
self,
flat=True,
styleSheet='* {font-size: 16pt; color: silver}'
'*:hover {color: black}')
temp_button.clicked.connect(lambda: self.remove_one(index.row()))
self.cond_list.setCellWidget(row, 3, temp_button)
self.remove_all_button.setDisabled(False)
self.set_new_operators(attr_combo, attr is not None, condition_type,
condition_value)
attr_combo.currentIndexChanged.connect(
lambda _: self.set_new_operators(attr_combo, False))
self.cond_list.resizeRowToContents(row)
def eventFilter(self, viewport, event):
view = viewport.parent()
if event.type() == QEvent.MouseButtonPress and event.button(
) == Qt.LeftButton:
index = view.indexAt(event.pos())
if index is not None:
self._pos = event.pos()
self._index = QPersistentModelIndex(index)
elif (event.type() == QEvent.MouseMove and self._pos is not None
and ((self._pos - event.pos()).manhattanLength() >=
QApplication.startDragDistance())):
if self._index.isValid():
# Map to a QModelIndex in the model.
index = self._index
index = index.model().index(index.row(), index.column(),
index.parent())
self._pos = None
self._index = None
self.dragStarted.emit(index)
return QObject.eventFilter(self, view, event)
def add_row(self, attr=None, condition_type=None, condition_value=None):
model = self.cond_list.model()
row = model.rowCount()
model.insertRow(row)
attr_combo = ComboBoxSearch(
minimumContentsLength=12,
sizeAdjustPolicy=QComboBox.AdjustToMinimumContentsLengthWithIcon)
attr_combo.setModel(self.variable_model)
attr_combo.row = row
attr_combo.setCurrentIndex(
self.variable_model.indexOf(attr) if attr else len(self.AllTypes) +
1)
self.cond_list.setCellWidget(row, 0, attr_combo)
index = QPersistentModelIndex(model.index(row, 3))
temp_button = QPushButton(
'×',
self,
flat=True,
styleSheet='* {font-size: 16pt; color: silver}'
'*:hover {color: black}')
temp_button.clicked.connect(lambda: self.remove_one(index.row()))
self.cond_list.setCellWidget(row, 3, temp_button)
self.remove_all_button.setDisabled(False)
self.set_new_operators(attr_combo, attr is not None, condition_type,
condition_value)
attr_combo.currentIndexChanged.connect(
lambda _: self.set_new_operators(attr_combo, False))
self.cond_list.resizeRowToContents(row)
def add_row(self, attr=None, condition_type=None, condition_value=None):
model = self.cond_list.model()
row = model.rowCount()
model.insertRow(row)
attr_combo = gui.OrangeComboBox(
minimumContentsLength=12,
sizeAdjustPolicy=QComboBox.AdjustToMinimumContentsLengthWithIcon)
attr_combo.row = row
for var in self._visible_variables(self.data.domain):
attr_combo.addItem(*gui.attributeItem(var))
attr_combo.setCurrentIndex(attr or 0)
self.cond_list.setCellWidget(row, 0, attr_combo)
index = QPersistentModelIndex(model.index(row, 3))
temp_button = QPushButton('×', self, flat=True,
styleSheet='* {font-size: 16pt; color: silver}'
'*:hover {color: black}')
temp_button.clicked.connect(lambda: self.remove_one(index.row()))
self.cond_list.setCellWidget(row, 3, temp_button)
self.remove_all_button.setDisabled(False)
self.set_new_operators(attr_combo, attr is not None,
condition_type, condition_value)
attr_combo.currentIndexChanged.connect(
lambda _: self.set_new_operators(attr_combo, False))
self.cond_list.resizeRowToContents(row)
class ItemViewDragStartEventListener(QObject):
dragStarted = Signal(QModelIndex)
def __init__(self, parent=None, **kwargs):
# type: (Optional[QObject], Any) -> None
super().__init__(parent, **kwargs)
self._pos = None # type: Optional[QPoint]
self._index = None # type: Optional[QPersistentModelIndex]
def eventFilter(self, viewport, event):
# type: (QObject, QEvent) -> bool
view = viewport.parent()
if event.type() == QEvent.MouseButtonPress and \
event.button() == Qt.LeftButton:
index = view.indexAt(event.pos())
if index is not None:
self._pos = event.pos()
self._index = QPersistentModelIndex(index)
elif event.type() == QEvent.MouseMove and self._pos is not None and \
((self._pos - event.pos()).manhattanLength() >=
QApplication.startDragDistance()):
assert self._index is not None
if self._index.isValid():
# Map to a QModelIndex in the model.
index = QModelIndex(self._index)
self._pos = None
self._index = None
self.dragStarted.emit(index)
return super().eventFilter(view, event)
def eventFilter(self, viewport, event):
view = viewport.parent()
if event.type() == QEvent.MouseButtonPress and \
event.button() == Qt.LeftButton:
index = view.indexAt(event.pos())
if index is not None:
self._pos = event.pos()
self._index = QPersistentModelIndex(index)
elif event.type() == QEvent.MouseMove and self._pos is not None and \
((self._pos - event.pos()).manhattanLength() >=
QApplication.startDragDistance()):
if self._index.isValid():
# Map to a QModelIndex in the model.
index = self._index
index = index.model().index(index.row(), index.column(),
index.parent())
self._pos = None
self._index = None
self.dragStarted.emit(index)
return super().eventFilter(view, event)
def itemData(
self, index: QModelIndex, roles: Sequence[int]
) -> Mapping[int, Any]:
"""
Return item data from `index` for `roles`.
The returned mapping is a read only view of *all* data roles accessed
for the index through this caching interface. It will contain at least
data for `roles`, but can also contain other ones.
"""
model = index.model()
if model is not self.__model:
self.setModel(model)
# NOTE: QPersistentModelIndex's hash changes when it is invalidated;
# it must be purged from __cache_data before that (`__connect_helper`)
key = QPersistentModelIndex(index)
try:
item = self.__cache_data[key]
except KeyError:
data = item_data(index, roles)
view = MappingProxy(data)
self.__cache_data[key] = data, view
else:
data, view = item
queryroles = tuple(filterfalse(data.__contains__, roles))
if queryroles:
data.update(item_data(index, queryroles))
return view
class ItemViewDragStartEventListener(QObject):
dragStarted = Signal(QModelIndex)
def __init__(self, parent=None):
QObject.__init__(self, parent)
self._pos = None
self._index = None
def eventFilter(self, viewport, event):
view = viewport.parent()
if event.type() == QEvent.MouseButtonPress and event.button() == Qt.LeftButton:
index = view.indexAt(event.pos())
if index is not None:
self._pos = event.pos()
self._index = QPersistentModelIndex(index)
elif (
event.type() == QEvent.MouseMove
and self._pos is not None
and ((self._pos - event.pos()).manhattanLength() >= QApplication.startDragDistance())
):
if self._index.isValid():
# Map to a QModelIndex in the model.
index = self._index
index = index.model().index(index.row(), index.column(), index.parent())
self._pos = None
self._index = None
self.dragStarted.emit(index)
return QObject.eventFilter(self, view, event)
def setSelection(self, selection):
"""
Set the model item selection.
Parameters
----------
selection : QtCore.QItemSelection
Item selection.
"""
if self.values_view.selectionModel() is not None:
indices = selection.indexes()
pind = defaultdict(list)
for index in indices:
parent = index.parent()
if parent.isValid():
if parent == self.__model.index(parent.row(),
parent.column()):
pind[parent.row()].append(QPersistentModelIndex(index))
else:
warnings.warn("Die Die Die")
else:
# top level index
pass
self.__selections = pind
self.__restoreSelection()
def __storeSelection(self, groupind, indices):
# Store current values selection for the current group
groupind = self.__currentIndex
indices = [
QPersistentModelIndex(ind)
for ind in self.values_view.selectedIndexes()
]
self.__selections[groupind] = indices
def createActionForItem(self, index):
"""Create the QAction instance for item at `index` (`QModelIndex`).
"""
action = QAction(item_icon(index),
item_text(index),
self,
toolTip=item_tooltip(index))
action.setData(QPersistentModelIndex(index))
return action
def data(self, index: QModelIndex, role: int) -> Any:
"""Return item data for `index` and `role`"""
model = index.model()
if model is not self.__model:
self.setModel(model)
key = QPersistentModelIndex(index)
try:
item = self.__cache_data[key]
except KeyError:
data = item_data(index, (role,))
view = MappingProxy(data)
self.__cache_data[key] = data, view
else:
data, view = item
if role not in data:
data[role] = model.data(index, role)
return data[role]
def save_and_commit():
if edit.text() and edit.text() in wordlist:
model = index.model()
pindex = QPersistentModelIndex(index)
if pindex.isValid():
new_index = pindex.sibling(pindex.row(),
pindex.column())
save = model.setData(new_index, edit.text(),
Qt.EditRole)
if save:
owwidget.commit()
return
edit.clear()