Subversion Repositories SmartDukaan

import pymongo
from datetime import datetime,time, timedelta, date
from dtr.main import refundToWallet
from dtr.storage import Mongo
from dtr.utils import utils
import math
import time
con = None
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

userAmountMap = {}

def calculateCashbackForUser(i):
    cashbackToBeCredited = 0 
    deliveredValue = i.get('delivered_order_value')
    maximumCashback = i.get('maxCashBack')
    target2CashBackPercentage = i.get('target2_cash_back_percetage')
    target1CashBackPercentage= i.get('target1_cash_back_percetage')
    target1Amount = i.get('target1')
    target2Amount = i.get('target2')
    if deliveredValue>=target1Amount and deliveredValue<target2Amount:
        cashbackToBeCredited = (target1CashBackPercentage/100.0)*deliveredValue
    elif deliveredValue>=target2Amount:
        cashbackToBeCredited = (target2CashBackPercentage/100.0)*deliveredValue
        if cashbackToBeCredited > maximumCashback:
            cashbackToBeCredited = maximumCashback
    else:
        pass
    return cashbackToBeCredited

def offerCashback():
    cursor = get_mongo_connection().Catalog.PromoOffer.find()
    global userAmountMap
    totalDeliveredSum = 0
    totalCashbackAmount = 0
    for i in cursor:
        if i.get('delivered_order_value') !=0:
            totalDeliveredSum = totalDeliveredSum + i.get('delivered_order_value') 
            cashbackAmount = calculateCashbackForUser(i)
            userId = i.get('user_id')
            if cashbackAmount!=0:
                if userAmountMap.has_key(userId):
                    userAmountMap[userId] = userAmountMap.get(userId)+ math.floor(cashbackAmount)
                else:
                    userAmountMap[userId] = math.floor(cashbackAmount)
#     for k,v in userAmountMap.iteritems():
#         print k,v
#         totalCashbackAmount = totalCashbackAmount + v
#     print totalDeliveredSum
#     print totalCashbackAmount
def main():
    offerCashback()
#     message = []
#     datetimeNow = datetime.now()
#     batchId = int(time.mktime(datetimeNow.timetuple()))
#     if refundToWallet(batchId,userAmountMap):
#         sum=0
#         for key, value in userAmountMap.iteritems():
#             sum += value
#             get_mongo_connection().Dtr.refund.insert({"userId": key, "batch":batchId, "userAmount":value, "timestamp":datetime.strftime(datetimeNow,"%Y-%m-%d %H:%M:%S"), "type":utils.CREDIT_TYPE_OFFER})
#             get_mongo_connection().Dtr.user.update({"userId":key}, {"$inc": { "credited": value, utils.CREDIT_TYPE_OFFER:value}}, upsert=True)
#             Mongo.sendNotification([key], 'Batch Credit','Cashback Credited for %ss'%(str(utils.CREDIT_TYPE_OFFER)),'Rs.%s has been added to your wallet'%(value),'url','http://api.profittill.com/cashbacks/mine?user_id=%s'%(key), '2999-01-01', True, "TRAN_SMS Dear Customer, Cashback Credited for %ss. Rs.%s has been added to your wallet"%(utils.CREDIT_TYPE_OFFER,value))
#         message.append("<b>Batch Id - %d</b><br><b>Total Amount Credited - %d</b><br>"%(batchId,sum))
#         utils.sendmail(['manas.kapoor@shop2020.in'], "".join(message), 'Cashback for April Offer Credited Successfully')
#     else:
#         print 'Cashback credit failed for all the users for batch id',batchId
if __name__ == '__main__':
    main()