Minor changes

1 files changed, 11 insertions, 3 deletions

diff --git a/snake/snake.py b/snake/snake.py
index 00cc261..98975cb 100755
--- a/snake/snake.py
+++ b/snake/snake.py
@@ -39,7 +39,7 @@ class Snake:
         self.screen.blit(text, (self.grid_pts/2,text_center[1]))
 
     def new_apple(self):
-        self.apple=(random.randint(0,self.grid_width),random.randint(0,self.grid_height))
+        self.apple=(random.randint(0,self.grid_width-1),random.randint(0,self.grid_height-1))
         while self.apple in self.snake:
             self.apple=(random.randint(0,self.grid_width),random.randint(0,self.grid_height))
 
@@ -76,7 +76,8 @@ class Snake:
             return(True)
         return(False)
 
-    def run(self):
+
+    def run(self, event_handler=None):
         clock = pygame.time.Clock()
         ignore_has_loose=True
         self.new_game()
@@ -95,7 +96,7 @@ class Snake:
                 if event.type == pygame.QUIT: 
                     pygame.quit()
                     sys.exit()
-                elif event.type == pygame.KEYDOWN:
+                elif event_handler==None and event.type == pygame.KEYDOWN:
                     if event.key == pygame.K_LEFT and self.direction != 3:
                         self.direction=9
                         break
@@ -108,6 +109,8 @@ class Snake:
                     elif event.key == pygame.K_DOWN and self.direction != 12:
                         self.direction=6
                         break
+            if event_handler!=None:
+                event_handler(self)
             self.move()
             # Check for eating apple
             if self.apple==self.snake[0]:
@@ -119,5 +122,10 @@ class Snake:
 
 
 game=Snake()
+
+def event_handler(game):
+    if game.snake[0][0]==10:
+        game.direction=6
+
 for i in range(0,10):
     game.run()
\ No newline at end of file