adding do support

This commit is contained in:
Paul Tagliamonte 2012-12-22 19:46:10 -05:00
parent b078155661
commit 5b50ea4907

View File

@ -69,10 +69,13 @@ def _ast_if(node, children, obj):
true = children.pop(0) true = children.pop(0)
flse = children.pop(0) flse = children.pop(0)
true = true if isinstance(true, list) else [true]
flse = flse if isinstance(flse, list) else [flse]
ret = ast.If( ret = ast.If(
test=cond, test=cond,
body=[true], body=true,
orelse=[flse] orelse=flse,
) )
return ret return ret