import os
import sys
import xlrd
import re

xlsarg = sys.argv[1]
catdir = sys.argv[2]

def getbook(xlsin):
    bk = xlrd.open_workbook(xlsin)
    sh = bk.sheet_by_index(0)
    return sh

def getretdesigns(sheet):
    designs = []
    for rx in range(sheet.nrows):
        designs.append(sheet.row_values(rx)[3])
    return designs

def fmtretdesigns(designlist):
    fmt = re.compile('([a-zA-Z]{1,3})(-?[0-9]{1,4})')
    fmtdlist = []
    for rx in designlist[2:]:
        fmtdlist.append(fmt.sub(r'\1-\2' + '.jpg', rx).strip())
    return fmtdlist

def getalldesigns(dir):
    path = dir
    dlist = []
    dirs = os.listdir(dir)
    for x in dirs:
        y = os.listdir(path + '/' + x)
        for z in y:
            dlist.append(z)
    return dlist

def gendellist(alldesigns, retiredlist):
    result = []
    for design in alldesigns:
        if design not in retiredlist:
            result.append(design)
    return result

def delnonret(nonret, dir):
    path = dir
    categories = os.listdir(dir)
    for category in categories:
        designs = os.listdir(path + '/' + category)
        for design in designs:
            if design in nonret:
                os.remove(path + '/' + category + '/' + design)

def executeall():
    ws = getbook(xlsarg)
    designs = getretdesigns(ws)
    fmtlist = fmtretdesigns(designs)
    dlist = getalldesigns(catdir)
    nonret = gendellist(dlist, fmtlist)
    delnonret(nonret, catdir)


executeall()