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

 def __constructDummyDealObject(skuBundleId):
     temp =[]
     deals = get_mongo_connection().Catalog.Deals.find({"skuBundleId":skuBundleId,"$or":[{"showDeal":1}, {"prepaidDeal":1}]},{'_id':1,'category_id':1,'brand':1,'totalPoints':1,'bestSellerPoints':1,'nlcPoints':1,'rank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1,'subCategoryId':1,'netPriceAfterCashBack':1})
     for d in deals:
         d['persPoints'] = 0
-        d['internalRank'] = 999999
+        d['dealRankPoints'] = 0
         temp.append(d)
     return temp
         else:
             temp = []
             temp_map[x.category_id] = 1 if x.status == 'show' else 0
             temp.append(temp_map)
             brandPrefMap[(x.brand).strip().upper()] = temp
     for x in session.query(user_actions).filter_by(user_id=userId).all():
         actionsMap[x.store_product_id] = 1 if x.action == 'like' else 0
+    session.close()
     all_deals = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query},{'_id':1,'category_id':1,'brand':1,'totalPoints':1,'bestSellerPoints':1,'nlcPoints':1,'rank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1, 'subCategoryId':1,'netPriceAfterCashBack':1}).sort([('totalPoints',pymongo.DESCENDING),('bestSellerPoints',pymongo.DESCENDING),('nlcPoints',pymongo.DESCENDING),('rank',pymongo.DESCENDING)]))
     mobile_deals = []
     tablet_deals = []
     #accessories_deals = []
     for deal in all_deals:
         #elif deal['category_id'] ==6:
         #    accessories_deals.append(deal)
         else:
             continue
-    session.close()
     mobile_deals = sorted(mobile_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
     tablet_deals = sorted(tablet_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
     #accessories_deals = sorted(accessories_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
     print "Populating accessory memcache for userId",userId
     outer_query = []
     outer_query.append({ "$or": [ { "showDeal": 1} , { "prepaidDeal": 1 } ] })
     query = {}
     query['$gt'] = 0
-    outer_query.append({'internalRank':query})
+    outer_query.append({'dealRankPoints':query})
     outer_query.append({'category_id':{"$in":[6]}})
-    accessories_deals = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query},{'_id':1,'category_id':1,'brand':1,'bestSellerPoints':1,'nlcPoints':1,'totalPoints':1,'internalRank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1, 'subCategoryId':1,'netPriceAfterCashBack':1}).sort([('internalRank',pymongo.ASCENDING),('bestSellerPoints',pymongo.DESCENDING)]))
+    accessories_deals = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query},{'_id':1,'category_id':1,'brand':1,'bestSellerPoints':1,'nlcPoints':1,'totalPoints':1,'dealRankPoints':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1, 'subCategoryId':1,'netPriceAfterCashBack':1}).sort([('internalRank',pymongo.ASCENDING),('bestSellerPoints',pymongo.DESCENDING)]))
+    brandPrefMap = {}
+    pricePrefMap = {}
     actionsMap = {}
+    brand_p = session.query(price_preferences).filter_by(user_id=userId).all()
-    actionRank = 3
+    for x in brand_p:
+        pricePrefMap[x.category_id] = [x.min_price,x.max_price]
-    for x in session.query(user_actions).filter_by(user_id=userId).order_by(asc(user_actions.created)).all():
+    for x in session.query(brand_preferences).filter_by(user_id=userId).all():
-        #actionsMap[x.store_product_id] = 1 if x.action == 'like' else 0
+        temp_map = {}
-        action = 1 if x.action == 'like' else 0
+        if brandPrefMap.has_key((x.brand).strip().upper()):
-        if action==1:
+            val = brandPrefMap.get((x.brand).strip().upper())
-            actionsMap[x.store_product_id] = {'action':action,'rank':actionRank}
+            temp_map[x.category_id] = 1 if x.status == 'show' else 0
-            actionRank = actionRank+3
+            val.append(temp_map)
         else:
+            temp = []
+            temp_map[x.category_id] = 1 if x.status == 'show' else 0
+            temp.append(temp_map)
+            brandPrefMap[(x.brand).strip().upper()] = temp
+    for x in session.query(user_actions).filter_by(user_id=userId).all():
-            actionsMap[x.store_product_id] = {'action':action,'rank':None}
+        actionsMap[x.store_product_id] = 1 if x.action == 'like' else 0
     session.close()
-    psuedo_count = 1
-    if actionsMap:
-        for k,v in actionsMap.iteritems():
-            if v['action'] == 0:
-                tmp_deal = get_mongo_connection().Catalog.Deals.find_one({'_id':k})
-                if tmp_deal is None:
-                    v['rank'] = 999999
-                    continue
-                max_rank = list(get_mongo_connection().Catalog.Deals.find({'subCategoryId':tmp_deal.get('subCategoryId')}).sort([('internalRank',pymongo.DESCENDING)]).limit(1))
-                if len(max_rank) == 0 or max_rank[0].get('internalRank') is None:
-                    v['rank'] = 999999
-                    continue
-                v['rank'] = max_rank[0]['internalRank'] + 1
     for deal in accessories_deals:
         try:
             fd_bundles.remove(deal.get('skuBundleId'))
         except:
             pass
-        if actionsMap and actionsMap.get(deal['_id'])!=None:
+        if actionsMap.get(deal['_id']) == 0:
+            fav_weight =.25
-            if actionsMap.get(deal['_id']).get('action') ==1:
+        elif actionsMap.get(deal['_id']) == 1:
+            fav_weight = 1.5
+        else:
+            fav_weight = 1
+        if brandPrefMap.get(deal['brand'].strip().upper()) is not None:
+            brand_weight = 1
-                actionRank = actionsMap.get(deal['_id']).get('rank')
+            for brandInfo in brandPrefMap.get(deal['brand'].strip().upper()):
-                if actionRank < psuedo_count:
+                if brandInfo.get(deal['category_id']) is not None:
+                    if brandInfo.get(deal['category_id']) == 1:
-                    deal['internalRank'] = actionRank
+                        brand_weight = 2.0
+        else:
+            brand_weight = 1
-            elif actionsMap.get(deal['_id']).get('action') ==0:
+        if pricePrefMap.get(deal['category_id']) is not None:
+            if deal['available_price'] >= pricePrefMap.get(deal['category_id'])[0] and deal['available_price'] <= pricePrefMap.get(deal['category_id'])[1]:
+                asp_weight = 1.5
-                actionRank = actionsMap.get(deal['_id']).get('rank')
+            elif  deal['available_price'] >= pricePrefMap.get(deal['category_id'])[0] - 0.5 * pricePrefMap.get(deal['category_id'])[0] and deal['available_price'] <= pricePrefMap.get(deal['category_id'])[1] + 0.5 * pricePrefMap.get(deal['category_id'])[1]:
-                deal['internalRank'] = actionRank
+                asp_weight = 1.2
             else:
-                pass
+                asp_weight = 1
+        else:
+            asp_weight = 1
+        persPoints = deal['dealRankPoints'] * fav_weight * brand_weight * asp_weight
-        psuedo_count = psuedo_count+1
+        deal['persPoints'] = persPoints
-    if actionsMap:
-        accessories_deals = sorted(accessories_deals, key = lambda x: (-x['internalRank'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints']),reverse=True)
+    accessories_deals = sorted(accessories_deals, key = lambda x: (x['persPoints'],x['dealRankPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints']),reverse=True)
     if len(fd_bundles) > 0:
         print "There is still bundle in feature deals.Lets add info"
         for skuBundleId in fd_bundles:
             dummyDealObjList = __constructDummyDealObject(skuBundleId)
             continue
         for d in dealList:
             item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))
             if len(item) ==0:
                 continue
-            item[0]['internalRankDeals'] = d.get('internalRank')
+            item[0]['dealRankPoints'] = d.get('dealRankPoints')
             if d['dealType'] == 1 and d['source_id'] ==1:
                 try:
                     manualDeal = list(get_mongo_connection().Catalog.ManualDeals.find({'sku':d['_id'],'startDate':{'$lte':to_java_date(datetime.now())},'endDate':{'$gte':to_java_date(datetime.now())}}))
                     item[0]['marketPlaceUrl'] = manualDeal[0]['dealUrl'].strip()
                 except:

Subversion Repositories SmartDukaan

(root)/trunk/PyDTR/src/dtr/storage/Mongo.py – Rev 19574 → 19575