diff options
author | Ezio Melotti <ezio.melotti@gmail.com> | 2013-03-30 03:10:28 (GMT) |
---|---|---|
committer | Ezio Melotti <ezio.melotti@gmail.com> | 2013-03-30 03:10:28 (GMT) |
commit | e66e7de5d69542d9f10010a83787ef4b87a70bf8 (patch) | |
tree | ad2c3abd8422c8ecf41414e62f5c0e51ba63e7dc /Lib | |
parent | f03aab7e519b2c801b193907086f064867ef53c1 (diff) | |
download | cpython-e66e7de5d69542d9f10010a83787ef4b87a70bf8.zip cpython-e66e7de5d69542d9f10010a83787ef4b87a70bf8.tar.gz cpython-e66e7de5d69542d9f10010a83787ef4b87a70bf8.tar.bz2 |
#17526: fix an IndexError raised while passing code without filename to inspect.findsource(). Initial patch by Tyler Doyle.
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/inspect.py | 2 | ||||
-rw-r--r-- | Lib/test/test_inspect.py | 6 |
2 files changed, 7 insertions, 1 deletions
diff --git a/Lib/inspect.py b/Lib/inspect.py index 66d5186..9e928b2 100644 --- a/Lib/inspect.py +++ b/Lib/inspect.py @@ -525,7 +525,7 @@ def findsource(object): file = getfile(object) sourcefile = getsourcefile(object) - if not sourcefile and file[0] + file[-1] != '<>': + if not sourcefile and file[:1] + file[-1:] != '<>': raise IOError('source code not available') file = sourcefile if sourcefile else file diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py index 30b1556..04dcfe9 100644 --- a/Lib/test/test_inspect.py +++ b/Lib/test/test_inspect.py @@ -404,6 +404,12 @@ class TestBuggyCases(GetSourceBase): self.assertEqual(inspect.findsource(co), (lines,0)) self.assertEqual(inspect.getsource(co), lines[0]) + def test_findsource_without_filename(self): + for fname in ['', '<string>']: + co = compile('x=1', fname, "exec") + self.assertRaises(IOError, inspect.findsource, co) + self.assertRaises(IOError, inspect.getsource, co) + class _BrokenDataDescriptor(object): """ |