Skip to content

Commit d51e6fc

Browse files
committed
tools: 迁移脚本可以删除空的 interrupt
1 parent 3594ef8 commit d51e6fc

1 file changed

Lines changed: 5 additions & 10 deletions

File tree

tools/migrate_pipeline_v5.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -204,8 +204,6 @@ def rebuild_json_with_comments(
204204
205205
使用更精确的方式在原文本上进行替换操作
206206
"""
207-
import json
208-
209207
result_text = original_text
210208

211209
# 遍历所有节点,应用修改
@@ -484,9 +482,7 @@ def add_prefix_to_specific_nodes(nodes: list, target_nodes: set, prefix: str) ->
484482
return result
485483

486484

487-
def migrate_node(
488-
node_name: str, node_data: dict, is_sub_nodes: set
489-
) -> tuple[dict, list]:
485+
def migrate_node(node_data: dict, is_sub_nodes: set) -> tuple[dict, list]:
490486
"""
491487
迁移单个节点的配置,保持字段顺序
492488
@@ -503,20 +499,19 @@ def migrate_node(
503499
# 收集需要处理的数据
504500
interrupt_list = []
505501
has_interrupt = "interrupt" in node_data
506-
has_is_sub = "is_sub" in node_data
507502

508503
if has_interrupt:
509504
interrupt_list = ensure_list(node_data.get("interrupt"))
510505

511506
# 按原始顺序遍历字段
512507
for key, value in node_data.items():
513508
if key == "interrupt":
514-
# 跳过 interrupt 字段,稍后处理
509+
# 跳过 interrupt 字段
510+
changes.append(f" - 删除 interrupt: {value}")
515511
continue
516512
elif key == "is_sub":
517513
# 跳过 is_sub 字段
518-
is_sub_value = value
519-
changes.append(f" - 删除 is_sub: {is_sub_value}")
514+
changes.append(f" - 删除 is_sub: {value}")
520515
continue
521516
elif key == "next":
522517
# 处理 next 字段
@@ -622,7 +617,7 @@ def migrate_pipeline_file(
622617
has_changes = False
623618

624619
for node_name, node_data in data.items():
625-
migrated_node, changes = migrate_node(node_name, node_data, global_is_sub_nodes)
620+
migrated_node, changes = migrate_node(node_data, global_is_sub_nodes)
626621
migrated_data[node_name] = migrated_node
627622

628623
if changes:

0 commit comments

Comments
 (0)