41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
import yaml
|
||
import os
|
||
|
||
|
||
# 单个文档
|
||
def get_yaml_data(yaml_file):
|
||
# 打开yaml文件
|
||
file = open(yaml_file, 'r', encoding='utf-8')
|
||
file_data = file.read()
|
||
file.close()
|
||
# 将字符串转化为字典或列表
|
||
data = yaml.safe_load(file_data) # safe_load，safe_load,unsafe_load
|
||
return data
|
||
|
||
|
||
# yaml文件中含多个文档时，分别获取文档中数据
|
||
def get_yaml_load_all(yaml_file):
|
||
# 打开文件
|
||
file = open(yaml_file, 'r', encoding='utf-8')
|
||
file_data = file.read()
|
||
file.close()
|
||
|
||
all_data = yaml.load_all(file_data, Loader=yaml.FullLoader)
|
||
for data in all_data:
|
||
print('data-----', data)
|
||
|
||
|
||
# 生成yaml文档
|
||
def generate_yaml_doc(yaml_file):
|
||
py_ob = {"school": "zhang", "students": ['a', 'b']}
|
||
file = open(yaml_file, 'w', encoding='utf-8')
|
||
yaml.dump(py_ob, file)
|
||
file.close()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
current_path = os.path.abspath("../../")
|
||
yaml_path = os.path.join(current_path, "config.yaml")
|
||
print('--------------------', yaml_path)
|
||
get_yaml_data(yaml_path)
|