From 5b3efc4a2170855b21fe141aa06b8fdaa81776b9 Mon Sep 17 00:00:00 2001
From: Loic Guegan <manzerbredes@mailbox.org>
Date: Wed, 2 Nov 2022 09:12:06 +0100
Subject: Minor changes

---
 snake.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'snake.py')

diff --git a/snake.py b/snake.py
index 0445e53..ca21129 100755
--- a/snake.py
+++ b/snake.py
@@ -150,6 +150,11 @@ class Snake:
         # Refresh screen
         self.draw()
         self.clock.tick(self.fps)
+        # Ensure we not quit
+        for event in pygame.event.get():
+                if event.type == pygame.QUIT: 
+                    pygame.quit()
+                    sys.exit()
         return(code)
     
     def play2(self,direction):
@@ -185,10 +190,7 @@ class Snake:
         while True:
             # Check inputs
             for event in pygame.event.get():
-                if event.type == pygame.QUIT: 
-                    pygame.quit()
-                    sys.exit()
-                elif event.type == pygame.KEYDOWN:
+                if event.type == pygame.KEYDOWN:
                     if event.key == pygame.K_LEFT and self.direction != 3:
                         self.direction=9
                         break
-- 
cgit v1.2.3