import turtle
from tkinter import *
import tkinter.messagebox

wn = turtle.Screen()
wn.setup(650,650)
wn.bgcolor("White")
wn.title("Turtle program")
geralt = turtle.Turtle()

def ujra():
    restart = raw_input("Új alakzatot akarsz? Ha igenre kattintasz visszamész a program legelejére!")
    if restart == "yes" or restart == "y":
        kerdo()
    if restart == "n" or restart == "no":
        vege()

def kerdo():
    while 1:
        tkinter.messagebox.showinfo("Választó","Válaszd ki a kívánt alakzatot!")
        answer = tkinter.messagebox.askquestion("Választó","Háromszöget választod?")

        if answer == "yes":
            var1 = "haromszog"
            print("Háromszöget választottad!")
            vege()
            break
        else:
            answer = tkinter.messagebox.askquestion("Választó","Négyszöget választod?")

        if answer == "yes":
            var1 = "negyszog"
            print("Négyszöget választottad!")
            vege()
            break
        else:
            answer = tkinter.messagebox.askquestion("Választó","Ötszöget választod?")

        if answer == "yes":
            var1 = "otszog"
            print("Ötszöget választottad!")
            vege()
            break
        else:
            answer = tkinter.messagebox.askquestion("Választó","Nem választottál alakzatot, a program újraindul!")

        if answer == "yes":
            continue
        else:
            continue

def vege():
    print ("A string értéke:", var1)
    if var1 == "haromszog":
        szelesseg=wn.numinput("Szélessége","")
        hosszusag=wn.numinput("Hosszúsága","")
        for i in range(3):
            geralt.forward(szelesseg)
            geralt.right(hosszusag)

    if var1 == "negyszog":
        szelesseg=wn.numinput("Szélessége","")
        hosszusag=wn.numinput("Hosszúsága","")
        for i in range(4):
            geralt.forward(szelesseg)
            geralt.right(hosszusag)

    if var1 == "otszog":
        print("IDE ÍRD A VÁLASZT!")
        lap=int(input("Lapok száma: "))
        hosszusag=int(input("Hosszúsága: "))
        szam=360
        szog = szam/lap
        for i in range(lap):
            geralt.forward(hosszusag)
            geralt.right(szog)
    ujra()

kerdo()