python - 使用带有 Yaml 文件的 Python f 字符串？

Question

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

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

如 x = 'val3' , 我希望 k2 的值反射(reflect) 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_constructor与 u'tag:yaml.org,2002:str'作为标签。这将使用您的默认字符串构造函数覆盖。