Files
my-v0-server/app/actions/update.py
T

78 lines
2.9 KiB
Python

"""在大屏 JSON 副本上按路径更新组件属性,用于后端校验动作能否执行"""
from copy import deepcopy
from app.actions.base import BaseActionHandler
from app.schemas.actions import UpdateAction
from app.schemas.screen import Screen
class UpdateHandler(BaseActionHandler[UpdateAction]):
def apply(self, screen: Screen, action: UpdateAction) -> Screen:
result, target = self.copy_target(screen, action.target_id)
# 未设置的 style 视为空对象,只允许直接添加末级字段
if action.section == "style" and target.style is None:
target.style = {}
current = getattr(target, action.section)
for index, segment in enumerate(action.path):
last = index == len(action.path) - 1
location = f"{action.section} {action.path[: index + 1]!r}"
if isinstance(current, dict):
if not isinstance(segment, str):
raise ValueError(f"对象路径必须使用字符串键: {location}")
if not last and segment not in current:
raise ValueError(f"中间路径不存在: {location}")
elif isinstance(current, list):
if type(segment) is not int:
raise ValueError(f"数组路径必须使用整数下标: {location}")
if not 0 <= segment < len(current):
raise ValueError(f"数组下标越界: {location}")
else:
raise ValueError(f"路径无法继续访问,当前值不是对象或数组: {location}")
if last:
# 新值也复制,避免后续修改结果时影响 Action
current[segment] = deepcopy(action.value)
else:
current = current[segment]
return result
if __name__ == "__main__":
screen = Screen.model_validate(
{
"canvas": {"width": 1920, "height": 1080},
"nodes": [
{
"id": "sales-chart",
"type": "chart",
"layout": {"x": 100, "y": 200, "width": 400, "height": 300},
"props": {"option": {"title": {"text": "销售额", "show": True}}},
}
],
}
)
action = UpdateAction(
target_id="sales-chart",
section="props",
path=["option", "title", "text"],
value="月度销售额",
)
handler = UpdateHandler()
final_screen = handler.apply(screen, action)
print(final_screen.model_dump_json(indent=2))
print("原始标题:", screen.nodes[0].props["option"]["title"]["text"])
try:
handler.apply(
screen,
UpdateAction(
target_id="sales-chart",
section="props",
path=["missing", "text"],
value="新标题",
),
)
except ValueError as error:
print("无效路径:", error)