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
         temp.append(d)
     return temp
     except Exception as e:
         print traceback.print_exc()
         featuredDeals = {3:{},5:{},6:{}}
+    #Not taking into account accessories
+    featuredDeals.pop(6,None)
     fd_bundles = []
     for catMap in featuredDeals.itervalues():
         for fd_map in catMap.itervalues():
             if not fd_map.get('skuBundleId') in fd_bundles:
     outer_query = []
     outer_query.append({ "$or": [ { "showDeal": 1} , { "prepaidDeal": 1 } ] })
     query = {}
     query['$gt'] = -100
     outer_query.append({'totalPoints':query})
+    outer_query.append({'category_id':{"$in":[3,5]}})
     brandPrefMap = {}
     pricePrefMap = {}
     actionsMap = {}
     brand_p = session.query(price_preferences).filter_by(user_id=userId).all()
     for x in brand_p:
     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
     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 = []
+    #accessories_deals = []
     for deal in all_deals:
         try:
             fd_bundles.remove(deal.get('skuBundleId'))
         except:
         if deal['category_id'] ==3:
             mobile_deals.append(deal)
         elif deal['category_id'] ==5:
             tablet_deals.append(deal)
-        elif deal['category_id'] ==6:
+        #elif deal['category_id'] ==6:
-            accessories_deals.append(deal)
+        #    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)
+    #accessories_deals = sorted(accessories_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
     if len(fd_bundles) > 0:
         print "There is still bundle in feature deals with points less than -100.Lets add info"
         for skuBundleId in fd_bundles:
             for dummyDealObj in dummyDealObjList:
                 if dummyDealObj['category_id'] ==3:
                     mobile_deals.append(dummyDealObj)
                 elif dummyDealObj['category_id'] ==5:
                     tablet_deals.append(dummyDealObj)
-                elif dummyDealObj['category_id'] ==6:
+                #elif dummyDealObj['category_id'] ==6:
-                    accessories_deals.append(dummyDealObj)
+                #    accessories_deals.append(dummyDealObj)
                 else:
                     pass
     for super_category, categoryFeaturedDeals in featuredDeals.iteritems():
         for val in categoryFeaturedDeals.itervalues():
                 if super_category ==3:
                     mobile_deals.append(dummyDealObj)
                 elif super_category ==5:
                     tablet_deals.append(dummyDealObj)
-                elif super_category ==6:
+                #elif super_category ==6:
-                    accessories_deals.append(dummyDealObj)
+                #    accessories_deals.append(dummyDealObj)
                 else:
                     pass
     sortedMapMobiles = {}
     rankMapMobiles = {}
     for rank in blockedRanksTablets:
         if blockedInfoTablets.get((featuredDealsTablets.get(rank+1)).get('skuBundleId')) is not None:
             rankMapTablets[rank] = blockedInfoTablets.get((featuredDealsTablets.get(rank+1)).get('skuBundleId'))
+#     sortedMapAccessories = {}
+#     rankMapAccessories = {}
+#     rankAccessories = 0
+#
+#     blockedRanksAccessories = []
+#     blockedSkuBundlesAccessories = []
+#     blockedInfoAccessories = {}
+#
+#     featuredDealsAccessories = featuredDeals.get(6)
+#
+#     for k, v in featuredDealsAccessories.iteritems():
+#         blockedRanksAccessories.append(k-1)
+#         blockedSkuBundlesAccessories.append(v.get('skuBundleId'))
+#
+#
+#     for sorted_deal in accessories_deals:
+#
+#         while(True):
+#             if rankAccessories in blockedRanksAccessories:
+#                 rankAccessories = rankAccessories +1
+#             else:
+#                 break
+#
+#         if sorted_deal['skuBundleId'] in blockedSkuBundlesAccessories:
+#             if blockedInfoAccessories.has_key(sorted_deal['skuBundleId']):
+#                 blockedInfoAccessories.get(sorted_deal['skuBundleId']).append(sorted_deal)
+#             else:
+#                 blockedInfoAccessories[sorted_deal['skuBundleId']] = [sorted_deal]
+#             continue
+#
+#         if sortedMapAccessories.get(sorted_deal['skuBundleId']) is None:
+#             sortedMapAccessories[sorted_deal['skuBundleId']] = {rankAccessories:[sorted_deal]}
+#             rankMapAccessories[rankAccessories] = (sortedMapAccessories[sorted_deal['skuBundleId']].values())[0]
+#             rankAccessories = rankAccessories +1
+#         else:
+#             for temp_list in sortedMapAccessories.get(sorted_deal['skuBundleId']).itervalues():
+#                 temp_list.append(sorted_deal)
+#             rankMapAccessories[(sortedMapAccessories.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
+#
+#
+#     for rank in blockedRanksAccessories:
+#         if blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId')) is not None:
+#             rankMapAccessories[rank] = blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId'))
+#
+    specialOffer = None
+    promoOffer = get_mongo_connection().Catalog.PromoOffer.find_one({'user_id':userId})
+    try:
+        if promoOffer is not None:
+            offer = get_mongo_connection().Catalog.Promotions.find_one({'_id':promoOffer.get('offer_id')})
+            if offer is None:
+                raise
+            promo = Promotion(offer.get('_id'),offer.get('offer_name'),offer.get('offer_description') , offer.get('categories_applicable'), offer.get('sub_categories_not_applicable'),
+                     offer.get('startDate'), offer.get('endDate'), promoOffer.get('target1'), promoOffer.get('target1_cash_back_percetage'), promoOffer.get('target2'), promoOffer.get('target2_cash_back_percetage'), promoOffer.get('maxCashBack'),offer.get('url'),
+                     promoOffer.get('pending_order_value'), promoOffer.get('delivered_order_value'), promoOffer.get('last_run_timestamp'))
+            specialOffer = promo
+    except:
+        traceback.print_exc()
+    mem_cache_val = {3:mobile_deals, 5:tablet_deals,"3_rankMap": rankMapMobiles,"5_rankMap": rankMapTablets, "specialOffer":specialOffer }
+    mc.set(str(userId), mem_cache_val)
+def __populateCacheForAccessories(userId):
+    try:
+        if mc.get("featured_deals") is None:
+            __populateFeaturedDeals()
+        featuredDeals = mc.get("featured_deals")
+    except Exception as e:
+        print traceback.print_exc()
+        featuredDeals = {3:{},5:{},6:{}}
+    #Not taking into account mobiles, tablets
+    featuredDeals.pop(3,None)
+    featuredDeals.pop(5,None)
+    fd_bundles = []
+    for catMap in featuredDeals.itervalues():
+        for fd_map in catMap.itervalues():
+            if not fd_map.get('skuBundleId') in fd_bundles:
+                fd_bundles.append(fd_map.get('skuBundleId'))
+    print "Populating accessory memcache for userId",userId
+    outer_query = []
+    outer_query.append({ "$or": [ { "showDeal": 1} , { "prepaidDeal": 1 } ] })
+    query = {}
+    query['$gte'] = 0
+    outer_query.append({'internalRank':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)]))
+    actionsMap = {}
+    actionRank = 3
+    for x in session.query(user_actions).filter_by(user_id=userId).order_by(asc(user_actions.created)).all():
+        #actionsMap[x.store_product_id] = 1 if x.action == 'like' else 0
+        action = 1 if x.action == 'like' else 0
+        if action==1:
+            actionsMap[x.store_product_id] = {'action':action,'rank':actionRank}
+            actionRank = actionRank+3
+        else:
+            actionsMap[x.store_product_id] = {'action':action,'rank':None}
+    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']).get('action') ==1:
+                actionRank = actionsMap.get(deal['_id']).get('rank')
+                if actionRank < psuedo_count:
+                    deal['internalRank'] = actionRank
+            elif actionsMap.get(deal['_id']).get('action') ==0:
+                actionRank = actionsMap.get(deal['_id']).get('rank')
+                deal['internalRank'] = actionRank
+            else:
+                pass
+        psuedo_count = psuedo_count+1
+    if actionsMap:
+        accessories_deals = sorted(accessories_deals, key = lambda x: (-x['internalRank'],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)
+            for dummyDealObj in dummyDealObjList:
+                if dummyDealObj['category_id'] ==6:
+                    accessories_deals.append(dummyDealObj)
+    for super_category, categoryFeaturedDeals in featuredDeals.iteritems():
+        for val in categoryFeaturedDeals.itervalues():
+            dummyDealObjList = __constructDummyDealObject(val.get('skuBundleId'))
+            for dummyDealObj in dummyDealObjList:
+                if super_category == dummyDealObj['category_id']:
+                    continue
+                if super_category ==6:
+                    accessories_deals.append(dummyDealObj)
     sortedMapAccessories = {}
     rankMapAccessories = {}
     rankAccessories = 0
     blockedRanksAccessories = []
     blockedSkuBundlesAccessories = []
     blockedInfoAccessories = {}
     featuredDealsAccessories = featuredDeals.get(6)
     for k, v in featuredDealsAccessories.iteritems():
         blockedRanksAccessories.append(k-1)
         blockedSkuBundlesAccessories.append(v.get('skuBundleId'))
     for sorted_deal in accessories_deals:
         while(True):
             if rankAccessories in blockedRanksAccessories:
                 rankAccessories = rankAccessories +1
             else:
                 break
         if sorted_deal['skuBundleId'] in blockedSkuBundlesAccessories:
             if blockedInfoAccessories.has_key(sorted_deal['skuBundleId']):
                 blockedInfoAccessories.get(sorted_deal['skuBundleId']).append(sorted_deal)
             else:
                 blockedInfoAccessories[sorted_deal['skuBundleId']] = [sorted_deal]
             continue
         if sortedMapAccessories.get(sorted_deal['skuBundleId']) is None:
             sortedMapAccessories[sorted_deal['skuBundleId']] = {rankAccessories:[sorted_deal]}
             rankMapAccessories[rankAccessories] = (sortedMapAccessories[sorted_deal['skuBundleId']].values())[0]
             rankAccessories = rankAccessories +1
         else:
             for temp_list in sortedMapAccessories.get(sorted_deal['skuBundleId']).itervalues():
                 temp_list.append(sorted_deal)
             rankMapAccessories[(sortedMapAccessories.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
     for rank in blockedRanksAccessories:
         if blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId')) is not None:
             rankMapAccessories[rank] = blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId'))
     specialOffer = None
     promoOffer = get_mongo_connection().Catalog.PromoOffer.find_one({'user_id':userId})
     try:
         if promoOffer is not None:
             offer = get_mongo_connection().Catalog.Promotions.find_one({'_id':promoOffer.get('offer_id')})
                      offer.get('startDate'), offer.get('endDate'), promoOffer.get('target1'), promoOffer.get('target1_cash_back_percetage'), promoOffer.get('target2'), promoOffer.get('target2_cash_back_percetage'), promoOffer.get('maxCashBack'),offer.get('url'),
                      promoOffer.get('pending_order_value'), promoOffer.get('delivered_order_value'), promoOffer.get('last_run_timestamp'))
             specialOffer = promo
     except:
-        pass
+        traceback.print_exc()
-    mem_cache_val = {3:mobile_deals, 5:tablet_deals, 6: accessories_deals,"3_rankMap": rankMapMobiles,"5_rankMap": rankMapTablets,"6_rankMap": rankMapAccessories, "specialOffer":specialOffer }
+    mem_cache_val = {6:accessories_deals,"6_rankMap": rankMapAccessories, "specialOffer":specialOffer }
-    mc.set(str(userId), mem_cache_val)
+    mc.set(str(userId)+'_acc', mem_cache_val)
 def __populateFeaturedDeals():
     print "Populating featured deals....."
     featuredDealsMobiles = {}
                 elif item[0]['cash_back_type'] ==2:
                     item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(item[0]['cash_back'])
                 else:
                     item[0]['netPriceAfterCashBack'] = item[0]['available_price']
+            temp.append(item[0])
+        if len(temp) > 1:
+            temp = sorted(temp, key = lambda x: (x['available_price']),reverse=False)
+        dealsListMap.append(temp)
+    return dealsListMap
+def getAccesoryDeals(userId, category_id, offset, limit, sort, direction, filterData=None):
+    if mc.get("category_cash_back") is None or not bool(mc.get("category_cash_back")):
+        populateCashBack()
+    dealsListMap = []
+    user_specific_deals = mc.get(str(userId)+'_acc')
+    if user_specific_deals is None:
+        __populateCacheForAccessories(userId)
+        user_specific_deals = mc.get(str(userId)+'_acc')
+    else:
+        print "Getting user deals from accessory cache"
+    category_specific_deals = user_specific_deals.get(category_id)
+    insert_featured_deals = False
+    if sort is None or direction is None:
+        insert_featured_deals = True
+        rankMap = user_specific_deals.get(str(category_id)+"_rankMap")
+    else:
+        if sort == "bestSellerPoints":
+            sorted_deals = sorted(category_specific_deals, key = lambda x: (x['bestSellerPoints'], x['internalRank'], x['nlcPoints']),reverse=True)
+        else:
+            if direction == -1:
+                rev = True
+            else:
+                rev = False
+            sorted_deals = sorted(category_specific_deals, key = lambda x: (x['available_price']),reverse=rev)
+    filtered_deals = []
+    dataExist = True
+    if filterData is not None:
+        try:
+            filtered_deals = filterDeals(category_specific_deals, filterData)
+            if len(filtered_deals)==0:
+                dataExist = False
+        except:
+            traceback.print_exc()
+    if dataExist:
+        if not insert_featured_deals:
+            sortedMap = {}
+            rankMap = {}
+            rank = 0
+            for sorted_deal in sorted_deals:
+                if len(filtered_deals) > 0 and  sorted_deal['skuBundleId'] not in filtered_deals:
+                    continue
+                if sortedMap.get(sorted_deal['skuBundleId']) is None:
+                    sortedMap[sorted_deal['skuBundleId']] = {rank:[sorted_deal]}
+                    rankMap[rank] = (sortedMap[sorted_deal['skuBundleId']].values())[0]
+                    rank = rank +1
+                else:
+                    for temp_list in sortedMap.get(sorted_deal['skuBundleId']).itervalues():
+                        temp_list.append(sorted_deal)
+                    rankMap[(sortedMap.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
+        else:
+            filterMap = {}
+            rank = 0
+            if len(filtered_deals) > 0:
+                try:
+                    od = collections.OrderedDict(sorted(rankMap.items()))
+                except:
+                    od = ordereddict.OrderedDict(sorted(rankMap.items()))
+                for k, v in od.iteritems():
+                    if v[0].get('skuBundleId') in filtered_deals:
+                        filterMap[rank] = v
+                        rank =rank+1
+                rankMap = filterMap
+        rankCounter = offset
+        if filterData is None:
+            if mc.get("featured_deals_deal_object") is None or not bool(mc.get("featured_deals_deal_object")):
+                try:
+                    __populateFeaturedDealsForDealObject()
+                    fd_dealObject = mc.get("featured_deals_deal_object")
+                except:
+                    fd_dealObject = {3:{},5:{},6:{}}
+            else:
+                fd_dealObject = mc.get("featured_deals_deal_object")
+    else:
+        rankMap = {}
+    specialOffer = user_specific_deals.get('specialOffer')
+    for dealList in [rankMap.get(k, []) for k in range(offset, offset+limit)]:
+        if filterData is None:
+            while(True):
+                rankCounter = rankCounter+1
+                if (fd_dealObject.get(category_id).get(rankCounter)) is not None:
+                    if (fd_dealObject.get(category_id).get(rankCounter).get('offer_id') !=0):
+                        #Check whether user can see offer
+                        if specialOffer is None or specialOffer.endDate < to_java_date(datetime.now()):
+                            continue
+                    deal_object_temp = []
+                    deal_object_temp.append(fd_dealObject.get(category_id).get(rankCounter).get('object'))
+                    dealsListMap.append(deal_object_temp)
+                else:
+                    break
+        temp = []
+        if dealList is None or len(dealList)==0:
+            continue
+        for d in dealList:
+            item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))
+            if len(item) ==0:
+                continue
+            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:
+                    pass
+            elif d['source_id'] ==3:
+                item[0]['marketPlaceUrl'] = item[0]['marketPlaceUrl']+'?supc='+item[0].get('identifier')
+            else:
+                pass
+            try:
+                cashBack = getCashBack(item[0]['_id'], item[0]['source_id'], item[0]['category_id'])
+                if not cashBack or cashBack.get('cash_back_status')!=1:
+                    item[0]['cash_back_type'] = 0
+                    item[0]['cash_back'] = 0
+                else:
+                    if cashBack.get('maxCashBack') is not None:
+                        if cashBack.get('cash_back_type') ==1 and (float(cashBack.get('cash_back'))*item[0]['available_price'])/100 > cashBack.get('maxCashBack'):
+                            cashBack['cash_back_type'] = 2
+                            cashBack['cash_back'] = cashBack['maxCashBack']
+                        elif cashBack.get('cash_back_type') ==2 and cashBack.get('cash_back') > cashBack.get('maxCashBack'):
+                            cashBack['cash_back'] = cashBack['maxCashBack']
+                    item[0]['cash_back_type'] = int(cashBack['cash_back_type'])
+                    item[0]['cash_back'] = cashBack['cash_back']
+            except:
+                print "Error in adding cashback to deals"
+                item[0]['cash_back_type'] = 0
+                item[0]['cash_back'] = 0
+            try:
+                item[0]['dp'] = d['dp']
+                item[0]['showDp'] = d['showDp']
+            except:
+                item[0]['dp'] = 0.0
+                item[0]['showDp'] = 0
+            try:
+                item[0]['thumbnail'] = item[0]['thumbnail'].strip()
+            except:
+                pass
+            try:
+                if item[0]['showVideo']==0:
+                    item[0]['videoLink'] =""
+            except:
+                item[0]['videoLink'] =""
+            try:
+                if item[0]['quantity'] >1:
+                    ppq = float(item[0]['available_price'])/item[0]['quantity']
+                    if ppq == round(ppq):
+                        item[0]['ppq'] = int(ppq)
+                    else:
+                        item[0]['ppq'] = round(ppq,2)
+                else:
+                    item[0]['ppq'] = 0
+            except:
+                item[0]['ppq'] = 0
+            if filterData is None:
+                if item[0]['category_id']!=6:
+                    try:
+                        item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
+                        item[0]['filterTextBrand'] = (item[0]['brand'])
+                    except:
+                        pass
+                else:
+                    try:
+                        if item[0]['subCategory']=='' or item[0]['subCategory'] is None or item[0]['subCategoryId']==0:
+                            raise
+                        item[0]['filterLinkSubCategory'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['subCategory'])+'&filter=subcategory&subcategories='+str(int(item[0]['subCategoryId']))
+                        item[0]['filterTextSubCategory'] = (item[0]['subCategory'])
+                        item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
+                        item[0]['filterTextBrand'] = (item[0]['brand'])
+                    except:
+                        pass
+            try:
+                if item[0]['source_id']==4:
+                    item_availability_info = mc.get("item_availability_"+str(item[0]['identifier']).strip())
+                    if item_availability_info is None or len(item_availability_info)==0:
+                        item[0]['availabilityInfo'] = [{}]
+                    else:
+                        item[0]['availabilityInfo'] = item_availability_info
+            except:
+                item[0]['availabilityInfo'] = [{}]
+            if item[0]['source_id']==4:
+                for av_info in item[0]['availabilityInfo']:
+                    if not av_info:
+                        continue
+                    av_info['cash_back'] = item[0]['cash_back']
+                    av_info['cash_back_type'] = item[0]['cash_back_type']
+                    netPriceAfterCashBack = av_info['sellingPrice']
+                    if av_info['cash_back_type'] ==1:
+                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(float(av_info['sellingPrice'])*av_info['cash_back']/100)
+                    elif av_info['cash_back_type'] ==2:
+                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(av_info['cash_back'])
+                    else:
+                        pass
+                    av_info['netPriceAfterCashBack'] = netPriceAfterCashBack
+            #item[0]['netPriceAfterCashBack'] = d['netPriceAfterCashBack']
+            if item[0]['source_id'] != SOURCE_MAP.get("PAYTM.COM"):
+                if item[0]['codAvailable'] ==1:
+                    paytmPrice = item[0]['available_price']
+                else:
+                    paytmPrice = item[0]['gross_price']
+                if item[0]['cash_back_type'] ==1:
+                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(float(paytmPrice)*item[0]['cash_back']/100)
+                elif item[0]['cash_back_type'] ==2:
+                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(item[0]['cash_back'])
+                else:
+                    item[0]['netPriceAfterCashBack'] = paytmPrice
+            else:
+                if item[0]['cash_back_type'] ==1:
+                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(float(item[0]['available_price'])*item[0]['cash_back']/100)
+                elif item[0]['cash_back_type'] ==2:
+                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(item[0]['cash_back'])
+                else:
+                    item[0]['netPriceAfterCashBack'] = item[0]['available_price']
 #             if specialOffer is not None and item[0]['source_id']==4:
 #                 if item[0]['category_id'] in specialOffer.categories_applicable and item[0]['subCategoryId'] not in specialOffer.sub_categories_not_applicable:
 #                     item[0]['specialOffer'] = specialOffer.__dict__
 #             else:
 #                 item[0]['specialOffer'] = {}
         if len(temp) > 1:
             temp = sorted(temp, key = lambda x: (x['available_price']),reverse=False)
         dealsListMap.append(temp)
     return dealsListMap
 def filterDeals(deals, filterData):
     print "Filter data ", filterData
     brandsFiltered = []
     subCategoryFiltered = []
     filterArray = filterData.split('|')
         return {0:"Data not updated."}
 def resetCache(userId):
     try:
         mc.delete(userId)
+        mc.delete(userId+'_acc')
         return {1:'Cache cleared.'}
     except:
         return {0:'Unable to clear cache.'}
 def updateCollection(data):
         data['identifier'] = str(data['identifier'])
         data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
         data['model_name'] = str(data['model_name'])
         data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
         data['shippingCost'] = float(data['shippingCost'])
+        data['internalRank'] = int(data['internalRank'])
         if data['category_id'] ==6 and data['subCategoryId'] ==0:
             return {0:'Sub-Category not selected'}
         elif data['category_id'] !=6 and data['subCategoryId'] !=0:
             return {0:'Sub-Category not valid'}
         else:
         data['identifier'] = str(data['identifier'])
         data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
         data['model_name'] = str(data['model_name'])
         data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
         data['shippingCost'] = float(data['shippingCost'])
+        data['internalRank'] = int(data['internalRank'])
         if data['category_id'] ==6 and data['subCategoryId'] ==0:
             return {0:'Sub-Category not selected'}
         elif data['category_id'] !=6 and data['subCategoryId'] !=0:
             return {0:'Sub-Category not valid'}
         else:
             data['updatedOn'] = to_java_date(datetime.now())
             data['identifier'] = str(data['identifier'])
             data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
             data['model_name'] = str(data['model_name'])
             data['shippingCost'] = float(data['shippingCost'])
+            data['internalRank'] = int(data['internalRank'])
             data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
             if data['category_id'] ==6 and data['subCategoryId'] ==0:
                 return {0:'Sub-Category not selected'}
             elif data['category_id'] !=6 and data['subCategoryId'] !=0:
                 return {0:'Sub-Category not valid'}
             else:
                 pass
             data['subCategory'] = SUB_CATEGORY_MAP.get(data['subCategoryId'])
             get_mongo_connection().Catalog.MasterData.update({'_id':_id},{"$set":data},upsert=False)
+            get_mongo_connection().Catalog.MasterData.update({'skuBundleId':skuBundleId},{"$set":{'internalRank':data['internalRank']}},upsert=False, multi=True)
             return {1:'Data updated successfully.'}
         else:
             _id = data.pop('_id')
             skuBundleId = data.pop('skuBundleId')
             data['updatedOn'] = to_java_date(datetime.now())
             data['identifier'] = str(data['identifier'])
             data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
             data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
             data['model_name'] = str(data['model_name'])
             data['shippingCost'] = float(data['shippingCost'])
+            data['internalRank'] = int(data['internalRank'])
             if data['category_id'] ==6 and data['subCategoryId'] ==0:
                 return {0:'Sub-Category not selected'}
             elif data['category_id'] !=6 and data['subCategoryId'] !=0:
                 return {0:'Sub-Category not valid'}
             else:
                 item['status'] = data['status']
                 item['maxPrice'] = data['maxPrice']
                 item['brand_id'] = data['brand_id']
                 item['subCategoryId'] = data['subCategoryId']
                 item['subCategory'] = data['subCategory']
+                item['internalRank'] = data['internalRank']
                 similar_item_id = item.pop('_id')
                 get_mongo_connection().Catalog.MasterData.update({'_id':similar_item_id},{"$set":item},upsert=False)
             return {1:'Data updated successfully.'}
     except:
         return {0:'Unable to update data'}

Subversion Repositories SmartDukaan

(root)/trunk/PyDTR/src/dtr/storage/Mongo.py – Rev 19556 → 19559