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

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)