【工具】列出指定文件夹下所有的目录和文件

帮我写一个python代码列出指定文件夹下所有的目录和文件 可选是否列出子文件夹及文件扩展名 也可以排除扩展名 并输出到文件

代码说明:1. 该函数通过递归遍历目录结构,列出所有文件和子目录。
2. 支持控制是否包含子目录、指定包含或排除特定扩展名的文件。
3. 可将结果输出到指定文件或直接打印到控制台。
4. 处理了权限不足的情况,避免程序崩溃。
5. 使用缩进表示目录层级结构,使输出更清晰易读。

import os

def list_files_and_dirs(root_dir, include_subdirs=True, include_extensions=None, exclude_extensions=None, output_file=None):
    """
    列出指定文件夹下的所有目录和文件
    :param root_dir: 根目录路径
    :param include_subdirs: 是否包含子目录
    :param include_extensions: 包含的文件扩展名列表,None表示所有
    :param exclude_extensions: 排除的文件扩展名列表,None表示不排除
    :param output_file: 输出文件路径,None表示输出到控制台
    """
    result = []
    
    def traverse(path, level=0):
        try:
            items = os.listdir(path)
        except PermissionError:
            result.append(f"{'  ' * level}! 权限不足: {path}")
            return
        
        for item in items:
            item_path = os.path.join(path, item)
            if os.path.isdir(item_path):
                result.append(f"{'  ' * level}📁 {item}/")
                if include_subdirs:
                    traverse(item_path, level + 1)
            else:
                ext = os.path.splitext(item)[1]
                if include_extensions is not None and ext not in include_extensions:
                    continue
                if exclude_extensions is not None and ext in exclude_extensions:
                    continue
                result.append(f"{'  ' * level}📄 {item}")
    
    traverse(root_dir)
    
    output_content = "\n".join(result)
    
    if output_file:
        with open(output_file, 'w', encoding='utf-8') as f:
            f.write(output_content)
        print(f"结果已输出到 {output_file}")
    else:
        print(output_content)

if __name__ == "__main__":
    # 示例调用
    list_files_and_dirs(
        root_dir=".",
        include_subdirs=True,
        include_extensions=None,
        exclude_extensions=['.tmp', '.log'],
        output_file="file_list.txt"
    )
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值