summaryrefslogtreecommitdiffstats
path: root/Lib/inspect.py
diff options
context:
space:
mode:
authorBenjamin Peterson <benjamin@python.org>2011-06-11 20:53:11 (GMT)
committerBenjamin Peterson <benjamin@python.org>2011-06-11 20:53:11 (GMT)
commit0eb4ac4c6258ecbcbcb65022f3987fe30f31f527 (patch)
tree24b81f9adb51190ddff71d3d115293393de7b573 /Lib/inspect.py
parent6abd97e3b2adbce537c65099d480e088bb348ea0 (diff)
downloadcpython-0eb4ac4c6258ecbcbcb65022f3987fe30f31f527.zip
cpython-0eb4ac4c6258ecbcbcb65022f3987fe30f31f527.tar.gz
cpython-0eb4ac4c6258ecbcbcb65022f3987fe30f31f527.tar.bz2
allow "fake" filenames in findsource (closes #9284)
This allows findsource() to work in doctests. A patch from Dirkjan Ochtman.
Diffstat (limited to 'Lib/inspect.py')
-rw-r--r--Lib/inspect.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/Lib/inspect.py b/Lib/inspect.py
index b0b2d3a..c727530 100644
--- a/Lib/inspect.py
+++ b/Lib/inspect.py
@@ -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__)