【Python】Pyyaml和ruamel.yaml

目录

PYYAML

读取yaml

保存yaml

读取保存的yaml文件 

 yaml文件规则

yaml文件数据结构

ruamel.yaml

格式化保存yaml

使用ruamel.yaml读取yaml

使用ruamel.yaml时python中符号对应于yaml中符号

---

PYYAML

config.yaml文件

username: zxx
age: 18
orther:
  height: 175CM
  weitht: 107KG
JobHistory:
  - name: IBM
    date: 2015-2017
  - name: GA
    date: 2017-now

读取yaml

read_yaml.py文件

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# load-yaml.py

import os
import yaml

userInfo = yaml.load(open(r"D:\myproject\MyUtils\test_yaml\config.yaml", 'r'))
print(userInfo)

结果为：

{'JobHistory': [{'name': 'IBM', 'date': '2015-2017'}, {'name': 'GA', 'date': '2017-now'}], 'username': 'zxx', 'orther': {'weitht': '107KG', 'height': '175CM'}, 'age': 18}

保存yaml

import yaml

data = {"cookie1":{'domain': '.yiyao.cc', 'expiry': 1521558688.480118, 'httpOnly': False}}

f = open(r'D:\myproject\MyUtils\test_yaml\save_config.yaml','a')
print(yaml.dump(data, f))
f.close()

结果为

{'cookie1': {'domain': '.yiyao.cc', 'expiry': 1521558688.480118, 'httpOnly': False}}

save_config.yaml文件内容为

读取保存的yaml文件 

import yaml

f = open(r'D:\myproject\MyUtils\test_yaml\save_config.yaml','r')
# 读取文件
cont = f.read()
# 加载数据
x = yaml.load(cont)
# 打印数据
print(x)
f.close()

结果为：

{'cookie1': {'expiry': 1521558688.480118, 'domain': '.yiyao.cc', 'httpOnly': False}}

 yaml文件规则

• 区分大小写；
• 使用缩进表示层级关系；
• 使用空格键缩进，而非Tab键缩进
• 缩进的空格数目不固定，只需要相同层级的元素左侧对齐；
• 文件中的字符串不需要使用引号标注，但若字符串包含有特殊字符则需用引号标注；
• 注释标识为#

yaml文件数据结构

• 对象：键值对的集合（简称 "映射或字典"）
  键值对用冒号 “:” 结构表示，冒号与值之间需用空格分隔
• 数组：一组按序排列的值（简称 "序列或列表"）
  数组前加有 “-” 符号，符号与值之间需用空格分隔
• 纯量(scalars)：单个的、不可再分的值（如：字符串、bool值、整数、浮点数、时间、日期、null等）
  None值可用null可 ~ 表示

ruamel.yaml

从上面pyyaml保存yaml的处理中可以发现，保存出来的save_config.yaml是字典形式，不是yaml的标准形式，这对于追求一致性的强迫症来说，有点难受。这时发现有一个库ruamel.yaml可以完美解决这个问题。

格式化保存yaml

from __future__ import print_function

import ruamel.yaml
import os

def generate_yaml_doc_ruamel(yaml_file):
    from ruamel import yaml
    py_object = {'school': 'zhang',
                 'students': ['a', 'b']}
    file = open(yaml_file, 'w', encoding='utf-8')
    yaml.dump(py_object, file, Dumper=yaml.RoundTripDumper)
    file.close()
current_path = os.path.abspath(".")
yaml_path = os.path.join(current_path, "save_ruamel_config.yaml")
generate_yaml_doc_ruamel(yaml_path)

结果为

使用ruamel.yaml读取yaml

import os

# 通过from ruamel import yaml读取yaml文件
def get_yaml_data_ruamel(yaml_file):
    from ruamel import yaml
    file = open(yaml_file, 'r', encoding='utf-8')
    data = yaml.load(file.read(), Loader=yaml.Loader)
    file.close()
    print(data)
current_path = os.path.abspath(".")
yaml_path = os.path.join(current_path, "save_ruamel_config.yaml")
get_yaml_data_ruamel(yaml_path)

使用ruamel.yaml时python中符号对应于yaml中符号

python中代码

py_object = {'school': 'zhang',
                 'students': ['a', 'b'],
                 'age':{'old':34,'young':10},
                 'test':{'test1':123, 'test2':456, 'test3':{'test3_0':0, 'test3_1':[1,1,1,1]}}}

结果为 

age:
  young: 10
  old: 34
school: zhang
test:
  test3:
    test3_0: 0
    test3_1:
    - 1
    - 1
    - 1
    - 1
  test2: 456
  test1: 123
students:
- a
- b