python迭代文件目录

import os

# 第一种方式
def recu_list_dirs_by_dictionary_order(path, indent = 0, maxi = -1):
    '''
        按字典序递归输出目录结构
        :param path:   str 文件路径
        :param indent: int 首次缩进空格(默认为 0，一般不用改变)
        :param maxi:   int 最大展开层数(默认为 -1，表示全部展开)
    '''
    if maxi != 0:
        try:
            lsdir = os.listdir(path)
        except PermissionError: # 对于权限不够的文件不作处理
            pass
        else:
            for item in lsdir:
                full_path = os.path.join(path, item)
                if os.path.isdir(full_path):
                    print(' ' * indent, '+', item)
                    recu_list_dirs_by_dictionary_order(full_path, indent + 4, maxi - 1)
                if os.path.isfile(full_path):
                    print(' ' * indent, '-', item)
# recu_list_dirs_by_dictionary_order('/home/ly/Desktop/vedio_to_img')

# 第二种方式
def recu_list_dirs_by_file_type(path, indent = 0, maxi = -1):
    '''
        按文件类型递归输出目录结构
        :param path:   str 文件路径
        :param indent: int 首次缩进空格(默认为 0，一般不用改变)
        :param maxi:   int 最大展开层数(默认为 -1，表示全部展开)
    '''
    if maxi != 0:
        try:
            lsdir = os.listdir(path)
        except PermissionError: # 对于权限不够的文件不作处理
            pass
        else:
            dirs = [item for item in lsdir if os.path.isdir(os.path.join(path, item))]
            files = [item for item in lsdir if os.path.isfile(os.path.join(path, item))]
            for item in dirs:
                print(' ' * indent, '+', item)
                recu_list_dirs_by_file_type(os.path.join(path, item), indent + 4, maxi - 1)
            for item in files:
                print(' ' * indent, '-', item)

if __name__ == '__main__':
    file_path = input('请输入路径：')
    print('\n---按字典序递归输出目录结构---')
    recu_list_dirs_by_dictionary_order(file_path, 0, 2)
    print('\n---按文件类型递归输出目录结构---')
    recu_list_dirs_by_file_type(file_path, 0, 2)