Travel directory tree with limited recursion depth

by

I think the easiest and most stable approach would be to copy the functionality of os.walk straight out of the source and insert your own depth-controlling parameter.

import os
import os.path as path

def walk(top, topdown=True, onerror=None, followlinks=False, maxdepth=None):
    islink, join, isdir = path.islink, path.join, path.isdir

    try:
        names = os.listdir(top)
    except OSError, err:
        if onerror is not None:
            onerror(err)
        return

    dirs, nondirs = [], []
    for name in names:
        if isdir(join(top, name)):
            dirs.append(name)
        else:
            nondirs.append(name)

    if topdown:
        yield top, dirs, nondirs

    if maxdepth is None or maxdepth > 1:
        for name in dirs:
            new_path = join(top, name)
            if followlinks or not islink(new_path):
                for x in walk(new_path, topdown, onerror, followlinks, None if maxdepth is None else maxdepth-1):
                    yield x
    if not topdown:
        yield top, dirs, nondirs

for root, dirnames, filenames in walk(args.directory, maxdepth=2):
    #...

If you’re not interested in all those optional parameters, you can pare down the function pretty substantially:

import os

def walk(top, maxdepth):
    dirs, nondirs = [], []
    for name in os.listdir(top):
        (dirs if os.path.isdir(os.path.join(top, name)) else nondirs).append(name)
    yield top, dirs, nondirs
    if maxdepth > 1:
        for name in dirs:
            for x in walk(os.path.join(top, name), maxdepth-1):
                yield x

for x in walk(".", 2):
    print(x)

More Related Contents:

  1. How to open a file within python?
  2. A simple maths operation on a CSV column using python [closed]
  3. Why am getting an "elif" syntax error showing invalid syntax in python [closed]
  4. Why is the id of a Python class not unique when called quickly?
  5. Freezing/Hanging tkinter GUI in waiting for the thread to complete
  6. How can I connect to MySQL in Python 3 on Windows?
  7. ‘str’ object has no attribute ‘decode’. Python 3 error?
  8. Using multiple Python engines (32Bit/64bit and 2.7/3.5)
  9. Importing class from another file [duplicate]
  10. Getting TypeError: __init__() missing 1 required positional argument: ‘on_delete’ when trying to add parent table after child table with entries
  11. What is the meaning of “Failed building wheel for X” in pip install?
  12. When should iteritems() be used instead of items()?
  13. Clear QLineEdit on click event
  14. How to set the timezone in Django
  15. How do I convert a string into an f-string?
  16. Understand Python swapping: why is a, b = b, a not always equivalent to b, a = a, b?
  17. pip no longer working after update error ‘module’ object is not callable
  18. What is the correct syntax for ‘else if’?
  19. os.walk without hidden folders
  20. Python – Printing a dictionary as a horizontal table with headers
  21. TypeError: ‘zip’ object is not subscriptable
  22. Read .csv file from URL into Python 3.x – _csv.Error: iterator should return strings, not bytes (did you open the file in text mode?)
  23. How to find cube root using Python? [duplicate]
  24. When using cx_Freeze and tkinter I get: “DLL load failed: The specified module could not be found.” (Python 3.5.3)
  25. How do you change the value of one attribute by changing the value of another? (dependent attributes)
  26. Need the path for particular files using os.walk() [duplicate]
  27. How to force Sphinx to use Python 3.x interpreter
  28. sys.stdin.readline() and input(): which one is faster when reading lines of input, and why?
  29. Weird MRO result when inheriting directly from typing.NamedTuple
  30. How to combine Celery with asyncio?

Leave a Comment