import time
import pyautogui
from numpy import asarray


def get_sizes(color, arr):
    size = 0
    for i in arr[0]:
        if all(i == color):
            size += 1
    return size


def game():
    img = pyautogui.screenshot('game_size.png', region=(710, 921, 500, 1))

    test = asarray(img)

    figure_need_len = get_sizes(test[0][250], test) - 10
    if figure_need_len < 110:
        print('small')
        figure_need_len -= 53
    wait_time = 0.004255 * figure_need_len

    pyautogui.mouseDown(button='left', x=960, y=873)  # press the right button down
    time.sleep(wait_time)
    pyautogui.mouseUp(button='left')
    time.sleep(0.75)
    fig = pyautogui.screenshot('fig.png', region=(710, 921, 500, 1))
    fig_arr = asarray(fig)
    fig_size = get_sizes(fig_arr[0][250], fig_arr)

    pyautogui.screenshot('end.png')


for _ in range(100):
    time.sleep(1)
    game()