Issue #123 fixed for bpython-urwid

mmaker · mmaker · commit e84680f96106 · 2010-07-18T19:35:24.000+02:00
diff --git a/bpython/urwid.py b/bpython/urwid.py
@@ -391,7 +391,12 @@ def __init__(self, config, statusbar=None):
 
     def confirm(self, q):
         """Ask for yes or no and return boolean"""
-        return self.statusbar.prompt(q).lower().startswith('y')
+        try:
+            reply = self.statusbar.prompt(q)
+        except ValueError:
+            return False
+
+        return reply.lower() in ('y', 'yes')
 
     def notify(self, s, n=10):
         return self.statusbar.message(s, n)
@@ -597,7 +602,12 @@ def getstdout(self):
 
     def ask_confirmation(self, q):
         """Ask for yes or no and return boolean"""
-        return self.statusbar.prompt(q).lower().startswith('y')
+        try:
+            reply = self.statusbar.prompt(q)
+        except ValueError:
+            return False
+
+        return reply.lower() in ('y', 'yes')
 
     def reevaluate(self):
         """Clear the buffer, redraw the screen and re-evaluate the history"""
