class TestPlayers(TestCase):
def setUp(self):
self.player = Player()
self.codemaker = CodeMaker()
def test_player(self):
self.assertEqual(self.player.points, 0)
def test_add_points(self):
self.player.add_point()
self.assertEqual(self.player.points, 1)
def test_new_codemaker(self):
self.assertTrue(self.codemaker.code)
class GameController():
def __init__(self):
""" Initializing what instances of classes """
self.snake = Snake()
self.view = GameView()
self.apple = Apple()
self.poison = Poison()
self.player = Player()
self.gameover = GameOverController()
self.gamestart = GameStartController()
def run(self):
""" Run the game """
clock = pygame.time.Clock()
# Game is running
running = True
# Show start screen and get the player's name
players_name = self.gamestart.run(self.view.window)
if not players_name:
running = False
else:
# assigning player name that's typed in to the player instance
self.player.name = players_name
while running:
clock.tick(20)
# Apple
apples = pygame.sprite.Group()
apples.add(self.apple)
# Poison
poisons = pygame.sprite.Group()
poisons.add(self.poison)
# Show the game screen
self.view.display(self.snake, self.player.score)
apples.draw(self.view.window)
poisons.draw(self.view.window)
# checking to see if snake has eaten apple
eaten_apple = self.apple.apple_eaten(self.snake.group)
# if apple has been eaten, generate new apple, make snake longer, and add a point to score
if eaten_apple:
self.snake.add_body()
self.player.add_point()
# checking to see if new apple overlaps static poison
self.apple.overlap_poison_with_apple(self.poison, poisons)
# checking to see if new apple is placed under snake body, if it is, generate new apple
self.apple.overlap_snake_new_apple(self.snake.group)
# checking to see if poison has been eaten by snake
overlap_snake = self.poison.poison_eaten(self.snake.group)
# if snake has eaten the poison
if overlap_snake == True:
# Ask the player to continue
if not self._game_restart():
running = False
# When the snake hits the wall
if self.snake.check_hit_wall(self.view.window):
# Ask the player to continue
if not self._game_restart():
running = False
# Catch the event
for event in pygame.event.get():
# If the window is closed
if event.type == pygame.locals.QUIT:
running = False
# If the direction keys are typed
if event.type == pygame.KEYDOWN:
if event.key == pygame.locals.K_RIGHT:
self.snake.turn("Right")
elif event.key == pygame.locals.K_LEFT:
self.snake.turn("Left")
elif event.key == pygame.locals.K_UP:
self.snake.turn("Up")
elif event.key == pygame.locals.K_DOWN:
self.snake.turn("Down")
# Move the snake
self.snake.move()
pygame.display.update()
def _game_restart(self):
"""
Send the score to the server, and ask the player to continue
"""
# Send the score to the server
post_result = self.player.post_score()
# Show the gameover screen
game_continue = self.gameover.run(self.view.window, self.player,
post_result)
# If restarting the game
if game_continue:
# Recreate the snake and gameview
self.snake = Snake()
self.view = GameView()
# Reset the score
self.player.score = 0
return True
else:
return False
