给出一个字符串如
"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
而不是将它们视为“关键字”。
我在看 是否有用于处理复杂数学集的Python库(使用数学集合构建器表示法构造)? 并试图弄清楚如何最好地将set-builder符号解析为lambdas以提供给约束求解器。
我基本上希望ast.literal_eval也可以识别变量。
理想情况下,给定 i >= 20
我想回去 ((lambda x: x >= 20), ['i'])
我可以直接喂到 constraint
。
如果您的输入来自a 可靠来源, 的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)
你正在寻找替代品 eval
, 但为什么?你接受任意代码并执行它,所以为什么不使用 eval
?避免的唯一理由 eval
是因为它很危险,但你最终创造的lambda也同样危险。
另外,请记住, 你真的不能在CPython中做到这一点