import os
a=1
count=0
lastfn=0
while(a<=38):
    dir = 'C:/Users/Administrator/Desktop/IMAGES_BIN/'+str(a)+'/'
    label = a
    files = os.listdir(dir)
    files.sort()
    train = open('D:/Users/Administrator/PycharmProjects/TwoStreamFingerVeinReco/CityUFV/Test1.txt','a')
    test = open('D:/Users/Administrator/PycharmProjects/TwoStreamFingerVeinReco/CityUFV/Train1.txt', 'a')
    for file in files:
        if a<28:
            fileType = os.path.split(file)
            if fileType[1] == '.txt':
                continue
            thisfn=file.split(".")[2];
            if(lastfn==None):
                count=count+0;
            elif(thisfn!=lastfn):
                count=count+1
            name =  str(dir) +  file + ' ' + str(int(count)) +'\n'
            train.write(name)
            lastfn=file.split(".")[2];
        else:
            fileType = os.path.split(file)
            if fileType[1] == '.txt':
                continue
            thisfn = file.split(".")[2];
            if (thisfn != lastfn):
                count = count + 1
            name = str(dir) +file + ' ' + str(int(count)) +'\n'
            test.write(name)
            lastfn = file.split(".")[2];
    train.close()
    print(a)
    a = a + 1