Catch IndexError when calling inspect.getsourcelines() and pydoc.getdoc().

Trundle · Trundle · commit 8b4c0cc8eea6 · 2010-09-13T21:10:26.000+02:00
This closes issue bpython#94.
diff --git a/bpython/inspection.py b/bpython/inspection.py
@@ -156,7 +156,9 @@ def fixlongargs(f, argspec):
     keys = argspec[0][-len(values):]
     try:
         src = inspect.getsourcelines(f)
-    except IOError:
+    except (IOError, IndexError):
+        # IndexError is raised in inspect.findsource(), can happen in
+        # some situations. See issue #94.
         return
     signature = ''.join(src[0])
     kwparsed = parsekeywordpairs(signature)
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -534,9 +534,15 @@ def complete(self, tab=False):
         if not self.get_args():
             self.argspec = None
         elif self.current_func is not None:
-            self.docstring = pydoc.getdoc(self.current_func)
-            if not self.docstring:
+            try:
+                self.docstring = pydoc.getdoc(self.current_func)
+            except IndexError:
                 self.docstring = None
+            else:
+                # pydoc.getdoc() returns an empty string if no
+                # docstring was found
+                if not self.docstring:
+                    self.docstring = None
 
         cw = self.cw()
         cs = self.current_string()
