add gi_code builtin property for generators - get access to code attributes like co_cellvars / co_freevars

Author: cosminbasca

graalpython/com.oracle.graal.python.test/src/tests/test_generators.py

@@ -168,3 +168,11 @@ def f():
 
 if sys.version_info.minor == 4 and sys.version_info.micro < 3:
     del ExceptionTest
+
+
+def test_unboundlocalerror_gen():
+    def func(args):
+        args = (arg for arg in args)
+        return set(args)
+
+    assert func([1, 2, 3, 4]) == {1, 2, 3, 4}

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/generator/GeneratorBuiltins.java

@@ -27,6 +27,7 @@
 
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__ITER__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__NEXT__;
+import static com.oracle.graal.python.runtime.exception.PythonErrorType.NotImplementedError;
 import static com.oracle.graal.python.runtime.exception.PythonErrorType.StopIteration;
 import static com.oracle.graal.python.runtime.exception.PythonErrorType.TypeError;
 
@@ -45,6 +46,7 @@
 import com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinNode;
 import com.oracle.graal.python.nodes.function.builtins.PythonUnaryBuiltinNode;
+import com.oracle.graal.python.runtime.PythonParseResult;
 import com.oracle.graal.python.runtime.exception.PException;
 import com.oracle.truffle.api.dsl.Cached;
 import com.oracle.truffle.api.dsl.GenerateNodeFactory;
@@ -173,4 +175,19 @@ Object sendThrow(PGenerator self, @SuppressWarnings("unused") PythonClass typ, P
             return resumeGenerator(self);
         }
     }
+
+    @Builtin(name = "gi_code", minNumOfArguments = 1, maxNumOfArguments = 2, isGetter = true, isSetter = true)
+    @GenerateNodeFactory
+    public abstract static class GetCodeNode extends PythonBuiltinNode {
+        @Specialization(guards = {"isNoValue(none)"})
+        Object getCode(PGenerator self, @SuppressWarnings("unused") PNone none) {
+            return new PythonParseResult(self.getGeneratorRootNode(), getCore());
+        }
+
+        @SuppressWarnings("unused")
+        @Specialization
+        Object setCode(PGenerator self, PythonParseResult code) {
+            throw raise(NotImplementedError, "setting gi_code");
+        }
+    }
 }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/generator/PGenerator.java

@@ -36,6 +36,7 @@
 import com.oracle.truffle.api.frame.FrameDescriptor;
 import com.oracle.truffle.api.frame.FrameSlot;
 import com.oracle.truffle.api.frame.MaterializedFrame;
+import com.oracle.truffle.api.nodes.RootNode;
 
 public final class PGenerator extends PythonBuiltinObject {
 
@@ -92,6 +93,10 @@ public RootCallTarget getCallTarget() {
         return callTarget;
     }
 
+    public RootNode getGeneratorRootNode() {
+        return callTarget.getRootNode();
+    }
+
     public Object[] getArguments() {
         return arguments;
     }