如何使用Ruamel.yaml在某些数据之前添加空行

时间:2017-02-13 07:08:00

标签: ruamel.yaml

我似乎无法弄清楚如何使用Ruamel.yaml在数据之间添加空行。

假设我有数据:

---
a: 1

b: 2

我需要添加这个,以便我有:

---
a: 1

b: 2

c: 3

我知道空白行是以CommentToken实现的:

Comment(comment=None,
  items={'data': [None, None, CommentToken(value=u'\n\n'), None], 'b': [None, None, CommentToken(value=u'\n\n'), None]})

我不知道如何操纵该结构。

1 个答案:

答案 0 :(得分:3)

Comment对象不是来自您提供的输入,因为data不是映射中的关键字,应该是a

import ruamel.yaml

yaml_strs = [
"""\
---
a: 1

b: 2
""",
"""\
---
a: 1

b: 2

c: 3
"""]

for yaml_str in yaml_strs:
    data = ruamel.yaml.round_trip_load(yaml_str)
    print(data.ca)

给出:

Comment(comment=None,
  items={'a': [None, None, CommentToken(), None]})
Comment(comment=None,
  items={'a': [None, None, CommentToken(), None], 'b': [None, None, CommentToken(), None]})

比较上述评论应该可以让您了解要尝试的内容:

import sys
import ruamel.yaml

yaml_str = """\
---
a: 1

b: 2
"""

data = ruamel.yaml.round_trip_load(yaml_str)
data['c'] = 3
ct = data.ca.items['a'][2]
data.ca.items['b'] = [None, None, ct, None]
ruamel.yaml.round_trip_dump(data, sys.stdout)

给出:

a: 1

b: 2

c: 3

CommentToken ct也可以从头开始构建:

ct = ruamel.yaml.tokens.CommentToken('\n\n', ruamel.yaml.error.CommentMark(0), None)

原样,例如在ruamel.yaml.comments.CommentedBase.yaml_set_start_comment()完成。

0 CommentMark()参数是评论缩进的距离,这在空行的情况下并不重要,但仍需要提供。