加载 yaml 字符串时跳过转换实体（使用 PyYAML）

Question

在将 YAML 字符串加载到 yaml 包时，是否有一种很好的方法可以防止实体转换为 python 对象？特别是，我不希望将时间戳字符串转换为日期时间对象。

这是一个例子：

import yaml
yaml.load("""d: 2018-06-17\nn: 42""")

给了

{'d': datetime.date(2018, 6, 17), 'n': 42}

但我想拥有

{'d': '2018-06-17', n: 42}

其中日期字符串保留为字符串，其他类型被转换。我不想更改输入字符串，例如，通过指定特定的数据类型。也许有一个替代的 YAML 加载器/解析器包。我正在使用 python3.6 和 PyYAML==3.12。

Answer 1

YAML 有几个模式，由于您在 PyYAML 中使用默认（不安全）加载，您可以获得它支持的 Python 对象的所有构造，包括您想要的 int 和 @987654322 的一个@你不想要的。

由于要转换整数，所以不能使用 baseloader：

import yaml
data = yaml.load("""d: 2018-06-17\nn: 42""", Loader=yaml.BaseLoader)
print(data)

因为这会在任何地方提供字符串：

{'d': '2018-06-17', 'n': '42'}

将匹配的日期时间对象作为字符串处理可能是最简单的。在我的 ruamel.yaml 库中，您可以使用：

import ruamel.yaml

yaml = ruamel.yaml.YAML(typ='safe')
yaml.constructor.yaml_constructors[u'tag:yaml.org,2002:timestamp'] = \
   yaml.constructor.yaml_constructors[u'tag:yaml.org,2002:str']
data = yaml.load("""d: 2018-06-17\nn: 42""")
print(data)

如果你只需要支持旧的 YAML 1.1 规范，你可以在 PyYAML 中做同样的事情：

import yaml
import yaml.constructor
yaml.constructor.SafeConstructor.yaml_constructors[u'tag:yaml.org,2002:timestamp'] = \
    yaml.constructor.SafeConstructor.yaml_constructors[u'tag:yaml.org,2002:str']

data = yaml.safe_load("""d: 2018-06-17\nn: 42""")
print(data)

两个打印：

{'d': '2018-06-17', 'n': 42}

Answer 2

我认为这种行为深深植根于 YAML 规范中，我尝试了ruamel，但结果是一样的。

from ruamel.yaml import YAML
from datetime import date

yaml=YAML(typ="unsafe")
assert yaml.load("""d: 2018-06-17\nn: 42""") == \
       {'d': date(2018, 6, 17), 'n': 42}

如果我遇到这个问题，我会尝试在需要的地方将 YAML 导入转换回 .strftime('YYYY-MM-DD')。