add restart whenever

diff --git a/game.py b/game.py
index d59d3f0e8ecc8098405e05b490eaee824a2f2e41..feb2276d59ab9e38733cf9280f4d6d7c846709b5 100644 (file)
--- a/game.py
+++ b/game.py
@@ -8,13 +8,15 @@ class Game:
 
        def __init__(self):
                self.grid = Grid(20, 20, 50)
-
                self.game_over = False
 
        def update(self, delta):
                for event in pygame.event.get():
                        if event.type == pygame.QUIT:
                                return COMMAND_EXIT
+                       if event.type == pygame.KEYDOWN:
+                               if event.key == pygame.K_r:
+                                       return COMMAND_RESTART
 
                mouse_pos = pygame.mouse.get_pos()
                mouse_button = pygame.mouse.get_pressed()