adding do support

2012-12-22 19:46:10 -05:00 · 2012-12-22 19:46:10 -05:00 · 5b50ea4907
commit 5b50ea4907
parent b078155661
1 changed files with 5 additions and 2 deletions
--- a/hy/compiler/ast27.py
+++ b/hy/compiler/ast27.py
@ -69,10 +69,13 @@ def _ast_if(node, children, obj):
    true = 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(
        test=cond,
-        body=[true],
-        orelse=[flse]
+        body=true,
+        orelse=flse,
    )
    return ret