如何将数据附加到 YAML 文件

Question

我有一个文件 *.yaml内容如下：

bugs_tree:
  bug_1:
    html_arch: filepath
    moved_by: user1
    moved_date: '2018-01-30'
    sfx_id: '1'

我想在节点下的这个文件中添加一个新的子元素 [bugs_tree] 我尝试执行此操作如下所示：

if __name__ == "__main__":
    new_yaml_data_dict = {
        'bug_2': {
            'sfx_id': '2', 
            'moved_by': 'user2', 
            'moved_date': '2018-01-30', 
            'html_arch': 'filepath'
        }
    }

    with open('bugs.yaml','r') as yamlfile:
        cur_yaml = yaml.load(yamlfile)
        cur_yaml.extend(new_yaml_data_dict)
        print(cur_yaml)

然后文件应该看起来：

bugs_tree:
  bug_1:
    html_arch: filepath
    moved_by: username
    moved_date: '2018-01-30'
    sfx_id: '1234'
  bug_2:
    html_arch: filepath
    moved_by: user2
    moved_date: '2018-01-30'
    sfx_id: '2'

当我尝试执行.append()OR .extend()OR 时.insert()出现错误

cur_yaml.extend(new_yaml_data_dict)
AttributeError: 'dict' object has no attribute 'extend'

Answer 1

如果您想更新文件，仅读取是不够的。您还需要再次写入该文件。像这样的事情会起作用：

with open('bugs.yaml','r') as yamlfile:
    cur_yaml = yaml.safe_load(yamlfile) # Note the safe_load
    cur_yaml['bugs_tree'].update(new_yaml_data_dict)

if cur_yaml:
    with open('bugs.yaml','w') as yamlfile:
        yaml.safe_dump(cur_yaml, yamlfile) # Also note the safe_dump

我没有对此进行测试，但他的想法是使用read来读取文件并使用 write来写入文件。使用safe_load并safe_dump像Anthon所说的那样：

“绝对没有必要使用 load()，它被记录为不安全。请改用 safe_load()”

Answer 2

你需要使用update

cur_yaml.update(new_yaml_data_dict)

结果代码

with open('bugs.yaml','r') as yamlfile:
        cur_yaml = yaml.load(yamlfile)
        cur_yaml.update(new_yaml_data_dict)
        print(cur_yaml)

with open('bugs.yaml','w') as yamlfile:
        yaml.safe_dump(cur_yaml, yamlfile) # Also note the safe_dump