| 15543 |
kshitij.so |
1 |
from dtr.utils.utils import get_mongo_connection
|
|
|
2 |
import xlrd
|
|
|
3 |
|
|
|
4 |
synMap = {}
|
|
|
5 |
|
|
|
6 |
class __Synonym:
|
|
|
7 |
|
|
|
8 |
def __init__(self, brand, synonyms):
|
|
|
9 |
self.brand = brand
|
|
|
10 |
self.synonyms = synonyms
|
|
|
11 |
|
|
|
12 |
class __ModelNameSynonyms:
|
|
|
13 |
|
|
|
14 |
def __init__(self, skuBundleId, synonyms):
|
|
|
15 |
self.skuBundleId = skuBundleId
|
|
|
16 |
self.synonyms = synonyms
|
|
|
17 |
|
|
|
18 |
|
|
|
19 |
#def read_data(file):
|
|
|
20 |
# workbook = xlrd.open_workbook(file)
|
|
|
21 |
# sheet = workbook.sheet_by_index(0)
|
|
|
22 |
# num_rows = sheet.nrows
|
|
|
23 |
# count =1
|
|
|
24 |
# for rownum in range(1, num_rows):
|
|
|
25 |
# print sheet.row_values(rownum)
|
|
|
26 |
# syn = __Synonym(sheet.row_values(rownum)[0],None)
|
|
|
27 |
# temp = []
|
|
|
28 |
# for data in sheet.row_values(rownum)[1:]:
|
|
|
29 |
# if len(data) ==0:
|
|
|
30 |
# continue
|
|
|
31 |
# temp.append(data)
|
|
|
32 |
# syn.synonyms = temp
|
|
|
33 |
# if len(temp) != 0:
|
|
|
34 |
# get_mongo_connection().Catalog.BrandSynonyms.insert(syn.__dict__)
|
|
|
35 |
|
|
|
36 |
def read_data(file):
|
|
|
37 |
workbook = xlrd.open_workbook(file)
|
|
|
38 |
sheet = workbook.sheet_by_index(0)
|
|
|
39 |
num_rows = sheet.nrows
|
|
|
40 |
count =1
|
|
|
41 |
for rownum in range(1, num_rows):
|
|
|
42 |
if synMap.has_key(sheet.row_values(rownum)[1]):
|
|
|
43 |
synonyms = synMap.get(sheet.row_values(rownum)[1])
|
|
|
44 |
for tempSyn in sheet.row_values(rownum)[3:]:
|
|
|
45 |
if tempSyn not in synonyms:
|
|
|
46 |
synonyms.append(tempSyn)
|
|
|
47 |
else:
|
|
|
48 |
synMap[sheet.row_values(rownum)[1]] = sheet.row_values(rownum)[3:]
|
|
|
49 |
|
|
|
50 |
|
|
|
51 |
|
|
|
52 |
# syn = __ModelNameSynonyms(sheet.row_values(rownum)[1],None)
|
|
|
53 |
# temp = []
|
|
|
54 |
# for data in sheet.row_values(rownum)[3:]:
|
|
|
55 |
# if len(str(data)) ==0:
|
|
|
56 |
# continue
|
|
|
57 |
# temp.append(str(data))
|
|
|
58 |
# syn.synonyms = temp
|
|
|
59 |
# if len(temp) != 0:
|
|
|
60 |
# get_mongo_connection().Catalog.ModelNameSynonyms.insert(syn.__dict__)
|
|
|
61 |
for x, y in synMap.iteritems():
|
|
|
62 |
syn = __ModelNameSynonyms(x,None)
|
|
|
63 |
temp = []
|
|
|
64 |
for z in y:
|
|
|
65 |
if len(str(z)) ==0:
|
|
|
66 |
continue
|
|
|
67 |
temp.append(z)
|
|
|
68 |
syn.synonyms = temp
|
|
|
69 |
if len(temp) != 0:
|
|
|
70 |
get_mongo_connection().Catalog.ModelNameSynonyms.insert(syn.__dict__)
|
|
|
71 |
|
|
|
72 |
|
|
|
73 |
|
|
|
74 |
|
|
|
75 |
|
|
|
76 |
def main():
|
|
|
77 |
#read_data("/home/kshitij/Desktop/brand-synonym.xls")
|
|
|
78 |
read_data("/home/kshitij/Desktop/Spice.xls")
|
|
|
79 |
|
|
|
80 |
if __name__=='__main__':
|
|
|
81 |
main()
|