Added 'globals' to the context

burakozturkdot · Dec 28, 2016 · 792b266 · 792b266
1 parent 992d91a
commit 792b266
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/pythonlua/context.py b/pythonlua/context.py
@@ -10,6 +10,7 @@ def __init__(self, values=None):
             "token_end_mode": TokenEndMode.LINE_FEED,
             "class_name": "",
             "locals": SymbolsStack(),
+            "globals": SymbolsStack(),  # Not working yet
         }
 
         self.ctx_stack = [values]

diff --git a/pythonlua/nodevisitor.py b/pythonlua/nodevisitor.py
@@ -263,6 +263,12 @@ def visit_For(self, node):
         self.emit(line.format(**values))
         self.visit_all(node.body)
         self.emit("end")
+
+    def visit_Global(self, node):
+        """Visit globals"""
+        last_ctx = self.context.last()
+        for name in node.names:
+            last_ctx["globals"].add_symbol(name)
 
     def visit_If(self, node):
         """Visit if"""

diff --git a/tests/global.py b/tests/global.py
@@ -0,0 +1,9 @@
+foo = 42
+
+def bar():
+    global foo
+    foo = 34
+
+print("foo = ", foo)
+bar()
+print("foo = ", foo)
diff --git a/tests/global.py.expected b/tests/global.py.expected