dump truffle ast builtin: in case of assertRaises

Author: cosminbasca

graalpython/com.oracle.graal.python.test/src/graalpytest.py

@@ -52,6 +52,13 @@
 verbose = False
 
 
+def dump_truffle_ast(func):
+    try:
+        print(__dump_truffle_ast__(func))
+    except:
+        pass
+
+
 class SkipTest(BaseException):
     pass
 
@@ -152,7 +159,6 @@ def assertSequenceEqual(self, expected, actual, msg=None):
             assert expected_value == next(actual_iter), msg
 
     class assertRaises():
-
         def __init__(self, exc_type, function=None, *args, **kwargs):
             self.function = function
             if self.function is None:
@@ -163,6 +169,7 @@ def __init__(self, exc_type, function=None, *args, **kwargs):
                 except exc_type:
                     pass
                 else:
+                    dump_truffle_ast(self.__init__)
                     assert False, "expected '%r' to raise '%r'" % (self.function, exc_type)
 
         def __enter__(self):

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/BuiltinFunctions.java

@@ -54,6 +54,8 @@
 import static com.oracle.graal.python.nodes.BuiltinNames.SETATTR;
 import static com.oracle.graal.python.nodes.BuiltinNames.SUM;
 import static com.oracle.graal.python.nodes.BuiltinNames.__BREAKPOINT__;
+import static com.oracle.graal.python.nodes.BuiltinNames.__BUILTIN__;
+import static com.oracle.graal.python.nodes.BuiltinNames.__DUMP_TRUFFLE_AST__;
 import static com.oracle.graal.python.nodes.HiddenAttributes.ID_KEY;
 import static com.oracle.graal.python.nodes.SpecialAttributeNames.__NAME__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__INSTANCECHECK__;
@@ -91,6 +93,7 @@
 import com.oracle.graal.python.builtins.objects.function.PythonCallable;
 import com.oracle.graal.python.builtins.objects.ints.PInt;
 import com.oracle.graal.python.builtins.objects.list.PList;
+import com.oracle.graal.python.builtins.objects.method.PMethod;
 import com.oracle.graal.python.builtins.objects.module.PythonModule;
 import com.oracle.graal.python.builtins.objects.object.PythonObject;
 import com.oracle.graal.python.builtins.objects.set.PFrozenSet;
@@ -155,6 +158,7 @@
 import com.oracle.truffle.api.frame.VirtualFrame;
 import com.oracle.truffle.api.nodes.ExplodeLoop;
 import com.oracle.truffle.api.nodes.Node;
+import com.oracle.truffle.api.nodes.NodeUtil;
 import com.oracle.truffle.api.nodes.NodeVisitor;
 import com.oracle.truffle.api.nodes.RootNode;
 import com.oracle.truffle.api.nodes.UnexpectedResultException;
@@ -1312,7 +1316,7 @@ private Object iterateGeneric(Object iterator, Object start, ConditionProfile er
         }
     }
 
-    @Builtin(name = "__builtin__", fixedNumOfPositionalArgs = 1)
+    @Builtin(name = __BUILTIN__, fixedNumOfPositionalArgs = 1)
     @GenerateNodeFactory
     public abstract static class BuiltinNode extends PythonUnaryBuiltinNode {
         @Child GetItemNode getNameNode = GetItemNode.create();
@@ -1376,4 +1380,26 @@ public boolean visit(Node node) {
             return factory().createBuiltinMethod(builtinModule, builtinFunc);
         }
     }
+
+    @Builtin(name = __DUMP_TRUFFLE_AST__, fixedNumOfPositionalArgs = 1)
+    @GenerateNodeFactory
+    public abstract static class DumpTruffleAstNode extends PythonUnaryBuiltinNode {
+        @Specialization
+        @TruffleBoundary
+        public String doIt(PFunction func) {
+            return NodeUtil.printTreeToString(func.getCallTarget().getRootNode());
+        }
+
+        @Specialization
+        @TruffleBoundary
+        public String doIt(PMethod method) {
+            return NodeUtil.printTreeToString(method.getCallTarget().getRootNode());
+        }
+
+        @Fallback
+        @TruffleBoundary
+        public Object doit(Object object) {
+            return "truffle ast dump not supported for " + object.toString();
+        }
+    }
 }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/nodes/BuiltinNames.java

@@ -132,4 +132,7 @@ public abstract class BuiltinNames {
     public static final String HASH = "hash";
     public static final String MEMORYVIEW = "memoryview";
     public static final String SET = "set";
+    // truffle specific builtins
+    public static final String __BUILTIN__ = "__builtin__";
+    public static final String __DUMP_TRUFFLE_AST__ = "__dump_truffle_ast__";
 }