make other frontends work

thomasballinger · thomasballinger · commit f9fa7e8d60ed · 2014-09-15T16:22:27.000-04:00
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -736,6 +736,7 @@ def update_completion(self, tab=False):
         self.list_win_visible = BpythonRepl.complete(self, tab)
 
     def predicted_indent(self, line):
+        #TODO get rid of this! It's repeated code! Combine with Repl.
         logger.debug('line is %r', line)
         indent = len(re.match(r'[ ]*', line).group())
         if line.endswith(':'):
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -902,7 +902,8 @@ def undo(self, n=1):
         entries = list(self.rl_history.entries)
 
         self.history = self.history[:-n]
-        if n == 1 and not self.done:
+        if (n == 1 and self.buffer and
+            hasattr(self, 'take_back_buffer_line')):
             self.take_back_buffer_line()
         else:
             self.reevaluate()
