condition.py

import turtle

shape = input("What shape do you want?: (square/triangle/circle) ")
if shape == "square":
  print("Making a square")
  pen = turtle.Turtle()
  screen = turtle.Screen()
  pen.color("black")
  for i in range(4):
    pen.forward(100)
    pen.left(90)
    pen.hideturtle()
elif shape == "triangle":
  print("Making a triangle")
  pen = turtle.Turtle()
  screen = turtle.Screen()
  pen.color("black")
  for i in range(3):
    pen.forward(100)
    pen.left(120)
  pen.hideturtle()
elif shape == "circle":
  print("Making a circle")
  pen = turtle.Turtle()
  screen = turtle.Screen()
  pen.color("black")
  for i in range(36):
    pen.forward(15)
    pen.left(10)
  pen.hideturtle()
else:
  print("Invalid shape given: " + shape + " (square/triangle/circle)")