Subversion Repositories SmartDukaan

from dtr.utils.utils import get_mongo_connection
import xlrd

synMap = {}

class __Synonym:
    
    def __init__(self, brand, synonyms):
        self.brand = brand
        self.synonyms = synonyms
        
class __ModelNameSynonyms:
    
    def __init__(self, skuBundleId, synonyms):
        self.skuBundleId = skuBundleId
        self.synonyms = synonyms
    

#def read_data(file):
#    workbook = xlrd.open_workbook(file)
#    sheet = workbook.sheet_by_index(0)
#    num_rows = sheet.nrows
#    count =1
#    for rownum in range(1, num_rows):
#        print sheet.row_values(rownum)
#        syn = __Synonym(sheet.row_values(rownum)[0],None)
#        temp = []
#        for data in sheet.row_values(rownum)[1:]:
#            if len(data) ==0:
#                continue
#            temp.append(data)
#        syn.synonyms = temp
#        if len(temp) != 0:
#            get_mongo_connection().Catalog.BrandSynonyms.insert(syn.__dict__)

def read_data(file):
    workbook = xlrd.open_workbook(file)
    sheet = workbook.sheet_by_index(0)
    num_rows = sheet.nrows
    count =1
    for rownum in range(1, num_rows):
        if synMap.has_key(sheet.row_values(rownum)[1]):
            synonyms = synMap.get(sheet.row_values(rownum)[1])
            for tempSyn in sheet.row_values(rownum)[3:]:
                if tempSyn not in synonyms:
                    synonyms.append(tempSyn)
        else:
            synMap[sheet.row_values(rownum)[1]] = sheet.row_values(rownum)[3:]
         
        
        
#        syn = __ModelNameSynonyms(sheet.row_values(rownum)[1],None)
#        temp = []
#        for data in sheet.row_values(rownum)[3:]:
#            if len(str(data)) ==0:
#                continue
#            temp.append(str(data))
#        syn.synonyms = temp
#        if len(temp) != 0:
#            get_mongo_connection().Catalog.ModelNameSynonyms.insert(syn.__dict__)
    for x, y in synMap.iteritems():
        syn = __ModelNameSynonyms(x,None)
        temp = []
        for z in y:
            if len(str(z)) ==0:
                continue
            temp.append(z)
        syn.synonyms = temp
        if len(temp) != 0:
            get_mongo_connection().Catalog.ModelNameSynonyms.insert(syn.__dict__)
        


        

def main():
    #read_data("/home/kshitij/Desktop/brand-synonym.xls")
    read_data("/home/kshitij/Desktop/Spice.xls")

if __name__=='__main__':
    main()