【Python】数据写入之yaml文件排序和中文问题

数据写入 yaml 文件时遇到的一些问题总结,主要是中文问题和字典的排序问题。

环境准备

python

PyYAML

使用示例

将一段 python 的字典类型,转成 yaml 文件。

import yaml

data = {
    "get请求": {
        "name": "GET请求示例",
        "sleep": 3,
        "request": {
            "method": "GET",
            "url": "/get"
        }
    }
}

# 写入yaml 文件
with open("data.yml", 'w', encoding="utf-8") as fp:
    yaml.safe_dump(data, fp, indent=4)

safe_dump 方法默认传2个传参,第一个是需要转的数据,第二个是 stream。
indent = 4 是设置缩进为4个空格
生成的yaml 文件如下

"get\u8BF7\u6C42":
    name: "GET\u8BF7\u6C42\u793A\u4F8B"
    request:
        method: GET
        url: /get
    sleep: 3

解决中文问题

查看 safe_dump 源码

def safe_dump(data, stream=None, **kwds):
    """
    Serialize a Python object into a YAML stream.
    Produce only basic YAML tags.
    If stream is None, return the produced string instead.
    """
    return dump_all([data], stream, Dumper=SafeDumper, **kwds)

调用的 dump_all 方法

def dump_all(documents, stream=None, Dumper=Dumper,
        default_style=None, default_flow_style=False,
        canonical=None, indent=None, width=None,
        allow_unicode=None, line_break=None,
        encoding=None, explicit_start=None, explicit_end=None,
        version=None, tags=None, sort_keys=True):
    """
    Serialize a sequence of Python objects into a YAML stream.
    If stream is None, return the produced string instead.
    """

allow_unicode 设置为 True 即可解决中文的问题

# 写入yaml 文件
with open("data.yml", 'w', encoding="utf-8") as fp:
    yaml.safe_dump(data, fp, indent=4,
        encoding='utf-8',
        allow_unicode=True,
        )

运行后生成的yaml结构

get请求:
    name: GET请求示例
    request:
        method: GET
        url: /get
    sleep: 3

排序问题

python的字典是有序的,生成yaml 后发现key 是按a-z 的顺序排列,并不是python里面定义的字典的顺序

加上 sort_keys=Flase 可以解决排序问题

# 写入yaml 文件
with open("data.yml", 'w', encoding="utf-8") as fp:
    yaml.safe_dump(data, fp, indent=4,
        # default_flow_style=False,
        encoding='utf-8',
        allow_unicode=True,
        sort_keys=False
        )

运行后

get请求:
    name: GET请求示例
    sleep: 3
    request:
        method: GET
        url: /get

需注意的是 sort_keys=False 在一些比较低的 PyYAML 库中并没有这个参数,升级到最新版就行了。
default_flow_style=False 是设置yaml格式的风格,默认为False就可以了,新版的PyYAML 已经默认为False 了。

原文链接:https://mp.weixin.qq.com/s/YegFouCnCwtFClL1vua9ZA

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
禁止转载,如需转载请通过简信或评论联系作者。

相关阅读更多精彩内容

友情链接更多精彩内容