def test_down(self):
dir = Direction()
self.assertFalse(dir.get_down())
dir.down()
self.assertTrue(dir.get_down())
class Game:
width = 80
height = 80
title = "SNAKE GAME"
def __init__(self):
min = Point(0, 10)
max = Point(self.width - 1, self.height - 1)
pyxel.init(self.width, self.height, caption=self.title, fps=18)
pyxel.load("my_resource.pyxres")
self.snake = Snake(self.width / 2, self.height / 2)
self.fruits = [
Fruit(min, max, 8),
Fruit(min, max, 9),
Fruit(min, max, 10)
]
self.direction = Direction()
self._reset()
pyxel.run(self.update, self.draw)
def _reset(self):
self.direction.up()
self.points = 0
self.snake.reset()
for fruit in self.fruits:
fruit.reset()
def update(self):
if pyxel.btnp(pyxel.KEY_Q):
pyxel.quit()
if not self.snake.alive and pyxel.btnp(pyxel.KEY_R):
self._reset()
for fruit in self.fruits:
self._checkFruitEaten(fruit)
self._changeDirection()
self._moveSnake()
def draw(self):
if self.snake.alive:
pyxel.cls(0)
if not self.snake.alive:
pyxel.cls(5)
self._drawHeader()
self._drawSnake()
self._drawGameOver()
if self.snake.alive:
for fruit in self.fruits:
self._drawFruit(fruit)
def _drawSnake(self):
if self.snake.alive:
snake_colors = [6, 12, 5, 1]
idx_color = 1
for part in self.snake.body:
pyxel.pset(part.x, part.y, snake_colors[idx_color - 1])
idx_color = min((idx_color + 1), len(snake_colors))
def _drawGameOver(self):
if not self.snake.alive:
for part in self.snake.body:
pyxel.pset(part.x, part.y, 8)
text = "PRESS R TO RESTART"
pyxel.text(5, 40, text, 1)
def _drawFruit(self, fruit):
if fruit.eaten:
fruit.reset()
pyxel.pset(fruit.x, fruit.y, fruit.color)
def _drawHeader(self):
pyxel.rect(0, 0, 80, 9, 2)
pyxel.text(12, 2, self.title, 8)
points = str(self.points)
x = 79 - (len(points) * 5) + len(points)
pyxel.text(x, 2, points, 10)
pyxel.blt(2, 0, 0, 0, 0, 8, 8, 0)
def _checkFruitEaten(self, fruit):
if self.snake.head == fruit.position:
fruit.eaten = True
self.points += 1
self.snake.grow()
pyxel.playm(0, loop=False)
def _changeDirection(self):
if pyxel.btnp(pyxel.KEY_UP):
self.direction.up()
if pyxel.btnp(pyxel.KEY_RIGHT):
self.direction.right()
if pyxel.btnp(pyxel.KEY_DOWN):
self.direction.down()
if pyxel.btnp(pyxel.KEY_LEFT):
self.direction.left()
def _moveSnake(self):
if (self.snake.alive):
self.snake.move(self.direction)
self._checkSnakeHitWalls()
def _checkSnakeHitWalls(self):
head = self.snake.head
if head.x >= self.width or head.x < 0 or head.y >= self.height or head.y < 9:
self.snake.alive = False
