python 复制文件或文件夹

import os
import shutil

def copy_file(src_file, dst_file):
    # 如果是文件，进行复制
    if os.path.isfile(src_file):
        _dir = os.path.dirname(dst_file)
        os.makedirs(_dir, exist_ok=True)
        shutil.copy2(src_file, dst_file)


def copy_dir(source_dir, target_dir):
    # 确保目标目录存在
    os.makedirs(target_dir, exist_ok=True)

    # 遍历源目录中的所有文件和目录
    for item in os.listdir(source_dir):
        src_path = os.path.join(source_dir, item)
        dst_path = os.path.join(target_dir, item)

        # 如果是文件，进行复制
        if os.path.isfile(src_path):
            shutil.copy2(src_path, dst_path)
        # 如果是目录，进行目录复制
        elif os.path.isdir(src_path):
            shutil.copytree(src_path, dst_path, dirs_exist_ok=True)  # dirs_exist_ok 在 Python 3.8 及以上可用