From: Raymond Hettinger <python@rcn.com>
Date: Tue, 14 Jan 2003 02:19:36 +0000 (+0000)
Subject: SF bug #661184: inspect.getsource bug
X-Git-Tag: v2.3c1~2460
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=2d375f78a570dc627daf61df9d12473633055cdf;p=thirdparty%2FPython%2Fcpython.git

SF bug #661184: inspect.getsource bug

inspect.getsource would crash with one line definitions like:
   def f(x): return x
or
   f = lambda x: x
---

diff --git a/Lib/inspect.py b/Lib/inspect.py
index d3fd3ad195a6..77129fd7324d 100644
--- a/Lib/inspect.py
+++ b/Lib/inspect.py
@@ -417,7 +417,7 @@ def findsource(object):
         if not hasattr(object, 'co_firstlineno'):
             raise IOError, 'could not find function definition'
         lnum = object.co_firstlineno - 1
-        pat = re.compile(r'^\s*def\s')
+        pat = re.compile(r'^(\s*def\s)|(.*\slambda(:|\s))')
         while lnum > 0:
             if pat.match(lines[lnum]): break
             lnum = lnum - 1
@@ -508,6 +508,8 @@ def getblock(lines):
         tokenize.tokenize(ListReader(lines).readline, BlockFinder().tokeneater)
     except EndOfBlock, eob:
         return lines[:eob.args[0]]
+    # Fooling the indent/dedent logic implies a one-line definition
+    return lines[:1]
 
 def getsourcelines(object):
     """Return a list of source lines and starting line number for an object.