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 datetime import timedelta

from elixir import *

from sqlalchemy.sql import func

import urllib

import httplib

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_recharge_transaction_status, get_next_invoice_number

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

from shop2020.model.v1.order.impl.DataService import RechargeTransaction, HotspotStore,\

    WalletForCompany, WalletHistoryForCompany

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="")

    parser.add_option("-a", "--authorized", dest="authorized",

                      action="store_true",

                      help="")

    parser.add_option("-c", "--collection", dest="collection",

                      action="store_true",

                      help="")

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

                      action="store_true",

                      help="")    

    parser.add_option("-t", "--txn-id", dest="txn_id",

                   type="int",

                   help="mark the transaction(recharge order id) TXN_ID as successful",

                   metavar="TXN_ID")

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

    if options.authorized:

        processAuthorizedTransactions(options.txn_id)

    if options.collection:

        d = datetime.datetime.now()

        d = d + timedelta(days = -1)

        compute_recharge_collection(d)

    if options.topup:

        topup_company_wallet(1,100000)

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:

                amount = order.totalAmount

                isStoreOrder = False

            else:

                order = RechargeTransaction.get_by(spiceTID = key)

                isStoreOrder = True

                amount = order.amount

            if order.status == RechargeOrderStatus.RECHARGE_FAILED_REFUNDED:

                print "Refund is already processed."

                continue

            if order.status not in (RechargeOrderStatus.RECHARGE_SUCCESSFUL, RechargeOrderStatus.PAYMENT_SUCCESSFUL):

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

                continue

            if amount != refundAmount:

                print "Refund amount is not same as transaction amount"

                continue

            if isStoreOrder:

                update_recharge_transaction_status(order.id, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED, refundDate)

            else:

                update_recharge_order_status(order.id, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED, refundDate)

def processUnknownTransactions():    

    orders = get_recharge_orders_for_status(RechargeOrderStatus.PAYMENT_SUCCESSFUL)

    for order in orders:

        try:

            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)

        except:

            print "Do Nothing"

def processAuthorizedTransactions(txn_id):

    update_recharge_order_status(txn_id, RechargeOrderStatus.PAYMENT_SUCCESSFUL)

def compute_recharge_collection(cdate):

    todate = datetime.datetime(cdate.year, cdate.month, cdate.day)

    tomorrow = todate + timedelta(days=1)

    txns = session.query(RechargeTransaction.storeId, RechargeTransaction.payMethod, RechargeTransaction.status, func.sum(RechargeTransaction.amount), func.sum(RechargeTransaction.discount)).filter(RechargeTransaction.status.in_([RechargeOrderStatus.RECHARGE_SUCCESSFUL, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED])).filter(RechargeTransaction.transactionTime >= todate).filter(RechargeTransaction.transactionTime < tomorrow).group_by(RechargeTransaction.storeId, RechargeTransaction.payMethod, RechargeTransaction.status).order_by(RechargeTransaction.storeId).all()

    storeData = {}

    for txn in txns:

        print txn

        if not storeData.has_key(txn[0]):

            data = [0,0,0,0,0]

            storeData[txn[0]] = data

        else:

            data = storeData[txn[0]]

        if txn[1] == 0:

            data[0] += int(txn[3]) - int(txn[4])

        if txn[1] == 1:

            data[1] += int(txn[3]) - int(txn[4])

        if txn[2] in (RechargeOrderStatus.RECHARGE_SUCCESSFUL, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED):

            data[2] += int(txn[3])

            data[3] += int(txn[4])

            data[4] += int(txn[3]) - int(txn[4])

        storeData[txn[0]] = data

    reftxns = session.query(RechargeTransaction.storeId, RechargeTransaction.payMethod, RechargeTransaction.status, func.sum(RechargeTransaction.amount), func.sum(RechargeTransaction.discount)).filter(RechargeTransaction.status == RechargeOrderStatus.RECHARGE_FAILED_REFUNDED).filter(RechargeTransaction.responseTime >= todate).filter(RechargeTransaction.responseTime < tomorrow).group_by(RechargeTransaction.storeId, RechargeTransaction.payMethod, RechargeTransaction.status).order_by(RechargeTransaction.storeId).all()  

    for txn in reftxns:

        print txn

        if not storeData.has_key(txn[0]):

            data = [0,0,0,0,0]

            storeData[txn[0]] = data

        else:

            data = storeData[txn[0]]

        if txn[1] == 0:

            data[0] += int(txn[3]) - int(txn[4])

        if txn[1] == 1:

            data[1] += int(txn[3]) - int(txn[4])

        data[2] -= int(txn[3])

        data[3] -= int(txn[4])

        data[4] -= int(txn[3]) - int(txn[4])

    print storeData

    push_recharge_collection_to_ocr(storeData, todate)

def push_recharge_collection_to_ocr(storeData, todate):

    for storeId in storeData.keys():

        store = HotspotStore.get_by(id = storeId)

        store.collectedAmount = 0

        store.availableLimit = store.creditLimit

        session.commit()

        data = storeData.get(storeId)

        store_string = "<Store>" + store.hotspotId + "</Store>"

        date_string = "<ReconDate>" + todate.strftime("%Y%m%d") + "</ReconDate>"

        cash_string = "<Cash>" + str(data[0]) + "</Cash>"

        card_string = "<Hdfc>" + str(data[1]) + "</Hdfc>"

        type_string = "<Type>RechargeSale</Type>"

        amount_string = "<GrossRechargeAmount>" + str(data[2]) + "</GrossRechargeAmount><Discount>" + str(data[3]) +  "</Discount><NetCollection>" + str(data[4]) + "</NetCollection>";

        #SaholicRechargeSaleTransfer(string Store, int ReconDate, decimal Cash, decimal Hdfc, string Type, decimal GrossRechargeAmount, decimal Discount, decimal NetCollection)

        conn = httplib.HTTPConnection("182.71.104.182")

        XML="""

        <soap:Envelope xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:soap="http://schemas.xmlsoap.org/soap/envelope/">

          <soap:Body>

            <SaholicRechargeSaleTransfer xmlns="http://tempuri.org/">

        """

        XML = XML + store_string + date_string + cash_string + card_string + type_string + amount_string 

        footer = """

            </SaholicRechargeSaleTransfer>

          </soap:Body>

        </soap:Envelope>

        """

        XML = XML + footer

        print XML

        params = urllib.urlencode({'op': 'SaholicRechargeSaleTransfer'})

        headers = { "Content-type": "text/xml", "Content-Length": "%d" % len(XML)}

        conn.request("POST", "/loaddetect/Service.asmx?"+params, "", headers)

        conn.send(XML)

        response = conn.getresponse()

        print response.status, response.reason

        resp = response.read()

        conn.close()

        print resp

def topup_company_wallet(companyId, amount):

    wallet = WalletForCompany.query.filter(WalletForCompany.id == companyId).with_lockmode("update").one()

    wh = WalletHistoryForCompany()

    wh.walletId = wallet.id

    wh.openingBal = wallet.amount

    wh.closingBal = wallet.amount +  amount

    wh.amount = amount

    wh.transactionTime = datetime.datetime.now()

    wh.referenceNumber =  get_next_invoice_number(5)

    wh.description = "Wallet Credited"

    wallet.amount += amount

    session.commit()

if __name__ == '__main__':

    main()