python - 将 Python f 字符串与 Yaml 文件结合使用？

Question

如果我有一个 yaml 文件，其中包含与 python f-strings 合作使用的带有括号符号 {} 的字符串，那么在这里如何利用 f-string 插值？以这个简单的 yaml 文件为例：

# tmp.yaml
k1: val1
k2: val2 as well as {x}

如果x = 'val3'，我希望 k2 的值能够反映val2 as well as val3

# app.py
x = 'val3'
with open('tmp.yaml', 'rt') as f:
    conf = yaml.safe_load(f)

print(conf)
{'k1': 'val1', 'k2': 'val2 as well as {x}'}

这可以很容易地用格式字符串来完成......

print(conf['k2'].format(x=x))
val2 as well as val3

但是如何对 f-strings 做同样的事情呢？

Answer 1

您可以定义自定义构造函数：

import yaml

values = { 'x': 'val3' }

def format_constructor(loader, node):
  return loader.construct_scalar(node).format(**values)

yaml.SafeLoader.add_constructor(u'!format', format_constructor)

conf = yaml.safe_load("""
k1: val1
k2: !format val2 as well as {x}
""")

print(conf)

如果不想使用标签!format，也可以使用add_constructorwith u'tag:yaml.org,2002:str'as 标签。这将用你的覆盖默认的字符串构造函数。

Answer 2

我发现它jinja2为这个问题提供了最简单的解决方案。

源 yaml 文件：

# tmp.yaml
k1: val1
k2: val2 as well as {{ x }}

读取文件并使用 jinja 模板进行渲染：

with open('tmp.yaml', 'rt') as f:
    conf = f.read().rstrip()

print(conf)
# 'k1: val1\nk2: val2 as well as {{ x }}'

import jinja2
template = Template(conf)
conf = template.render(x='val3')
config = yaml.safe_load(conf)

print(config)
# {'k1': 'val1', 'k2': 'val2 as well as val3'}