Using os.walk() to recursively traverse directories in Python

Using os.walk() to recursively traverse directories in Python

This will give you the desired result

#!/usr/bin/python

import os

# traverse root directory, and list directories as dirs and files as files
for root, dirs, files in os.walk(.):
    path = root.split(os.sep)
    print((len(path) - 1) * ---, os.path.basename(root))
    for file in files:
        print(len(path) * ---, file)

try this:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

FileTreeMaker.py: ...

__author__  = legendmohe

import os
import argparse
import time

class FileTreeMaker(object):

    def _recurse(self, parent_path, file_list, prefix, output_buf, level):
        if len(file_list) == 0 
            or (self.max_level != -1 and self.max_level <= level):
            return
        else:
            file_list.sort(key=lambda f: os.path.isfile(os.path.join(parent_path, f)))
            for idx, sub_path in enumerate(file_list):
                if any(exclude_name in sub_path for exclude_name in self.exn):
                    continue

                full_path = os.path.join(parent_path, sub_path)
                idc = ┣━
                if idx == len(file_list) - 1:
                    idc = ┗━

                if os.path.isdir(full_path) and sub_path not in self.exf:
                    output_buf.append(%s%s[%s] % (prefix, idc, sub_path))
                    if len(file_list) > 1 and idx != len(file_list) - 1:
                        tmp_prefix = prefix + ┃  
                    else:
                        tmp_prefix = prefix +     
                    self._recurse(full_path, os.listdir(full_path), tmp_prefix, output_buf, level + 1)
                elif os.path.isfile(full_path):
                    output_buf.append(%s%s%s % (prefix, idc, sub_path))

    def make(self, args):
        self.root = args.root
        self.exf = args.exclude_folder
        self.exn = args.exclude_name
        self.max_level = args.max_level

        print(root:%s % self.root)

        buf = []
        path_parts = self.root.rsplit(os.path.sep, 1)
        buf.append([%s] % (path_parts[-1],))
        self._recurse(self.root, os.listdir(self.root), , buf, 0)

        output_str = n.join(buf)
        if len(args.output) != 0:
            with open(args.output, w) as of:
                of.write(output_str)
        return output_str

if __name__ == __main__:
    parser = argparse.ArgumentParser()
    parser.add_argument(-r, --root, help=root of file tree, default=.)
    parser.add_argument(-o, --output, help=output file name, default=)
    parser.add_argument(-xf, --exclude_folder, nargs=*, help=exclude folder, default=[])
    parser.add_argument(-xn, --exclude_name, nargs=*, help=exclude name, default=[])
    parser.add_argument(-m, --max_level, help=max level,
                        type=int, default=-1)
    args = parser.parse_args()
    print(FileTreeMaker().make(args))

you will get this:

root:.
[.]
┣━[.idea]
┃  ┣━[scopes]
┃  ┃  ┗━scope_settings.xml
┃  ┣━.name
┃  ┣━Demo.iml
┃  ┣━encodings.xml
┃  ┣━misc.xml
┃  ┣━modules.xml
┃  ┣━vcs.xml
┃  ┗━workspace.xml
┣━[test1]
┃  ┗━test1.txt
┣━[test2]
┃  ┣━[test2-2]
┃  ┃  ┗━[test2-3]
┃  ┃      ┣━test2
┃  ┃      ┗━test2-3-1
┃  ┗━test2
┣━folder_tree_maker.py
┗━tree.py

Using os.walk() to recursively traverse directories in Python

Recursive walk through a directory where you get ALL files from all dirs in the current directory and you get ALL dirs from the current directory – because codes above dont have a simplicity (imho):

for root, dirs, files in os.walk(rootFolderPath):
    for filename in files:
        doSomethingWithFile(os.path.join(root, filename))
    for dirname in dirs:
        doSomewthingWithDir(os.path.join(root, dirname))