util-遍历一个文件夹下的所有文件夹和文件

最新推荐文章于 2024-11-25 16:19:50 发布

原创最新推荐文章于 2024-11-25 16:19:50 发布 · 352 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

常用轮子专栏收录该内容

6 篇文章

订阅专栏

这段代码定义了两个函数，get_all_folders和get_all_files，分别用于获取指定基目录下的所有子文件夹路径和所有文件路径。在主程序中，它打印出这些路径。代码适用于文件管理系统或数据处理场景。

import os

def get_all_folders(base_dir):
    file_list = []
    folder_list = []
    for root,dir_names,file_names in os.walk(base_dir):

        for dir_name in dir_names:
            dir_path = os.path.join(root,dir_name)
            folder_list.append(dir_path)

        for file_name in file_names:
            file_path = os.path.join(root,file_name)
            file_list.append(file_path)
    
    return folder_list

def get_all_files(base_dir):
    file_list = []
    folder_list = []
    for root,dir_names,file_names in os.walk(base_dir):

        for dir_name in dir_names:
            dir_path = os.path.join(root,dir_name)
            folder_list.append(dir_path)

        for file_name in file_names:
            file_path = os.path.join(root,file_name)
            file_list.append(file_path)
    
    return file_list

if __name__ =="__main__":
    base_dir = "/home/wzg/data2/pypi-package-test/"

    file_list = get_all_files(base_dir)
    for file_path in file_list:
        print(file_path)

    folder_list = get_all_folders(base_dir)
    for folder_path in folder_list:
        print(folder_path)