modify the NameError message displayed by show_source

mlauter · mlauter · commit 31db4c7e27bc · 2014-09-24T16:10:58.000-04:00
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -974,8 +974,10 @@ def p_key(self, key):
                     source = format(PythonLexer().get_tokens(source),
                                     TerminalFormatter())
                 page(source)
-            except (ValueError, NameError), e:
+            except (ValueError), e:
                 self.statusbar.message(_(e))
+            except (NameError), e:
+                self.statusbar.message(_('Cannot get source: %s' % e))
             except (AttributeError, IOError, TypeError), e:
                 self.statusbar.message(_('Failed to get source: %s' % e))
             return ''
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -1306,8 +1306,10 @@ def show_source(self):
                 source = format(PythonLexer().get_tokens(source),
                                 TerminalFormatter())
             self.pager(source)
-        except (ValueError, NameError), e:
+        except (ValueError), e:
             self.status_bar.message(_(e))
+        except (NameError), e:
+            self.status_bar.message(_('Cannot get source: %s' % e))
         except (AttributeError, IOError, TypeError), e:
             self.status_bar.message(_('Failed to get source: %s' % e))
 
