Parsowanie YAML, powrót z numerem linii

Question

Robię generator dokumentów z danych YAML, który określa, z której linii pliku YAML jest generowany każdy element. Jaki jest najlepszy sposób na zrobienie tego? Więc jeśli plik YAML jest tak:

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

chcę coś takiego:

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

obecnie używam PyYAML, ale każda inna biblioteka jest OK, jeśli można go używać z poziomu Pythona.

Answer 1

Zrobiłem go poprzez dodanie haki do Composer.compose_node i 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) 

Answer 2

W celu uzyskania dalszych inspiracji, tu jest mój kod do tego. Zawiera więcej informacji niż wymagane powyżej, ponieważ zgłasza informacje o położeniu za pomocą start_mark, end_mark na każdym dict/list/unicode (odpowiednio za pomocą dict_node, list_node, subklasses unicode_node).

https://gist.github.com/dagss/5008118

Answer 3

Jeśli używasz ruamel.yaml> = 0,9 (z których jestem autorem) i użyć RoundTripLoader, można uzyskać dostęp do właściwości lc na przedmioty zbiórki uzyskać wiersz i kolumnę, gdzie rozpoczęła się w źródło YAML:

def test_item_04(self): 
    data = load(""" 
    # testing line and column based on SO 
    # http://stackoverflow.com/questions/13319067/ 
    - key1: item 1 
     key2: item 2 
    - key3: another item 1 
     key4: another item 2 
     """) 
    assert data[0].lc.line == 2 
    assert data[0].lc.col == 2 
    assert data[1].lc.line == 4 
    assert data[1].lc.col == 2 

(liczba linii i kolumna liczona od 0).

This answer Pokaż, jak dodać atrybut lc do typów ciągów podczas ładowania.