python os模块 复制和删除目录及目录下的文件

用python内置模块os模块对目录及其内部的文件及目录进行复制和删除操作。

本文用到的os模块内置函数如下：

os.mkdir(path)                    # 创建path指定的目录，该参数不能省略
os.rmdir(path)                    # 删除path指定的目录，该参数不能省略
os.listdir(path)                  # 列出path目录下所有的文件和目录名 
os.remove(path)                   # 删除path指定的文件，该参数不能省略
os.path.isfile(path)              # 判断路径是否为文件
os.path.isdir(path)               # 判断路径是否为目录
os.path.join(path1[, path2[, ]])  # 把目录和文件名合成一个路径

代码部分：

import os


def copy(path_1, path_2):
    """
    将路径path_1下的所有目录及文件复制到路径path_2下
    path_1: 待复制的目录或者文件路径
    path_2: 目标路径
    """
    if os.path.isdir(path_1):  # path_1是目录

        list_1 = os.listdir(path_1)
        if not list_1:  # 复制目录，仅仅是复制空目录
            os.mkdir(path_2)
        else:
            os.mkdir(path_2)  # 先复制最上层空目录
            for i in list_1:
                path_r = os.path.join(path_1, i)  # 下层目录或文件的绝对路径
                path_w = os.path.join(path_2, i)  # 目标目录或文件的绝对路径
                if os.path.isfile(path_r):  # 是文件则直接进行复制
                    with open(path_r, 'rb') as rstream:
                        container = rstream.read()
                        with open(path_w, 'wb') as wstream:
                            wstream.write(container)
                else:  # 是目录则调用本函数
                    copy(path_r, path_w)

    else:  # path_1是文件
        with open(path_1, 'rb') as rstream:  # 是文件直接复制文件
            container = rstream.read()
            file_name = os.path.basename(path_1)
            path_2 = os.path.join(path_2, file_name)
            with open(path_2, 'wb') as wstream:
                wstream.write(container)


def delete(path_3):
    """
    删除路径path_3下的所有目录及文件
    path: 待删除目录或文件的路径
    """
    if os.path.isdir(path_3):  # path_3是目录
        list_1 = os.listdir(path_3)
        if not list_1:
            os.rmdir(path_3)  # 空目录直接删除
        else:  # 非空目录
            for index, i in enumerate(list_1):
                # 获得目录下所有目录及文件的绝对路径
                list_1[index] = os.path.join(path_3, i)

            for i in list_1:
                if os.path.isfile(i):  # 是文件直接删除
                    os.remove(i)
                else:  # 是目录循环继续执行本函数
                    delete(i)
            os.rmdir(path_3)
    else:  # path_3是文件
        os.remove(path_3)


# 复制目录
path1 = r'C:\Users\XJQ\Music\aa'
path2 = r'C:\Users\XJQ\Music\bb'
copy(path1, path2)

# 复制文件
path3 = r'C:\Users\XJQ\Music\001.txt'
path4 = r'C:\Users\XJQ\Music\cc'
copy(path3, path4)

# 删除目录
delete(path1)

# 删除文件
delete(path3)