Let yes/no dialog accept 'y' or 'n' as input

Change-Id: I6680e13c55568f42e07f62eeb496bc7d23a322f7
2014-05-05 13:45:02 -07:00 · 2014-05-05 13:45:02 -07:00 · d3db343b39
commit d3db343b39
parent 80cacc5759
1 changed files with 9 additions and 0 deletions
--- a/gertty/mywid.py
+++ b/gertty/mywid.py
@ -89,3 +89,12 @@ class YesNoDialog(ButtonDialog):
                             lambda button:self._emit('no'))
        super(YesNoDialog, self).__init__(title, message, buttons=[yes_button,
                                                                   no_button])
+    def keypress(self, size, key):
+        r = super(YesNoDialog, self).keypress(size, key)
+        if r in ('Y', 'y'):
+            self._emit('yes')
+            return None
+        if r in ('N', 'n'):
+            self._emit('no')
+            return None
+        return r