mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-17 23:31:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

allow "fake" filenames in findsource (closes #9284)

Browse source 
This allows findsource() to work in doctests.

A patch from Dirkjan Ochtman.
This commit is contained in:
Benjamin Peterson 2011-06-11 15:53:11 -05:00
parent 6abd97e3b2
commit 0eb4ac4c62
3 changed files with 26 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Lib/inspect.py
View file

@ -524,9 +524,13 @@ def findsource(object):
or code object. The source code is returned as a list of all the lines
in the file and the line number indexes a line in that list. An IOError
is raised if the source code cannot be retrieved."""
file = getsourcefile(object)
if not file:
file = getfile(object)
sourcefile = getsourcefile(object)
if not sourcefile and file[0] + file[-1] != '<>':
raise IOError('source code not available')
file = sourcefile if sourcefile else file
module = getmodule(object, file)
if module:
lines = linecache.getlines(file, module.__dict__)