给出一个字符串,如
"2*(i+j) <= 100"
我想生成相应的lambda函数,
fn = lambda i,j: 2*(i+j) <= 100
我可以用eval
做到这一点,但我正在寻找一种不那么邪恶的方法。
我找到了
import ast
f = ast.Lambda('i,j', '2*(i+j) <= 100')
但我还没弄清楚如何执行结果!
理想情况下,我想自动拉出参数列表('i','j') - 现在,我只是使用re.findall('\ w +'),但我会我希望能够正确使用cos
等现有功能,而不是将其视为“关键字”。
我正在查看Is there a Python library for handling complicated mathematical sets (constructed using mathematical set-builder notation)?并尝试找出如何最好地将set-builder符号解析为lambdas以提供给约束求解器。
我基本上希望ast.literal_eval也能识别变量。
理想情况下,考虑到i >= 20
,我想返回((lambda x: x >= 20), ['i'])
,然后我可以直接将其提供给constraint
。
答案 0 :(得分:14)
您正在寻找eval
的替代品,但为什么?你接受任意代码并执行它,所以为什么不使用eval
?避免eval
的唯一原因是因为它很危险,但你最终创建的lambda也同样危险。
答案 1 :(得分:3)
如果您的输入来自可信来源,则 eval()是最简单,最清晰,最可靠的方式。
如果您的输入不信任,则需要清理。
一种合理的方法是使用正则表达式。确保字符串中没有函数调用,属性查找或双下划线。
或者,更复杂的方法是遍历AST解析树以确定是否存在任何令人反感的调用。
第三种方法是遍历AST解析树并直接执行它。这使您可以完全控制调用的内容。 ast.literal_eval函数采用这种方法。也许你从源代码开始,为你想要支持的任何操作做一些扩建:
def literal_eval(node_or_string):
"""
Safely evaluate an expression node or a string containing a Python
expression. The string or node provided may only consist of the following
Python literal structures: strings, numbers, tuples, lists, dicts, booleans,
and None.
"""
_safe_names = {'None': None, 'True': True, 'False': False}
if isinstance(node_or_string, basestring):
node_or_string = parse(node_or_string, mode='eval')
if isinstance(node_or_string, Expression):
node_or_string = node_or_string.body
def _convert(node):
if isinstance(node, Str):
return node.s
elif isinstance(node, Num):
return node.n
elif isinstance(node, Tuple):
return tuple(map(_convert, node.elts))
elif isinstance(node, List):
return list(map(_convert, node.elts))
elif isinstance(node, Dict):
return dict((_convert(k), _convert(v)) for k, v
in zip(node.keys, node.values))
elif isinstance(node, Name):
if node.id in _safe_names:
return _safe_names[node.id]
elif isinstance(node, BinOp) and \
isinstance(node.op, (Add, Sub)) and \
isinstance(node.right, Num) and \
isinstance(node.right.n, complex) and \
isinstance(node.left, Num) and \
isinstance(node.left.n, (int, long, float)):
left = node.left.n
right = node.right.n
if isinstance(node.op, Add):
return left + right
else:
return left - right
raise ValueError('malformed string')
return _convert(node_or_string)