take back line working for bpythno-curtsies

thomasballinger · thomasballinger · commit 192cbe9ca4b5 · 2014-09-08T16:49:16.000-04:00
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -1193,6 +1193,13 @@ def reprint_line(self, lineno, tokens):
         logger.debug("calling reprint line with %r %r", lineno, tokens)
         if self.config.syntax:
             self.display_buffer[lineno] = bpythonparse(format(tokens, self.formatter))
+
+    def take_back_buffer_line(self):
+        self.display_buffer.pop()
+        self.buffer.pop()
+        self.cursor_offset = 0
+        self.current_line = ''
+
     def reevaluate(self, insert_into_history=False):
         """bpython.Repl.undo calls this"""
         if self.watcher: self.watcher.reset()
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -902,8 +902,10 @@ def undo(self, n=1):
         entries = list(self.rl_history.entries)
 
         self.history = self.history[:-n]
-
-        self.reevaluate()
+        if n == 1 and not self.done:
+            self.take_back_buffer_line()
+        else:
+            self.reevaluate()
 
         self.rl_history.entries = entries
 
