mirror of
https://github.com/yuruotong1/autoMate.git
synced 2026-03-22 13:07:17 +08:00
71 lines
2.9 KiB
Python
71 lines
2.9 KiB
Python
from PyQt6.QtGui import QUndoCommand
|
|
from PyQt6 import QtCore
|
|
|
|
from utils.global_util import GlobalUtil
|
|
class ActionListAddCommand(QUndoCommand):
|
|
def __init__(self, action_list, row, action_item):
|
|
super().__init__()
|
|
self.action_list = action_list
|
|
self.action_item = action_item
|
|
self.row = row
|
|
self.tmp = None
|
|
self.setText(f"Add data to {row}")
|
|
|
|
|
|
def redo(self):
|
|
self.action_list.insertItem(self.row, self.action_item)
|
|
if self.action_list.level > 0:
|
|
self.action_list.get_parent().action.args.action_list.append(self.action_item.action)
|
|
self.action_item.render()
|
|
self.action_list.adjust_ui()
|
|
|
|
|
|
def undo(self):
|
|
self.action_list = GlobalUtil.all_widget["action_list"][self.action_list.uuid]
|
|
self.action_list.takeItem(self.row)
|
|
if self.action_item.type == "include":
|
|
# GlobalUtil.all_widget["action_list"].remove(self.action_item.data(QtCore.Qt.ItemDataRole.UserRole))
|
|
del GlobalUtil.all_widget["action_list"][self.action_item.data(QtCore.Qt.ItemDataRole.UserRole).uuid]
|
|
del GlobalUtil.all_widget["action_list_item"][self.action_item.uuid]
|
|
if self.action_list.level > 0:
|
|
self.action_list.get_parent().action.args.action_list.remove(self.action_item.action)
|
|
from actions.action_list_item import ActionListItem
|
|
# 重新加载，避免被 pyqt GC
|
|
action_list_item = ActionListItem.load(self.action_item.dump())
|
|
self.action_item = action_list_item
|
|
self.action_list.adjust_ui()
|
|
|
|
|
|
class ActionListDeleteCommand(QUndoCommand):
|
|
def __init__(self, action_list, row):
|
|
super().__init__()
|
|
self.action_list = action_list
|
|
self.delete_action_list_item = None
|
|
self.row = row
|
|
self.tmp = None
|
|
self.setText(f"Add data to {row}")
|
|
|
|
|
|
def redo(self):
|
|
from actions.action_list_item import ActionListItem
|
|
# 重新加载，避免被 pyqt GC
|
|
self.delete_action_list_item = ActionListItem.load(self.action_list.item(self.row).dump())
|
|
if self.delete_action_list_item.type == "include":
|
|
del GlobalUtil.all_widget["action_list"][self.action_list.item(self.row).data(QtCore.Qt.ItemDataRole.UserRole).uuid]
|
|
del GlobalUtil.all_widget["action_list_item"][self.delete_action_list_item.uuid]
|
|
if self.action_list.level > 0:
|
|
self.action_list.get_parent().action.args.action_list.remove(self.delete_action_list_item.action)
|
|
self.action_list.takeItem(self.row)
|
|
self.action_list.adjust_ui()
|
|
|
|
|
|
|
|
def undo(self):
|
|
self.action_list.insertItem(self.row, self.delete_action_list_item)
|
|
if self.action_list.level > 0:
|
|
self.action_list.get_parent().action.args.action_list.append(self.delete_action_list_item.action)
|
|
self.delete_action_list_item.render()
|
|
self.action_list.adjust_ui()
|
|
|
|
|