WebSVN – SmartDukaan – Diff – /trunk/PyDTR/src/scripts/DealRecomendation.py

 import pymongo
 from elixir import *
 from shop2020.model.v1.catalog.impl import DataService
 from shop2020.model.v1.catalog.impl.DataService import Item
 from shop2020.clients.InventoryClient import InventoryClient
-from dtr.utils.utils import to_java_date
+from dtr.utils.utils import to_java_date, getCashBack
 from datetime import datetime, timedelta
 import time
 import optparse
 from dtr.storage.MemCache import MemCache
 import traceback
 class __SkuInfo:
     def __init__(self, _id, skuBundleId, category_id, mrp, available_price, source_id, rank, maxNlc, minNlc, schemeAmount, minDiscount, \
                  maxDiscount, discountType, dp, nlcPoints, status, in_stock, maxprice, brand, dealType, brand_id, manualDealThresholdPrice,\
-                 codAvailable,showDp,gross_price, subCategoryId,subCategory,shippingCost):
+                 codAvailable,showDp,gross_price, subCategoryId,subCategory,shippingCost,netPriceAfterCashBack):
         self._id = _id
         self.skuBundleId = skuBundleId
         self.category_id = category_id
         self.mrp = mrp
         self.available_price = available_price
         self.showDp = showDp
         self.gross_price = gross_price
         self.subCategoryId = subCategoryId
         self.subCategory = subCategory
         self.shippingCost = shippingCost
+        self.netPriceAfterCashBack = netPriceAfterCashBack
 def get_mongo_connection(host=options.mongoHost, port=27017):
     global con
     if con is None:
             break
         #topSkus = collection.find( {'_id':664})
         for sku in topSkus:
             """Fix this """
             #TODO Compute deal flags else where.
+            if sku['source_id'] != SOURCE_MAP.get('PAYTM.COM'):
+                netPriceAfterCashBack = getNetPriceForItem(sku['_id'], sku['source_id'], sku['category_id'], sku['available_price'])
+            else:
+                if sku['codAvailable'] ==0:
+                    netPriceAfterCashBack = getNetPriceForItem(sku['_id'], sku['source_id'], sku['category_id'], sku['gross_price'])
+                else:
+                    netPriceAfterCashBack = getNetPriceForItem(sku['_id'], sku['source_id'], sku['category_id'], sku['available_price'])
             info = __SkuInfo(sku['_id'], sku['skuBundleId'], sku['category_id'], sku['mrp'], sku['available_price'], sku['source_id'], sku['rank'], None, None, 0.0, None, \
                              None, None, None, None, sku['status'], sku['in_stock'],sku['maxPrice'],sku['brand'].strip().upper(), 0, sku['brand_id'], None, sku['codAvailable'], 0, sku['gross_price'], sku['subCategoryId'], \
-                             sku['subCategory'],sku['shippingCost'])
+                             sku['subCategory'],sku['shippingCost'],netPriceAfterCashBack)
             exceptionalNlc = list(get_mongo_connection().Catalog.ExceptionalNlc.find( {"$and" : [ {'skuBundleId':info.skuBundleId}, {'overrideNlc':1} ]} ))
             if len(exceptionalNlc) > 0:
                 """Exceptional nlc found, no need to calculate max and min R-nlc"""
                 info.maxNlc = exceptionalNlc[0]['maxNlc']
                 info.minNlc = exceptionalNlc[0]['minNlc']
 def eliminateSimilarDeals():
     allItems = get_mongo_connection().Catalog.Deals.find().distinct('skuBundleId')
     for skuBundleId in allItems:
         print skuBundleId
-        similarItems = list(get_mongo_connection().Catalog.Deals.find({'skuBundleId':skuBundleId}).sort([('available_price',pymongo.ASCENDING)]))
+        similarItems = list(get_mongo_connection().Catalog.Deals.find({'skuBundleId':skuBundleId}).sort([('netPriceAfterCashBack',pymongo.ASCENDING)]))
         bestPrice = float("inf")
         bestOne = None
         bestSellerPoints = 0
         toUpdate = []
         prepaidBestPrice = float("inf")
                     get_mongo_connection().Catalog.Deals.update({ '_id' : similarItem['_id'] }, {'$set':{'showDeal':0, 'prepaidDeal':0 }})
                     continue
                 if similarItem['source_id'] == SOURCE_MAP.get('SHOPCLUES.COM') and similarItem['rank']==0 and similarItem['category_id']!=6:
                     get_mongo_connection().Catalog.Deals.update({ '_id' : similarItem['_id'] }, {'$set':{'showDeal':0,'prepaidDeal':0 }})
                     continue
-                if similarItem['available_price'] < bestPrice:
+                if similarItem['netPriceAfterCashBack'] < bestPrice:
                     bestOne = similarItem
-                    bestPrice = similarItem['available_price']
+                    bestPrice = similarItem['netPriceAfterCashBack']
                     bestSellerPoints = similarItem['bestSellerPoints']
-                elif similarItem['available_price'] == bestPrice and bestSellerPoints < similarItem['bestSellerPoints']:
+                elif similarItem['netPriceAfterCashBack'] == bestPrice and bestSellerPoints < similarItem['bestSellerPoints']:
                     bestOne = similarItem
-                    bestPrice = similarItem['available_price']
+                    bestPrice = similarItem['netPriceAfterCashBack']
                     bestSellerPoints = similarItem['bestSellerPoints']
                 else:
                     pass
             else:
                 if mc.get("negative_deals") is None:
                     get_mongo_connection().Catalog.Deals.update({ '_id' : similarItem['_id'] }, {'$set':{'showDeal':0, 'prepaidDeal':0 }})
                     continue
                 if similarItem['source_id'] == SOURCE_MAP.get('SHOPCLUES.COM') and similarItem['rank']==0 and similarItem['category_id']!=6:
                     get_mongo_connection().Catalog.Deals.update({ '_id' : similarItem['_id'] }, {'$set':{'showDeal':0,'prepaidDeal':0 }})
                     continue
-                if similarItem['source_id'] == SOURCE_MAP.get('PAYTM.COM'):
-                    similarItem['available_price'] = similarItem['gross_price']
-                if similarItem['available_price'] < prepaidBestPrice:
+                if similarItem['netPriceAfterCashBack'] < prepaidBestPrice:
                     prepaidBestOne = similarItem
-                    prepaidBestPrice = similarItem['available_price']
+                    prepaidBestPrice = similarItem['netPriceAfterCashBack']
                     prepaidBestSellerPoints = similarItem['bestSellerPoints']
-                elif similarItem['available_price'] == prepaidBestPrice and prepaidBestSellerPoints < similarItem['bestSellerPoints']:
+                elif similarItem['netPriceAfterCashBack'] == prepaidBestPrice and prepaidBestSellerPoints < similarItem['bestSellerPoints']:
                     prepaidBestOne = similarItem
-                    prepaidBestPrice = similarItem['available_price']
+                    prepaidBestPrice = similarItem['netPriceAfterCashBack']
                     prepaidBestSellerPoints = similarItem['bestSellerPoints']
                 else:
                     pass
         if bestOne is not None or prepaidBestOne is not None:
             for similarItem in similarItems:
             if bestOne is not None:
                 toUpdate.remove(bestOne['_id'])
                 get_mongo_connection().Catalog.Deals.update({ '_id' : bestOne['_id'] }, {'$set':{'showDeal':1,'prepaidDeal':0 }})
             if prepaidBestOne is not None:
                 if bestOne is not None:
-                    if prepaidBestOne['available_price'] < bestOne['available_price']:
+                    if prepaidBestOne['netPriceAfterCashBack'] < bestOne['netPriceAfterCashBack']:
                         toUpdate.remove(prepaidBestOne['_id'])
                         get_mongo_connection().Catalog.Deals.update({ '_id' : prepaidBestOne['_id'] }, {'$set':{'showDeal':0,'prepaidDeal':1 }})
                 else:
                     toUpdate.remove(prepaidBestOne['_id'])
                     get_mongo_connection().Catalog.Deals.update({ '_id' : prepaidBestOne['_id'] }, {'$set':{'showDeal':0,'prepaidDeal':1 }})
 def populateDealPointsBundle():
     global DEAL_POINTS_MAP
     activeDealPoints = get_mongo_connection().Catalog.DealPoints.find({'startDate':{'$lte':to_java_date(datetime.now())},'endDate':{'$gte':to_java_date(datetime.now())}})
     for activeDeal in activeDealPoints:
         dealPoints = __DealPoints(activeDeal['dealThresholdPrice'], activeDeal['dealPoints'])
         DEAL_POINTS_MAP[activeDeal['skuBundleId']] = dealPoints
+def getNetPriceForItem(itemId, source_id, category_id ,price):
+    cash_back_type = 0
+    cash_back = 0
+    try:
+        cashBack = getCashBack(itemId, source_id, category_id, mc, options.mongoHost)
+        if not cashBack or cashBack.get('cash_back_status')!=1:
+            cash_back_type = 0
+            cash_back = 0
+        else:
+            if cashBack['cash_back_type'] in (1,2):
+                if cashBack.get('maxCashBack') is not None:
+                    if cashBack.get('cash_back_type') ==1 and (float(cashBack.get('cash_back'))*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']
+                    else:
+                        pass
+                cash_back_type = cashBack['cash_back_type']
+                cash_back = float(cashBack['cash_back'])
+    except Exception as cashBackEx:
+        pass
+    if cash_back_type ==1:
+        return (price - float(cash_back)*price/100)
+    elif cash_back_type ==2:
+        return (price - cash_back)
+    else:
+        return price
+# def calculateNetPriceAfterCashback():
+#     for data in LATEST_UPDATED_ITEMS:
+#         master = list(get_mongo_connection().Catalog.MasterData.find({'_id':data._id}))
+#         if master[0]['source_id'] != SOURCE_MAP.get('PAYTM.COM'):
+#             netPriceAfterCashBack = getNetPriceForItem(data._id, master[0]['source_id'], master[0]['category_id'], master[0]['available_price'])
+#             get_mongo_connection().Catalog.Deals.update({'_id':data._id},{"$set":{'available_price':master[0]['available_price'],'in_stock':master[0]['in_stock'],'netPriceAfterCashBack':netPriceAfterCashBack}})
+#         else:
+#             if master[0]['codAvailable'] ==0:
+#                 netPriceAfterCashBack = getNetPriceForItem(data._id, master[0]['source_id'], master[0]['category_id'], master[0]['gross_price'])
+#             else:
+#                 netPriceAfterCashBack = getNetPriceForItem(data._id, master[0]['source_id'], master[0]['category_id'], master[0]['available_price'])
+#             get_mongo_connection().Catalog.Deals.update({'_id':data._id},{"$set":{'available_price':master[0]['available_price'],'gross_price':master[0]['gross_price'],'in_stock':master[0]['in_stock'],'netPriceAfterCashBack':netPriceAfterCashBack}})
+#
 def main():
     try:
         populateStuff()
         calculateNlc()

Subversion Repositories SmartDukaan

(root)/trunk/PyDTR/src/scripts/DealRecomendation.py – Rev 18189 → 19144