Дедушка Сапёра

import pygame
from random import randint
 
 
class Board:
    def __init__(self, width, height):
        self.width = width
        self.height = height
        self.board = [[0] * width for _ in range(height)]
        self.left = 10
        self.top = 10
        self.cell_size = 30
 
    def set_view(self, left, top, cell_size):
        self.left = left
        self.top = top
        self.cell_size = cell_size
 
    def render(self, screen):
        colors = [
            (0, 0, 0),
            (255, 0, 0),
            (0, 0, 255)
        ]
        border = (255, 255, 255)
        for row in range(self.height):
            for col in range(self.width):
                x = self.left + self.cell_size * col
                y = self.top + self.cell_size * row
                side = self.cell_size
                color = colors[self.board[row][col]]
                pygame.draw.rect(screen, color, (x, y, side, side))
                pygame.draw.rect(screen, border, (x, y, side, side), 1)
 
    def get_cell(self, mouse_pos):
        left = mouse_pos[0] - self.left
        top = mouse_pos[1] - self.top
        if left < 0 or top < 0:
            return None
        row = top // self.cell_size
        col = left // self.cell_size
        if row >= self.height or col >= self.width:
            return None
        return col, row
 
    def on_click(self, cell_coords):
        if cell_coords:
            col, row = cell_coords
            self.board[row][col] += 1
            self.board[row][col] %= 3
 
    def get_click(self, mouse_pos):
        cell = self.get_cell(mouse_pos)
        self.on_click(cell)
 
 
class Minesweeper(Board):
    def __init__(self, width, height, mines):
        super().__init__(width, height)
        for row in range(self.height):
            for col in range(self.width):
                self.board[row][col] = -1
        for _ in range(mines):
            row = randint(0, self.height - 1)
            col = randint(0, self.width - 1)
            while self.board[row][col] == 10:
                row = randint(0, self.height - 1)
                col = randint(0, self.width - 1)
            self.board[row][col] = 10
 
    def render(self, screen):
        colors = [
            (0, 0, 0),
            (255, 0, 0)
        ]
        border = (255, 255, 255)
        for row in range(self.height):
            for col in range(self.width):
                x = self.left + self.cell_size * col
                y = self.top + self.cell_size * row
                side = self.cell_size
                index = 1 if self.board[row][col] == 10 else 0
                color = colors[index]
                pygame.draw.rect(screen, color, (x, y, side, side))
                pygame.draw.rect(screen, border, (x, y, side, side), 1)
                if 0 <= self.board[row][col] <= 8:
                    font = pygame.font.Font(None, (self.cell_size * 2) // 3)
                    text = font.render(str(self.board[row][col]), True, (0, 255, 0))
                    tab = self.cell_size // 15
                    screen.blit(text, (x + tab, y + tab))
 
    def on_click(self, cell_coords):
        if cell_coords:
            col, row = cell_coords
            if self.board[row][col] == -1:
                self.open_cell(cell_coords)
 
    def open_cell(self, cell_coords):
        col, row = cell_coords
        count = 0
        for i in range(-1, 2):
            for j in range(-1, 2):
                good_row = 0 <= row + i < self.height
                good_col = 0 <= col + j < self.width
                if good_row and good_col:
                    if self.board[row + i][col + j] == 10:
                        count += 1
        self.board[row][col] = count
 
 
if __name__ == '__main__':
    pygame.init()
    pygame.display.set_caption('Дедушка сапёра')
    size = width, height = 620, 620
    screen = pygame.display.set_mode(size)
    minesweeper = Minesweeper(10, 10, 10)
    minesweeper.set_view(10, 10, 60)
    running = True
    while running:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False
            if event.type == pygame.MOUSEBUTTONDOWN:
                minesweeper.get_click(event.pos)
        screen.fill((0, 0, 0))
        minesweeper.render(screen)
        pygame.display.flip()
    pygame.quit()