-
Notifications
You must be signed in to change notification settings - Fork 4
/
queryUser.py
32 lines (28 loc) · 911 Bytes
/
queryUser.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# -*- coding: utf-8 -*-
"""
Created on Tue Jul 05 11:09:29 2016
@author: Drew
"""
import sys
def queryUser(question):
"""
Ask a simple yes/no question of the user.
"""
valid = {"yes": True, "y": True, "ye": True,
"no": False, "n": False}
while True:
try:
sys.stdout.write(question + '[y/n]')
choice = raw_input().lower()
if choice == '':
return valid['y']
elif choice in valid:
return valid[choice]
else:
sys.stdout.write("Please respond with 'yes' or 'no' "
"(or 'y' or 'n').\n")
except KeyboardInterrupt:
# turns out this doesn't fix the problem with IPython console
# console freezes if Ctrl-C during raw-input()
sys.stdout.write("'No' answer assumed.")
return False