Python实现指定目录拷贝文件与移动文件

此代码拷贝过去可以直接能用。

"""
参考文章 https://blog.csdn.net/longshaonihaoa/article/details/105679517
"""


# -*- coding:utf-8 -*-
import os
import shutil
#from glob import glob
class File():
    
    def __init__(self):
        self.fileList = []

    """
    递归列表文件
    """
    def recursion_file(self, filepath):
        files = os.listdir(filepath)
        for file in files:
            fi_d = os.path.join(filepath, file)
            if os.path.isdir(fi_d):
                self.recursion_file(fi_d)
            else:
                self.fileList.append(fi_d)

    """
    获取文件列表
    """
    def get_file_list(self, filepath):
        self.recursion_file(filepath)
        return self.fileList


#指定文件路径获取文件最后文件的路径包含文件
#如：D:\test\file.txt 返回的结果为：file.txt
def get_file_name(path):
    return os.path.basename(path)
    
   
# 创建目录
def mkdir(path):
    # 去除尾部 \ 符号 
    pathx = path.strip().rstrip("\\")
    #print(f'pathx={pathx}')
    
    # 判断路径是否存在
    # 存在     True
    # 不存在   False
    isExists = os.path.exists(pathx)
    # 判断结果
    if not isExists:
        # 如果不存在则创建目录创建目录操作函数
        os.makedirs(path)
        print(path + ' 创建成功')
        return True
    else:
        # 如果目录存在则不创建，并提示目录已存在
        print(path + ' 目录已存在')
        return False

#移动文件,移动文件比拷贝文件速度快
#src_dir 指定要拷贝或者移动文件的源目录
#dst_dir 指定要拷贝或者移动文件的目标目录
#flag True为拷贝，False 代表移动
def copy_or_copy_files(src_dir,dst_dir,flag):

    fileObj = File()
    #指定新的路径不存在就创建
    mkdir(dst_dir)
    for file_path in fileObj.get_file_list(src_dir):
        if(flag==True):
            shutil.copy(file_path, dst_dir+'\\' + get_file_name(file_path))   
        else:
            shutil.move(file_path, dst_dir+'\\' + get_file_name(file_path))
           
   

# __main__方法
if __name__ == '__main__':
    src_dir = r'D:\test'
    dst_dir = r'D:\test\new'

    # 复制文件
    copy_or_copy_files(src_dir, dst_dir,False)                      
    
    print('拷贝或者移动文件完成....')