Commit 2d375f78 authored by Raymond Hettinger's avatar Raymond Hettinger

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
parent 9896ea24
......@@ -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.
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment