Subversion Repositories SmartDukaan

#!/usr/bin/python

'''

It is used to process transactions for which the payment 

was received but the order was not processed.

@author: Rajveer

'''

import optparse

import sys

import datetime

from elixir import *

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

    import os

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

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

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

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

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

from shop2020.model.v1.order.impl.RechargeService import checkTransactionStatus, getRefunds

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("-r", "--refund", dest="refund",

                      action="store_true",

                      help="")

    parser.add_option("-u", "--unknown", dest="unknown",

                      action="store_true",

                      help="")

    (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?")

    if options.refund:

        processRefunds()

    if options.unknown:

        processUnknownTransactions()

def processRefunds():

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

    todate = datetime.datetime.now()

    for i in range(15):

        orderDate = todate + datetime.timedelta(days= -i)

        refunds = getRefunds(orderDate)

        for key in refunds.keys():

            refund = refunds.get(key)

            refundAmount = refund[0]

            refundDate = refund[1]

            order = RechargeOrder.get_by(spiceTID = key)

            if order.status == RechargeOrderStatus.RECHARGE_FAILED_REFUNDED:

                print "Refund is already processed."

                continue

            if order.status != RechargeOrderStatus.RECHARGE_SUCCESSFUL:

                print "Recharge is not successful. There is something wrong."

                continue

            if order.totalAmount != refundAmount:

                print "Refund amount is not same as transaction amount"

                continue

            order.status = RechargeOrderStatus.RECHARGE_FAILED_REFUNDED

            order.responseTimestamp = refundDate

            session.commit()

def processUnknownTransactions():    

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

    orders = get_recharge_orders_for_status(RechargeOrderStatus.PAYMENT_SUCCESSFUL)

    for order in orders:

        if order.creationTimestamp + datetime.timedelta(minutes=10) < datetime.datetime.now():

            status, description = checkTransactionStatus('', str(order.id))

            print status, description

            if status:

                update_recharge_order_status(order.id, RechargeOrderStatus.RECHARGE_SUCCESSFUL)

            else:

                update_recharge_order_status(order.id, RechargeOrderStatus.RECHARGE_FAILED)

if __name__ == '__main__':

    main()