Remove duplication of statement writing

1 files changed, 12 insertions, 3 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonWriter.java b/src/main/java/org/zwobble/couscous/backends/python/PythonWriter.java
index 2551700..781db71 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonWriter.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonWriter.java
@@ -47,14 +47,23 @@ public class PythonWriter {
         depth--;
     }
 
-    public void startStatement() {
+    public void writeStatement(Action action) {
+        startStatement();
+        action.run();
+        endStatement();
+    }
+
+    private void startStatement() {
         builder.append(Strings.repeat(" ", depth * SPACES_PER_INDENT));
     }
 
-    public void endStatement() {
+    private void endStatement() {
         if (builder.charAt(builder.length() - 1) != '\n') {
             builder.append("\n");   
         }
     }
-
+    
+    interface Action {
+        void run();
+    }
 }