Dump py.Objects with __str__ if possible

Author: ncw

ast/dump.go

@@ -51,6 +51,8 @@ func Dump(ast Ast) string {
 			case Pos:
 			case Ast:
 				args = append(args, fmt.Sprintf("%s=%s", fname, Dump(x)))
+			case py.I__str__:
+				args = append(args, fmt.Sprintf("%s=%s", fname, x.M__str__()))
 			default:
 				args = append(args, fmt.Sprintf("%s=%v", fname, x))
 			}

ast/dump_test.go

@@ -20,6 +20,7 @@ func TestDump(t *testing.T) {
 		{&Module{}, `Module(body=[])`},
 		{&Module{Body: []Stmt{&Pass{}}}, `Module(body=[Pass()])`},
 		{&Module{Body: []Stmt{&ExprStmt{Value: &Tuple{}}}}, `Module(body=[Expr(value=Tuple(elts=[], ctx=UnknownExprContext(0)))])`},
+		{&NameConstant{Value: py.True}, `NameConstant(value=True)`},
 	} {
 		out := Dump(test.in)
 		if out != test.out {