Python读写yaml排版混乱还丢失注释？我来告诉你解决办法！

Python实战社群

Java实战社群

长按识别下方二维码，按需求添加

扫码关注添加客服

进Python社群▲

扫码关注添加客服

进Java社群▲

作者丨王翔丨 

来源丨清风Python

Python读写Yaml

日常我们在使用Python读写Yaml时，都是使用推荐的Pyyaml模块。安装：pip install pyyaml导入：import yaml至于操作，简直不要太简单... yaml只有两个方法load、dump，而且使用完全和json模块一样。但真的如此吗？显然不是...

Yaml安全告警

由于Yaml数据存在安全隐患，在使用pyyaml进行load时，会给出提示：

YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsafe. Pleaseread https://msg.pyyaml.org/load for full details.

所以我们有以下方式解决：

• 添加Loaderinfo = yaml.load(data,Loader=yaml.SafeLoader)

• 使用语法糖info = yaml.safe_load(data)

同时pyyaml支持多段yaml内容的文本读写功能，即load_all() dump_all()。当我们使用safe_load获取多节点yaml文件时，会出发yaml.composer.ComposerError的错误，所以更优雅的导入方式应该是：

import yaml
from yaml.composer import ComposerError

with open(file, 'r') as f:
    data = f.read()
try:
    return yaml.safe_load(data)
except ComposerError:
    return yaml.safe_load_all(data)

要注意，load_all 返回的是一个generator。当然用了这么久yaml真的很少见多段yaml非要写在一个文件中的情况。

pyyaml的劣势

上面介绍的主要是pyyaml的读取功能，可是在pyyaml将读取后的内容进行二次写入时，你就会遇到很多的问题，比如：

1. 原本的yaml文件字段顺序，被重新调整

2. 所有的注释内容，全部丢失

顺序这个勉强还可以容忍，但是注释全部丢失真的是叔叔能忍，婶婶都忍不啊！让我们看一个yaml文件读写前后对比：

我们该如何解决这个问题呢？

ruamel.yaml

ruamel.yaml作为pyyaml的衍生版，在继承了传统pyyaml的基础上，增加了RoundTrip模式。在读取yaml的时候，将yaml转化为ordereddict。众所周知，collections中的ordereddict是一个有序的字典，这样就保证了我们yaml读写后的顺序一致。安装：pip install ruamel.yaml导入：from ruamel import yaml来看看效果吧：

from ruamel import yaml

class YamlLoader:
    def __init__(self, file):
        self.file = file

    def file_load(self):
        with open(self.file, 'r', encoding='utf-8') as f:
            data = f.read()
        return yaml.load(data, Loader=yaml.RoundTripLoader)

    def file_dump(self, data):
        with open('newer_%s' % self.file, 'w',
                  encoding='utf-8') as f:
            yaml.dump(data, f,
                      Dumper=yaml.RoundTripDumper)


yml = YamlLoader('single.yaml')
yaml_info = yml.file_load()
print(yaml_info)
yml.file_dump(yaml_info)

output:
ordereddict([('Users', [ordereddict([('Lilei', ordereddict([('age', 18), ('sex', 'male')]))]), ordereddict([('Hanmeimei', ordereddict([('age', 17), ('sex', 'female')]))])]), ('Code', ['Python', 'Java', 'Ruby'])])

程序员专栏 扫码关注填加客服 长按识别下方二维码进群

近期精彩内容推荐：   看黄片，起诉网站，可尼玛太秀了 知乎高赞：35岁失业的中年人都去了哪儿？ 求求你别再用offset和limit分页了 Python 代码实现验证码识别，很稳


在看点这里好文分享给更多人↓↓