#按照檔名設定標籤並將其插入檔名中
import os
import shutil
#獲取目標資料夾的路徑
a= r'C:\Users\yy\Desktop\tianchi大賽\guangdong_round1_train2_20180910\無瑕疵樣本'
#b=r'C:\Users\yy\Desktop\tianchi大賽\trian'
tag_dict={'不導電':'t00','凸粉':'t01','擦花':'t02','桔皮':'t03','橫條壓凹':'t04','塗層開裂':'t05','漏底':'t06',
          '碰傷':'t07','髒點':'t08','起坑':'t09','不導電負樣本':'f00','凸粉負樣本':'f01','擦花負樣本':'f02','桔皮負樣本':'f03',
          '橫條壓凹負樣本':'f04','塗層開裂負樣本':'f05','漏底負樣本':'f06','碰傷負樣本':'f07','髒點負樣本':'f08','起坑負樣本':'f09'}
#獲取當前資料夾中的檔名稱列表  
filenames=os.listdir(a)
for i in filenames:
    if i!='其他':
        print(i)
        imgdir=a+'\\'+i
        imgnames=os.listdir(imgdir)
        for imgname in imgnames:
            #print(imgdir+'\\'+imgname)
            
            imgdirname=imgdir+'\\'+imgname
            site=imgdirname.find('.')
            #imgname[0:len(i)]=tag_dict[i]
            os.rename(imgdirname,imgdirname[0:site]+tag_dict[i+'負樣本']+'.jpg')
            #print(imgdirname[0:site]+tag_dict[i]+'.jpg')
        
        #f=open(b+'\\'+filename,'a')
'''        filepath=a+'\\'+filename
        shutil.copyfile(filepath,b+str(i)+'\\'+filename+'.jpg')'''
        #遍歷單個檔案，讀取行數