]> granicus.if.org Git - python/commitdiff
TarFile.__init__() no longer fails if no name argument is passed and
authorLars Gustäbel <lars@gustaebel.de>
Tue, 28 Aug 2007 12:33:15 +0000 (12:33 +0000)
committerLars Gustäbel <lars@gustaebel.de>
Tue, 28 Aug 2007 12:33:15 +0000 (12:33 +0000)
the fileobj argument has no usable name attribute (e.g. StringIO).

(backported from r57616)

Lib/tarfile.py
Lib/test/test_tarfile.py
Misc/NEWS

index 8afea475d8be5d858045da19ae5e5eb17ebc85ed..45997cbf1ae69c68ca54abdc89f990ba16827912 100644 (file)
@@ -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
index dc262f656da105d791704c0a0f00db2975f1405e..5a5099604378c2cebc11a395e74ddc51bb45b2f2 100644 (file)
@@ -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
index 0e2985c9cb784181aae79b4d8b3aa324996527ec..a859e408c15150e8a71b5b5a0c7d91e55ad95dbc 100644 (file)
--- 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.