From 34d5106804f793eef61d0aa1c35e084f9e63063b Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Tue, 20 Aug 2019 22:05:20 -0600
Subject: [PATCH] Added _get_valid_answers() to Menu()

---
 scripts/wk/std.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/scripts/wk/std.py b/scripts/wk/std.py
index cde2595c..6e3a0a2e 100644
--- a/scripts/wk/std.py
+++ b/scripts/wk/std.py
@@ -137,6 +137,23 @@ class Menu():
     # Done
     return self.separator * separator_length
 
+  def _get_valid_answers(self):
+    """Get valid answers based on menu items, returns list."""
+    valid_answers = []
+
+    # Numbered items
+    max_value = 0
+    for section in (self.sets, self.toggles, self.options):
+      max_value += len(section)
+    valid_answers.extend([str(x+1) for x in range(max_value)])
+
+    # Action items
+    for name in self.actions.keys():
+      valid_answers.append(name[:1].upper())
+
+    # Done
+    return valid_answers
+
   def _update(self, single_selection=True):
     """Update menu items in preparation for printing to screen."""
     index = 0