diff --git a/cheat/utils.py b/cheat/utils.py index 4010c15..28ba2eb 100644 --- a/cheat/utils.py +++ b/cheat/utils.py @@ -1,3 +1,4 @@ +from __future__ import print_function import os import sys @@ -47,10 +48,18 @@ def editor(): def prompt_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) - return raw_input('[y/n] ') == 'y' + return get_input('[y/n] ') == 'y' def warn(message): """ Prints a message to stderr """ - print >> sys.stderr, (message) + print((message), file=sys.stderr)