console_snake/frame.py

from settings import *
from sys import platform

if platform == "win32":
    from ctypes import *
    STD_OUTPUT_HANDLE = -11
    STDHANDLE = windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)

    class COORDSET(Structure):
        _fields_ = [("X", c_long), ("Y", c_long)]

    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:
        print(f"\033[{x};{y}H")


class Frame:
    def __init__(self, width: int, height: int) -> None:
        self.width = width
        self.height = height
        self.matrix = [[SPACE] * width for line in range(height)]

    def __str__(self) -> str:
        return (
            f"Width:\n  {self.width}\n"
            f"Height:\n  {self.height}"
        )

    def draw(
        self,

        x: int,
        y: int,

        value: int = WALL,
        width: int = 1,
        height: int = 1
    ) -> None:
        if not isinstance(value, int):
            raise TypeError("Value must be int")

        for line in range(height):
            for column in range(width):
                self.matrix[y + line][x + column] = value

    def show(self) -> None:
        set_cursor_position(0, 0)

        out_string = f"{WALL_COLOR}┍{'━' * (self.width * 2)}┑{RESET_COLOR}\n"

        for line in self.matrix:
            to_str = ''

            to_str += f'{WALL_COLOR}│{RESET_COLOR}'
            for elem in line:
                if elem == SPACE:
                    to_str += TT_SPACE
                elif elem == WALL:
                    to_str += TT_WALL
                elif elem == FOOD:
                    to_str += TT_FOOD
                elif elem == WALL_FOOD:
                    to_str += TT_WALL_FOOD
                elif elem == HEAD:
                    to_str += TT_HEAD
            to_str += f'{WALL_COLOR}│{RESET_COLOR}\n'

            out_string += to_str
        out_string += f"{WALL_COLOR}┕{'━' * (self.width * 2)}┙{RESET_COLOR}"

        print(out_string)

    def see(self, x, y) -> str:
        return self.matrix[y][x]