wait for keypress to clear URL containing messages

thomasballinger · thomasballinger · commit 70ec8e98512c · 2015-01-24T15:06:10.000-05:00
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -321,7 +321,7 @@ def confirm(self, q):
         return reply.lower() in (_('y'), _('yes'))
 
 
-    def notify(self, s, n=10):
+    def notify(self, s, n=10, wait_for_keypress=False):
         return self.statusbar.message(s, n)
 
     def file_prompt(self, s):
diff --git a/bpython/curtsiesfrontend/interaction.py b/bpython/curtsiesfrontend/interaction.py
@@ -54,11 +54,12 @@ def pop_permanent_message(self, msg):
     def has_focus(self):
         return self.in_prompt or self.in_confirm or self.waiting_for_refresh
 
-    def message(self, msg):
+    def message(self, msg, schedule_refresh=True):
         """Sets a temporary message"""
         self.message_start_time = time.time()
         self._message = msg
-        self.schedule_refresh(time.time() + self.message_time)
+        if schedule_refresh:
+            self.schedule_refresh(time.time() + self.message_time)
 
     def _check_for_expired_message(self):
         if self._message and time.time() > self.message_start_time + self.message_time:
@@ -128,10 +129,10 @@ def should_show_message(self):
         return bool(self.current_line)
 
     # interaction interface - should be called from other greenlets
-    def notify(self, msg, n=3):
+    def notify(self, msg, n=3, wait_for_keypress=False):
         self.request_greenlet = greenlet.getcurrent()
         self.message_time = n
-        self.message(msg)
+        self.message(msg, schedule_refresh=wait_for_keypress)
         self.waiting_for_refresh = True
         self.request_refresh()
         self.main_greenlet.switch(msg)
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -240,7 +240,7 @@ def __init__(self, config, statusbar=None):
     def confirm(self, s):
         raise NotImplementedError
 
-    def notify(self, s, n=10):
+    def notify(self, s, n=10, wait_for_keypress=False):
         raise NotImplementedError
 
     def file_prompt(self, s):
diff --git a/bpython/urwid.py b/bpython/urwid.py
@@ -556,7 +556,7 @@ def callback_wrapper(result):
 
         self.prompt(q, callback_wrapper, single=True)
 
-    def notify(self, s, n=10):
+    def notify(self, s, n=10, wait_for_keypress=False):
         return self.statusbar.message(s, n)
 
     def prompt(self, s, callback=None, single=False):
