Git/cheat
1
0
Fork 0
mirror of https://github.com/cheat/cheat.git synced 2024-12-18 18:55:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:erikcox/cheat into erikcox-master

Browse Source 
* 'master' of github.com:erikcox/cheat:
  Added a condition in prompt_yes_or_no() to use input or raw_input
  First attempt at making utils.py compatible with Python 2 and 3.
This commit is contained in:
Chris Lane 2014-10-19 11:02:32 -04:00
commit 6efae113cf
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cheat/utils.py
View File

@ -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)