使用所有键的行号解析 YAML

Question

我想解析一个结构如下的 YAML 文件：

key1:
    key2: 10
    key3:
        key4:
            key5:
                "value1"
                "value2"

目前，我使用以下代码：

data = yaml.load(file, Loader=ruamel.yaml.RoundTripLoader)
print(data['key1'].lc.line) #I get line number of key1
print(data['key1']['key3'].lc.line) #I get line number of key3
print(data['key1']['key3']['key4'].lc.line) #I get line number of key4

现在我无法获取 key2 和 key5 的行号。我发现问题是由于data[key][key2] 不是字典，而是int。同样，data[key1][key3][key4][key5] 是一个列表而不是字典。

在这种情况下，有没有办法获取 所有 键的行号？

Answer 1

key1:
  key2: 10
  key3:
    key4:
      key5:
        - "value1"
        - "value2"
      key6: 20
      key7:
        key8:
          - "value3"
          - "value4"

import yaml

with open('test.yml') as f:
    data = yaml.full_load(f)

result = {}
current = 1


def keyId(val, current):
    ks = val.keys()
    for key in ks:
        result[key] = current
        current += 1
        if isinstance(val[key], type([])):
            current += len(val[key])
        if isinstance(val[key], type({})):
            keyId(val[key], current)

    return


keyId(data, current)
print(result)

{'key1': 1, 'key2': 2, 'key3': 3, 'key4': 4, 'key5': 5, 'key6': 8, 'key7': 9, 'key8': 10}

Answer 2

这在多个层面都是错误的：

• 您正在使用旧的ruamel.yaml 接口，并且在您的代码中同时包含yaml. 和ruamel.yaml。 您是否将 RoundTripLoader 与 PyYAML 一起使用？

• 您的输入不是有效的 YAML，因为您不能使用双引号 标量 ("value1") 后跟另一个（您当然可以在 Python）。既然你说的是“是一个列表”，但是没有序列 在将创建一个列表的输入中，您可能已经遗漏了 破折号。

• 您将获得的数字与键的行号混淆了 在您的各种映射中。虽然可能很难看到 计数从 0 开始，很明显，如果你添加一个额外的 输入中的（空）行

• 完全可以打印 data['key1']['key3']['key4']['key5'].lc.line，因为该值是 列表（子类型）虽然这当然是该列表开头的行号 不适用于key5

• 由于data['key1']['key2'] 是int，我不明白你是怎么做到的 可以期望一个属性指示该类型的行号（例如 如果 int 恰好是 dict 值，则它本身或某个键）。有 你曾尝试在像int 这样的内置类上设置属性吗？ 有时间试试10.lc = 24。

import ruamel.yaml 

yaml_str = """\
key1:
    key2: 10
    key3:

        key4:
            key5:
                - "value1"
                - "value2"
"""

yaml = ruamel.yaml.YAML()
data = yaml.load(yaml_str)
assert isinstance(data['key1']['key2'], int)
print(data['key1'].lc.line) # the starting line number of the value associated with of key1
print(data['key1']['key3'].lc.line) # the starting line number of the value associated with key3
print(data['key1']['key3']['key4'].lc.line) # the starting line number of the value of key4
print(data['key1']['key3']['key4']['key5'].lc.line) # the starting line number of the value of key5

给出：

1
4
5
6

由于输入中多了一行，输出从1 3 4 5变为1 4 5 6 应该很清楚 你没有得到的不是任何键的行号。

所以很明显，您根本没有获得键的行号。

是的，有一种方法可以获取所有键的行号。
（如果您意识到您想知道如何做到这一点：您应该从方法 construct_mapping 的所有 key_node 分配中获取该信息 RoundTripLoader 类（它是子类并将其提供给 yaml 加载前的实例）并将其附加到某些属性 你自己在该方法的maptyp 变量中。）

Answer 3

key1:
  key4: 10
  key3:
    key8:

      key5:
        - "value1"
        - "value2"
      key6: 20
      key7:

        key2:
          - "value3"
          - "value4"

with open('test.yml') as f:
    lines = list(map(lambda x: x.strip().split(':')[0] if x.find(':') != -1 else "", f.readlines()))
    print(lines)

['key1', 'key4', 'key3', 'key8', '', 'key5', '', '', 'key6', 'key7', '', 'key2', '', '']

这里每个键都存储在 yml 文件中出现的索引处