Ggthjy

I did a project in Python where I combined user-input, random, and the turtle module to make a sort of "Mandala Generator", or a program that can generate a simple design either randomly or by the specifications of users. Is there any way I can shorten my code without drastically changing it?

#custom mandala project Python

import turtle

import random

import time



#is the game in progress?

drawing = False



#define turtle

pen = turtle.Turtle()

pen.setheading(90)

pen.pensize(3)

pen.speed(12)



#colors empty list

colors = []



#define variables

LINE1, LINE2 = "~" * 36, "~" * 24



#define check input function

def get_input(response, values):

    #make sure the input is in the choices

    while True:

        value = input(response).strip().lower()

        if value in values:

            return value



#define main function

def main():

    print(LINE1 + "nCustom Mandala Generatorn" + LINE1)

    name = input("What is your name? ")

    print("In this custom mandala generator, you get to choose the size and colors of your mandala, or have the computer generate a random one,", name, ".n" + LINE2)

    time.sleep(1)

    ready = input("Are you ready or not? ")



    if ready[0].lower() == "y":

        answer = get_input("Would you like to use our random generator function or our custom function? {random/custom} ", {"random", "custom"})



        #universal variables

        size = random.uniform(1.5, 2.5)

        fd = 75 * size



        if answer == "random":

            colors.extend(("darkred", "red", "yellow", "darkgreen", "green", "lightgreen", "darkblue", "blue", "purple"))

            #rt = right turn

            rt = random.uniform(100, 300)



        elif answer == "custom":

            print(LINE2 + "nYou can choose any colors from this list for your Mandala: nDarkred, red, yellow, darkgreen, green, lightgreen, darkblue, blue, purple")

            color1 = get_input("What is the first color? ", {"darkred", "red", "yellow", "darkgreen", "green", "lightgreen", "darkblue", "blue", "purple"})

            color2 = get_input("What is the second color? ", {"darkred", "red", "yellow", "darkgreen", "green", "lightgreen", "darkblue", "blue", "purple"})

            colors.extend((color1, color2))

            rt = int(input("What angle would you like your Mandala to turn at? "))



        #pre-draw sequence

        print(LINE2 + "nInitializing...")

        print("Determining features... nColor... nSize...")

        time.sleep(3)

        print(LINE1 + "nYour final result should be drawing as of this point!nThanks for using the generator,", name, "!")



        for counter in range(50):

            selection = random.choice(colors)

            pen.color(selection)

            pen.forward(fd)

            pen.right(rt)



    else:

        print("Come and use the custom generator later!")





main()