Python循环解析含!Element自定义标签的YAML文件并生成独立YAML文件方案咨询

先梳理下你现有代码里的几个关键问题：

• 使用了不安全的yaml.load()，存在代码注入风险，应该换成更安全的加载方式
• 错误地尝试通过elements_yml["elements"]["!Element"]获取元素，实际上elements是一个列表，每个列表项就是带!Element标签的元素对象
• 创建目录的路径不符合需求（你需要的是new_folder/element1这种嵌套结构），且未处理目录已存在的情况
• 缺少核心的文件写入逻辑，也没正确处理YAML内容的序列化

接下来是修正后的完整代码，我会一步步拆解说明：

import yaml
import os
from yaml.loader import SafeLoader

# 自定义构造器处理!Element标签，只提取标签对应的内容
def element_constructor(loader, node):
    return loader.construct_mapping(node)

# 给SafeLoader注册自定义标签的解析规则
yaml.add_constructor('!Element', element_constructor, Loader=SafeLoader)

# 读取原始YAML文件
with open('elements.yml', 'r') as elements_file:
    elements_yml = yaml.safe_load(elements_file)

# 遍历所有带!Element标签的元素
for element in elements_yml['elements']:
    element_name = element['name']
    # 拼接目标目录和文件路径
    target_dir = os.path.join('new_folder', element_name)
    target_file = os.path.join(target_dir, f'{element_name}_config.yml')
    
    # 创建目录，exist_ok=True避免目录已存在时报错
    os.makedirs(target_dir, exist_ok=True)
    
    # 将当前元素内容写入新YAML文件
    with open(target_file, 'w') as f:
        # 用default_flow_style=False保证输出是易读的块级格式
        yaml.dump(element, f, default_flow_style=False)

关键细节说明： #

• 自定义标签处理：因为原始YAML包含!Element自定义标签，默认SafeLoader无法解析，所以注册构造器来提取标签对应的字典内容，确保所有元素能正常加载
• 安全加载：使用yaml.safe_load()替代yaml.load()，彻底避免恶意YAML文件的代码执行风险
• 路径兼容性：用os.path.join拼接路径，自动适配Windows、Linux等不同系统的路径分隔符
• 容错处理：os.makedirs的exist_ok=True参数，保证重复运行代码时不会因为目录已存在而崩溃
• 格式一致性：yaml.dump时设置default_flow_style=False，让生成的YAML文件和你示例中的格式完全一致（块级缩进，而非紧凑的行内格式）

运行这段代码后，就会在new_folder目录下为每个!Element元素生成对应的子目录和配置文件，每个文件仅保留对应元素的内容，完全符合你的需求。

内容的提问来源于stack exchange，提问作者Shalomi90