revert to shortening messages thrown by inspect.getsource before raising

mlauter · mlauter · commit dab1ccbe62c2 · 2014-09-24T17:05:28.000-04:00
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -600,7 +600,15 @@ def get_source_of_current_name(self):
                 obj = self.get_object(line)
             if obj is None:
                 raise NameError("%s is not defined" % line)
-        return inspect.getsource(obj) #throws an exception that we'll catch
+        try: 
+            inspect.getsource(obj)
+        except TypeError, e:
+            msg = e.msg
+            if "built-in" in msg:
+                msg.split(">")
+                raise TypeError("Cannot access source of <built-in function %s>" % self.current_line)
+            else:
+                raise TypeError("No source code found for %s" % self.current_line)
 
     def set_docstring(self):
         self.docstring = None
