Mark entry as selected in make_single_selection()

This commit is contained in:
2Shirt 2019-05-30 21:55:07 -06:00
parent 0623b070c3
commit e34455beb0
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C

View file

@ -80,6 +80,10 @@ class MenuState():
# Safety Check
assert self.entries, 'No menu entries defined.'
# Reset selections
for entry in self.entries.values():
entry.selected = False
# Build Menu
i = 1
for name, entry in self.entries.items():
@ -120,7 +124,8 @@ class MenuState():
print('\n'.join(display_list))
_answer = input('Please make a selection: ')
# Save last selection
# Mark and save selection
self.entries[valid_answers[_answer.upper()]].selected = True
self.last_sel = valid_answers[_answer.upper()]
def make_multiple_selections(self):