Add if-statement node

1 files changed, 15 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 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("(");