Add suffixes

python · miss-islington · May 14, 2021 · May 14, 2021 · May 14, 2021 · May 14, 2021
commit 65ded7cf50bd3550a458393aa5602e0c7721ccdc
diff --git a/Lib/test/test_zipfile.py b/Lib/test/test_zipfile.py
@@ -3112,6 +3112,28 @@ def test_suffix(self, alpharep):
         d = root / "d"
         assert d.suffix == ""
 
+    @pass_alpharep
+    def test_suffixes(self, alpharep):
+        """
+        The suffix of the root should be the suffix of the zipfile.
+        The suffix of each nested file is the final component's last suffix, if any.
+        Includes the leading period, just like pathlib.Path.
+        """
+        root = zipfile.Path(alpharep)
+        assert root.suffixes == ['.zip'] == root.filename.suffixes
+
+        b = root / 'b.txt'
+        assert b.suffixes == ['.txt']
+
+        c = root / 'c' / 'filename.tar.gz'
+        assert c.suffixes == ['.tar', '.gz']
+
+        d = root / 'd'
+        assert d.suffixes == []
+
+        e = root / '.hgrc'
+        assert e.suffixes == []
+
     @pass_alpharep
     def test_stem(self, alpharep):
         """

diff --git a/Lib/zipfile.py b/Lib/zipfile.py
@@ -2346,6 +2346,10 @@ def name(self):
     def suffix(self):
         return pathlib.Path(self.at).suffix or self.filename.suffix
 
+    @property
+    def suffixes(self):
+        return pathlib.Path(self.at).suffixes or self.filename.suffixes
+
     @property
     def stem(self):
         return pathlib.Path(self.at).stem or self.filename.stem