python实现将a目录下所有某类型文件移动到b目录

简介

python实现将a目录下所有某类型文件移动到b目录

代码

import argparse
import os

def move_files(old_path,new_path,file_type):
    for file in os.listdir(old_path):
        if os.path.isdir(os.path.join(old_path,file))==False:
            if file[-len(file_type):] == file_type:
                os.rename(src=os.path.join(old_path,file), dst=os.path.join(new_path,file))

def mv_files_by_type(old_path,new_path,file_type,deal_child_folder):
    if os.path.exists(old_path):
        if os.path.exists(new_path):
            move_files(old_path,new_path,file_type)
            if deal_child_folder:
                for file in os.listdir(old_path):
                    if os.path.isdir(os.path.join(old_path, file)):
                        mv_files_by_type(os.path.join(old_path,file),new_path,file_type,deal_child_folder)
        else:
            print(new_path+' is not existed!')
    else:
        print(old_path + ' is not existed!')

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--old_path', type=str, default='/home/luban/jone/util/order_csv_0509.csv')
    parser.add_argument('--new_path', type=str, default='/home/luban/jone/util/order_csv_0509.csv')
    parser.add_argument('--file_type', type=str, default='.csv')
    parser.add_argument('--deal_child_folder', type=bool, default=True)
    args = parser.parse_args()

    mv_files_by_type(args.old_path,args.new_path,args.file_type,args.deal_child_folder)