Ruamel.yaml: Как сохранить структуру dict в YAML

Как указал Ник Бейли в комментариях, это стилистическое изменение, а не структурное. То есть данные одни и те же, просто представлены по-разному.

Теперь, что касается этого стиля, YAML имеет два стиля представления структур данных:

• Стиль блока: Каждый ключ/значение начинается с новой строки, а списки и словари (сопоставления) запускаются и останавливаются с помощью отступа. Обычно это предпочтительный стиль, так как он более удобочитаем.

• Стиль потока: списки/сопоставления начинаются и заканчиваются квадратными скобками, а несколько ключей/значений разделяются запятыми, как в JSON. Разрывы строк не требуются между парами ключ/значение, но также не запрещены. Этот формат чаще используется для небольших и простых структур данных, особенно в одной строке, поскольку он может сэкономить место.

Исходный YAML, который вы показали, представляет собой одну пару ключ/значение в более крупном отображении блочного стиля, но само значение имеет блочный стиль не; это просто стиль потока с добавлением дополнительных разрывов строк. Я думаю, вы, вероятно, захотите этого, полностью в блочном стиле:

test:
  parameters:
    "provision_pipeline": "provision-integrations"
    "enable_sfcd_ds_argo_operator": "false"
    "clustermanagement_helm_values_path": "sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml"

ruamel.yaml в режиме по умолчанию (туда и обратно) сохраняет стиль потока или блока, в зависимости от того, что вы ему зададите, но я не знаю, как заставить его запоминать определенные разрывы строк, которые вы добавили в разделе потока. Посмотрите это сравнение:

import sys
from ruamel.yaml import YAML

yaml_string_1 = """\
test:
  parameters: {
    "provision_pipeline": "provision-integrations",
    "enable_sfcd_ds_argo_operator": "false",
    "clustermanagement_helm_values_path": "sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml"
  }
"""
yaml_string_2 = """\
test:
  parameters:
    "provision_pipeline": "provision-integrations"
    "enable_sfcd_ds_argo_operator": "false"
    "clustermanagement_helm_values_path": "sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml"
"""

yaml = YAML()
for yaml_string in [yaml_string_1, yaml_string_2]:
    output = yaml.load(yaml_string)
    yaml.dump(output, sys.stdout)
    print()

Выход:

test:
  parameters: {provision_pipeline: provision-integrations, enable_sfcd_ds_argo_operator: 'false',
    clustermanagement_helm_values_path: sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml}

test:
  parameters:
    provision_pipeline: provision-integrations
    enable_sfcd_ds_argo_operator: 'false'
    clustermanagement_helm_values_path: sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml

Вы также можете, конечно, добавить preserve_quotes и любые другие опции, которые вам нужны.

ruamel.yaml не сохраняет новые строки между элементами сопоставления стилей потока. Единственное на них влияет yaml.width, поэтому вы получаете перенос строк, которые становятся слишком длинными. Например. с вашим вводом, если вы установите ширину на 40, вы получите:

parameters: {"provision_pipeline": "provision-integrations",
  "enable_sfcd_ds_argo_operator": "false",
  "clustermanagement_helm_values_path": "sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml"}

Но нет элемента управления, который даст вам первую пару ключ-значение на новой строке или закрытие. фигурная скобка на отдельной строке.

Ваше добавление ryaml.default_flow_style = False влияет только на совершенно новые диктовки и списки, которые вы добавляете в структура данных.

Вам следует подумать о переходе на блочный стиль и отбросить все несущественные кавычки, что делает YAML и менее подробный, и более читаемый. Для программы, которая загружает данные, это не имеет значения, и преобразование легко выполняется путем загрузки в обычном безопасном режиме (который не устанавливает информацию о блочном/потоковом стиле). на загруженных данных):

import sys
import pathlib
import ruamel.yaml

basefile = pathlib.Path('input.yaml')

data = ruamel.yaml.YAML(typ='safe').load(basefile)
yaml = ruamel.yaml.YAML()
yaml.dump(data, sys.stdout)

который дает:

parameters:
  provision_pipeline: provision-integrations
  enable_sfcd_ds_argo_operator: 'false'
  clustermanagement_helm_values_path: sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml

Строковый скаляр 'false' должен быть заключен в кавычки, чтобы его не путали с логическим значением false.

Если вышеуказанное улучшение неприемлемо, т.е. если дальнейшая обработка выполняется с чем-то кроме полного синтаксического анализатора YAML, вы можете выполнить постобработку вывода:

import sys
import pathlib
import ruamel.yaml

basefile = pathlib.Path('input.yaml')

def splitflowmap(s):
    res = []
    for line in s.splitlines():
        if ': {' in line and line[-1] == '}':
            start, rest = line.split(': {', 1)
            start = start + ': {'
            indent = '  '  # two spaces more than the start
            for idx, ch in enumerate(start):
                if ch != ' ':
                    break
                indent += ' '
            res.append(start)
            rest = rest[:-1]  # cut of }\n
            for x in rest.split(', '):  # if you always have quotes it is safer to split on '", "'
                res.append(f'{indent}{x},')
            res[-1] = res[-1][:-1]  # delete trailing comma
            res.append(f'{indent[2:]}}}')  # re-add the cut of }\n on a line of its own
            continue
        res.append(line)
    return '\n'.join(res) + '\n'

yaml = ruamel.yaml.YAML()
yaml.preserve_quotes = True
yaml.width = 2**16
data = yaml.load(basefile)
yaml.dump(data, sys.stdout, transform=splitflowmap)

который дает:

parameters: {
  "provision_pipeline": "provision-integrations",
  "enable_sfcd_ds_argo_operator": "false",
  "clustermanagement_helm_values_path": "sam/sam-helm-charts/kube-node-recycler-0.0.4-273/values.nodepool.yaml"
}