Git
/
cheat
mirror of https://github.com/cheat/cheat.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added a condition in prompt_yes_or_no() to use input or raw_input

This commit is contained in:
Erik Cox 2014-09-13 15:46:57 -07:00
parent d1d5d84e4c
commit dde3acf109
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cheat/utils.py
View File

@ -48,8 +48,16 @@ def editor():
def prompt_yes_or_no(question): def prompt_yes_or_no(question):
""" Prompts the user with a yes-or-no question """ """ Prompts the user with a yes-or-no question """
# Support Python 2 and 3 input
# Default to Python 2's input()
get_input = raw_input
# If this is Python 3, use input()
if sys.version_info[:2] >= (3, 0):
get_input = input
print(question) print(question)
return raw_input('[y/n] ') == 'y' return get_input('[y/n] ') == 'y'
def warn(message): def warn(message):