mirror of https://github.com/python/cpython
Patch #1504073: Fix tarfile.open() for mode "r" with a fileobj argument.
Backport from rev. 53161.
This commit is contained in:
parent
12e087a1b1
commit
f9a2c63c79
|
@ -1133,9 +1133,13 @@ class TarFile(object):
|
|||
# Find out which *open() is appropriate for opening the file.
|
||||
for comptype in cls.OPEN_METH:
|
||||
func = getattr(cls, cls.OPEN_METH[comptype])
|
||||
if fileobj is not None:
|
||||
saved_pos = fileobj.tell()
|
||||
try:
|
||||
return func(name, "r", fileobj)
|
||||
except (ReadError, CompressionError):
|
||||
if fileobj is not None:
|
||||
fileobj.seek(saved_pos)
|
||||
continue
|
||||
raise ReadError("file could not be opened successfully")
|
||||
|
||||
|
|
|
@ -625,6 +625,15 @@ class FileModeTest(unittest.TestCase):
|
|||
self.assertEqual(tarfile.filemode(0755), '-rwxr-xr-x')
|
||||
self.assertEqual(tarfile.filemode(07111), '---s--s--t')
|
||||
|
||||
class OpenFileobjTest(BaseTest):
|
||||
# Test for SF bug #1496501.
|
||||
|
||||
def test_opener(self):
|
||||
fobj = StringIO.StringIO("foo\n")
|
||||
try:
|
||||
tarfile.open("", "r", fileobj=fobj)
|
||||
except tarfile.ReadError:
|
||||
self.assertEqual(fobj.tell(), 0, "fileobj's position has moved")
|
||||
|
||||
if bz2:
|
||||
# Bzip2 TestCases
|
||||
|
@ -670,6 +679,7 @@ def test_main():
|
|||
|
||||
tests = [
|
||||
FileModeTest,
|
||||
OpenFileobjTest,
|
||||
ReadTest,
|
||||
ReadStreamTest,
|
||||
ReadDetectTest,
|
||||
|
|
Loading…
Reference in New Issue