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

---
 .../couscous/backends/python/PythonSerializer.java | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

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

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 bc5fb19..728b71a 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -4,6 +4,8 @@ import org.zwobble.couscous.backends.python.ast.PythonAssignmentNode;
 import org.zwobble.couscous.backends.python.ast.PythonBlock;
 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.PythonExpressionNode;
 import org.zwobble.couscous.backends.python.ast.PythonFunctionDefinitionNode;
 import org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode;
 import org.zwobble.couscous.backends.python.ast.PythonModuleNode;
@@ -61,6 +63,20 @@ public class PythonSerializer implements PythonNodeVisitor<Void> {
         return null;
     }
 
+    @Override
+    public Void visit(PythonConditionalExpressionNode conditional) {
+        writeParenthesised(conditional.getTrueValue());
+        writer.writeSpace();
+        writer.writeKeyword("if");
+        writer.writeSpace();
+        writeParenthesised(conditional.getCondition());
+        writer.writeSpace();
+        writer.writeKeyword("else");
+        writer.writeSpace();
+        writeParenthesised(conditional.getFalseValue());
+        return null;
+    }
+
     @Override
     public Void visit(PythonReturnNode pythonReturn) {
         writer.startStatement();
@@ -119,6 +135,12 @@ public class PythonSerializer implements PythonNodeVisitor<Void> {
         return null;
     }
 
+    private void writeParenthesised(PythonExpressionNode expression) {
+        writer.writeSymbol("(");
+        write(expression);
+        writer.writeSymbol(")");
+    }
+
     private void writeArgumentNames(PythonFunctionDefinitionNode functionDefinition) {
         if (!functionDefinition.getArgumentNames().isEmpty()) {
             writer.writeIdentifier(functionDefinition.getArgumentNames().get(0));
-- 
cgit v1.2.3