diff options
| -rw-r--r-- | .hgignore | 2 | ||||
| -rw-r--r-- | bots.py | 16 | ||||
| -rw-r--r-- | common.py | 4 | ||||
| -rw-r--r-- | snake.py | 46 |
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 @@ -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) + |