register decorators as possible cells

Author: timfel

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

@@ -88,3 +88,13 @@ def test_lambda_no_args_with_nested_lambdas():
 
 def test_byte_numeric_escapes():
     assert eval('b"PK\\005\\006\\00\\11\\22\\08"') == b'PK\x05\x06\x00\t\x12\x008'
+
+
+def test_decorator_cell():
+    foo = lambda x: "just a string, not %s" % x.__name__
+    def run_me():
+        @foo
+        def func():
+            pass
+        return func
+    assert run_me() == "just a string, not func", run_me()

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/parser/ScopeTranslator.java

@@ -100,11 +100,14 @@ public T visitEval_input(Python3Parser.Eval_inputContext ctx) {
         }
     }
 
+    @Override
+    public T visitDecorator(Python3Parser.DecoratorContext ctx) {
+        registerPossibleCell(ctx.dotted_name().NAME(0).getText());
+        return super.visitDecorator(ctx);
+    }
+
     @Override
     public T visitFuncdef(Python3Parser.FuncdefContext ctx) {
-        if (ctx.parent instanceof Python3Parser.DecoratedContext) {
-            // TODO: get the decorators
-        }
         environment.createLocal(ctx.NAME().getText());
         ArgListCompiler<T> argListCompiler = new ArgListCompiler<>(errors);
         argListCompilers.add(argListCompiler);