This commit is contained in:
Paul Tagliamonte 2012-12-22 19:38:54 -05:00
parent a0842d278a
commit b078155661

View File

@ -77,6 +77,10 @@ def _ast_if(node, children, obj):
return ret return ret
def _ast_do(node, children, obj):
return children
def _ast_return(node, children, obj): def _ast_return(node, children, obj):
return ast.Return(value=children[-1]) return ast.Return(value=children[-1])
@ -102,6 +106,7 @@ special_cases = {
"if": _ast_if, "if": _ast_if,
"return": _ast_return, "return": _ast_return,
"do": _ast_do,
} }
@ -153,7 +158,9 @@ class AST27Converter(object):
for child in args: for child in args:
c.append(self.render(child)) c.append(self.render(child))
body = [c[-1]] cont = c[-1] # XXX: Wrong...
body = cont if isinstance(cont, list) else [cont]
if doc: if doc:
# Shim in docstrings # Shim in docstrings
body.insert(0, ast.Expr(value=ast.Str(s=str(doc)))) body.insert(0, ast.Expr(value=ast.Str(s=str(doc))))