Игра "Жизнь"

import pygame as pg
 
# Константы
BLACK = pg.Color(0, 0, 0)
DEAD = pg.Color(70, 33, 26)
ALIVE = pg.Color(186, 85, 54)
RULES = {'born': [3],
         'survive': [2, 3]}
TICKRATE = 100
 
 
# Класс клеточки
class Cell:
    def __init__(self, x: int, y: int, size: int, life: bool = False) -> None:
        self.x = x
        self.y = y
        self.size = size
        self.rect = x, y, size, size
        self.life = life
 
    # Использовал этот метод для обработки нажатия ЛКМ
    def is_pressed(self, pos) -> bool:
        x, y = pos
        if self.x < x < self.x + self.size and self.y < y < self.y + self.size:
            return True
 
    def change(self) -> None:
        self.life = not self.life
 
    def render(self, scr: pg.Surface) -> None:
        if self.life:
            pg.draw.rect(scr, ALIVE, self.rect)
        else:
            pg.draw.rect(scr, DEAD, self.rect)
 
 
# Функция подсчёта живых соседей
def get_nei(tar: Cell, cells: list) -> int:
    count = 0
    for cell in cells:
        if cell.life:
            x_sym = abs(tar.x - cell.x)
            y_sym = abs(tar.y - cell.y)
 
            # Горизонтальное, вертикальное или диагональное соседство
            h_nei = x_sym == 20 and tar.y == cell.y and cell.life
            v_nei = y_sym == 20 and tar.x == cell.x and cell.life
            d_nei = x_sym == 20 and y_sym == 20 and cell.life
            if h_nei or v_nei or d_nei:
                count += 1
    return count
 
 
# Просто рендер живых клеток
def update(scr: pg.Surface, cells: list) -> None:
    for cell in cells:
        cell.render(scr)
 
 
# Обработка следующего поколения
def get_next_gen(cells: list) -> None:
    for cell in cells:
        neibs = get_nei(cell, cells)
        if cell.life:
            if neibs not in RULES['survive']:
                cell.change()
        else:
            if neibs in RULES['born']:
                cell.change()
 
 
# Тут хотел обрабатывать позицию нажатия ЛКМ (более оптимизированно)
def react(pos):
    pass
 
 
# Главная функция
def main() -> None:
    pg.init()
    size = w, h = 1280, 960
    scr = pg.display.set_mode(size)
    pg.display.set_caption('Cell Machine')
 
    # Внесение всех клеточек в массив
    cell_size = 20
    cells = []
    for x in range(0, w, cell_size):
        for y in range(0, h, cell_size):
            cells.append(Cell(x, y, cell_size))
 
    pause = True
    clock = pg.time.Clock()
 
    running = True
    while running:
        for event in pg.event.get():
            if event.type == pg.QUIT:
                running = False
 
            # По нажанию на ЛКМ клетка должна менять своё состояние
            # Живая / мёртвая
            if event.type == pg.MOUSEBUTTONDOWN:
                if event.button == 1:
                    # react(event.pos)
                    for cell in cells:
                        if cell.is_pressed(event.pos):
                            cell.change()
 
            if event.type == pg.KEYDOWN:
                if event.key == pg.K_SPACE:
                    pause = not pause
 
        # Попытка реализоваться паузу
        if not pause:
            get_next_gen(cells)
            pg.display.set_caption('PLAY')
            clock.tick(TICKRATE)
        else:
            pg.display.set_caption('PAUSE')
 
        scr.fill(DEAD)
        update(scr, cells)
 
        pg.display.flip()
    pg.quit()
 
 
if __name__ == '__main__':
    main()