21 lines
666 B
Python
21 lines
666 B
Python
from collections.abc import Iterator
|
|
from itertools import chain
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Callable
|
|
|
|
|
|
def visit_path(
|
|
path: Path, *, filter: Callable[[Path], bool], recurse: Callable[[Path], bool]
|
|
) -> Iterator[Path]:
|
|
"""
|
|
Implements the interface of ``py.path.local.visit()`` for Path objects,
|
|
to simplify porting the code over from ``py.path.local``.
|
|
"""
|
|
for dirpath, dirnames, filenames in os.walk(path):
|
|
dirnames[:] = [x for x in dirnames if recurse(Path(dirpath, x))]
|
|
for name in chain(dirnames, filenames):
|
|
p = Path(dirpath, name)
|
|
if filter(p):
|
|
yield p
|