动态构造 YAML 对象

Question

我正在使用 PyYAML 库动态生成 YAML 文件。

例如，

import yaml

with open(r'content.txt', 'r') as a_file:


  with open(r'file.yaml', 'a') as file:

    file.write("  root:")
    file.write("\n")
       
  for line in a_file:
    
    stripped_line = line.strip()
    txt = stripped_line
    x = txt.split("=")
    
    with open(r'file.yaml', 'a') as file:
       yaml.dump([{'child-1':x[0],'child-2':x[1]}])

content.txt 文件可能包含以下形式的数据：

a=b
c=d

所需的最终 YAML 文件应如下所示：

  root:
  - child-1: a
    child-2: b

  - child-1: c
    child-2: d

请注意根对象的缩进，假设它嵌套在另一个根对象中

但是上面的代码使得输出为：

  root:
-   child-1: a
    child-2: b

-   child-1: c
    child-2: d

这不是有效的 YAML。 在yaml.dump() 命令中提及根对象会重复它：

#for line in content.txt
#yaml.dump({'root':[{'child-1': x[0], 'child-2':x[1]}])

  root:
  - child-1: a
    child-2: b
  root
  - child-1: c
    child-2: d

由于pythonyaml.dump()函数要求我们将对象连同其子对象一起完整地提及，因此分开处理这2个对象似乎很困难。

有没有办法分开调用这些对象并稍后附加/链接根子对象？

Answer 1

您没有使用 PyYAML 写入 YAML 文档，因为您忘记向 dump() 提供流参数，不幸的是，这在 PyYAML 中是可选的，并且因为您不使用其结果（字符串）值，所以可以替换为pass。

除此之外，您应该一次性将 YAML 文档写入流，而不是尝试附加到它 ('a')。因此，立即创建并转储您想要的整个数据结构：

import sys


import yaml   

children = []
data = dict(root=children)

with open('content.txt', 'r') as a_file:
    for line in a_file:
        x = line.strip().split('=')
        children.append({'child-1': x[0], 'child-2': x[1]})

with open('file.yaml', 'w') as o_file:
    yaml.dump(data, o_file)

with open('file.yaml', 'r') as i_file:
     print(i_file.read(), end='')

给出：

root:
- child-1: a
  child-2: b
- child-1: c
  child-2: d

您还应该注意，PyYAML 仅支持 YAML 1.1 版，尽管 最新版本是 YAML 1.2，因此您可能需要切换到 ruamel.yaml 库，它支持这个标准（已经有 10 多年的历史了。免责声明我是 ruamel.yaml 的开发者）。 为了做到这一点，替换

import yaml

与

 import ruamel.yaml
 yaml = ruamel.yaml.YAML(type='safe')

这对于大文件也更快。