Алгоритм для сапера

from random import randint
from termcolor import colored
 
class Field():
    def __init__(self, size, bombs):
        self.field = generatorAI(size, bombs)
        self.openned_field = [['?']*size for i in range(size)]
        self.endgame = 0
 
 
    def generatorAI(self, n, b):
        if b > n**2 // 4:
            return "Error: too many bombs. Count of bombs must be smaller than 1/4 of cells' number"
        used = []
        field = [[[] for i in range(n)] for i in range(n)]
        while b != 0:
            x = randint(0, n - 1)
            y = randint(0, n - 1)
            if (x, y) not in used:
                used.append((x, y))
                field[x][y].append("B")
                b -= 1
        for i in range(n):
            s = 0
            for j in range(n):
                s = 0
                if "B" in field[i][j]:
                    continue
                if i < n - 1:
                    if "B" in field[i + 1][j]:
                        s += 1
                    if j < n - 1:
                        if "B" in field[i + 1][j + 1]:
                            s += 1
                        if "B" in field[i][j + 1]:
                            s += 1
                    if j > 0:
                        if "B" in field[i + 1][j - 1]:
                            s += 1
 
                elif j < n - 1:
                    if "B" in field[i][j + 1]:
                            s += 1
 
                if i > 0:
                    if "B" in field[i - 1][j]:
                        s += 1
                    if j > 0:
                        if "B" in field[i - 1][j - 1]:
                            s += 1
                        if "B" in field[i][j - 1]:
                            s += 1
                    if j < n - 1:
                        if "B" in field[i - 1][j + 1]:
                            s += 1
 
                elif j > 0:
                    if "B" in field[i][j - 1] :
                            s += 1
 
                field[i][j].append(s)
 
        return field
 
 
print('@ ' * 12)
for i in Field.generatorAI(10, 10):
    print('@', end=' ')
    for j in i:
        if 0 in j:
            print(' ', end = ' ')
        elif "B" in j:
            print(colored(str(j[0]), 'red'), end = ' ')
        elif 1 in j:
            print(colored(str(j[0]), 'white'), end = ' ')
        elif 2 in j:
            print(colored(str(j[0]), 'yellow'), end = ' ')
        elif 3 in j:
            print(colored(str(j[0]), 'green'), end = ' ')
        elif j[0] >= 4:
            print(colored(str(j[0]), 'grey'), end = ' ')
    print('@')
print('@ ' * 12)