From d3db343b39d18c3bb1b5892fe2b422ce403148e4 Mon Sep 17 00:00:00 2001
From: "James E. Blair" <jeblair@openstack.org>
Date: Mon, 5 May 2014 13:45:02 -0700
Subject: [PATCH] Let yes/no dialog accept 'y' or 'n' as input

Change-Id: I6680e13c55568f42e07f62eeb496bc7d23a322f7
---
 gertty/mywid.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/gertty/mywid.py b/gertty/mywid.py
index 77df6b7..aee227b 100644
--- 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