76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
from ..Qt import QtCore, QtWidgets
|
|
from .._py.move_dialog import Ui_MoveDialog
|
|
|
|
|
|
class QMover(QtWidgets.QDialog, Ui_MoveDialog):
|
|
moveData = QtCore.pyqtSignal(list, str, str)
|
|
copyData = QtCore.pyqtSignal(list, str)
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent=parent)
|
|
self.setupUi(self)
|
|
|
|
self.entries = {}
|
|
|
|
self.fromcomboBox.currentIndexChanged.connect(self.change_graph)
|
|
self.buttonBox.clicked.connect(self.button_clicked)
|
|
|
|
def setup(self, entries: dict):
|
|
self.fromcomboBox.blockSignals(True)
|
|
self.tocomboBox.blockSignals(True)
|
|
for k, v in entries.items():
|
|
self.entries[k[0]] = v
|
|
self.fromcomboBox.addItem(k[1], userData=k[0])
|
|
self.tocomboBox.addItem(k[1], userData=k[0])
|
|
self.fromcomboBox.blockSignals(False)
|
|
self.tocomboBox.blockSignals(False)
|
|
self.change_graph(0)
|
|
|
|
@QtCore.pyqtSlot(int)
|
|
def change_graph(self, idx: int):
|
|
self.listWidget.clear()
|
|
idd = self.fromcomboBox.itemData(idx)
|
|
if idd is not None:
|
|
for i, j in self.entries[idd]:
|
|
it = QtWidgets.QListWidgetItem(j)
|
|
it.setData(QtCore.Qt.UserRole, i)
|
|
self.listWidget.addItem(it)
|
|
|
|
@QtCore.pyqtSlot(QtWidgets.QAbstractButton)
|
|
def button_clicked(self, btn: QtWidgets.QAbstractButton):
|
|
if self.buttonBox.buttonRole(btn) in [QtWidgets.QDialogButtonBox.ApplyRole,
|
|
QtWidgets.QDialogButtonBox.AcceptRole]:
|
|
from_graph = self.fromcomboBox.itemData(self.fromcomboBox.currentIndex())
|
|
to_graph = self.tocomboBox.itemData(self.tocomboBox.currentIndex())
|
|
if from_graph != to_graph:
|
|
moving = []
|
|
for idx in self.listWidget.selectedIndexes():
|
|
it = self.listWidget.itemFromIndex(idx)
|
|
moving.append(it.data(QtCore.Qt.UserRole))
|
|
it_data = (it.data(QtCore.Qt.UserRole), it.text())
|
|
if self.move_button.isChecked():
|
|
self.entries[from_graph].remove(it_data)
|
|
self.entries[to_graph].append(it_data)
|
|
self.change_graph(self.fromcomboBox.currentIndex())
|
|
|
|
if self.move_button.isChecked():
|
|
self.moveData.emit(moving, to_graph, from_graph)
|
|
else:
|
|
self.copyData.emit(moving, to_graph)
|
|
|
|
if self.buttonBox.buttonRole(btn) == QtWidgets.QDialogButtonBox.AcceptRole:
|
|
self.close()
|
|
|
|
elif self.buttonBox.buttonRole(btn) == QtWidgets.QDialogButtonBox.RejectRole:
|
|
self.close()
|
|
else:
|
|
return
|
|
|
|
def close(self):
|
|
self.listWidget.clear()
|
|
self.tocomboBox.clear()
|
|
self.fromcomboBox.clear()
|
|
self.entries = {}
|
|
|
|
super().close()
|