(original) (raw)

changeset: 88265:ef2b2ddd27c8 user: Antoine Pitrou solipsis@pitrou.net date: Fri Jan 03 00:07:17 2014 +0100 files: Lib/pathlib.py Lib/test/test_pathlib.py Misc/NEWS description: Issue #20111: pathlib.Path.with_suffix() now sanity checks the given suffix. diff -r 3e75f649e93b -r ef2b2ddd27c8 Lib/pathlib.py --- a/Lib/pathlib.py Thu Jan 02 16:48:24 2014 -0600 +++ b/Lib/pathlib.py Fri Jan 03 00:07:17 2014 +0100 @@ -755,6 +755,12 @@ def with_suffix(self, suffix): """Return a new path with the file suffix changed (or added, if none).""" # XXX if suffix is None, should the current suffix be removed? + drv, root, parts = self._flavour.parse_parts((suffix,)) + if drv or root or len(parts) != 1: + raise ValueError("Invalid suffix %r" % (suffix)) + suffix = parts[0] + if not suffix.startswith('.'): + raise ValueError("Invalid suffix %r" % (suffix)) name = self.name if not name: raise ValueError("%r has an empty name" % (self,)) diff -r 3e75f649e93b -r ef2b2ddd27c8 Lib/test/test_pathlib.py --- a/Lib/test/test_pathlib.py Thu Jan 02 16:48:24 2014 -0600 +++ b/Lib/test/test_pathlib.py Fri Jan 03 00:07:17 2014 +0100 @@ -528,9 +528,16 @@ self.assertEqual(P('/a/b').with_suffix('.gz'), P('/a/b.gz')) self.assertEqual(P('a/b.py').with_suffix('.gz'), P('a/b.gz')) self.assertEqual(P('/a/b.py').with_suffix('.gz'), P('/a/b.gz')) + # Path doesn't have a "filename" component self.assertRaises(ValueError, P('').with_suffix, '.gz') self.assertRaises(ValueError, P('.').with_suffix, '.gz') self.assertRaises(ValueError, P('/').with_suffix, '.gz') + # Invalid suffix + self.assertRaises(ValueError, P('a/b').with_suffix, 'gz') + self.assertRaises(ValueError, P('a/b').with_suffix, '/') + self.assertRaises(ValueError, P('a/b').with_suffix, '/.gz') + self.assertRaises(ValueError, P('a/b').with_suffix, 'c/d') + self.assertRaises(ValueError, P('a/b').with_suffix, '.c/.d') def test_relative_to_common(self): P = self.cls @@ -920,10 +927,23 @@ self.assertEqual(P('c:/a/b').with_suffix('.gz'), P('c:/a/b.gz')) self.assertEqual(P('c:a/b.py').with_suffix('.gz'), P('c:a/b.gz')) self.assertEqual(P('c:/a/b.py').with_suffix('.gz'), P('c:/a/b.gz')) + # Path doesn't have a "filename" component self.assertRaises(ValueError, P('').with_suffix, '.gz') self.assertRaises(ValueError, P('.').with_suffix, '.gz') self.assertRaises(ValueError, P('/').with_suffix, '.gz') self.assertRaises(ValueError, P('//My/Share').with_suffix, '.gz') + # Invalid suffix + self.assertRaises(ValueError, P('c:a/b').with_suffix, 'gz') + self.assertRaises(ValueError, P('c:a/b').with_suffix, '/') + self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\') + self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:') + self.assertRaises(ValueError, P('c:a/b').with_suffix, '/.gz') + self.assertRaises(ValueError, P('c:a/b').with_suffix, '\\.gz') + self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c:.gz') + self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c/d') + self.assertRaises(ValueError, P('c:a/b').with_suffix, 'c\\d') + self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c/d') + self.assertRaises(ValueError, P('c:a/b').with_suffix, '.c\\d') def test_relative_to(self): P = self.cls diff -r 3e75f649e93b -r ef2b2ddd27c8 Misc/NEWS --- a/Misc/NEWS Thu Jan 02 16:48:24 2014 -0600 +++ b/Misc/NEWS Fri Jan 03 00:07:17 2014 +0100 @@ -44,6 +44,8 @@ Library ------- +- Issue #20111: pathlib.Path.with_suffix() now sanity checks the given suffix. + - Fix breakage in TestSuite.countTestCases() introduced by issue #11798. - Issue #20108: Avoid parameter name clash in inspect.getcallargs(). /solipsis@pitrou.net