diff options
Diffstat (limited to 'snakegame/viewers/pygame.py')
-rw-r--r-- | snakegame/viewers/pygame.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/snakegame/viewers/pygame.py b/snakegame/viewers/pygame.py index 8b8fca4..3f32da1 100644 --- a/snakegame/viewers/pygame.py +++ b/snakegame/viewers/pygame.py @@ -59,7 +59,13 @@ class Viewer(object): xscale = self.board_width / self.columns yscale = self.board_height / self.rows - self.apple = load_image('images/apple.png', xscale, yscale) + self.items = { + common.APPLE : 'images/apple.png', + common.ICE_CREAM : 'images/icecream.png', + common.SHRINK_POTION : 'images/shrinkpotion.png' + } + for item in self.items: + self.items[item] = load_image(self.items[item], xscale, yscale) self.eyes = load_image('images/eyes.png', xscale, yscale) def draw_board(self, board): @@ -80,8 +86,8 @@ class Viewer(object): self.EDGE_WIDTH) # Draw the things on the square. - if cell == common.APPLE: - self.surface.blit(self.apple, r.topleft) + if cell in self.items: + self.surface.blit(self.items[cell], r.topleft) elif common.is_snake(cell): bot = self.engine.bots[cell.lower()] |