import os
import shutil


# 复制一级目录下的所有文件到指定目录
def coverFiles(sourceDir, targetDir):
    for file in os.listdir(sourceDir):
        sourceFile = os.path.join(sourceDir, file)
        targetFile = os.path.join(targetDir, file)
        if os.path.isfile(sourceFile):
            open(targetFile, "wb").write(open(sourceFile, "rb").read())


# 删除目录下所有文件
def removeFileInFirstDir(targetDir):
    for file in os.listdir(targetDir):
        targetFile = os.path.join(targetDir, file)
        if os.path.isfile(targetFile):
            os.remove(targetFile)


# 目录下的所有文件复制到指定目录中
def copyFiles(sourceDir, targetDir):
    for file in os.listdir(sourceDir):
        # 判断拷贝目录是否存在，不存在则创建
        if not os.path.exists(targetDir):
            os.makedirs(targetDir)

        # 拼接路径
        sourceFile = os.path.join(sourceDir, file)
        targetFile = os.path.join(targetDir, file)

        if os.path.isfile(sourceFile):
            open(targetFile, "wb").write(open(sourceFile, "rb").read())

        if os.path.isdir(sourceFile):
            copyFiles(sourceFile, targetFile)


def copyFiles2(sourceDir, targetDir):
    shutil.copytree(sourceDir, targetDir)


if __name__ == '__main__':
    # coverFiles(r"C:\Users\Administrator\Desktop\my\w1\day1", r"C:\Users\Administrator\Desktop\my\新建文件夹\麻瓜")
    # removeFileInFirstDir(r"C:\Users\Administrator\Desktop\my\新建文件夹\麻瓜")
    copyFiles2(r"C:\Users\Administrator\Desktop\my\szpython1801", r"C:\Users\Administrator\Desktop\my\新建文件夹\麻瓜")