1 files changed, 74 insertions, 0 deletions
diff --git a/console_snake.py b/console_snake.py
new file mode 100755
index 0000000..8c327e5
--- /dev/null
+++ b/console_snake.py
@@ -0,0 +1,74 @@
+#!/usr/bin/env python
+
+from __future__ import division
+
+import time
+
+import curses
+
+from common import *
+from snake import SnakeEngine
+
+class ConsoleSnakeEngine(SnakeEngine):
+    def new_game(self, *args):
+        super(ConsoleSnakeEngine, self).new_game(*args)
+
+        self.window = curses.initscr()
+        curses.start_color()
+
+        curses.init_pair(1, curses.COLOR_RED, curses.COLOR_BLACK)
+        curses.init_pair(4, curses.COLOR_BLUE, curses.COLOR_BLACK)
+
+        self.EMPTY_COLOUR = curses.color_pair(0)
+        self.APPLE_COLOUR = curses.color_pair(1)
+        self.SNAKE_COLOUR = curses.color_pair(4)
+
+    def draw_board(self):
+        # Draw grid.
+        for y, row in enumerate(self.board):
+            for x, cell in enumerate(row):
+                char = '.'
+                colour = self.EMPTY_COLOUR
+
+                # Draw the things on the square.
+                if cell == Squares.APPLE:
+                    char = '@'
+                    colour = self.APPLE_COLOUR
+
+                elif cell.isalpha(): # Snake...
+#                    colour = self.bots[cell.lower()][1]
+                    char = cell
+                    colour = self.SNAKE_COLOUR
+
+                self.window.addstr(y, x, char, colour)
+
+    def run(self):
+        while self.bots:
+            # Clear the screen.
+            self.window.erase()
+
+            # Draw the board.
+            self.draw_board()
+
+            # Update the display.
+            self.window.refresh()
+            time.sleep(0.025)
+
+            # Let the snakes move!
+            self.update_snakes()
+
+def main(*args):
+    from bots import *
+    from oldbot import BotWrapper
+
+    game = ConsoleSnakeEngine(25, 25, 50)
+    game.add_bot(right_bot)
+    game.add_bot(random_bot)
+    game.add_bot(random_bounds_bot)
+    game.add_bot(random_square_bot)
+    game.add_bot(BotWrapper('oldbots/peter.py'))
+    game.run()
+
+if __name__ == '__main__':
+    curses.wrapper(main)
+