这是我在kadee's elegant answer 对类似问题的启发下对 f 字符串进行更稳健评估的尝试。
不过,我想避免eval 方法的一些基本缺陷。例如,只要模板包含撇号,eval(f"f'{template}'") 就会失败,例如the string's evaluation 变为 f'the string's evaluation',其计算结果出现语法错误。第一个改进是使用三撇号:
eval(f"f'''{template}'''")
现在在模板中使用撇号(大部分)是安全的,只要它们不是三撇号。 (三引号也可以。)一个值得注意的例外是字符串末尾的撇号:whatcha doin' 变为 f'''whatcha doin'''',在连续第四个撇号处出现语法错误。下面的代码通过去除字符串末尾的撇号并在评估后将它们放回原处来避免这个特殊问题。
import builtins
def fstr_eval(_s: str, raw_string=False, eval=builtins.eval):
r"""str: Evaluate a string as an f-string literal.
Args:
_s (str): The string to evaluate.
raw_string (bool, optional): Evaluate as a raw literal
(don't escape \). Defaults to False.
eval (callable, optional): Evaluation function. Defaults
to Python's builtin eval.
Raises:
ValueError: Triple-apostrophes ''' are forbidden.
"""
# Prefix all local variables with _ to reduce collisions in case
# eval is called in the local namespace.
_TA = "'''" # triple-apostrophes constant, for readability
if _TA in _s:
raise ValueError("Triple-apostrophes ''' are forbidden. " + \
'Consider using """ instead.')
# Strip apostrophes from the end of _s and store them in _ra.
# There are at most two since triple-apostrophes are forbidden.
if _s.endswith("''"):
_ra = "''"
_s = _s[:-2]
elif _s.endswith("'"):
_ra = "'"
_s = _s[:-1]
else:
_ra = ""
# Now the last character of s (if it exists) is guaranteed
# not to be an apostrophe.
_prefix = 'rf' if raw_string else 'f'
return eval(_prefix + _TA + _s + _TA) + _ra
不指定求值函数,这个函数的局部变量是可以访问的,所以
print(fstr_eval(r"raw_string: {raw_string}\neval: {eval}\n_s: {_s}"))
打印
raw_string: False
eval: <built-in function eval>
_s: raw_string: {raw_string}\neval: {eval}\n_s: {_s}
虽然前缀 _ 减少了意外碰撞的可能性,但可以通过传递适当的评估函数来避免该问题。例如,可以通过lambda 传递当前全局命名空间:
fstr_eval('{_s}', eval=lambda expr: eval(expr))#NameError: name '_s' is not defined
或者更一般地,通过将合适的 globals 和 locals 参数传递给 eval,例如
fstr_eval('{x+x}', eval=lambda expr: eval(expr, {}, {'x': 7})) # 14
我还包含了一种机制,可以通过“原始字符串文字”机制选择是否应将\ 视为转义字符。例如,
print(fstr_eval(r'x\ny'))
产量
x
y
同时
print(fstr_eval(r'x\ny', raw_string=True))
产量
x\ny
可能还有其他我没有注意到的陷阱,但出于许多目的,我认为这就足够了。