import pyautogui, sys
import time
import pyperclip
import random
import string

print('Press Ctrl-Alt-Del to stop the bot.')

correctAnswers = []
amountCorrect = len(correctAnswers)
texts = []
locations = []
choices = []
last = ['none','none','none','none','none']
wrong = []
laskuri = 0

while True:
    print("Montako kysymystä?")
    questions = int(input())
    break

while (len(correctAnswers)) < questions-1:
    time.sleep(1)
    for i in range(5):
        pyautogui.press('pgdn')
    pos = None
    done = False
    while pos is None:
        for pos in pyautogui.locateAllOnScreen('blue2.png'):
            if done == False:
                pyautogui.moveTo(pos[0]-20,pos[1]+5)
                pyautogui.dragRel(700, 45, 0.2, button='left')
                pyautogui.hotkey('ctrl', 'c')
                valinta = pyperclip.paste()
                valinta = valinta.split(".")
                valinta = valinta[0]
                valinta.splitlines()
                texts.append(valinta)
                last.append(valinta)
                locations.append((pos[0]+5,pos[1]+5))
                choices.append((pos[0]+5,pos[1]+5))

    for i in range(correctAnswers):
        for j in texts:
            if correctAnswers[laskuri] == j:
                laskuri += 1
                pyautogui.click(locations[texts.index(j)])
                pyautogui.click(pyautogui.locateCenterOnScreen('next.png'))
                time.sleep(0.5)
                pyautogui.click(pyautogui.locateCenterOnScreen('next2.png'))
                pos = None
                while pos is None:
                    for pos in pyautogui.locateAllOnScreen('blue2.png'):
                        pyautogui.moveTo(pos[0]-20,pos[1]+5)
                        pyautogui.dragRel(700, 45, 0.2, button='left')
                        pyautogui.hotkey('ctrl', 'c')
                        valinta = pyperclip.paste()
                        valinta = valinta.split(".")
                        valinta = valinta[0]
                        valinta.splitlines()
                        texts.append(valinta)
                        last.append(valinta)
                        locations.append((pos[0]+5,pos[1]+5))
                        choices.append((pos[0]+5,pos[1]+5))



        #print("Correct answers: ",correctAnswers,"\n")
        #print("Choices :",texts,"\n")
        count = len(texts)
        #print("Valintoja yhteensä: ", count)
        #print(wrong)
        texts = []
        locations = []

    while True:
        if last[-1] in wrong:
            pass
            if last[-2] in wrong:
                pass
                if last[-3] in wrong:
                    pass
                    if last[-4] in wrong:
                        pyautogui.click(choices[-5])
                    else:
                        pyautogui.click(choices[-4])
                else:
                    pyautogui.click(choices[-3])
            else:
                pyautogui.click(choices[-2])
        else:
            pyautogui.click(choices[-1])

        if pyautogui.locateOnScreen('green.png') != None:
            print("Correct answer\n")
            laskuri += 1
            if last[-1] in correctAnswers:
                pyautogui.click(pyautogui.locateCenterOnScreen('next.png'))
                time.sleep(0.5)
                pyautogui.click(pyautogui.locateCenterOnScreen('next2.png'))
                break
            else:
                wrong = []
                #last = []
                choices = []
                green = pyautogui.locateOnScreen('green.png')
                pyautogui.moveTo(green[0],green[1])
                pyautogui.dragRel(700, 45, 0.2, button='left')
                pyautogui.hotkey('ctrl', 'c')
                rightAnswer = pyperclip.paste()
                rightAnswer = rightAnswer.split(".")
                rightAnswer = rightAnswer[0]
                rightAnswer = rightAnswer.splitlines()
                if len(rightAnswer) > 1:
                    rightAnswer = rightAnswer[1]
                correctAnswers.append(rightAnswer)
                pyautogui.click(pyautogui.locateCenterOnScreen('next.png'))
                time.sleep(0.5)
                pyautogui.click(pyautogui.locateCenterOnScreen('next2.png'))
                break

        if pyautogui.locateOnScreen('green.png') == None:
            print("Wrong answer\n")
            laskuri = 0
            wrong.append(last[-1])
            pyautogui.click(pyautogui.locateCenterOnScreen('restart.png'))
            for i in range(5):
                pyautogui.press('pgdn')
            break