GameOfLife.py

from random import randint as ra


class GameOfLife:
    def __init__(self, width, height, cells):
        self.game = {}
        self.generate_random(0, 0, width, height, cells)

    def update(self):
        to_update = {}
        for coord in self.game:
            to_update[coord] = 0

        for x, y in self.game:
            row_p_1 = x + 1
            row_m_1 = x - 1
            col_p_1 = y + 1
            col_m_1 = y - 1
            to_update[(row_p_1, y)] = to_update.get((row_p_1, y), 0) + 1
            to_update[(row_p_1, col_m_1)] = to_update.get((row_p_1, col_m_1), 0) + 1
            to_update[(x, col_m_1)] = to_update.get((x, col_m_1), 0) + 1
            to_update[(row_m_1, col_m_1)] = to_update.get((row_m_1, col_m_1), 0) + 1
            to_update[(row_m_1, y)] = to_update.get((row_m_1, y), 0) + 1
            to_update[(row_m_1, col_p_1)] = to_update.get((row_m_1, col_p_1), 0) + 1
            to_update[(x, col_p_1)] = to_update.get((x, col_p_1), 0) + 1
            to_update[(row_p_1, col_p_1)] = to_update.get((row_p_1, col_p_1), 0) + 1

        for key, value in to_update.items():
            if (key in self.game) and (value < 2 or value > 3):
                self.game.remove(key)
            elif not (key in self.game) and value == 3:
                self.game.add(key)

    def generate_random(self, start_x, start_y, width, height, alive_cells):
        self.game = {(ra(start_x, start_x + width), ra(start_y, start_y + height)) for _ in range(alive_cells)}