from stackA import Stack
from random import randint
from datetime import datetime

COLOURS = ["Red", "Green", "Blue"]

rand = lambda: randint(0, 2)

# Single stack
stacks = [Stack() for i in range(5)]

points = 1

win = False

while not win:
    print("Points:", points)

    for stack in stacks:
        print("Stacks:\n", stack)

    block = rand()
    blSt = COLOURS[block]
    print("New block:", blSt)
    time = datetime.now()
    if input("Accept?(y/n)").lower() == "y":
        # Continue
        i = input("Push to stack:[1..5]")
        try:
            i = int(i)-1
        finally:
            if blSt == stacks[i].top():
                stacks[i].pop()
            else:
                stacks[i].push(blSt)
                if stacks[i].length() > 9:
                    win = True
                points += 1
    elif (datetime.now() - time).seconds > 5:
        print("Time limit exceeded. pushing ")
        stacks[0].push(blSt)
        if stacks[0].length() > 9:
            win = True

    print("\n")

print(f"You got {points} points")