python测试

import os
import shutil

def check_dir_exist(dir) :
    # 如果目标路径不存在原文件夹的话就创建
    if not os.path.exists(dir):
        os.makedirs(dir)
        
def remove_dir(dir) :
     # 如果目标路径存在原文件夹的话就先删除
    if os.path.exists(dir):
        shutil.rmtree(dir)

def copy_tree(src, dst) :
    remove_dir(dst)
    shutil.copytree(src, dst)
    
def copy_file_recursion(source_path, target_path) :
    check_dir_exist(target_path)
    if os.path.exists(source_path):
        # root 所指的是当前正在遍历的这个文件夹的本身的地址
        # dirs 是一个 list，内容是该文件夹中所有的目录的名字(不包括子目录)
        # files 同样是 list, 内容是该文件夹中所有的文件(不包括子目录)
        for root, dirs, files in os.walk(source_path):
            for file in files:
                src_file = os.path.join(root, file)
                shutil.copy(src_file, target_path)


def main():
    MyPrintf("1111")
    copy_tree("./src", "./dst")
    copy_file_recursion("./src", "./dst1")

def MyPrintf(param):
    print(param)

if __name__ == '__main__':
    main()