WebSVN – SmartDukaan – Diff – /trunk/PyDTR/src/dtr/storage/Mongo.py

     except:
         pass
     return specialOffer
 def getBrandSubCategoryInfo(category_id, id_list, type):
+    try:
-    id_list_mapping = {}
+        id_list_mapping = {}
-    if type == "brandInfo":
+        if type == "brandInfo":
-        brands = [int(brand_id) for brand_id in id_list.split('^')]
+            brands = [int(brand_id) for brand_id in id_list.split('^')]
-        for brand in brands:
+            for brand in brands:
-            tm_brand = get_mongo_connection().Catalog.Deals.find_one({'brand_id':brand},{'brand':1})
+                tm_brand = get_mongo_connection().Catalog.Deals.find_one({'brand_id':brand},{'brand':1})
-            id_list_mapping[brand] = tm_brand.get('brand')
+                id_list_mapping[brand] = tm_brand.get('brand')
-    elif type == "subCategoryInfo":
+        elif type == "subCategoryInfo":
-        subCategories = [int(subCategoryId) for subCategoryId in id_list.split('^')]
+            subCategories = [int(subCategoryId) for subCategoryId in id_list.split('^')]
-        for subCat in subCategories:
+            for subCat in subCategories:
-            id_list_mapping[subCat] = SUB_CATEGORY_MAP.get(subCat)
+                id_list_mapping[subCat] = SUB_CATEGORY_MAP.get(subCat)
-    else:
+        else:
-        return {}
+            return {}
-    if mc.get("brandSubCategoryInfo") is None:
-        collection =  get_mongo_connection().Catalog.Deals
-        result = collection.aggregate(
-                [
-                    {'$match':{'category_id':category_id,'showDeal':1}},
-                    {"$group": { "_id": { "subCategoryId": "$subCategoryId", "brand_id": "$brand_id", "brand":"$brand","subCategory":"$subCategory" }, "count": {"$sum": 1} } }
-                ]
-            );
-        #from pprint import pprint
-        #pprint(result)
-        subCategoryMap = {}
-        brandMap = {}
-        for data in result['result']:
-            result_set = data['_id']
-            count = data['count']
-            subCategoryId = result_set['subCategoryId']
-            brand_id = result_set['brand_id']
+        if mc.get("brandSubCategoryInfo") is None:
-            subCategory = result_set['subCategory']
+            collection =  get_mongo_connection().Catalog.Deals
-            brand = result_set['brand']
+            result = collection.aggregate(
+                    [
-            if subCategoryMap.has_key(subCategoryId):
+                        {'$match':{'category_id':category_id,'showDeal':1}},
-                subCategoryMap.get(subCategoryId).append({'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand})
+                        {"$group": { "_id": { "subCategoryId": "$subCategoryId", "brand_id": "$brand_id", "brand":"$brand","subCategory":"$subCategory" }, "count": {"$sum": 1} } }
-            else:
+                    ]
-                subCategoryMap[subCategoryId] = [{'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand}]
+                );
-            if brandMap.has_key(brand_id):
-                brandMap.get(brand_id).append({'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand})
-            else:
-                brandMap[brand_id] = [{'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand}]
-        mc.set("brandSubCategoryInfo",{'subCategoryMap':subCategoryMap, 'brandMap': brandMap}, 600)
-    returnMap = {}
-    brandSubCategoryInfo = mc.get("brandSubCategoryInfo")
-    if brandSubCategoryInfo is None:
-        return {'idListInfo':id_list_mapping,'result':[]}
-    if type == "subCategoryInfo":
-        if brandSubCategoryInfo.get('subCategoryMap') is None:
-            return {'idListInfo':id_list_mapping,'result':[]}
-        subCategories = [int(subCategoryId) for subCategoryId in id_list.split('^')]
-        subCategoryMap = brandSubCategoryInfo.get('subCategoryMap')
-        for subCategory in subCategories:
-            tempList = subCategoryMap.get(subCategory)
-            print tempList
+            #from pprint import pprint
-            if tempList is None:
-                continue
-            for v in tempList:
+            #pprint(result)
-                if returnMap.has_key(v['brand_id']):
-                    existing = returnMap.get(v['brand_id'])
-                    existing['count'] = existing['count'] + v['count']
-                else:
-                    returnMap[v['brand_id']] = {'brand_id':v['brand_id'],'brand':v['brand'],'count':v['count']}
+            subCategoryMap = {}
+            brandMap = {}
-    elif type == "brandInfo":
+            for data in result['result']:
-        if brandSubCategoryInfo.get('brandMap') is None:
+                result_set = data['_id']
-            return {'idListInfo':id_list_mapping,'result':[]}
+                count = data['count']
-        brands = [int(brand_id) for brand_id in id_list.split('^')]
-        brandMap = brandSubCategoryInfo.get('brandMap')
+                subCategoryId = result_set['subCategoryId']
-        for brand_id in brands:
+                brand_id = result_set['brand_id']
-            tempList = brandMap.get(brand_id)
+                subCategory = result_set['subCategory']
-            print tempList
-            if tempList is None:
+                brand = result_set['brand']
-                continue
-            for v in tempList:
-                if returnMap.has_key(v['subCategoryId']):
+                if subCategoryMap.has_key(subCategoryId):
-                    existing = returnMap.get(v['subCategoryId'])
+                    subCategoryMap.get(subCategoryId).append({'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand})
-                    existing['count'] = existing['count'] + v['count']
                 else:
+                    subCategoryMap[subCategoryId] = [{'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand}]
+                if brandMap.has_key(brand_id):
+                    brandMap.get(brand_id).append({'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand})
+                else:
+                    brandMap[brand_id] = [{'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand}]
+            mc.set("brandSubCategoryInfo",{'subCategoryMap':subCategoryMap, 'brandMap': brandMap}, 600)
+        returnMap = {}
+        brandSubCategoryInfo = mc.get("brandSubCategoryInfo")
+        if brandSubCategoryInfo is None:
+            return {'idListInfo':id_list_mapping,'result':[]}
+        if type == "subCategoryInfo":
+            if brandSubCategoryInfo.get('subCategoryMap') is None:
+                return {'idListInfo':id_list_mapping,'result':[]}
+            subCategories = [int(subCategoryId) for subCategoryId in id_list.split('^')]
+            subCategoryMap = brandSubCategoryInfo.get('subCategoryMap')
+            for subCategory in subCategories:
+                tempList = subCategoryMap.get(subCategory)
+                print tempList
+                if tempList is None:
+                    continue
+                for v in tempList:
+                    if returnMap.has_key(v['brand_id']):
+                        existing = returnMap.get(v['brand_id'])
+                        existing['count'] = existing['count'] + v['count']
+                    else:
+                        returnMap[v['brand_id']] = {'brand_id':v['brand_id'],'brand':v['brand'],'count':v['count']}
+        elif type == "brandInfo":
+            if brandSubCategoryInfo.get('brandMap') is None:
+                return {'idListInfo':id_list_mapping,'result':[]}
+            brands = [int(brand_id) for brand_id in id_list.split('^')]
+            brandMap = brandSubCategoryInfo.get('brandMap')
+            for brand_id in brands:
+                tempList = brandMap.get(brand_id)
+                print tempList
+                if tempList is None:
+                    continue
+                for v in tempList:
+                    if returnMap.has_key(v['subCategoryId']):
+                        existing = returnMap.get(v['subCategoryId'])
+                        existing['count'] = existing['count'] + v['count']
+                    else:
-                    returnMap[v['subCategoryId']] = {'subCategoryId':v['subCategoryId'],'subCategory':v['subCategory'],'count':v['count']}
+                        returnMap[v['subCategoryId']] = {'subCategoryId':v['subCategoryId'],'subCategory':v['subCategory'],'count':v['count']}
-    return {'idListInfo':id_list_mapping,'result':returnMap.values()}
+        return {'idListInfo':id_list_mapping,'result':returnMap.values()}
+    except:
+        traceback.print_exc()
+        return {}
 def main():
     #getDealsForSearchText("20", 'Samsung', 0, 10)

Subversion Repositories SmartDukaan

(root)/trunk/PyDTR/src/dtr/storage/Mongo.py – Rev 19401 → 19402