Как обновить файл yaml с помощью python

Пакет ruamel.yaml был специально улучшен (я начал с PyYAML) для этого вид туда и обратно, программный, обновление.

Если вы начнете с (обратите внимание, что я удалил лишние начальные пробелы):

init_config: {}
instances:
    - host: <IP>              # update with IP
      username: <username>    # update with user name
      password: <password>    # update with password

и запустите:

import ruamel.yaml

file_name = 'input.yaml'
config, ind, bsi = ruamel.yaml.util.load_yaml_guess_indent(open(file_name))

instances = config['instances']
instances[0]['host'] = '1.2.3.4'
instances[0]['username'] = 'Username'
instances[0]['password'] = 'Password'

with open('output.yaml', 'w') as fp:
    yaml.dump(config, fp)

Вывод будет:

init_config: {}
instances:
    - host: 1.2.3.4           # update with IP
      username: Username      # update with user name
      password: Password      # update with password

Порядок ключей сопоставления (host, username и password), стиль и комментарии сохраняются без каких-либо дополнительных действий.

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

yaml = ruamel.yaml.YAML()
yaml.indent(mapping=6, sequence=4)
with open(file_name) as fp:
    config = yaml.load(fp)

Если вы посмотрите на историю этого ответа, вы увидите, как это сделать с помощью более ограниченного API, похожего на PyYAML.

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

import yaml

fname = "some.yaml"

stream = open(fname, 'r')
data = yaml.load(stream)

data['instances'][0]['host'] = '1.2.3.4'
data['instances'][0]['username'] = 'Username'
data['instances'][0]['password'] = 'Password'

with open(fname, 'w') as yaml_file:
    yaml_file.write( yaml.dump(data, default_flow_style=False))

Я не знаю, нужен ли вам YAML. Помимо использования тега YAML, похоже, документ YAML вас не интересует. Так почему бы не использовать Jinja2 или какой-нибудь язык шаблонов?

from jinja2 import Template

tmpl = Template(u'''\
    init_config: {}
    instances:
         - host: {{ IP }}
           username: {{ username }}
           password: {{ password }}
''')

print tmpl.render(
     IP=u"1.2.3.4",
     username=u"Username",
     password=u"Password"
)

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

Бонус: пример использования

Я работал с очень сложными документами YAML, для которых есть неизвестные теги

...
  propertiesIDs: { 1, 2, 3, 4 }
  globalID: !myapplication.InterfaceID &primitiveID

replication: !myapplication.replication
  beginDate: 2012-09-10T20:00:03
  endDate: 2020-09-10T20:00:04
  replicant_uuid:
    ? 17169504-B6AB-11E4-8437-36E258BB2172
    ? 206B5842-B6AB-11E4-AAC3-36E258BB2172
...

Выполнение корректного разбора этого документа сложно и требует много времени. Мне нужно только заполнить некоторые значения, а YAML отправляется в стороннее приложение. Таким образом, вместо того, чтобы анализировать YAML или пытаться сгенерировать действительный документ напрямую с помощью pyyaml, проще (более эффективно с точки зрения времени и менее подвержено ошибкам) ​​создавать его непосредственно с помощью шаблонов. Кроме того, языки шаблонов можно легко использовать с циклами для заполнения полей с динамическим размером.

Вот как я создаю шаблоны docker-crane для разработки, производства, сцены и т. д.

1. мкдир кран_шаблоны
2. коснитесь крана_templates/init.py
3. Добавьте содержимое шаблона с помощью nanocrane_templates/some.yaml
4. Нано-кран_gen.py

--- Crane_gen.py ---

#!/usr/bin/env python
from jinja2 import Environment, PackageLoader

env = Environment(loader=PackageLoader('crane_templates', './'))
tmpl = env.get_template('crane.yaml.tmpl')

result = tmpl.render(
     IP=u"1.2.3.4",
     username=u"Username",
     password=u"Password"
)

5. Питон Crane_gen.py > результат.yaml

Ответ, вдохновленный @MariusSiuram

Вот пример использования PyYaml. Я так понимаю, у вас что-то вроде шаблона в формате yaml, и вы должны заменить места в угловых скобках реальными значениями.

import yaml

s = """
    init_config: {}
    instances:
        - host: <IP>
          username: <username>
          password: <password>
"""

dict_obj = yaml.load(s) # loads string in internal data structure - dict
dict_obj['instances'][0]['host'] = 'localhost' # change values
print yaml.dump(dict_obj) # dumps dict to yaml format back