Update

2023/scripts/animation_tools/dwpicker/commands.py

@@ -0,0 +1,159 @@
+from copy import deepcopy
+from .pyside import QtWidgets, QtCore
+from .templates import COMMAND, MENU_COMMAND
+from .qtutils import icon
+from .dialog import CommandEditorDialog, MenuCommandEditorDialog
+
+
+class CommandItemWidget(QtWidgets.QWidget):
+    editRequested = QtCore.Signal(object)
+    deletedRequested = QtCore.Signal(object)
+
+    def __init__(self, command, parent=None):
+        super(CommandItemWidget, self).__init__(parent)
+
+        self.command = command
+        self.label = QtWidgets.QLabel(self.get_label())
+        self.edit = QtWidgets.QPushButton(icon('edit2.png'), '')
+        self.edit.released.connect(lambda: self.editRequested.emit(self))
+        self.edit.setFixedSize(25, 25)
+        self.delete = QtWidgets.QPushButton(icon('delete2.png'), '')
+        self.delete.setFixedSize(25, 25)
+        self.delete.released.connect(lambda: self.deletedRequested.emit(self))
+        layout = QtWidgets.QHBoxLayout(self)
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setSpacing(0)
+        layout.addWidget(self.edit)
+        layout.addWidget(self.delete)
+        layout.addSpacing(10)
+        layout.addWidget(self.label)
+
+    def get_label(self):
+        language = '<a style="color: #FFFF00"><i>({0})</i></a>'.format(
+            self.command['language'])
+        touchs = [self.command['button'] + 'Click']
+        touchs.extend([m for m in ('ctrl', 'shift') if self.command[m]])
+        return '{} {}'.format('+'.join(touchs), language)
+
+    def update_label(self):
+        self.label.setText(self.get_label())
+
+
+class MenuCommandItemWidget(CommandItemWidget):
+
+    def get_label(self):
+        language = '<a style="color: #FFFF00"><i>({0})</i></a>'.format(
+            self.command['language'])
+        return '{} {}'.format(self.command['caption'], language)
+
+
+class CommandsEditor(QtWidgets.QWidget):
+    valueSet = QtCore.Signal(object)
+    edit_dialog_constructor = CommandEditorDialog
+    picker_command_key = 'action.commands'
+    template = COMMAND
+    item_constructor = CommandItemWidget
+
+    def __init__(self, parent=None):
+        super(CommandsEditor, self).__init__(parent)
+        self.warning = QtWidgets.QLabel('Select only one shape')
+
+        self.commands = QtWidgets.QListWidget()
+        self.commands.setSelectionMode(QtWidgets.QAbstractItemView.NoSelection)
+        self.commands.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOn)
+        self.commands.setHorizontalScrollBarPolicy(
+            QtCore.Qt.ScrollBarAlwaysOff)
+
+        self.add_command = QtWidgets.QPushButton('Add command')
+        self.add_command.released.connect(self.call_create_command)
+        self.add_command.setEnabled(False)
+
+        layout = QtWidgets.QVBoxLayout(self)
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setSpacing(0)
+        layout.addWidget(self.warning)
+        layout.addWidget(self.commands)
+        layout.addWidget(self.add_command)
+
+    def set_options(self, options):
+        self.commands.clear()
+        if len(options) != 1:
+            self.warning.setVisible(True)
+            self.add_command.setEnabled(False)
+            return
+        self.warning.setVisible(False)
+        self.add_command.setEnabled(True)
+        for command in options[0][self.picker_command_key]:
+            self.call_add_command(command)
+
+    def call_create_command(self):
+        command = deepcopy(self.template)
+        dialog = self.edit_dialog_constructor(command)
+        if not dialog.exec_():
+            return
+        self.call_add_command(dialog.command_data())
+        self.valueSet.emit(self.commands_data())
+
+    def call_add_command(self, command=None):
+        widget = self.item_constructor(command)
+        widget.editRequested.connect(self.edit_command)
+        widget.deletedRequested.connect(self.delete_command)
+        item = QtWidgets.QListWidgetItem()
+        item.widget = widget
+        item.setSizeHint(
+            QtCore.QSize(
+                self.commands.width() -
+                self.commands.verticalScrollBar().width(),
+                widget.sizeHint().height()))
+        self.commands.addItem(item)
+        self.commands.setItemWidget(item, widget)
+
+    def edit_command(self, widget):
+        for r in range(self.commands.count()):
+            item = self.commands.item(r)
+            if item.widget != widget:
+                continue
+            dialog = self.edit_dialog_constructor(item.widget.command)
+            if not dialog.exec_():
+                return
+            widget.command = dialog.command_data()
+            widget.update_label()
+            self.valueSet.emit(self.commands_data())
+
+    def delete_command(self, widget):
+        for r in range(self.commands.count()):
+            item = self.commands.item(r)
+            if item.widget != widget:
+                continue
+            self.commands.takeItem(r)
+            self.valueSet.emit(self.commands_data())
+            return
+
+    def commands_data(self):
+        return [
+            self.commands.item(r).widget.command
+            for r in range(self.commands.count())]
+
+
+class MenuCommandsEditor(CommandsEditor):
+    edit_dialog_constructor = MenuCommandEditorDialog
+    picker_command_key = 'action.menu_commands'
+    template = MENU_COMMAND
+    item_constructor = MenuCommandItemWidget
+
+
+class GlobalCommandsEditor(CommandsEditor):
+    edit_dialog_constructor = MenuCommandEditorDialog
+    picker_command_key = 'menu_commands'
+    template = MENU_COMMAND
+    item_constructor = MenuCommandItemWidget
+
+    def __init__(self, parent=None):
+        super(GlobalCommandsEditor, self).__init__(parent)
+        self.warning.hide()
+        self.add_command.setEnabled(True)
+
+    def set_options(self, options):
+        self.commands.clear()
+        for command in options[self.picker_command_key]:
+            self.call_add_command(command)