Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- def walklevel(some_dir, level=1):
- some_dir = some_dir.rstrip(os.path.sep)
- assert os.path.isdir(some_dir)
- num_sep = some_dir.count(os.path.sep)
- for root, dirs, files in os.walk(some_dir):
- yield root, dirs, files
- num_sep_this = root.count(os.path.sep)
- if num_sep + level <= num_sep_this:
- del dirs[:]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement