add pause and refactor

frame.py

@@ -9,10 +9,10 @@ if platform == "win32":
     class COORDSET(Structure):
         _fields_ = [("X", c_long), ("Y", c_long)]
 
-    def _set_cursor_position(x: int, y: int) -> None:
+    def set_cursor_position(x: int, y: int) -> None:
         windll.kernel32.SetConsoleCursorPosition(STDHANDLE, COORDSET(x, y))
 else:
-    def _set_cursor_position(x: int, y: int) -> None:
+    def set_cursor_position(x: int, y: int) -> None:
         print(f"\033[{x};{y}H")
 
 
@@ -40,7 +40,7 @@ class Frame:
                 self.matrix[y + line][x + column] = value
 
     def show(self) -> None:
-        _set_cursor_position(0, 0)
+        set_cursor_position(0, 0)
 
         out_string = f"┍{'━' * (self.width * 2)}┑\n"
 

main.py

@@ -11,22 +11,21 @@ class Game:
         self.running = True
         self.pl = Player()
         self.out = Out(self.pl)
+        self.is_pause = False
 
         kb.add_hotkey(QUIT_BUTTON, self.stop_game)
+        kb.add_hotkey(PAUSE_BUTTON, self.pause_game)
         kb.add_hotkey(LEFT_BUTTON, self.pl.left)
         kb.add_hotkey(RIGHT_BUTTON, self.pl.right)
         kb.add_hotkey(UP_BUTTON, self.pl.up)
         kb.add_hotkey(DOWN_BUTTON, self.pl.down)
 
+    def pause_game(self) -> None:
+        self.is_pause = not self.is_pause
+
     def stop_game(self) -> None:
         self.running = False
 
-    def update(self) -> None:
-        pass
-
-    def output(self) -> None:
-        pass
-
     def play(self) -> None:
         system("clear||cls")
         time_to_sleep = 1 / FPS
@@ -34,16 +33,19 @@ class Game:
         while self.running:
             try:
                 sleep(time_to_sleep)
+
+                if not self.is_pause:
                     self.pl.input()
                     self.pl.update()
                     self.out.draw()
+
             except IndexError as e:
                 self.running = False
                 print(e)
             except KeyboardInterrupt:
                 return
 
-        input("Press enter to leave from game")
+        input("Press enter to leave from game.\n")
 
 
 def main() -> None:

out.py

@@ -7,7 +7,6 @@ class Out:
     def __init__(self, pl: Player) -> None:
         self.width = WIDTH
         self.height = HEIGHT
-        self.fps = FPS
         self.pl = pl
 
     def draw(self) -> None:
@@ -15,11 +14,6 @@ class Out:
         pl = self.pl
 
         for point in pl.body:
-            if point.x < 0 or point.y < 0:
-                raise IndexError("Snake has collision with walls")
-            elif point.x >= WIDTH or point.y >= HEIGHT:
-                raise IndexError("Snake has collision with walls")
-
             frame.draw(
                 x=point.x, y=point.y,
                 value=WALL,

player.py

@@ -23,6 +23,16 @@ class Player:
         elif self.direction == D_RIGHT:
             self.body.append(Point(self.body[-1].x + 1, self.body[-1].y))
 
+    def _check_collision(body: list[Point]) -> None:
+        for point in body:
+            if point.x < 0 or point.y < 0:
+                raise IndexError("Snake has collision with walls")
+            elif point.x >= WIDTH or point.y >= HEIGHT:
+                raise IndexError("Snake has collision with walls")
+
+        if len(set(body)) != len(body):
+            raise IndexError("Player has collision with self")
+
     def update(self) -> None:
         if self.body[-1] == self.food:
             self.food.generate_new()
@@ -30,8 +40,7 @@ class Player:
         else:
             self.body.pop(0)
 
-        if len(set(self.body)) != len(self.body):
+        Player._check_collision(self.body)
-            raise IndexError("Player has collision with self")
 
     def left(self) -> None:
         self.direction = D_LEFT

settings.py

@@ -5,6 +5,7 @@ FPS = 5
 
 # Buttons
 QUIT_BUTTON = 'q'
+PAUSE_BUTTON = 'p'
 LEFT_BUTTON = 'a'
 RIGHT_BUTTON = 'd'
 UP_BUTTON = 'w'