Subversion Repositories SmartDukaan

#!/usr/bin/python

from elixir import *

import optparse

import sys

if __name__ == '__main__' and __package__ is None:

    import os

    sys.path.insert(0, os.getcwd())

from datetime import timedelta, datetime

from shop2020.thriftpy.model.v1.order.ttypes import RechargeOrderStatus

from shop2020.model.v1.order.impl.DataAccessors import update_recharge_order_status,\

    update_recharge_transaction_status

from shop2020.model.v1.order.impl import DataService, RechargeService

from shop2020.model.v1.order.impl.DataService import RechargeTransaction

from shop2020.model.v1.order.impl.model.RechargeOrder import RechargeOrder

def processRechargeTransactions(deltaTime):

    #Lets first get all recharge unknown orders irrespective of recharge mode in delta minutes time window

    try:

        timeFilter = datetime.now() -timedelta(minutes=deltaTime)

        unknownOrders = RechargeTransaction.query.filter(RechargeTransaction.status.in_([RechargeOrderStatus.RECHARGE_UNKNOWN, RechargeOrderStatus.INIT])).filter(RechargeTransaction.transactionTime >=timeFilter).all()

        for unknownOrder in unknownOrders:

            try:

                status, description = RechargeService.checkTransactionStatus('', str(unknownOrder.id))

                print status, description

                if status:

                    update_recharge_transaction_status(unknownOrder.id, RechargeOrderStatus.RECHARGE_SUCCESSFUL)

                else:

                    update_recharge_transaction_status(unknownOrder.id, RechargeOrderStatus.RECHARGE_FAILED)

            except:

                print "Still unknown"

    finally:

        session.close()

    try:

        timeFilter = datetime.now() -timedelta(minutes=2,seconds=30)

        inprocessOrders = RechargeTransaction.query.filter(RechargeTransaction.status == RechargeOrderStatus.RECHARGE_IN_PROCESS).filter(RechargeTransaction.responseTime < timeFilter).all()

        #Lets mark all recharges in process in above time window as payment successful

        for inprocessOrder in inprocessOrders:

            inprocessOrder.status = RechargeOrderStatus.RECHARGE_UNKNOWN

            inprocessOrder.responseTime = datetime.now()

        session.commit()

        for inprocessOrder in inprocessOrders:

            try:

                status, description = RechargeService.checkTransactionStatus('', str(inprocessOrder.id))

                print status, description

                if status:

                    update_recharge_transaction_status(inprocessOrder.id, RechargeOrderStatus.RECHARGE_SUCCESSFUL)

                else:

                    update_recharge_transaction_status(inprocessOrder.id, RechargeOrderStatus.RECHARGE_FAILED)

            except:

                print "Still unknown"

    finally:

        session.close()

def processRechargeOrders(deltaTime):

    #Lets first get all payment successful orders irrespective of recharge mode in delta minutes time window

    try:

        paymentSuccessfulOrders = RechargeOrder.query.filter(RechargeOrder.status == RechargeOrderStatus.PAYMENT_SUCCESSFUL).all()

        for paymentSuccessfulOrder in paymentSuccessfulOrders:

            try:

                status, description = RechargeService.checkTransactionStatus('', str(paymentSuccessfulOrder.id))

                print status, description

                if status:

                    update_recharge_order_status(paymentSuccessfulOrder.id, RechargeOrderStatus.RECHARGE_SUCCESSFUL)

                else:

                    update_recharge_order_status(paymentSuccessfulOrder.id, RechargeOrderStatus.RECHARGE_FAILED)

            except:

                print "Still unknown"

    finally:

        session.close()

    try:

        timeFilter = datetime.now() -timedelta(minutes=5)

        unknownOrders = RechargeOrder.query.filter(RechargeOrder.status == RechargeOrderStatus.RECHARGE_UNKNOWN).filter(RechargeOrder.responseTimestamp < timeFilter).all()

        #Lets mark all recharges unknown in above time window as payment successful

        for unknownOrder in unknownOrders:

            unknownOrder.status = RechargeOrderStatus.PAYMENT_SUCCESSFUL

            unknownOrder.responseTimestamp = datetime.now()

        session.commit()

        for unknownOrder in unknownOrders:

            try:

                status, description = RechargeService.checkTransactionStatus('', str(unknownOrder.id))

                print status, description

                if status:

                    update_recharge_order_status(unknownOrder.id, RechargeOrderStatus.RECHARGE_SUCCESSFUL)

                else:

                    update_recharge_order_status(unknownOrder.id, RechargeOrderStatus.RECHARGE_FAILED)

            except:

                print "Still unknown"

    finally:

        session.close()

def main():

    parser = optparse.OptionParser()

    parser.add_option("-H", "--host", dest="hostname",

                      default="localhost",

                      type="string", help="The HOST where the DB server is running",

                      metavar="HOST")

    parser.add_option("-T", "--time", dest="time",

                      default=60,

                      type="int", help="Time in minutes to filter recharge orders/transaction",

                      metavar="TIME")

    (options, args) = parser.parse_args()

    if len(args) != 0:

        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")

    DataService.initialize(db_hostname=options.hostname, echoOn=True)

    processRechargeOrders(options.time)

    processRechargeTransactions(options.time)

if __name__=='__main__':

    main()