正しく標準入力されるまで関数を呼び出し続けるための例外処理

confirm(message: str) -> boolという任意の文章を引数に取り、それに対する回答がyesかnoかを真偽値で返す関数を書きました。

Python
1def confirm(message: str) -> bool:
2    return input(f"{message} [y/n] ").lower()[0] == "y"

何度か実行して気づいたのですが、うっかり空白のままエンターキーを押してしまうと[0]に値するものがないためにIndexError: string index out of rangeが送出されます。
そこで、非空白の入力が行われるまで繰り返し問い続けるように例外処理を使ってみたのですが、うまく機能しません。
具体的には、1度でも空行のままの入力があると(except文が実行されると)返り値がNoneになります。
再帰的に呼び出す形になっているのが問題なのかと目星を付けてはいるのですが、どうすればよいかが分かりません。
よろしくお願いします。

Python
1# 例外処理版
2def confirm(message: str) -> str:
3    try:
4        response = input(f"{message} [y/n] ").lower()[0]
5    except IndexError:
6        confirm(message)
7    else:
8        return response == "y"
9
10res = confirm("")
11print(f"{res}")
12
13>  [y/n]  
14>  [y/n]  y
15> None

行動規範の内容に同意します

回答3件

Enterだけを入力した後でも正常動作するようにし、さらに
ctrl-C や ctrl-D を入力した場合、つねに False を返すようにしてみました。
i.py

python3
1def confirm(message: str) -> str:
2    try:
3        while True:
4            response = input(f"{message} [y/n] ")
5            if len(response) > 0:
6                break;
7        return response[0].lower() == "y"
8    except EOFError:
9        return False
10    except KeyboardInterrupt:
11        return False
12
13res = confirm("")
14print(f"{res}")

投稿2019/12/06 13:34

katoy

総合スコア22324

ChaosRepainter

2019/12/06 14:12 編集

if len(response) > 0: は if response: でおｋ res = confirm("") print(f"{res}") format文でその変数しかprintしないなら print(confirm("")) でおｋ

退会済みユーザー

2019/12/06 20:07

確かにコンソールなのでctrl-C, ctrl-Dに対応したほうが良いですね。ありがとうございます

行動規範の内容に同意します

ベストアンサー

再帰呼び出しの際、returnを忘れがちです。

Python
1def confirm(message: str) -> str:
2    try:
3        response = input(f"{message} [y/n] ").lower()[0]
4    except IndexError:
5        return confirm(message)
6    else:
7        return response == "y"

こんなふうに組むこともできます。

Python
1def confirm(message: str) -> bool:   # 返り値はboolでは？
2    response = ''
3    while not response:
4        response = input(f"{message} [y/n] ").lower()
5
6    return response[0] == 'y'