import os
import shutil
#--------------------------------------------------------

filelist = os.listdir(os.getcwd() + "/" + "new") # 获取当前文件夹中的所有文件名称(包括文件夹)

flist2 = []
i0 = 0
for filename in filelist:
    i0 += 1
    f1 = filename  + "/" + filename
    f2 = os.getcwd() + "/" + "new" + "/" +  f1 # 文件夹下面还有名称相同的文件夹，这里进行拼接，组合成正确的文件路径，方便后面获取其中的文件
    flist2.append(f2)
print("原始文件夹数量为 %d"%i0)

os.chdir(os.getcwd())# 确定用户执行路径
tarfolename = "AFold" # 新建一个文件夹来存储这些文件，消除掉原本有两层的结构
folder = os.path.exists(tarfolename)
if not folder:
    os.mkdir(tarfolename) # 在当前路径下新建一个文件夹，用来存放新的文件
# else:
    # print("文件夹已经存在")
f11 = os.getcwd() + "/" + tarfolename # 文件复制的目标路径

i0 = 0
for f1 in flist2: #开始遍历每个文件夹，这里已经考虑了文件夹下面还有文件夹
    f2 = os.listdir(f1 + "/") # 获取文件夹中的所有文件
    i0 += 1
    for f3 in f2: # 遍历文件夹中的文件
        src = os.path.join(f1,f3) # 文件路径

        folder = os.path.exists(f11 + "/" + os.path.basename(f1)) # 先判断文件夹是否存在

        if not folder: # 如果不存在就新建这个文件夹，然后才能copy文件进来
            os.mkdir(f11 + "/" + os.path.basename(f1)) # 在当前路径下新建一个文件夹，用来存放新的文件
        # else:
            # print("文件夹已经存在")

        tar = os.path.join(f11 + "/" + os.path.basename(f1),f3) # 需要移动到哪个位置
        shutil.copyfile(src,tar)
print("复制文件夹数量为 %d"%i0)