欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2023-11-07 17:16:16
首先安装pip install ruamel.yaml用于修改yaml文件#coding:utf-8from ruamel import yamldef up_yml(ip_server): with...

首先安装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

补充知识:解决 python ruamel.yaml 读写 yaml 文档 format 格式不一致问题

使用 ruamel.yaml 过程中发现输入输出格式不一致,网上找到一个解决方案:

input:

chuatony: 
 公司: 文传
 联络: 
 - 公司: '1422031030'
 - 手机: '18650144578'
 - 住家: '2233068479'
 住址: 
 - 公司: 褔建。
 - 住家: 台中。
 交通: 无
 附件: 
tom: 
 公司: 天褔
 联络: 
 - 公司: '1422031030'
 - 手机: '18650123456'
 - 住家: '2233061234'
 住址: 
 - 公司: 褔建。
 - 住家: 台北。
 交通: 自行车
 附件: 
output:
chuatony:
 交通: 无
 住址:
 - 公司: 褔建。
 - 住家: 台中。
 公司: 文传
 联络:
 - 公司: '1422031030'
 - 手机: '18650144578'
 - 住家: '2233068479'
 附件: null
tom:
 交通: 自行车
 住址:
 - 公司: 褔建。
 - 住家: 台北。
 公司: 天褔
 联络:
 - 公司: '963852741'
 - 手机: '18650123456'
 - 住家: '2233061234'
 附件: null

格式不一致

使用:

ruamel.yaml.load(loader=ruamel.yaml.roundtriploader) ruamel.yaml.dump(dumper=ruamel.yaml.roundtripdumper)

ruamel.yaml.round_trip_load() ruamel.yaml.round_trip_dump()

来解决

修改程式码如下:

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
' a yaml module '
 
__author__ = 'chua tony'
 
import ruamel.yaml
 
#开档
with open("人资.yaml", "r",encoding="utf-8") as docs:
	try:
		#alldata = ruamel.yaml.load(docs, loader=ruamel.yaml.roundtriploader) #使用 loader=ruamel.yaml.roundtriploader
		alldata = ruamel.yaml.round_trip_load(docs)
	except ruamel.yaml.yamlerror as exc:
		print(exc)
 
#印出
for data in alldata:
	print(alldata[data]['联络'])
 
#修改
alldata['tom']['联络'][0]['公司']='963852741'
 
#写档
with open('人资1.yaml', 'w+', encoding='utf8') as outfile:
	#ruamel.yaml.dump(alldata, outfile, default_flow_style=false, allow_unicode=true, dumper=ruamel.yaml.roundtripdumper, indent=2, block_seq_indent=2) #使用 dumper=ruamel.yaml.roundtripdumper
	ruamel.yaml.round_trip_dump(alldata, outfile, default_flow_style=false, allow_unicode=true, indent=2, block_seq_indent=2) #indent:缩进 block_seq_indent:区块缩进

output:

chuatony:
 公司: 文传
 联络:
 - 公司: '1422031030'
 - 手机: '18650144578'
 - 住家: '2233068479'
 住址:
 - 公司: 褔建。
 - 住家: 台中。
 交通: 无
 附件:
tom:
 公司: 天褔
 联络:
 - 公司: '963852741'
 - 手机: '18650123456'
 - 住家: '2233061234'
 住址:
 - 公司: 褔建。
 - 住家: 台北。
 交通: 自行车
 附件:

搞定!!!

以上这篇python读取yaml文件后修改写入本地实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。