Python pathlib2-based recursive directory iterator with depth limits

dig.py

def _dig(p, max_depth, depth=1):
    for sp in p.iterdir():
        if sp.is_file():
            yield sp
        elif sp.is_dir() and depth <= max_depth:
            depth += 1
            for ssp in _dig(sp, max_depth, depth):
                yield ssp