From 1448733a03e965d5cb9261a3519370ff5b34122d Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 5 Oct 2015 17:21:55 +0100
Subject: Support ternary conditionals in Python backend

---
 .../python/ast/PythonConditionalExpressionNode.java     | 17 +++++++++++++++++
 .../backends/python/ast/visitors/PythonNodeVisitor.java |  2 ++
 2 files changed, 19 insertions(+)
 create mode 100644 src/main/java/org/zwobble/couscous/backends/python/ast/PythonConditionalExpressionNode.java

(limited to 'src/main/java/org/zwobble/couscous/backends/python/ast')

diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/PythonConditionalExpressionNode.java b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonConditionalExpressionNode.java
new file mode 100644
index 0000000..9fd406c
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonConditionalExpressionNode.java
@@ -0,0 +1,17 @@
+package org.zwobble.couscous.backends.python.ast;
+
+import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
+
+import lombok.Value;
+
+@Value(staticConstructor="pythonConditionalExpression")
+public class PythonConditionalExpressionNode implements PythonExpressionNode {
+    PythonExpressionNode condition;
+    PythonExpressionNode trueValue;
+    PythonExpressionNode falseValue;
+    
+    @Override
+    public <T> T accept(PythonNodeVisitor<T> visitor) {
+        return visitor.visit(this);
+    }
+}
diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
index f5cd04f..1f5b1a2 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
@@ -3,6 +3,7 @@ package org.zwobble.couscous.backends.python.ast.visitors;
 import org.zwobble.couscous.backends.python.ast.PythonAssignmentNode;
 import org.zwobble.couscous.backends.python.ast.PythonBooleanLiteralNode;
 import org.zwobble.couscous.backends.python.ast.PythonClassNode;
+import org.zwobble.couscous.backends.python.ast.PythonConditionalExpressionNode;
 import org.zwobble.couscous.backends.python.ast.PythonFunctionDefinitionNode;
 import org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode;
 import org.zwobble.couscous.backends.python.ast.PythonModuleNode;
@@ -16,6 +17,7 @@ public interface PythonNodeVisitor<T> {
     T visit(PythonStringLiteralNode stringLiteral);
     T visit(PythonBooleanLiteralNode booleanLiteral);
     T visit(PythonVariableReferenceNode reference);
+    T visit(PythonConditionalExpressionNode conditional);
     
     T visit(PythonReturnNode pythonReturn);
     T visit(PythonPassNode pass);
-- 
cgit v1.2.3