python讀取yaml文件后修改寫入本地實例

更新時間：2020年04月27日 09:00:13 作者：哇葫蘆娃

這篇文章主要介紹了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文件中的參數(shù)
  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

補(bǔ)充知識：解決 Python ruamel.yaml 讀寫 yaml 文檔 format 格式不一致問題

使用 ruamel.yaml 過程中發(fā)現(xiàn)輸入輸出格式不一致，網(wǎng)上找到一個解決方案：

input：

ChuaTony: 
 公司: 文傳
 聯(lián)絡(luò): 
 - 公司: '1422031030'
 - 手機(jī): '18650144578'
 - 住家: '2233068479'
 住址: 
 - 公司: 褔建。
 - 住家: 臺中。
 交通: 無
 附件: 
Tom: 
 公司: 天褔
 聯(lián)絡(luò): 
 - 公司: '1422031030'
 - 手機(jī): '18650123456'
 - 住家: '2233061234'
 住址: 
 - 公司: 褔建。
 - 住家: 臺北。
 交通: 自行車
 附件:

output：
ChuaTony:
 交通: 無
 住址:
 - 公司: 褔建。
 - 住家: 臺中。
 公司: 文傳
 聯(lián)絡(luò):
 - 公司: '1422031030'
 - 手機(jī): '18650144578'
 - 住家: '2233068479'
 附件: null
Tom:
 交通: 自行車
 住址:
 - 公司: 褔建。
 - 住家: 臺北。
 公司: 天褔
 聯(lián)絡(luò):
 - 公司: '963852741'
 - 手機(jī): '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]['聯(lián)絡(luò)'])
 
#修改
alldata['Tom']['聯(lián)絡(luò)'][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:縮進(jìn) block_seq_indent:區(qū)塊縮進(jìn)

output：

ChuaTony:
 公司: 文傳
 聯(lián)絡(luò):
 - 公司: '1422031030'
 - 手機(jī): '18650144578'
 - 住家: '2233068479'
 住址:
 - 公司: 褔建。
 - 住家: 臺中。
 交通: 無
 附件:
Tom:
 公司: 天褔
 聯(lián)絡(luò):
 - 公司: '963852741'
 - 手機(jī): '18650123456'
 - 住家: '2233061234'
 住址:
 - 公司: 褔建。
 - 住家: 臺北。
 交通: 自行車
 附件:

搞定！?。?/p>

以上這篇python讀取yaml文件后修改寫入本地實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: