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

2020-09-23 本文已影响0人清风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. Please
read https://msg.pyyaml.org/load for full details.

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

添加Loader
info = 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将读取后的内容进行二次写入时，你就会遇到很多的问题，比如：

原本的yaml文件字段顺序，被重新调整
所有的注释内容，全部丢失

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

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'])])

ruamel.yaml效果

今天的分享就到这里，希望对大家有所帮助。如果有收获，记得转发+点再看支持下，谢谢。

The End

期待你关注我的公众号清风Python，如果你觉得不错，希望能动动手指转发给你身边的朋友们。
我的github地址：https://github.com/BreezePython