編寫在 CLI 執行的程式, 很多時都會用到詢問使用者 yes / no 是否執行, 以下記錄一下在 Python 的寫法:
例如:
1 2 3 4 5 6 |
#!/usr/bin/python action = raw_input('continue?') if action == 'yes': print 'Hello World!' |
上面程式很簡單詢問是否繼續, 如果輸入 “yes” 則會輸出 “Hello World!”.
上面程式必須輸入 “yes” 才會輸出 “Hello World!”, 如果輸入 “Yes” 則不會繼續, 其中一個方法是將 if 判斷式加入多一個選項, 例如改成這樣:
1 2 3 4 5 6 |
#!/usr/bin/python action = raw_input('continue?') if action == 'yes' or action == 'Yes': print 'Hello World!' |
上面程式碼對輸入 “yes” 及 “Yes” 均有效, 但如果使用者輸入 “YES”, “YEs”, “yEs”…… 則不會繼續, 所以較好的寫法是用 lower() 將輸入轉換為英文小寫, 再進行判斷, 例如:
1 2 3 4 5 6 |
#!/usr/bin/python action = raw_input('continue?') if action.lower() == 'yes': print 'Hello World!' |
以上程式碼只要是 “yes” 的輸入均有效, 那一個是大小寫英文字母也可以。