python で yes/no で実行する/しないの処理のメモ

• 変更2020-06-09: python2.xの raw_input が python3.x で input に変更になっているので修正

コード

hoge.py

def yes_no_input():
    while True:
        choice = input("Please respond with 'yes' or 'no' [y/N]: ").lower()
        if choice in ['y', 'ye', 'yes']:
            return True
        elif choice in ['n', 'no']:
            return False


if __name__ == '__main__':
    if yes_no_input():
        print('OK!')

実行

~/example $ python hoge.py
Please respond with 'yes' or 'no' [y/N]: y
OK!
~/example $ python hoge.py
Please respond with 'yes' or 'no' [y/N]: N
~/example $

参考

python - APT command line interface-like yes/no input? - Stack Overflow