python readline 返回行号_python – 解析YAML,返回行号

我通过添加钩子到Composer.compose_node和Constructor.construct_mapping：

import yaml

from yaml.composer import Composer

from yaml.constructor import Constructor

def main():

loader = yaml.Loader(open('data.yml').read())

def compose_node(parent, index):

# the line number where the previous token has ended (plus empty lines)

line = loader.line

node = Composer.compose_node(loader, parent, index)

node.__line__ = line + 1

return node

def construct_mapping(node, deep=False):

mapping = Constructor.construct_mapping(loader, node, deep=deep)

mapping['__line__'] = node.__line__

return mapping

loader.compose_node = compose_node

loader.construct_mapping = construct_mapping

data = loader.get_single_data()

print(data)