6
def __init__(self, char, action, help, default=False):
12
class UserInteractor(object):
21
def add_item(self, item):
22
self.items.append(item)
23
self.__total_items = len(self.items)
25
def set_items(self, item_list):
26
self.items = item_list
27
self.__total_items = len(self.items)
29
def set_item_callback(self, cb):
30
self.item_callback = cb
32
def set_start_callback(self, cb):
33
self.start_callback = cb
35
def set_end_callback(self, cb):
36
self.end_callback = cb
38
def set_prompt(self, prompt):
41
def set_options(self, opt_list):
43
self._option_dict = {}
44
for option in opt_list:
45
self.add_option(option)
47
def add_option(self, option):
48
self._options.append(option)
49
self._option_dict[option.char] = option
51
def get_option(self, char):
52
return self._option_dict[char]
54
def __do_action(self, action, item):
55
if type(action) is int:
56
if action == self.QUIT:
58
self.__finished = True
59
elif action == self.RESTART:
61
elif action == self.FINISH:
62
self.__finished = True
67
def __select_loop(self):
70
while i < len(self.items):
73
self.item_callback(item, i + 1)
75
if self.__ask_once(item, i + 1):
78
if self.__quit or self.__finished:
83
self.__finished = False
85
while not self.__quit and not self.__finished:
86
self.__restart = False
92
if self.end_callback():
95
self.__ask_once(None, self.__total_items)
96
while not self.__finished and not self.__restart:
97
self.__ask_once(None, -1)
99
return not self.__quit
101
def __ask_once(self, item, count):
102
args = {'count': count, 'total' : self.__total_items}
105
sys.stdout.write(self.prompt % args)
106
sys.stdout.write(' [')
107
for opt in self._options:
110
sys.stdout.write(opt.char)
111
sys.stdout.write('?] (%s): ' % default.char)
113
response = self.__getchar()
115
# default, which we see as newline, is 'n'
116
if response in ['\n', '\r', '\r\n']:
117
response = default.char
119
print response # because echo is off
121
for opt in self._options:
122
if opt.char == response:
123
return self.__do_action(opt.action, item)
127
def __show_help(self):
128
for opt in self._options:
129
print ' %s - %s' % (opt.char, opt.help)
131
if sys.platform == "win32":
134
return msvcrt.getche()
139
fd = sys.stdin.fileno()
140
settings = termios.tcgetattr(fd)
143
ch = sys.stdin.read(1)
145
termios.tcsetattr(fd, termios.TCSADRAIN, settings)
added
removed
userinteractor.py
