From 2d375f78a570dc627daf61df9d12473633055cdf Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Tue, 14 Jan 2003 02:19:36 +0000 Subject: [PATCH] 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 --- Lib/inspect.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/Lib/inspect.py b/Lib/inspect.py index d3fd3ad195a..77129fd7324 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.