import xlrd, os, shutil

#set name of excel sheet
loc="ai.xlsx"


# input folder is /some/folder/
#outfolder is ./some/folder
def picformatter(folder, outfolder, sheet, startcell, endcell):
    wb=xlrd.open_workbook(loc)
    sheet = wb.sheet_by_index(int(sheet))

    #create directories for categories if they don't exist.
    for i in range(int(startcell)-1, int(endcell)):
        if not os.path.exists(str(outfolder+'/'+sheet.cell_value(i,2))):
            os.makedirs(str(outfolder+'/'+sheet.cell_value(i,2)))
# format as type_size_quality_brand, as well as move images to folders for the type
    for i in range(int(startcell)-1, int(endcell)):
        shutil.copyfile(str(folder+sheet.cell_value(i,6)), str(outfolder+'/'+ sheet.cell_value(i,2) +'/'
            + sheet.cell_value(i,2)+'_' #type
            + sheet.cell_value(i,3)+'_' #size
            + sheet.cell_value(i,5)+'_' #quality
            + sheet.cell_value(i,4)+'.JPG' #brand + file extension
            ))

# run our picformatter function on each bin folder we have.
picformatter("./AI Project/Bin Photos/Bin #1 Pits Defect/","./sorted/Bin #1 Pits Defect", 0,3,49) #1
picformatter("./AI Project/Bin Photos/Bin #2 Stains/","./sorted/Bin #2 Stains", 1,3,66) #2
picformatter("./AI Project/Bin Photos/Bin #3 Collars/","./sorted/Bin #3 Collars", 2,3,50) #3
picformatter("./AI Project/Bin Photos/Bin #4 Pockets/","./sorted/Bin #4 Pockets", 3,3,7) #4
picformatter("./AI Project/Bin Photos/Bin #5 Ripped & Torn/","./sorted/Bin #5 Ripped & Torn", 4,3,29) #5
picformatter("./AI Project/Bin Photos/Bin #6 Cuffs & Hems/","./sorted/Bin #6 Cuffs & Hems", 5,3,43) #6
picformatter("./AI Project/Bin Photos/Bin #7 Pilling/","./sorted/Bin #7 Pilling", 6,3,150) #7
picformatter("./AI Project/Bin Photos/Bin #8 Luxury brands/","./sorted/Bin #8 Luxury brands", 7,3,34) #8