python读取yaml文件后修改写入本地

首先安装pip install ruamel.yaml
用于修改yaml文件

#coding:utf-8
from ruamel import yaml

def up_yml(ip_server):
    with open('./../docker-compose-demo.yml', encoding="utf-8") as f:
        content = yaml.load(f, Loader=yaml.RoundTripLoader)
        # 修改yml文件中的参数
        content['service']['memo_query']['server_ip'][0] = 'mysql_host={}'.format(ip_server)
    with open('./../docker-compose.yml', 'w', encoding="utf-8") as nf:
        yaml.dump(content, nf, Dumper=yaml.RoundTripDumper)

if __name__ == '__main__':
    up_yml(ip_server='0.0.0.0')

修改前：

version: '1'
service:
  memo_query:
    image: python:2.0
    restart: always
    server_ip:
    - mysql_host=192.168.1.1

修改后：

version: '1'
service:
  memo_query:
    image: python:2.0
    restart: always
    server_ip:
    - mysql_host=0.0.0.0