python - 在python中评估if的条件

Question

我有 python 程序的 AST，想手动评估 if 语句的条件。

cond = node.test
b = eval(compile(cond,"<string>","eval"))
print b

node 是 If-Node，给我 TypeError: expected Expression node, got Compare，即使根据 ast 的 python 文档中的语法，Compare 是一个表达式。

有什么想法吗？

Answer 1

你有一个 ast.expr 子类，不是 ast.Expression 顶级节点。

compile() 只能接受一个 mod 对象，所以 Module、Interactive 或 Expression，取决于 compile() 的第三个参数。对于 'eval'，使用 ast.Expression()。

您可以创建一个包含 ast.Compare 节点:

expr = ast.Expression(cond)

因为 abstract grammar定义为:

Expression(expr body)

你可以编译:

compile(expr, '<file>', 'eval')

演示:

>>> import ast
>>> code = "if foo == 'bar': pass"
>>> tree = ast.parse(code, '<file>', 'exec')
>>> cond = tree.body[0].test
>>> expr = ast.Expression(cond)
>>> compile(expr, '<file>', 'eval')
<code object <module> at 0x1067f6230, file "<file>", line 1>
>>> foo = 'baz'
>>> eval(compile(expr, '<file>', 'eval'))
False
>>> foo = 'bar'
>>> eval(compile(expr, '<file>', 'eval'))
True