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,\
    update_amount_in_wallet
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?")
    DataService.initialize(db_hostname=options.hostname, echoOn=True)
        
    if options.refund:
        processRefunds()
    if options.unknown:
        processUnknownTransactions()

def processRefunds():
    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()
            update_amount_in_wallet(order.userId, order.walletAmount, order.id)
            update_amount_in_wallet(order.userId, order.totalAmount - order.walletAmount, order.id)
            

def processUnknownTransactions():    
    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()