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

from shop2020.utils.EmailAttachmentSender import mail, mail_html

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

    import os

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

from shop2020.clients.HelperClient import HelperClient

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

    OrderType

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, RechargeCollection, Company, HotspotServiceMatrix

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

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

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:

        wallet = WalletForCompany.query.filter(WalletForCompany.id == 3).one()

        oldBalance = wallet.amount

        newBalance = getBalance()  

        wallet.amount = newBalance

        session.commit()

        d = datetime.datetime.now()

        d = d + timedelta(days = -1)

        compute_recharge_collection(d)

        compute_website_recharge_collection(d, oldBalance, newBalance)

    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)

                if not order:

                    continue

                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, RechargeOrderStatus.RECHARGE_UNKNOWN):

                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)

            else:

                update_recharge_order_status(order.id, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED)

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"

    ## For store transactions

    rorders = RechargeTransaction.query.filter(RechargeTransaction.status.in_([RechargeOrderStatus.RECHARGE_UNKNOWN, RechargeOrderStatus.INIT])).all()

    for order in rorders:

        try:

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

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

                print status, description

                if status:

                    update_recharge_transaction_status(order.id, RechargeOrderStatus.RECHARGE_SUCCESSFUL)

                elif order.status == RechargeOrderStatus.INIT:

                    update_recharge_transaction_status(order.id, RechargeOrderStatus.RECHARGE_FAILED)

                else:

                    update_recharge_transaction_status(order.id, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED)

        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, RechargeOrderStatus.RECHARGE_UNKNOWN])).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] == 1:

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

        if txn[1] == 2:

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

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

            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] == 1:

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

        if txn[1] == 2:

            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

    wallet = WalletForCompany.query.filter(WalletForCompany.id == 1).one()    

    dt = session.query(func.sum(RechargeTransaction.amount)).filter(RechargeTransaction.status.in_([RechargeOrderStatus.RECHARGE_SUCCESSFUL])).one()

#    if int(dt[0]) != wallet.amount:

#        mail("cnc.center@shop2020.in", "5h0p2o2o", ["rajveer.singh@shop2020.in", "anupam.singh@shop2020.in"], "Wallet amount: " + str(wallet.amount) + " does not match with transaction amount: " + str(int(dt[0])) , "", [], [], [])    

    maildata = "<html><body><table border='1'><thead><th>StoreId</th><th>Gross</th><th>Discount</th><th>Net</th></thead><tbody>"

    trecharge = 0

    hotspotServiceMatrices = HotspotServiceMatrix.query.all()

    hotspotServiceMatrixMap = {}

    for hotspotServiceMatrix in hotspotServiceMatrices:

        hotspotServiceMatrixMap[hotspotServiceMatrix.storeId] = hotspotServiceMatrix

    for storeId in storeData.keys():

        store = HotspotStore.get_by(id = storeId)

        if hotspotServiceMatrixMap.has_key(storeId):

            del hotspotServiceMatrixMap[storeId]

        store.collectedAmount = 0

        store.availableLimit = store.creditLimit

        session.commit()

        data = storeData.get(storeId)

        rc = RechargeCollection()

        rc.hotspotId = store.hotspotId

        rc.reconDate = int(todate.strftime("%Y%m%d"))

        rc.cash = data[0]

        rc.hdfc = data[1]

        rc.grossAmount = data[2]

        rc.discount = data[3]

        rc.netCollection = data[4]

        rc.addedAt = datetime.datetime.now()

        rc.pushedToOcr = False

        session.commit()

        maildata += "<tr><td>" + store.hotspotId + "</td><td>" + str(data[2]) + "</td><td>" + str(data[3]) + "</td><td>" + str(data[4]) + "</td></tr>"

        trecharge +=  data[2]

    dit = session.query(func.sum(RechargeCollection.grossAmount)).one()

    dit2 = session.query(func.sum(WalletHistoryForCompany.amount)).filter(WalletHistoryForCompany.walletId == 1).filter(WalletHistoryForCompany.amount >= 100000).one()

    wamt = int(dit2[0])- int(dit[0])

    wallet.amount = wamt

    session.commit()

    maildata += "</tbody></table></body></html>"

    mail_html("cnc.center@shop2020.in", "5h0p2o2o", ["rajveer.singh@shop2020.in", "anupam.singh@shop2020.in", "Ashwani.Kumar@spiceretail.co.in","parveen.mittal@spiceretail.co.in","pardeep.panwar@spiceretail.co.in","gagan.sharma@spiceretail.co.in","j.p.gupta@shop2020.in", "rajneesh.arora@shop2020.in", "amit.tyagi@spiceretail.co.in"], "MIS :- SpiceRetail  (Date - " + todate.strftime("%d-%m-%Y") + ")   (Wallet Amount - " + str(wallet.amount) + ")    (Total Recharge - " + str(trecharge) + ")", maildata, []) 

    try:

        push_recharge_collection_to_ocr()

    except:

        mail_html("cnc.center@shop2020.in", "5h0p2o2o", ["rajveer.singh@shop2020.in", "anupam.singh@shop2020.in", "rajneesh.arora@shop2020.in"], "Problem while pushing recharge collection to OCR", "", [])

    finally:

        for storeId in hotspotServiceMatrixMap.keys():

            if hotspotServiceMatrixMap.get(storeId).rechargeService:

                store = HotspotStore.get_by(id = storeId)

                helper_client = HelperClient().get_client()

                helper_client.saveUserEmailForSending([store.clusterEmail], "cnc.center@shop2020.in", "No Recharge happened for store " + store.hotspotId + " yesterday", "", "NRM", "NoRechargeMail", [store.email], ["anupam.singh@shop2020.in"])

def push_recharge_collection_to_ocr():

    rcs = RechargeCollection.query.filter(RechargeCollection.pushedToOcr == False).all()

    for rc in rcs:

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

        date_string = "<ReconDate>" + str(rc.reconDate) + "</ReconDate>"

        cash_string = "<Cash>" + str(rc.cash) + "</Cash>"

        card_string = "<Hdfc>" + str(rc.hdfc) + "</Hdfc>"

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

        amount_string = "<GrossRechargeAmount>" + str(rc.grossAmount) + "</GrossRechargeAmount><Discount>" + str(rc.discount) +  "</Discount><NetCollection>" + str(rc.netCollection) + "</NetCollection>";

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

        conn = httplib.HTTPConnection("182.71.104.186")

        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

        if "Saved Successfully" in resp:

            rc.pushedAt = datetime.datetime.now()

            rc.pushedToOcr = True

            session.commit()

        elif "Error in Saving Data" in resp:

            mail_html("cnc.center@shop2020.in", "5h0p2o2o", ["rajveer.singh@shop2020.in", "anupam.singh@shop2020.in", "rajneesh.arora@shop2020.in"], "Problem while pushing recharge collection to OCR", resp, [])

        else: 

            mail_html("cnc.center@shop2020.in", "5h0p2o2o", ["rajveer.singh@shop2020.in", "anupam.singh@shop2020.in", "rajneesh.arora@shop2020.in"], "Problem while pushing recharge collection to OCR", resp, [])

def topup_company_wallet(companyId, amount):

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

    company = Company.get_by(id = companyId)

    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(OrderType.WALLETCREDIT)

    wh.description = "Wallet Credited"

    wallet.amount += amount

    session.commit()

    mail("cnc.center@shop2020.in", "5h0p2o2o", ["rajveer.singh@shop2020.in", "anupam.singh@shop2020.in", "Ashwani.Kumar@spiceretail.co.in","parveen.mittal@spiceretail.co.in","pardeep.panwar@spiceretail.co.in","gagan.sharma@spiceretail.co.in","j.p.gupta@shop2020.in", "rajneesh.arora@shop2020.in", "amit.tyagi@spiceretail.co.in"] , company.name + " wallet topped up by " +  str(amount) + " rupees.", "", [], [], [])

def compute_website_recharge_collection(cdate, oldBalance, newBalance):

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

    tomorrow = todate + timedelta(days=1)

    txns = session.query(RechargeOrder.status, func.sum(RechargeOrder.totalAmount)).filter(RechargeOrder.status.in_([RechargeOrderStatus.RECHARGE_SUCCESSFUL, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED, RechargeOrderStatus.PAYMENT_SUCCESSFUL])).filter(RechargeOrder.creationTimestamp >= todate).filter(RechargeOrder.creationTimestamp < tomorrow).group_by(RechargeOrder.status).all()

    tamount = 0

    for txn in txns:

        tamount += int(txn[1])

    reftxns = session.query(RechargeOrder.status, func.sum(RechargeOrder.totalAmount)).filter(RechargeOrder.status == RechargeOrderStatus.RECHARGE_FAILED_REFUNDED).filter(RechargeOrder.responseTimestamp >= todate).filter(RechargeOrder.responseTimestamp < tomorrow).group_by(RechargeOrder.status).all()  

    for txn in reftxns:

        tamount -= int(txn[1])

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

    d = datetime.datetime.now()

    wh = WalletHistoryForCompany()

    wh.walletId = wallet.id

    wh.openingBal = wallet.amount

    wh.closingBal = wallet.amount - tamount

    wh.amount = -tamount

    wh.transactionTime = d

    wh.referenceNumber =  int(d.strftime("%Y%m%d"))

    wh.description = "Wallet Credited"

    wallet.amount = wallet.amount - tamount

    session.commit()

    maildata = "Old Wallet Amount is : " + str(oldBalance) + "<br>New Wallet Amount is : " + str(newBalance)

    mail_html("cnc.center@shop2020.in", "5h0p2o2o", ["pardeep.panwar@spiceretail.co.in>", "amit.tyagi@spiceretail.co.in", "rajveer.singh@shop2020.in", "anupam.singh@shop2020.in", "j.p.gupta@shop2020.in", "rajneesh.arora@shop2020.in"], "MIS :- Saholic (Date - " + todate.strftime("%d-%m-%Y") + ")   (Wallet Amount - " + str(wallet.amount) + ")    (Total Recharge - " + str(tamount) + ")", maildata, []) 

if __name__ == '__main__':

    main()