如何在YAML中使用双引号和空格打印值?

时间:2017-11-28 22:58:12

标签: yaml python-3.5 pyyaml

我正在尝试将Python字典转储到YAML,其中有一些字符串作为值字段。

import yaml
str1 = "hello"
str2 = "world"
mystr = "\"" + str1 + str(" ") + str2 + "\""
mydict = {"a" : mystr}
f = open("temp.yaml", "w")
yaml.dump(mydict, f, default_flow_style = False, \
                    explicit_start = "---", explicit_end = "...", encoding = 'UTF-8')
f.close()
我得到的YAML是:

a: '"hello
 world"'

注意,值“hello world”溢出到下一行。 我使用的是python 3.5,YAML模块的版本是3.11

有人能告诉我如何让YAML看起来像下面的那样吗?

a: "hello world"

2 个答案:

答案 0 :(得分:2)

代码有点草率但它会给出你想要的结果。

global dict_keys

def mk_double_quote(dumper, data):
    if data in dict_keys:
        return dumper.represent_scalar('tag:yaml.org,2002:str', data, style='')
    else:
        return dumper.represent_scalar('tag:yaml.org,2002:str', data, style='"')

yaml.add_representer(str, mk_double_quote)

d = {'a': 'Hello World'}
dict_keys = set(d.keys())

f = open('temp.yaml', 'w')
yaml.dump(d, f, default_flow_style=False, encoding='utf-8')
f.close()

结果如下:a: "Hello World"

答案 1 :(得分:1)

如果你希望细粒度控制哪个字符串得到双(或单)引号,你应该使用ruamel.yaml(免责声明:我是该包的作者)。它是PyYAML的改进版本,PyYAML中存在许多长期存在的问题。

有了它,你可以这样做:

import sys
import ruamel.yaml

mystr = ruamel.yaml.scalarstring.DoubleQuotedScalarString('hello world')
mydict = dict(a=mystr)

yaml = ruamel.yaml.YAML()
yaml.dump(mydict, sys.stdout)

得到:

a: "hello world"

如果您从YAML文档开始,事情就更容易了,您可以表明您想要保留这些多余的引号:

yaml_str = """\
a: "hello world"
"""

yaml = ruamel.yaml.YAML()
yaml.preserve_quotes = True
data = yaml.load(yaml_str)
yaml.dump(data, sys.stdout)