fix show_source method of curtsiesfrontend/repl to catch exceptions from get_source_of_current_name

mlauter · mlauter · commit 13eb5188dd3f · 2014-09-24T15:41:59.000-04:00
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -1300,13 +1300,16 @@ def pager(self, text):
             self.focus_on_subprocess(command + [tmp.name])
 
     def show_source(self):
-        source = self.get_source_of_current_name()
-        if source is None:
-            self.status_bar.message(_('Cannot show source.'))
-        else:
+        try:
+            source = self.get_source_of_current_name()
             if self.config.highlight_show_source:
-                source = format(PythonLexer().get_tokens(source), TerminalFormatter())
+                source = format(PythonLexer().get_tokens(source),
+                                TerminalFormatter())
             self.pager(source)
+        except (ValueError, NameError), e:
+            self.status_bar.message(_(e))
+        except (AttributeError, IOError, TypeError), e:
+            self.status_bar.message(_('Failed to get source: %s' % e))
 
     def help_text(self):
         return (self.version_help_text() + '\n' + self.key_help_text()).encode('utf8')
