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

 import pymongo
 from dtr.utils.utils import to_java_date
 import optparse
 from dtr.storage.MemCache import MemCache
 from dtr.utils.utils import getCashBack
+from dtr.utils.utils import getNlcPoints
+import traceback
 dealsMap = {}
 con = None
 dealsCatalogIds = []
 itemCatalogMap = {}
 def getPrivateDeals():
     try:
         global dealsMap
         dealsMap = dict()
         all_active_items_query =  session.query(PrivateDeals).filter(PrivateDeals.isActive==True).filter(now().between(PrivateDeals.startDate, PrivateDeals.endDate))
-        print all_active_items_query
         all_active_private_deals = all_active_items_query.all()
         if all_active_private_deals is not None or all_active_private_deals!=[]:
             for active_private_deal in all_active_private_deals:
+                print active_private_deal.item_id
                 item = Item.get_by(id = active_private_deal.item_id)
                 if item.sellingPrice >  active_private_deal.dealPrice and item.status==3:
                     dealsMap[active_private_deal.item_id] = active_private_deal
     finally:
         session.close()
 def getItemsToUpdate():
     global dealsCatalogIds
     global itemCatalogMap
+    toScrapMap = {}
     bestSellers = list(get_mongo_connection().Catalog.MasterData.find({'rank':{"$gt":0}}))
     for bestSeller in bestSellers:
         saholicCatalogIds = list(get_mongo_connection().Catalog.MasterData.find({'skuBundleId':bestSeller['skuBundleId'],'source_id':4}))
         for d in saholicCatalogIds:
             if d['source_id']!=4:
                 continue
+            d['dealFlag'] = 0
+            d['dealType'] = 0
+            d['dealPoints'] = 0
+            d['manualDealThresholdPrice'] = None
+            toScrapMap[d['_id']] = d
             dealsCatalogIds.append(long(d['identifier'].strip()))
-#    dealFlagged = list(get_mongo_connection().Catalog.Deals.find({'source_id':4,'showDeal':1,'totalPoints':{'$gt':0}}))
+    dealFlagged = list(get_mongo_connection().Catalog.Deals.find({'source_id':4,'showDeal':1,'totalPoints':{'$gt':0}}))
-#    for deal in dealFlagged:
+    for deal in dealFlagged:
+        if not toScrapMap.has_key(deal['_id']):
+            data = list(get_mongo_connection().Catalog.MasterData.find({'_id':deal['_id']}))
+            data[0]['dealFlag'] = 0
+            data[0]['dealType'] = 0
+            data[0]['dealPoints'] = 0
+            data[0]['manualDealThresholdPrice'] = None
+            toScrapMap[deal['_id']] = data[0]
-#        if not (deal['_id']) in dealsCatalogIds:
+            if long(data[0]['identifier'].strip()) not in dealsCatalogIds:
-#            dealsCatalogIds.append(long(deal['identifier'].strip()))
+                dealsCatalogIds.append(long(data[0]['identifier'].strip()))
+    manualDeals = list(get_mongo_connection().Catalog.ManualDeals.find({'startDate':{'$lte':to_java_date(datetime.now())},'endDate':{'$gte':to_java_date(datetime.now())},'source_id':4}))
+    for manualDeal in manualDeals:
+        if not toScrapMap.has_key(manualDeal['sku']):
+            data = list(get_mongo_connection().Catalog.MasterData.find({'_id':manualDeal['sku']}))
+            if len(data) > 0:
+                data[0]['dealFlag'] = 1
+                data[0]['dealType'] = manualDeal['dealType']
+                data[0]['dealPoints'] = manualDeal['dealPoints']
+                data[0]['manualDealThresholdPrice'] = manualDeal['dealThresholdPrice']
+                toScrapMap[manualDeal['sku']] = data[0]
+                if long(data[0]['identifier'].strip()) not in dealsCatalogIds:
+                    dealsCatalogIds.append(long(data[0]['identifier'].strip()))
+        else:
+            data = toScrapMap.get(manualDeal['sku'])
+            data['dealFlag'] = 1
+            data['dealType'] = manualDeal['dealType']
+            data['dealPoints'] = manualDeal['dealPoints']
+            data['manualDealThresholdPrice'] = manualDeal['dealThresholdPrice']
     items = Item.query.filter(Item.catalog_item_id.in_(dealsCatalogIds)).all()
     for item in items:
         temp = []
         if not itemCatalogMap.has_key(item.catalog_item_id):
             temp.append(item)
             for l in val:
                 temp.append(l)
             temp.append(item)
             itemCatalogMap[item.catalog_item_id] = temp
-    for saholicCatalogId in bestSellers:
+    for saholicCatalogId in toScrapMap.itervalues():
         if saholicCatalogId['source_id']!=4:
             continue
         d_items = itemCatalogMap.get(long(saholicCatalogId['identifier'].strip()))
         available_price = None
         in_stock = 0
         else:
             get_mongo_connection().Catalog.MasterData.update({'_id':saholicCatalogId['_id']}, {'$set' : {'updatedOn':to_java_date(datetime.now()),'in_stock':in_stock,'priceUpdatedOn':to_java_date(datetime.now())}}, multi=True)
             get_mongo_connection().Catalog.Deals.update({'_id':saholicCatalogId['_id']}, {'$set' : {'in_stock':in_stock}}, multi=True)
         try:
-            recomputeDeal(saholicCatalogId['skuBundleId'])
+            recomputeDeal(saholicCatalogId)
         except:
             print "Unable to compute deal for ",saholicCatalogId['skuBundleId']
 def populateNegativeDeals():
     negativeDeals = get_mongo_connection().Catalog.NegativeDeals.find().distinct('sku')
     mc.set("negative_deals", negativeDeals, 600)
+def recomputePoints(item, deal):
+    try:
+        nlcPoints = getNlcPoints(item, deal['minNlc'], deal['maxNlc'], deal['available_price'])
+    except:
+        traceback.print_exc()
+        nlcPoints = deal['nlcPoints']
+    if item['manualDealThresholdPrice'] >= deal['available_price']:
+        dealPoints = item['dealPoints']
+    else:
+        dealPoints = 0
+    get_mongo_connection().Catalog.Deals.update({'_id':deal['_id']},{"$set":{'totalPoints':deal['totalPoints'] - deal['nlcPoints'] + nlcPoints - deal['dealPoints'] +dealPoints , 'nlcPoints': nlcPoints, 'dealPoints': dealPoints, 'manualDealThresholdPrice': item['manualDealThresholdPrice']}})
-def recomputeDeal(skuBundleId):
+def recomputeDeal(item):
     """Lets recompute deal for this bundle"""
-    print "Recomputing for bundleId",skuBundleId
+    print "Recomputing for bundleId",item.get('skuBundleId')
+    skuBundleId = item['skuBundleId']
     similarItems = list(get_mongo_connection().Catalog.Deals.find({'skuBundleId':skuBundleId}).sort([('available_price',pymongo.ASCENDING)]))
     bestPrice = float("inf")
     bestOne = None
     bestSellerPoints = 0
     toUpdate = []
     for similarItem in similarItems:
         if mc.get("negative_deals") is None:
             populateNegativeDeals()
-        try:
+#        try:
-            cashBack = getCashBack(similarItem['_id'], similarItem['source_id'], similarItem['category_id'], mc, options.mongoHost)
+#            cashBack = getCashBack(similarItem['_id'], similarItem['source_id'], similarItem['category_id'], mc, options.mongoHost)
-            if not cashBack or cashBack.get('cash_back_status')!=1:
+#            if not cashBack or cashBack.get('cash_back_status')!=1:
-                pass
+#                pass
-            else:
+#            else:
-                if cashBack['cash_back_type'] ==1:
+#                if cashBack['cash_back_type'] ==1:
-                    similarItem['available_price'] = similarItem['available_price'] - similarItem['available_price'] * float(cashBack['cash_back'])/100
+#                    similarItem['available_price'] = similarItem['available_price'] - similarItem['available_price'] * float(cashBack['cash_back'])/100
-                elif cashBack['cash_back_type'] ==2:
+#                elif cashBack['cash_back_type'] ==2:
-                    similarItem['available_price'] = similarItem['available_price'] - float(cashBack['cash_back'])
+#                    similarItem['available_price'] = similarItem['available_price'] - float(cashBack['cash_back'])
-                else:
+#                else:
-                    pass
+#                    pass
-        except Exception as cashBackEx:
+#        except Exception as cashBackEx:
-            print cashBackEx
+#            print cashBackEx
-            print "Error calculating cashback."
+#            print "Error calculating cashback."
+        if similarItem['_id'] == item['_id']:
+            try:
+                recomputePoints(item, similarItem)
+            except:
+                traceback.print_exc()
         if similarItem['in_stock'] == 0 or similarItem['maxprice'] is None or similarItem['maxprice'] < similarItem['available_price'] or similarItem['_id'] in mc.get("negative_deals"):
             get_mongo_connection().Catalog.Deals.update({ '_id' : similarItem['_id'] }, {'$set':{'showDeal':0 }})
             continue
         if similarItem['available_price'] < bestPrice:
             bestOne = similarItem
     getPrivateDeals()
     try:
         getItemsToUpdate()
     finally:
         session.close()
+    print "Done with saholic pricing"
 if __name__=='__main__':
     main()

Subversion Repositories SmartDukaan

(root)//trunk/PyDTR/src/scripts/SaholicPricing.py – Rev 14705 → 15268