Support passing arguments to Python runner

1 files changed, 7 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
index c6cf516..4893cb6 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -10,6 +10,7 @@ import org.zwobble.couscous.backends.python.ast.PythonNode;
 import org.zwobble.couscous.backends.python.ast.PythonPassNode;
 import org.zwobble.couscous.backends.python.ast.PythonReturnNode;
 import org.zwobble.couscous.backends.python.ast.PythonStringLiteralNode;
+import org.zwobble.couscous.backends.python.ast.PythonVariableReferenceNode;
 import org.zwobble.couscous.backends.python.ast.PythonWriter;
 import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
 
@@ -54,6 +55,12 @@ public class PythonSerializer implements PythonNodeVisitor<Void> {
     }
 
     @Override
+    public Void visit(PythonVariableReferenceNode reference) {
+        writer.writeIdentifier(reference.getName());
+        return null;
+    }
+
+    @Override
     public Void visit(PythonReturnNode pythonReturn) {
         writer.startStatement();
         writer.writeKeyword("return");