import os

import numpy as np

import re

from shutil import copyfile

def getlist(path,name):

    listall = []

    for n in open(path):

        location = re.search(name, n) #定位关键字

        bingli_Id = n[location.start():location.start()+7] #找到关键字后面7位长度字符

        listall.append(bingli_Id) #添加到列表中

    return set(listall) #去除重复内容

def mkdir(file1, bingId): #建立文件夹

    if not os.path.exists(file1):

        os.mkdir(file1) #建立一级目录

    for n in bingId:

        path = os.path.join(file1, n)

        path_1 = os.path.join(path, str(1)) #组装三级目录，此时我知道三级目录的内容，所以直接设置了常量

        path_7 = os.path.join(path, str(7))

        if not os.path.exists(path): #建立目录

            os.mkdir(path)

        if not os.path.exists(path_1):

            os.mkdir(path_1)

        if not os.path.exists(path_7):

            os.mkdir(path_7)

def copytoflods(path_all_list, name, file1):

    for n in open(path_all_list):

        location = re.search(name, n)

        bingId = n[location.start():location.start()+7]

        arr = n.split('_') #以‘_’分割字符串

        type_ = arr[2] # 取第3个元素

        root = os.path.join(file1, bingId, type_, arr[0][-20:]+ '.png') #组装文件路径

        copyfile(n[:-1], root) #n[:-1]：便捷地去除换行符号

if __name__ == '__main__':

    listall = getlist('./list.txt','181') #第一个参数为需要处理的文件路径列表， 181为关键字

    mkdir('dataset', listall)

    copytoflods('./list.txt','181', 'dataset') #dataset为一级目录名