Добавить несколько документов в файл yaml | PyYAML - PullRequest
Новая
       11

Добавить несколько документов в файл yaml | PyYAML

0 голосов
/ 05 ноября 2018

Я работаю над объектом, где первый питон читает YAML, вносит некоторые изменения и затем записывает их обратно в файл. Часть загрузки и обновления значений работает нормально, но когда я пишу файл, он создает списки, а не отдельные документы.

testing.yaml 

apiVersion: v1
data:
  databag1: try this
  databag2: then try this
kind: ConfigMap
metadata:
  name: data bag info
---
apiVersion: extensions/v1beta1
kind: Deployment
metadata:
  labels:
    app: data-bag-service
  name: data-bag-tagging

Кодовый блок 

import yaml
with open("./testing.yaml", "r") as stream:
    deployment_dict= list(yaml.safe_load_all(stream))

print(deployment_dict)
with open("./testing.yaml", "w") as service_config:
    yaml.dump(
        deployment_dict,
        service_config,
        default_flow_style=False
    )

Преобразование я получаю: testing.yaml 

- apiVersion: v1
  data:
    databag1: try this
    databag2: then try this
  kind: ConfigMap
  metadata:
    name: data bag info
- apiVersion: extensions/v1beta1
  kind: Deployment
  metadata:
    labels:
      app: data-bag-service
    name: data-bag-tagging

Как мне достичь исходного состояния с помощью --- индикаторов конца директивы?

Ответы [ 2 ]

0 голосов
/ 05 ноября 2018

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

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

  • поддержка YAML 1.2 (но может записывать / читать YAML 1.1 при необходимости)
  • сохранение комментариев, порядка ключей, имен якорей / псевдонимов, форматов с плавающей запятой / целых чисел
  • более точный контроль над отступами отображений и списков
  • нет необходимости загружать все документы, обрабатывать их и выгружать за один раз
  • необязательное сохранение кавычек и / или скаляров блочного стиля
  • безопасная загрузка по умолчанию и предупреждение, если вы используете небезопасный load в обратно совместимый API
  • исправление многих ошибок 


from pathlib import Path
from ruamel.yaml import YAML

path = Path('testing.yaml')
tmp_path = path.with_suffix('.yaml.tmp')


with YAML(output=tmp_path) as yaml:
    # yaml.indent(mapping=4, sequence=4, offset=2)
    # yaml.preserve_quotes = True
    for data in yaml.load_all(path):
        # update data
        yaml.dump(data)

path.unlink()
tmp_path.rename(path)

print(path.read_text(), end='')

, что дает: 

apiVersion: v1
data:
  databag1: try this
  databag2: then try this
kind: ConfigMap
metadata:
  name: data bag info
---
apiVersion: extensions/v1beta1
kind: Deployment
metadata:
  labels:
    app: data-bag-service
  name: data-bag-tagging

Обратите внимание, что вы не можете писать и читать из того же файла, что и вы обрабатывают документ за один раз. Отсюда временный файл, который имеет дополнительное преимущество, что если вы получаете ошибку при обновлении этот последний документ и ваша программа падает, вы не остаетесь с наполовину записанный поток YAML.

0 голосов
/ 05 ноября 2018

Согласно документам :

Если вам нужно выгрузить несколько документов YAML в один поток, используйте функцию yaml.dump_all. yaml.dump_all принимает список или генератор, выдающий 

yaml.dump_all(
    deployment_dict,
    service_config,
    default_flow_style=False
)

Вам все еще нужно default_flow_style=False, чтобы получить вывод стиля блока.

Пример кода: 

import yaml


with open("./testing.yaml", "r") as stream:
    d = list(yaml.safe_load_all(stream))

d.append(d[-1])

with open("./testing2.yaml", "w") as stream:
    yaml.dump_all(
        d,
        stream,
        default_flow_style=False
    )

testing2.yaml 

apiVersion: v1
data:
  databag1: try this
  databag2: then try this
kind: ConfigMap
metadata:
  name: data bag info
---
apiVersion: extensions/v1beta1
kind: Deployment
metadata:
  labels:
    app: data-bag-service
  name: data-bag-tagging
---
apiVersion: extensions/v1beta1
kind: Deployment
metadata:
  labels:
    app: data-bag-service
  name: data-bag-tagging
...