解析 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)

这篇关于解析 YAML，返回行号的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持跟版网！