[파이썬] 문제 : 터틀(turtle) 텍스트 T=삼각형, R=사각형 입력받아 그리기

(문제)

텍스트 입력 방법을 사용하여 
입력창에 T를 입력하여 정삼각형을 
그리고 "R"을 이용하여 사각형을 그립니다 
이때 다른 문자를 입력하면 쓰기 방법을 사용하여 
화면에 입력이 잘못되었씁니다.가 나타난다

 

(방법)

import turtle

t = turtle.Turtle()
t.speed(0)

def draw_polygon(pn):        # 다각형 그리기
    for i in range(pn):      # 다각형에서 변의 수만큼 반복
        t.forward(100)        # 한변 그리기
        t.left(int(360/pn))   # 360도를 4각형일때 4호 나누면 각도 구할 수 있음

while True:
    ss = input('(T, R) : ')   # 입력 받기
    
    n = 3                     # 몇 각형인지 변수
    if ss != 'T' and ss != 'R' and ss.upper() != 'Q':   # T, R, Q가 아니면 whlle 반복 재진행
        print('입력이 잘못되었씁니다')
        continue             # 아래 코드로 가지 않고 while 반복문 재실행
    elif ss == 'T':
        n = 3                 # T 일때 3각형
    elif ss == 'R':
        n = 4                 # R 일때 4각형
    elif ss.upper() == 'Q':   # Q, q 면 대문자로 확인해서 종료
        break
        
    t.clear()                 # 화면 클리어
    draw_polygon(n)           # 다각형 그리기


turtle.done()

try:
    turtle.bye()
except:
    print("bye")