summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--.hgignore2
-rw-r--r--bots.py16
-rw-r--r--common.py4
-rw-r--r--snake.py46
4 files changed, 68 insertions, 0 deletions
diff --git a/.hgignore b/.hgignore
new file mode 100644
index 0000000..85576b0
--- /dev/null
+++ b/.hgignore
@@ -0,0 +1,2 @@
+glob:*.pyc
+glob:*.orig
diff --git a/bots.py b/bots.py
new file mode 100644
index 0000000..cf23602
--- /dev/null
+++ b/bots.py
@@ -0,0 +1,16 @@
+import random
+
+def random_bot(board, (x, y)):
+ height = len(board)
+ width = len(board[0])
+ moves = []
+ if x > 0:
+ moves.append('L')
+ if x < width - 1:
+ moves.append('R')
+ if y > 0:
+ moves.append('U')
+ if y < height - 1:
+ moves.append('D')
+ return random.choice(moves)
+
diff --git a/common.py b/common.py
new file mode 100644
index 0000000..f19ebd9
--- /dev/null
+++ b/common.py
@@ -0,0 +1,4 @@
+class Squares(object):
+ EMPTY = '.'
+ APPLE = '*'
+
diff --git a/snake.py b/snake.py
new file mode 100644
index 0000000..341a242
--- /dev/null
+++ b/snake.py
@@ -0,0 +1,46 @@
+import random
+
+from common import *
+
+import pygame
+pygame.init()
+
+class SnakeEngine(object):
+ def __init__(self, rows, columns, n_apples, width=800, height=600, fullscreen=False):
+ super(SnakeEngine, self).__init__()
+ flags = 0
+ if fullscreen:
+ flags |= pygame.FULLSCREEN
+ pygame.display.set_mode((width, height), flags)
+
+ self.bots = {}
+
+ self.new_game(rows, columns, n_apples)
+
+ def new_game(self, rows, columns, n_apples):
+ self.board = [[Squares.EMPTY for x in xrange(columns)] for y in xrange(rows)]
+ for i in xrange(n_apples):
+ x = random.randint(0, columns - 1)
+ y = random.randint(0, rows - 1)
+ self.board[y][x] = Squares.APPLE
+
+ def add_bot(self, name, bot):
+ """
+ A bot is a callable object, with this method signature:
+ def bot_callable(
+ board=[[cell for cell in row] for row in board],
+ position=(snake_x, snake_y)
+ ):
+ return random.choice('RULD')
+ """
+ self.bots[name] = bot
+
+ def remove_bot(self, name):
+ del self.bots[name]
+
+ def run(self):
+ pass
+
+if __name__ == '__main__':
+ game = SnakeEngine(8, 16, 10)
+