zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add base class for directory visitor (#32008)

Browse Source
This commit is contained in:
Harmen Stoppels
2022-08-09 15:43:30 +02:00
committed by GitHub
parent b61187455a
commit f53b522572
3 changed files with 103 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/spack/spack/test/llnl/util/filesystem.py
View File

@@ -729,7 +729,7 @@ def test_lexists_islink_isdir(tmpdir):
assert fs.lexists_islink_isdir(symlink_to_symlink_to_file) == (True, True, False)
class RegisterVisitor(object):
class RegisterVisitor(fs.BaseDirectoryVisitor):
"""A directory visitor that keeps track of all visited paths"""
def __init__(self, root, follow_dirs=True, follow_symlink_dirs=True):
@@ -751,6 +751,9 @@ def visit_file(self, root, rel_path, depth):
self.check(root, rel_path, depth)
self.files.append(rel_path)
def visit_symlinked_file(self, root, rel_path, depth):
self.visit_file(root, rel_path, depth)
def before_visit_dir(self, root, rel_path, depth):
self.check(root, rel_path, depth)
self.dirs_before.append(rel_path)