From f7a7bfd4764f2787e022fc0a9dd8eb57c151e90d Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 7 Nov 2015 15:54:07 +0000
Subject: Add if-statement node

---
 .../couscous/backends/python/PythonSerializer.java        | 15 +++++++++++++++
 1 file changed, 15 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 2fb341f..4eea760 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -13,6 +13,7 @@ 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.PythonGetSliceNode;
+import org.zwobble.couscous.backends.python.ast.PythonIfStatementNode;
 import org.zwobble.couscous.backends.python.ast.PythonImportNode;
 import org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode;
 import org.zwobble.couscous.backends.python.ast.PythonModuleNode;
@@ -192,6 +193,20 @@ public class PythonSerializer implements PythonNodeVisitor {
             });
         });
     }
+
+    @Override
+    public void visit(PythonIfStatementNode ifStatement) {
+        writer.writeStatement(() -> {
+            writer.writeKeyword("if");
+            writer.writeSpace();
+            write(ifStatement.getCondition());
+            writeBlock(ifStatement.getTrueBranch());
+        });
+        writer.writeStatement(() -> {
+            writer.writeKeyword("else");
+            writeBlock(ifStatement.getFalseBranch()); 
+        });
+    }
     
     private void writeParenthesised(PythonExpressionNode expression) {
         writer.writeSymbol("(");
-- 
cgit v1.2.3