Início / coding / Perguntas / 79544385
Accepted
Marc Swingler
Asked: 2025-03-30 17:23:54 +0800 CST

Adicionar âncoras às chaves no dicionário antes de despejar com ruamel.yaml?

  • 772

Tenho um arquivo de configuração que (propositalmente) contém alguns aliases indefinidos:

base:
  foo: *args_foo
  bar: *args_bar
  fuz: fuz_val
  buz: buz_val

Enquanto isso, tenho um cli que recebe --foo e --bar como argumentos, onde --foo pode ser especificado várias vezes. Acabo com um dict parecido com:

args = {}
args["foo"] = ["foo_val_1", "foo_val_2", "foo_val_3"]
args["bar"] = "bar_val"

Gostaria de despejar os args para yaml, mas com âncoras personalizadas para cada chave em arg. Algo assim:

args:
  foo: &args_foo
  - foo_val_1
  - foo_val_2
  - foo_val_3
  bar: &args_bar "bar_val"

Eu poderia escrever meu próprio código para gerar o doc, mas eu preferiria de alguma forma dizer ao ruamel.yaml para lidar com a adição das âncoras antes da serialização. A questão é, isso é possível e, se sim, como?

A intenção é então concatenar os dois arquivos assim:

args:
  foo: &args_foo
  - foo_val_1
  - foo_val_2
  - foo_val_3
  bar: &args_bar "bar_val"
base:
  foo: *args_foo
  bar: *args_bar
  fuz: fuz_val
  buz: buz_val

e extrair "base" como a configuração final após carregar o documento concatenado.

ruamel.yaml

1 respostas

  1. Best Answer

    Como você não pode carregar seu arquivo de configuração (pois ele é um YAML inválido), o mais fácil, na minha opinião, é gerar um arquivo YAML com as definições de âncora e anexar o arquivo de configuração a ele.

    Recomendo primeiro fazer uma viagem de ida e volta da saída esperada:

    import sys
import ruamel.yaml

yaml_str = """\
args:
  foo: &args_foo
  - foo_val_1
  - foo_val_2
  - foo_val_3
  bar: &args_bar "bar_val"
base:
  foo: *args_foo
  bar: *args_bar
  fuz: fuz_val
  buz: buz_val
"""

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

    que dá:

    args:
  foo: &args_foo
  - foo_val_1
  - foo_val_2
  - foo_val_3
  bar: &args_bar bar_val
base:
  foo: *args_foo
  bar: *args_bar
  fuz: fuz_val
  buz: buz_val

    E então analisando data, especialmente para os tipos especiais que precisam ser criados (veja também esta resposta ):

    print('type', type(data['args']['foo']))
print('anchor', data['args']['foo'].anchor)
print('type', type(data['args']['bar']))
print('anchor', data['args']['foo'].anchor)

    que dá:

    type <class 'ruamel.yaml.comments.CommentedSeq'>
anchor Anchor('args_foo')
type <class 'ruamel.yaml.scalarstring.PlainScalarString'>
anchor Anchor('args_foo')

    Sua argsestrutura de dados precisa de alguma elaboração:

    args = {}
args["foo"] = map = ruamel.yaml.comments.CommentedSeq(["foo_val_1", "foo_val_2", "foo_val_3"])
map.yaml_set_anchor('args_foo', always_dump=True)
args["bar"] = scalar = ruamel.yaml.scalarstring.DoubleQuotedScalarString("bar_val")
scalar.yaml_set_anchor('args_bar', always_dump=True)
yaml.dump(args, sys.stdout)

    que dá:

    foo: &args_foo
- foo_val_1
- foo_val_2
- foo_val_3
bar: &args_bar "bar_val"

    Isso always_dumpé necessário, pois âncoras falsas normalmente não são descartadas.

    Agora você pode combinar o acima:

    import sys
import io
import ruamel.yaml
from pathlib import Path

config = Path('config.nonyaml')
config.write_text("""\
base:
  foo: *args_foo
  bar: *args_bar
  fuz: fuz_val
  buz: buz_val
""")

yaml = ruamel.yaml.YAML()

args = {}
args["foo"] = map = ruamel.yaml.comments.CommentedSeq(["foo_val_1", "foo_val_2", "foo_val_3"])
map.yaml_set_anchor('args_foo', always_dump=True)
args["bar"] = scalar = ruamel.yaml.scalarstring.DoubleQuotedScalarString("bar_val")
scalar.yaml_set_anchor('args_bar', always_dump=True)
buf = io.BytesIO()

yaml.dump(dict(args=args), buf)
buf.write(config.read_bytes())

print(buf.getvalue().decode('utf-8'))

# check if buf contains loadable YAML
data = yaml.load(buf.getvalue())
print('data:', data)

    que dá:

    args:
  foo: &args_foo
  - foo_val_1
  - foo_val_2
  - foo_val_3
  bar: &args_bar "bar_val"
base:
  foo: *args_foo
  bar: *args_bar
  fuz: fuz_val
  buz: buz_val

data: {'args': {'foo': ['foo_val_1', 'foo_val_2', 'foo_val_3'], 'bar': 'bar_val'}, 'base': {'foo': ['foo_val_1', 'foo_val_2', 'foo_val_3'], 'bar': 'bar_val', 'fuz': 'fuz_val', 'buz': 'buz_val'}}
    • 1

relate perguntas