正确地从字符串**创建一个lambda函数**

时间:2012-06-20 02:28:06

标签: python string lambda

给出一个字符串,如

"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

2 个答案:

答案 0 :(得分:14)

您正在寻找eval的替代品,但为什么?你接受任意代码并执行它,所以为什么不使用eval?避免eval的唯一原因是因为它很危险,但你最终创建的lambda也同样危险。

另请注意,you really can't make it safe to do this in CPython

答案 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)