Re-factor to put screen erase + screen draw as close as possible

Author: eriknyquist

examples/readme_example.py

@@ -23,8 +23,8 @@ def keypress_event(e):
 game = SnakeGame()
 
 while True:
-    new_state = game.process(last_direction)  # Produce new frame
-    sys.stdout.write("\033[2J\n")             # Clear terminal screen
-    sys.stdout.write(new_state.to_string())   # Print new game state
-    sys.stdout.flush()                        # Flush output
+    new_state = game.process(last_direction)           # Produce new frame
+    new_state_string = new_state.to_string()
+    sys.stdout.write("\033[2J\n" + new_state_string)   # Clear terminal screen and print new game state
+    sys.stdout.flush()                                 # Flush output
     time.sleep(0.05)

snakeng/__main__.py

@@ -35,18 +35,14 @@ def keypress_event(e):
     runtime_data['last_direction'] = ret
 
 
-def draw_screen(state):
-    sys.stdout.write("\033[2J\n")
-    sys.stdout.write(state.to_string())
-    sys.stdout.flush()
-
-
 def process_frame(game, frame_time):
     runtime_data["scheduler_event"] = scheduler.enterabs(time.time() + frame_time, 0, process_frame, argument=(game, frame_time))
 
     if not runtime_data['paused']:
         new_state = game.process(runtime_data['last_direction'])
-        draw_screen(new_state)
+        new_state_string = new_state.to_string()
+        sys.stdout.write("\033[2J\n" + new_state_string)
+        sys.stdout.flush()
 
         if new_state.dead:
             scheduler.cancel(runtime_data["scheduler_event"])