From f3bf334cac89842fe775742e46cd3d72bb49a13f Mon Sep 17 00:00:00 2001
From: Nicolas Sudres <nicolas.sudres@hotmail.fr>
Date: Sun, 31 Jul 2022 22:53:59 +0200
Subject: [PATCH] Add error handling

---
 Snake/main.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/Snake/main.py b/Snake/main.py
index a4ddfcd..fed567f 100644
--- a/Snake/main.py
+++ b/Snake/main.py
@@ -1,3 +1,4 @@
+from ast import Break
 import pygame
 from pygame.locals import *
 import src.level as level
@@ -19,11 +20,21 @@ def main():
 
     matrix = level.init_level(HEIGH, WIDTH)
     matrix[5][5] = 1
-    matrix[5][6] = 2
+    matrix[5][6] = 1
     level.spawn_apple(matrix, HEIGH, WIDTH)
     #print(matrix)
 
-    L_snake = snake.Snake_move(matrix, L_snake, "right")
+    try:
+        L_snake = snake.Snake_move(matrix, L_snake, "right")
+    except ValueError as Error:
+        if Error.args[0] == 1:
+            print("You take a wall !")
+        elif Error.args[0] == 2:
+            print("You ate your tale !")
+        
+        exit()
+    
+    
     print(L_snake)