解析 YAML，返回行号

2022-01-14 00:00:00 python parsing pyyaml yaml

问题描述

我正在用 YAML 数据制作一个文档生成器，它将指定每个项目是从 YAML 文件的哪一行生成的.做这个的最好方式是什么?所以如果 YAML 文件是这样的:

I'm making a document generator from YAML data, which would specify which line of the YAML file each item is generated from. What is the best way to do this? So if the YAML file is like this:

- key1: item 1
  key2: item 2
- key1: another item 1
  key2: another item 2

我想要这样的东西:

[
     {'__line__': 1, 'key1': 'item 1', 'key2': 'item 2'},
     {'__line__': 3, 'key1': 'another item 1', 'key2': 'another item 2'},
]

我目前正在使用 PyYAML，但如果我可以从 Python 中使用它，任何其他库都可以.

I'm currently using PyYAML, but any other library is OK if I can use it from Python.

解决方案

这是puzzlet的答案的改进版:

import yaml
from yaml.loader import SafeLoader

class SafeLineLoader(SafeLoader):
    def construct_mapping(self, node, deep=False):
        mapping = super(SafeLineLoader, self).construct_mapping(node, deep=deep)
        # Add 1 so line numbering starts at 1
        mapping['__line__'] = node.start_mark.line + 1
        return mapping

你可以这样使用它:

data = yaml.load(whatever, Loader=SafeLineLoader)

相关文章