Início / coding / Perguntas / 79540913
Accepted
julaine
Asked: 2025-03-28 16:41:31 +0800 CST

Como escrevo uma âncora usando ruamel.yaml?

  • 772

Quero criar dados em python e gravá-los em um arquivo como um documento yaml usando âncoras e mesclagens na saída.

Acredito que isso seria possível com o ruamel YAML, porque, conforme descrito nos exemplos oficiais do ruamels :

  • ele pode carregar documentos yaml com âncoras e mesclar de forma transparente (transparente no sentido de que o acesso aos dados é o mesmo, seja definido por meio de âncoras ou não)
  • ele pode fazer o roundtrip de arquivos yaml enquanto preserva âncoras (então carregar um arquivo yaml que inclui âncoras pode ser lido e reescrito no disco e ainda ter âncoras)

Isso significa que ruamel.yaml deve ter uma representação interna do yaml-data que inclua e entenda âncoras (ao contrário do PyYAML, que apenas lê âncoras, mas não as preserva). Não parece haver uma maneira documentada de criar tais âncoras a partir do código python.

O arquivo mais minimalista que eu quero criar seria algo como isto:

base: &ANCHOR
  x: 1

object:
  <<: *ANCHOR
  y: 2
python

1 respostas

  1. Best Answer

    Quando você faz o ida e volta dos seus dados usando o seguinte programa:

    import sys  
import ruamel.yaml

yaml_str = """\ 
base: &ANCHOR  
x: 1

object:  
<<: \*ANCHOR  
y: 2  
"""

yaml = ruamel.yaml.YAML()  
data = yaml.load(yaml_str)  
yaml.dump(data, sys.stdout)

    você vê que a saída corresponde à entrada:

    base: &ANCHOR
  x: 1

object:
  <<: *ANCHOR
  y: 2

    Você poderia realmente criar uma nova YAMLinstância para dumping e ainda obter a mesma saída. Isso significa que as informações sobre âncoras e mesclagem estão em algum lugar na estrutura de dados em data. Então você deve inspecionar vários itens.

    print('anchor', data['base'].anchor)
print('type', type(data['base'].anchor))
print('keys', list(data['object'].keys()))
print('merge', data['object'].merge)
print('merge type', type(data['object'].merge))
print('ids', id(data['object'].merge[0][1]), id(data['base']))

    que dá:

    anchor Anchor('ANCHOR')
type <class 'ruamel.yaml.anchor.Anchor'>
keys ['y', 'x']
merge [(0, {'x': 1})]
merge type <class 'list'>
ids 4304434048 4304434048

    O acima é normalmente um processo incremental (mesmo para mim, que tenho algum conhecimento sobre os internos). E ajuda olhar para a fonte, especialmente construct_mappingem constructor.pye CommentedMapemcomments. py

    Com as informações acima, vamos primeiro lidar com a âncora (você pode ter mesclagens sem âncora/alias, mas elas não fazem muito sentido).

    import sys
import ruamel.yaml

def CM(**kw):
    return ruamel.yaml.comments.CommentedMap(**kw)

common = CM(x=1)
common.yaml_set_anchor('ANCHOR')
data = CM(base=common, object=CM(tmp=common, y=2))
yaml = ruamel.yaml.YAML()
yaml.dump(data, sys.stdout)

    que dá:

    base: &ANCHOR
  x: 1
object:
  tmp: *ANCHOR
  y: 2

    A criação da chave de mesclagem pode ser feita definindo-a CommentedMapapenas com a ychave e adicionando o mergeatributo:

    import sys
import ruamel.yaml

def CM(**kw):
    return ruamel.yaml.comments.CommentedMap(**kw)

common = CM(x=1)
common.yaml_set_anchor('ANCHOR')
data = CM(base=common, object=CM(y=2))
setattr(data['object'], ruamel.yaml.comments.merge_attrib, [(0, common)])
yaml = ruamel.yaml.YAML()
yaml.dump(data, sys.stdout)

    que dá:

    base: &ANCHOR
  x: 1
object:
  <<: *ANCHOR
  y: 2
    1. Não há setter para o .mergeatributo
    2. esse atributo é uma lista de porque em YAML a chave de mesclagem pode ter um mapeamento ou uma lista de mapeamentos como valor. Se bem me lembro 0, o inteiro determina a ordenação.

    Fixe a versão que ruamel.yamlvocê está usando. Internos como esses mudarão sem aviso prévio

    • 1

relate perguntas