From a9bad98e3df5be4eb96f94bd52d5711c4404bd3e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lars=20Gust=C3=A4bel?= Date: Tue, 28 Aug 2007 12:33:15 +0000 Subject: [PATCH] TarFile.__init__() no longer fails if no name argument is passed and the fileobj argument has no usable name attribute (e.g. StringIO). (backported from r57616) --- Lib/tarfile.py | 11 ++++------- Lib/test/test_tarfile.py | 24 +++++++++++++++++++----- Misc/NEWS | 3 +++ 3 files changed, 26 insertions(+), 12 deletions(-) diff --git a/Lib/tarfile.py b/Lib/tarfile.py index 8afea475d8b..45997cbf1ae 100644 --- a/Lib/tarfile.py +++ b/Lib/tarfile.py @@ -1044,24 +1044,21 @@ class TarFile(object): can be determined, `mode' is overridden by `fileobj's mode. `fileobj' is not closed, when TarFile is closed. """ - self.name = name - if self.name is not None: - self.name = os.path.abspath(name) - if len(mode) > 1 or mode not in "raw": raise ValueError("mode must be 'r', 'a' or 'w'") self._mode = mode self.mode = {"r": "rb", "a": "r+b", "w": "wb"}[mode] if not fileobj: - fileobj = file(self.name, self.mode) + fileobj = file(name, self.mode) self._extfileobj = False else: - if self.name is None and hasattr(fileobj, "name"): - self.name = os.path.abspath(fileobj.name) + if name is None and hasattr(fileobj, "name"): + name = fileobj.name if hasattr(fileobj, "mode"): self.mode = fileobj.mode self._extfileobj = True + self.name = os.path.abspath(name) if name else None self.fileobj = fileobj # Init datastructures diff --git a/Lib/test/test_tarfile.py b/Lib/test/test_tarfile.py index dc262f656da..5a509960437 100644 --- a/Lib/test/test_tarfile.py +++ b/Lib/test/test_tarfile.py @@ -642,11 +642,25 @@ class OpenFileobjTest(BaseTest): except tarfile.ReadError: self.assertEqual(fobj.tell(), 0, "fileobj's position has moved") - def test_fileobj(self): - # Test for SF bug #1695229, opening a tarfile without - # a name argument. - tarfile.open(mode="r", fileobj=open(tarname(""))) - tarfile.TarFile(mode="r", fileobj=open(tarname(""))) + def test_no_name_argument(self): + fobj = open(testtar, "rb") + tar = tarfile.open(fileobj=fobj, mode="r") + self.assertEqual(tar.name, os.path.abspath(fobj.name)) + + def test_no_name_attribute(self): + data = open(testtar, "rb").read() + fobj = StringIO.StringIO(data) + self.assertRaises(AttributeError, getattr, fobj, "name") + tar = tarfile.open(fileobj=fobj, mode="r") + self.assertEqual(tar.name, None) + + def test_empty_name_attribute(self): + data = open(testtar, "rb").read() + fobj = StringIO.StringIO(data) + fobj.name = "" + tar = tarfile.open(fileobj=fobj, mode="r") + self.assertEqual(tar.name, None) + if bz2: # Bzip2 TestCases diff --git a/Misc/NEWS b/Misc/NEWS index 0e2985c9cb7..a859e408c15 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -26,6 +26,9 @@ Core and builtins Library ------- +- TarFile.__init__() no longer fails if no name argument is passed and + the fileobj argument has no usable name attribute (e.g. StringIO). + - Reverted the fix for bug #1548891 because it broke compatibility with arbitrary read buffers. Added a note in the documentation.