ユーザーが入力文字列で形状を指定し、問題を抱えているときに形状を出力する一連のコードに if/else ステートメントを追加しようとしています。形状コードに単純な If/else ステートメントを追加して、正しい入力文字列を含む形状のみを出力し、正しくない入力文字列が使用された場合に「形状が定義されていません」というエラー メッセージを出力する方法を教えてください。
私の形状コードの 1 つは、単純な正方形アルゴリズムです。
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'