Subversion Repositories SmartDukaan

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 datetime import datetime, timedelta

import time

DataService.initialize(db_hostname='localhost')

con = None

SOURCE_MAP = {'AMAZON':1,'FLIPKART':2,'SNAPDEAL':3,'SAHOLIC':4}

DISCOUNT_TYPE = {'MRP':1,'DP':2}

LATEST_UPDATED_ITEMS = []

now = datetime.now()

class __SkuInfo:

    def __init__(self, _id, skuBundleId, category_id, mrp, available_price, source_id, rank, maxNlc, minNlc, schemeAmount, minDiscount, \

                 maxDiscount, discountType, dp, nlcPoints, bestSellerPoints, totalPoints):

        self._id = _id

        self.skuBundleId = skuBundleId

        self.category_id = category_id

        self.mrp = mrp

        self.available_price = available_price

        self.source_id = source_id

        self.rank = rank

        self.maxNlc = maxNlc

        self.minNlc = minNlc

        self.schemeAmount = schemeAmount

        self.minDiscount = minDiscount

        self.maxDiscount = maxDiscount

        self.discountType = discountType

        self.dp = dp

        self.nlcPoints = nlcPoints

        self.bestSellerPoints = bestSellerPoints

        self.totalPoints = totalPoints 

def get_mongo_connection(host='localhost', port=27017):

    global con

    if con is None:

        print "Establishing connection %s host and port %d" %(host,port)

        try:

            con = pymongo.MongoClient(host, port)

        except Exception, e:

            print e

            return None

    return con

def populateStuff():

    print "Inside populate"

    global LATEST_UPDATED_ITEMS

    """Fetch latest updated items across portals

       and calculate max and min R-Nlc"""

    offset= 0

    while(True):

        print "Fetching records offset %d and limit %d" %(offset,300)

        topSkus = list(get_mongo_connection().Catalog.MasterData.find( {"$and":[{'updatedOn': { "$gt": to_java_date(now - timedelta(hours=4))} }, { 'source_id' : { "$in": SOURCE_MAP.values() } }] }).skip(offset).limit(300))

        if len((topSkus)) == 0:

            break

        #topSkus = collection.find( {'_id':664})

        for sku in topSkus:

            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, None, None)

            exceptionalNlc = list(get_mongo_connection().Catalog.ExceptionalNlc.find( {"$and" : [ {'sku':info._id}, {'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']

                LATEST_UPDATED_ITEMS.append(info)

                continue

            skuSchemeDetails = list(get_mongo_connection().Catalog.SkuSchemeDetails.find( {'sku':info._id}).sort([('addedOn',pymongo.DESCENDING)]).limit(1))

            if len(skuSchemeDetails) > 0:

                """Sku scheme details, populate scheme amount (Recently added)"""

                info.schemeAmount = float(skuSchemeDetails[0]['schemeAmount'])

            skuDealerPrices = list(get_mongo_connection().Catalog.SkuDealerPrices.find( {'sku':info._id} ) )

            if len(skuDealerPrices) > 0:

                info.dp = skuDealerPrices[0]['dp']

            skuDiscount = list(get_mongo_connection().Catalog.SkuDiscountInfo.find( {'sku':info._id} ) )

            if len(skuDiscount) > 0:

                """Sku rule found, populate max , min Discount and discount type"""

                info.maxDiscount = skuDiscount[0]['max_discount']

                info.minDiscount = skuDiscount[0]['min_discount']

                info.discountType = DISCOUNT_TYPE.get(skuDiscount[0]['discountType'].upper())

                LATEST_UPDATED_ITEMS.append(info)

                continue

            categoryDiscount = list(get_mongo_connection().Dtr.CategoryDiscount.find( {"$and" : [{'brand':sku['brand'].upper()}, {'category_id':sku['category_id']} ]} ))

            if len(categoryDiscount) > 0:

                info.maxDiscount = categoryDiscount[0]['max_discount']

                info.minDiscount = categoryDiscount[0]['min_discount']

                info.discountType = DISCOUNT_TYPE.get(categoryDiscount[0]['discountType'].upper())

            LATEST_UPDATED_ITEMS.append(info)

        offset = offset + 300

    for lol in LATEST_UPDATED_ITEMS:

        print lol.__dict__

def calculateNlc():

    global LATEST_UPDATED_ITEMS

    populated = 0

    while(populated <= len(LATEST_UPDATED_ITEMS)):

        inventory_client = InventoryClient().get_client()

        for obj in LATEST_UPDATED_ITEMS[populated:300+populated]:

            if obj.maxNlc > 0 and obj.minNlc > 0:

                continue

            saholic_sku = list(get_mongo_connection().Catalog.MasterData.find( {"$and":[{'skuBundleId': obj.skuBundleId}, { 'source_id' : SOURCE_MAP.get('SAHOLIC')}] }))

            identifier = None

            if len(saholic_sku) > 0:

                identifier = saholic_sku[0]['identifier']

            if obj.discountType == DISCOUNT_TYPE.get('MRP'):

                if obj.mrp == 0:

                    """Now mrp is zero, so we have to use saholic MRP"""

                    if identifier is not None:

                        it = Item.query.filter_by(catalog_item_id=identifier).first()

                        obj.mrp = it.mrp

                if obj.mrp > 0:

                    obj.minNlc = obj.mrp - (obj.mrp * obj.maxDiscount/100) - obj.schemeAmount

                    obj.maxNlc = obj.mrp - (obj.mrp * obj.minDiscount/100) - obj.schemeAmount 

            elif obj.discountType == DISCOUNT_TYPE.get('DP'):

                if obj.dp == 0:

                    """Now dp is zero, so we have to use saholic minimum dp for item"""

                    if identifier is not None:

                        it = Item.query.filter_by(catalog_item_id=identifier).first()

                        try:

                            vendorPricing = inventory_client.getAllItemPricing(it.id)

                            min_dp = min(pricing.dealerPrice for pricing in vendorPricing)

                            obj.dp = min_dp 

                        except:

                            pass

                if obj.dp > 0:

                    obj.minNlc = obj.dp - (obj.dp * obj.maxDiscount/100) - obj.schemeAmount

                    obj.maxNlc = obj.dp - (obj.dp * obj.minDiscount/100) - obj.schemeAmount 

            else:

                """No rule found, use saholic min nlc as max and min R-Nlc"""

                if identifier is not None:

                    it = Item.query.filter_by(catalog_item_id=identifier).first()

                    try:

                        vendorPricing = inventory_client.getAllItemPricing(it.id)

                        min_nlc = min(pricing.nlc for pricing in vendorPricing)

                        obj.maxNlc = min_nlc

                        obj.minNlc = min_nlc 

                    except:

                        pass

        populated = populated + 300

        time.sleep(10) 

def calculateNlcPoints():

    global LATEST_UPDATED_ITEMS

    for sku in LATEST_UPDATED_ITEMS:

        if sku.maxNlc and sku.minNlc:

            discFromMinNlc = (sku.minNlc - sku.available_price)/sku.available_price *100

            discFromMaxNlc = (sku.maxNlc - sku.available_price)/sku.available_price *100

            if discFromMinNlc > 0:

                nlcPoints = 100/5 * discFromMinNlc

            elif discFromMinNlc < 0 and discFromMaxNlc > 0:

                nlcPoints = 0

            else:

                nlcPoints = 100/5 * discFromMinNlc

            sku.nlcPoints = min(nlcPoints,300)

        else:

            sku.nlcPoints = 0

def commitData():

    global LATEST_UPDATED_ITEMS

    for sku in LATEST_UPDATED_ITEMS:

        #get_mongo_connection().Catalog.Deals.update({'_id':sku._id},{'$set' : sku.__dict__},upsert=True,multi=True)

        get_mongo_connection().Catalog.Deals.update({'_id':sku._id},sku.__dict__,upsert=True)

def addBestSellerPoints():

    allItems = list(get_mongo_connection().Catalog.Deals.find({})) 

    for sku in allItems:

        bestSellerPoints = list(get_mongo_connection().Catalog.BestSellerPoints.find( {"$and":[{'min_rank': { "$lte": sku['rank'] } }, {'max_rank': { "$gte": sku['rank'] } } , { 'category_id' : sku['category_id'] }] } ))

        if len(bestSellerPoints) > 0:

            sku['bestSellerPoints'] = bestSellerPoints[0]['points']

        else:

            sku['bestSellerPoints'] = -50

        sku['totalPoints'] = sku['bestSellerPoints'] + sku['nlcPoints']

        get_mongo_connection().Catalog.Deals.update({'_id':sku['_id']},{'$set':{'bestSellerPoints':sku['bestSellerPoints'],'totalPoints':sku['totalPoints']}},multi=False)

def main():

    populateStuff()

    calculateNlc()

    calculateNlcPoints()

    commitData()

    addBestSellerPoints()

if __name__=='__main__':

    main()