Subversion Repositories SmartDukaan

from datetime import datetime

from dtr.storage.DataService import user_app_cashbacks

import sys

import traceback

import getopt

from dtr.main import tprint, refundToWallet

import time

from dtr.storage import DataService

from elixir import *

DataService.initialize(db_hostname="localhost") 

def settleAppOffersPayBack(runtype='dry'):

    if runtype == 'live':

        currentTimestamp = datetime.today()

        currentMonth = currentTimestamp.month

        currentDay = currentTimestamp.day

        currentYear = currentTimestamp.year

        fortNight = (currentMonth - 1)*2 + (currentDay/15)

        userCashbacksToBeCredited = user_app_cashbacks.query.filter(user_app_cashbacks.status=='Approved').filter(user_app_cashbacks.fortnightOfYear<fortNight).filter(user_app_cashbacks.yearVal==currentYear).all()

        userCashbacksMap = {}

        for userCashback in userCashbacksToBeCredited:

            if userCashback.user_id >0:

                if userCashbacksMap.has_key(userCashback.user_id):

                    cashBackAmount = userCashbacksMap.get(userCashback.user_id) + userCashback.amount

                    userCashbacksMap[userCashback.user_id] = cashBackAmount

                else:

                    userCashbacksMap[userCashback.user_id] = userCashback.amount

        datetimeNow = datetime.now() 

        batchId = int(time.mktime(datetimeNow.timetuple()))

        if(refundToWallet(batchId, userCashbacksMap, 'App')):

            for userCashback in userCashbacksToBeCredited:

                userCashback.status = 'Credited'

                userCashback.batchCreditId = batchId

            session.commit()

def main(argv):

    runtype = 'dry'

    try:

        opts, args = getopt.getopt(argv,"ht:",["runtype="])

    except getopt.GetoptError:

        print 'settlementcron.py -t <dry|live>'

        sys.exit(2)

    for opt, arg in opts:

        if opt == '-h':

            print 'settlementcron.py -t <dry|live>'

            sys.exit()

        elif opt == '-t':

            runtype = arg

        tprint("Settling payback", type)

        try:

            settleAppOffersPayBack(runtype)

        except:

            tprint("Error")

            traceback.print_exc()

    if session.is_active():

        session.close()

if __name__=='__main__':

    main(sys.argv[1:])