在自动化过程中,需要使用配置文件储存数据,比如数据库信息、账号信息、域名等。
其中,yaml
文件是一种配置文件类型,相比较ini
,conf
配置文件来说,更加的简洁,操作也更加简单,同时可以存放不同类型的数据。
以下主要介绍yaml语法、yaml存储数据,封装类读写yaml配置文件。
一 yaml介绍及使用
YAML
语言( /ˈjæməl/ )的设计目标,就是方便人类读写。它实质上是一种通用的数据串行化格式。
YAML 是一种非常灵活的格式,几乎是 JSON
的超集。除了支持注释、换行符分隔、多行字符串、裸字符串和更灵活的类型系统之外,YAML 也支持引用文件,以避免重复代码。
基本语法规则:
▌对象
Map(属性和值)(键值对)的形式:
key:(空格)value :表示一对键值对,空格不可省略。
person: name: vivi age: 18
person:{name: vivi,age: 18}
{"name":"vivi","age":18}
▌数组
一组连词线开头的行,构成一个数组。数组前加有 “-” 符号,符号与值之间需用空格分隔。
color: - red - blue - green
color: [red,blue,green]
相当于JSON:
["red","blue","green"]
▌纯量
单个的、不可再分的值。(如:字符串、bool值、整数、浮点数、时间、日期、null等)
n1: 8 n2: 8.8 n3: true n4: false n5: 'vivi'
二 yaml配置文件的使用
在项目下新建一个目录config,在目录下新建一个文件config.yaml。
在config.yaml配置文件中写入数据库配置。
mysql: host: "127.0.0.1" port: 3306 user: "vivi" password: "123456" db: "ITester" charset: "utf8"
三 yaml配置文件读写
在python中读取yaml文件,需要用到第三方模块PyYaml。
load:
将yaml流转化为python字典;
dump:
将python对象转化为yaml流
03 读写yaml配置文件
将读写yaml配置文件的类进行封装。
在common目录下新建一个文件,config_handler.py用于读写yaml。
config_handler.py
import yaml class YamlHandler: def __init__(self, file): self.file = file def read_yaml(self, encoding='utf-8'): """读取yaml数据,Loader=yaml.FullLoader这个必须得加上更加安全,不加报错""" with open(self.file, encoding=encoding) as f: return yaml.load(f.read(), Loader=yaml.FullLoader) def write_yaml(self, data, encoding='utf-8'): """向yaml文件写入数据""" with open(self.file, encoding=encoding, mode='w') as f: return yaml.dump(data, stream=f, allow_unicode=True) if __name__ == '__main__': data = { "user": { "username": "vivi", "password": "123456" } } # 读取config.yaml配置文件数据 read_data = YamlHandler('../config/config.yaml').read_yaml() # 将data数据写入config1.yaml配置文件 write_data = YamlHandler('../config/config1.yaml').write_yaml(data) print(read_data)
运行结果为:
{'mysql': {'host': '127.0.0.1', 'port': 3306, 'user': 'vivi', 'password': '123456', 'db': 'ITester', 'charset': 'utf8'}}