笔记||Python3进阶之读取和写入yaml配置文件

yaml是专门用来写配置文件的语言，简洁强大，远比JSON格式方便，yaml在python语言中有PyYAML安装包。

- 首先需要pip安装：pip install pyyaml

- yaml基本语法规则：大小写敏感

使用缩进表示层级关系

缩进时不允许Tab键，只允许使用空格

#  表示注释

------------------------------------python代码读取yaml文件--------------------------------------

import  yaml

import os

# 获取当前脚本所在文件夹路径

curpath = os.path.dirname(os.path.realpath(__file__))

# print(curpath)

# 获取yaml文件路径

yamlpath = os.path.join(curpath, "cfgyaml.yaml")

# open方法打开文件直接读出来

f = open(yamlpath, 'r', encoding='utf-8')

cfg = f.read()

print(type(cfg))

# print(cfg)

# 用load方法转字典

d = yaml.load(cfg)

print(type(d))

print(d)

---------------------------------cfgyaml.yaml文件内容如下--------------------------------------

# yaml中的数组，相当于python中的列表

- admin1: 123456

- admin2: 234567

- admin3: 345678

# yaml中的键值对，相当于python中的字典

# test1:

#     user: username

#     pw: 123456

---------------------------------------------------------------------------------------------------------------------------------------------------------

用yaml模块写入字典嵌套字典这种复杂的数据，会出现大括号{}，不是真正的yaml文件数据，可以用ruamel模块解决。

安装：pip install ruamel.yaml

使用方法跟yaml差不多，只是在使用dump方法多一个参数：Dumper=yaml.RoundTripDumper

---------------------------python代码写入yaml文件----------------------------------

import os

from ruamel import yaml

# 将字典写入到yaml

desired_caps = {

'platformName': 'Android',

'platformVersion': '7.0',

'deviceName': 'ASRNW1111111111'

'appPackage': 'com.tencent.com',

'appActivity': 'ui.LauncherUI',

'automationName': 'Uiautomator2',

'unicodeKeyboard': True,

'resetKeyboard': True,

'noReset': True,

'chromeOptions': {'androidProcess': 'com.tencent.com'}

}

curpath = os.path.dirname(os.path.realpath(__file__))

yamlpath = os.path.join(curpath, "cfgyaml.yaml")

# 写入到yaml文件

with open(yamlpath, 'w', encoding="utf-8") as f:

yaml.dump(desired_caps, f, Dumper=yaml.RoundTripDumper)

--------------------------------------------------------------------------------------------------------------------

使用ruamel.yaml模块也能读yaml文件，使用方法相对之前的yaml.load方法多加了一个参数：Loader=yaml.Loader

如：

rea = open(yamlpath, 'r')

a = rea.load()

b = yaml.load(a, Loader=yaml.Loader)

print(b)