"""
Created on Thu May  7 18:00:39 2015

@author: frac
"""
import os
import shutil

def create_dict():
    f=open('species_list.txt','r')
    lines=f.readlines()
    dictio={}
    for key in lines:
        key=key.split('\n')[0]
        key=key.split('\t')
        dictio[str(key[1])]=key[0]
    f.close()
    return dictio
    
def isolerlabel(meta):
    metado=meta.readlines()
    metado=' '.join(metado)
    metado=metado.split('<ClassId>')[1]
    metado=metado.split('</ClassId>')[0]
    return metado
    
def cut_name(fname):
    return fname.split('_CMT.jpg')[0]
    
dictio=create_dict()
count=0
ls_file=os.listdir('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMT_TOUTFILESAPLAT')
for fname in ls_file:
    fnamebis=cut_name(fname)
    print fname
    try:
        meta=open('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/LIFECLEF2015BIRDCHALLENGE_TRAIN_TEST_SETS_XML_METADATA/xml/train/'+fnamebis+'.xml','r')
        metad=isolerlabel(meta)
        meta.close()
        num=dictio[metad]
        count=count+1
        if not os.path.exists('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMTfinalspectro_CHECKED/TRAIN'+str(num)):
            os.makedirs('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMTfinalspectro_CHECKED/TRAIN'+str(num))
        shutil.move('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMT_TOUTFILESAPLAT/'+fname, '/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMTfinalspectro_CHECKED/TRAIN'+str(num)+'/'+fname)
        print 'file moved : ',count
    except:
        a=0
 
       
#dictio=create_dict()

ls_file=os.listdir('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMT_TOUTFILESAPLAT')
for fname in ls_file:
    fnamebis=cut_name(fname)
    try:
        meta=open('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/LIFECLEF2015BIRDCHALLENGE_TRAIN_TEST_SETS_XML_METADATA/xml/test/'+fnamebis+'.xml','r')
        meta.close()
        count=count+1
        shutil.move('/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMT_TOUTFILESAPLAT/'+fname, '/NAS3/SABIOD/public_data/DIENS_DYNI/NEW_CHALLENGE_2015/CMTfinalspectro_CHECKED/TEST/'+fname)
        print 'filetest moved : ',count
    except:
        a=0