Добавление if/else для определения алгоритма для beginner

Question

Я пытаюсь добавить инструкции if/else в набор кода, который выводит форму, когда пользователь указывает форму с входной строкой и имеет трудности. Как бы добавить простой оператор If/else в код моей формы, чтобы он выводил только форму с правильной входной строкой и при использовании неправильной строки ввода выводил сообщение об ошибке «Форма не определена».

Один из моих кодов формы простой квадратный алгоритм:

def square(size, chr): 
    row = 1 
    while row <= size: 
     col = 1 
     while col <= size: 
      print chr, 
      col = col + 1 
     print '' 
     row = row + 1 
    print '' 

Обновленный код:

print('Please specify a shape to draw. Type S for square, T for triangle, R for reverse triangle, D for diagonal, and A for angle') 
shape = raw_input('Please specify a shape to draw:') 
size = input('Please enter a size:') 
chr = raw_input('Please enter a drawing character:') 
print'' 

valid_shape_codes =('"S":square, "T":triangle, "R":reversetriangle, "D":diagonal, "A":angle') 
if shape in valid_shape_codes: 
    valid_shape_codes[shape](size, chr) 
else: 
    print 'Shape is not defined' 

Answer 1

print('Please specify a shape to draw. Type S for square') 
shape = raw_input('Please specify a shape to draw:') 
size = input('Please enter a size:') 
chr = raw_input('Please enter a drawing character:') 
valid_shape_codes ={"S":square, "T":triangle, "R":reversetriangle, "D":diagonal, "A":angle} 

if shape in valid_shape_codes: 
    valid_shape_codes[shape](size, chr) 
else: 
    print "Shape is not defined" 
print''