summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGuido van Rossum <guido@python.org>2016-01-07 10:56:36 -0800
committerGuido van Rossum <guido@python.org>2016-01-07 10:56:36 -0800
commit71eea260396a79e645615fbfc13d9bdb350ad145 (patch)
tree9c5bd8da56f616162712a9f1e15cf818ce3d6058
parentfad6487381fb717c08f6fe8dd72f5eb4c8149ec3 (diff)
downloadcpython-71eea260396a79e645615fbfc13d9bdb350ad145.tar.gz
Add another try/except PermissionError to avoid depending on listdir order. Fix issues #24120 and #26012.
-rw-r--r--Lib/pathlib.py13
-rw-r--r--Lib/test/test_pathlib.py16
2 files changed, 16 insertions, 13 deletions
diff --git a/Lib/pathlib.py b/Lib/pathlib.py
index 1bd457a487..8fef2590b8 100644
--- a/Lib/pathlib.py
+++ b/Lib/pathlib.py
@@ -494,11 +494,14 @@ class _RecursiveWildcardSelector(_Selector):
def _iterate_directories(self, parent_path, is_dir, listdir):
yield parent_path
- for name in listdir(parent_path):
- path = parent_path._make_child_relpath(name)
- if is_dir(path) and not path.is_symlink():
- for p in self._iterate_directories(path, is_dir, listdir):
- yield p
+ try:
+ for name in listdir(parent_path):
+ path = parent_path._make_child_relpath(name)
+ if is_dir(path) and not path.is_symlink():
+ for p in self._iterate_directories(path, is_dir, listdir):
+ yield p
+ except PermissionError:
+ return
def _select_from(self, parent_path, is_dir, exists, listdir):
try:
diff --git a/Lib/test/test_pathlib.py b/Lib/test/test_pathlib.py
index 061c7645f2..a12af225c8 100644
--- a/Lib/test/test_pathlib.py
+++ b/Lib/test/test_pathlib.py
@@ -1240,7 +1240,7 @@ class _BasePathTest(object):
# | |-- dirD
# | | `-- fileD
# | `-- fileC
- # |-- dirE
+ # |-- dirE # No permissions
# |-- fileA
# |-- linkA -> fileA
# `-- linkB -> dirB
@@ -1396,13 +1396,13 @@ class _BasePathTest(object):
p = P(BASE)
it = p.rglob("fileA")
self.assertIsInstance(it, collections.Iterator)
- # XXX cannot test because of symlink loops in the test setup
- #_check(it, ["fileA"])
- #_check(p.rglob("fileB"), ["dirB/fileB"])
- #_check(p.rglob("*/fileA"), [""])
- #_check(p.rglob("*/fileB"), ["dirB/fileB"])
- #_check(p.rglob("file*"), ["fileA", "dirB/fileB"])
- # No symlink loops here
+ _check(it, ["fileA"])
+ _check(p.rglob("fileB"), ["dirB/fileB"])
+ _check(p.rglob("*/fileA"), [])
+ _check(p.rglob("*/fileB"), ["dirB/fileB", "dirB/linkD/fileB",
+ "linkB/fileB", "dirA/linkC/fileB"])
+ _check(p.rglob("file*"), ["fileA", "dirB/fileB",
+ "dirC/fileC", "dirC/dirD/fileD"])
p = P(BASE, "dirC")
_check(p.rglob("file*"), ["dirC/fileC", "dirC/dirD/fileD"])
_check(p.rglob("*/*"), ["dirC/dirD/fileD"])