util-文件移动

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

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

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #pytorch #开发语言

常用轮子专栏收录该内容

6 篇文章

订阅专栏

该代码实现了一个功能，从指定的`base_dir`目录下查找所有的whl和gz文件，将这些文件移动到`aim_dir`目标目录，并在移动后删除源目录下的空文件夹。主要使用了os和shutil模块来完成文件操作。

遍历一个文件夹下的所有whl和gz文件，移动到另外一个文件夹，并删除原文件夹下的空文件夹

import os
import shutil

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

def remove_empty_folder(folder_path):
    if len(os.listdir(folder_path)) == 0:
        os.rmdir(folder_path)

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

    # 获取所有文件
    file_list = get_all_files(base_dir)
    # 遍历文件，判断是否符合要求，
    for file_path in file_list:
        file_name = os.path.split(file_path)[-1]
        # print(file_name)
        file_type = file_name.split(".")[-1]
        # print(file_type)
        if file_type in ["whl","gz"]:
            aim_path = os.path.join(aim_dir,file_name)
            shutil.move(file_path,aim_path)
        #     print(file_path)

    # 遍历base_dir内的所有文件夹，删除空文件夹
    folder_list = get_all_folders(base_dir)
    for folder_path in folder_list:
        # print(folder_path)
        remove_empty_folder(folder_path)