Không bao gồm các thư mục trong os.walk

Tôi đang viết một tập lệnh đi xuống cây thư mục (sử dụng os.walk ()) và sau đó truy cập từng tệp khớp với một phần mở rộng tệp nhất định. Tuy nhiên, vì một số các cây thư mục đó công cụ của tôi sẽ được sử dụng trên cũng chứa thư mục con mà lần lượt chứa một LOT của vô dụng (đối với mục đích của kịch bản này) công cụ, tôi figured tôi muốn thêm một lựa chọn cho người sử dụng để xác định một danh sách các thư mục để loại trừ khỏi giao dịch.

Điều này là đủ dễ dàng với os.walk (). Rốt cuộc, tùy thuộc vào tôi để quyết định xem tôi có thực sự muốn truy cập các tệp / thư mục tương ứng do os.walk () mang lại hay chỉ cần bỏ qua chúng. Vấn đề là nếu tôi có, ví dụ, một cây thư mục như thế này:

root--
     |
     --- dirA
     |
     --- dirB
     |
     --- uselessStuff --
                       |
                       --- moreJunk
                       |
                       --- yetMoreJunk

và tôi muốn loại trừ uslessStuff và tất cả các con của nó, os.walk () vẫn sẽ truy cập vào tất cả các thư mục con (có khả năng hàng ngàn) của uslessStuff , không cần phải nói, làm chậm mọi thứ. Trong một thế giới lý tưởng, tôi có thể nói với os.walk () thậm chí không bận tâm đến việc sinh thêm bất kỳ đứa trẻ nào vô dụng , nhưng theo hiểu biết của tôi thì không có cách nào để làm điều đó (có phải không?).

Có ai có ý tưởng gì không? Có lẽ có một thư viện bên thứ ba cung cấp một cái gì đó như thế?

Sửa đổi dirs tại chỗ sẽ cắt tỉa các tệp và thư mục (tiếp theo) được truy cập bởi os.walk:

# exclude = set([...])
for root, dirs, files in os.walk(top, topdown=True):
    dirs[:] = [d for d in dirs if d not in exclude]

Từ trợ giúp (os.walk):

Khi topdown là true, người gọi có thể sửa đổi danh sách dirnames tại chỗ (ví dụ: thông qua phân bổ del hoặc lát) và walk sẽ chỉ lặp lại vào các thư mục con có tên vẫn còn trong dirnames; điều này có thể được sử dụng để cắt tỉa tìm kiếm ...

... một dạng khác của câu trả lời xuất sắc của @ unutbu đọc trực tiếp hơn một chút, với mục đích là loại trừ các thư mục, với chi phí là thời gian O (n ** 2) so với O (n).

(Tạo một bản sao của danh sách dirs list(dirs)là bắt buộc để thực hiện chính xác)

# exclude = set([...])
for root, dirs, files in os.walk(top, topdown=True):
    [dirs.remove(d) for d in list(dirs) if d in exclude]