Subversion Repositories SmartDukaan

import pymongo

from datetime import datetime

from dtr.utils.utils import to_java_date

from operator import itemgetter 

con = None

cashBackMap = {}

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 populateCashBack():

    global cashBackMap

    cashBack = list(get_mongo_connection().Catalog.CategoryCashBack.find())

    for row in cashBack:

        if cashBackMap.has_key(row['source_id']):

            temp_map = {}

            temp_list = []

            arr = cashBackMap.get(row['source_id'])

            for val in arr:

                temp_list.append(val)

            temp_map[row['category_id']] = row

            temp_list.append(temp_map)

            cashBackMap[row['source_id']] = temp_list 

        else:

            temp_map = {}

            temp_list = []

            temp_map[row['category_id']] = row

            temp_list.append(temp_map)

            cashBackMap[row['source_id']] = temp_list

    for x, y in cashBackMap.iteritems():

        print x

        print y

        print "================="

def addCategoryDiscount(data):

    collection = get_mongo_connection().Dtr.CategoryDiscount

    query = []

    data['brand'] = data['brand'].strip().upper()

    query.append({"brand":data['brand']})

    query.append({"category_id":data['category_id']})

    r = collection.find({"$and":query})

    if r.count() > 0:

        return {0:"Brand & Category info already present."}

    else:

        collection.insert(data)

        return {1:"Data added successfully"}

def getAllCategoryDiscount():

    data = []

    collection = get_mongo_connection().Dtr.CategoryDiscount

    cursor = collection.find()

    for val in cursor:

        data.append(val)

    return data

def addSchemeDetailsForSku(data):

    collection = get_mongo_connection().Dtr.SkuSchemeDetails

    data['addedOn'] = to_java_date(datetime.now())

    collection.insert(data)

    return {1:"Data added successfully"}

def getAllSkuWiseSchemeDetails():

    data = []

    collection = get_mongo_connection().Dtr.SkuSchemeDetails

    cursor = collection.find()

    for val in cursor:

        data.append(val)

    return data

def addSkuDiscountInfo(data):

    collection = get_mongo_connection().Dtr.SkuDiscountInfo

    cursor = collection.find({"sku":data['sku']})

    if cursor.count() > 0:

        return {0:"Sku information already present."}

    else:

        collection.insert(data)

        return {1:"Data added successfully"}

def getallSkuDiscountInfo():

    data = []

    collection = get_mongo_connection().Dtr.SkuDiscountInfo

    cursor = collection.find()

    for val in cursor:

        data.append(val)

    return data

def addExceptionalNlc(data):

    collection = get_mongo_connection().Dtr.ExceptionalNlc

    cursor = collection.find({"sku":data['sku']})

    if cursor.count() > 0:

        return {0:"Sku information already present."}

    else:

        collection.insert(data)

        return {1:"Data added successfully"}

def getAllExceptionlNlcItems():

    data = []

    collection = get_mongo_connection().Dtr.ExceptionalNlc

    cursor = collection.find()

    for val in cursor:

        data.append(val)

    return data

def getMerchantOrdersByUser(userId, page=1, window=50):

    if page==None:

        page = 1

    if window==None:

        window = 50

    result = {}

    skip = (page-1)*window

    collection = get_mongo_connection().Dtr.merchantOrder

    cursor = collection.find({"userId":userId})

    total_count = cursor.count()

    pages = total_count/window + (0 if total_count%window==0 else 1)  

    print "total_count", total_count

    if total_count > skip:

        cursor = cursor.skip(skip).limit(window)

        orders = []

        for order in cursor:

            del(order["_id"])

            orders.append(order)

        result['data'] = orders

        result['window'] = window

        result['totalCount'] = total_count 

        result['currCount'] = cursor.count()

        result['totalPages'] = pages

        result['currPage'] = page    

        return result

    else:

        return result

def getDeals(category_id, offset, limit, sort, direction):

    rank = 1

    deals = {}

    if sort is None or direction is None:

        sortField = "totalPoints"

        direct = -1

    else:

        sortField = sort

        direct = direction

    if category_id == 0:

        print "All categories deals"

        print category_id

        print sortField

        print offset

        print limit

        print direct

        data = list(get_mongo_connection().Catalog.Deals.find({'rank':{'$gt':0},'showDeal':1}).sort([(sortField,direct)]).skip(offset).limit(limit))

    else:

        print "selective category deals"

        print category_id

        print sortField

        print offset

        print limit

        print direct

        data = list(get_mongo_connection().Catalog.Deals.find({'rank':{'$gt':0},'category_id':category_id,'showDeal':1}).sort([(sortField,direct)]).skip(offset).limit(limit))

    print data[0]['_id']

    for d in data:

        item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))

        if not deals.has_key(item[0]['identifier']):

            item[0]['dealRank'] = rank 

            deals[item[0]['identifier']] = item[0]

            rank +=1

    return sorted(deals.values(), key=itemgetter('dealRank'))

def getItem(skuId):

    try:

        skuData = list(get_mongo_connection().Catalog.MasterData.find({'_id':int(skuId)}))

        return skuData

    except:

        return [{}]

def getCashBackDetails(identifier, source_id):

    """Need to add item level cashback, no data available right now."""

    if source_id in (1,2,4):

        skuData = list(get_mongo_connection().Catalog.MasterData.find({'identifier':identifier.strip(), 'source_id':source_id}))

    elif source_id == 3:

        skuData = list(get_mongo_connection().Catalog.MasterData.find({'secondaryIdentifier':identifier.strip(), 'source_id':source_id}))

    else:

        return {}

    if len(skuData) > 0:

        cashback = list(get_mongo_connection().Catalog.CategoryCashBack.find({'category_id':skuData[0]['category_id'], 'source_id':source_id}))

        if len(cashback) > 0:

            return cashback[0]

        else:

            return {} 

    else:

        return {}

def main():

    populateCashBack()

if __name__=='__main__':

    main()