Subversion Repositories SmartDukaan

Rev

Rev 7144 | Rev 7148 | Go to most recent revision | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 7144 Rev 7147
Line 20... Line 20...
20 
    import os
 20 
    import os
21 
    sys.path.insert(0, os.getcwd())
 21 
    sys.path.insert(0, os.getcwd())
22 
 
 22 
 
23 
from shop2020.thriftpy.model.v1.order.ttypes import RechargeOrderStatus
 23 
from shop2020.thriftpy.model.v1.order.ttypes import RechargeOrderStatus
24 
from shop2020.model.v1.order.impl.DataAccessors import get_recharge_orders_for_status, update_recharge_order_status,\
 24 
from shop2020.model.v1.order.impl.DataAccessors import get_recharge_orders_for_status, update_recharge_order_status,\
25 
    update_amount_in_wallet, update_recharge_transaction_status
 25 
    update_recharge_transaction_status, get_next_invoice_number
26 
from shop2020.model.v1.order.impl import DataService
 26 
from shop2020.model.v1.order.impl import DataService
27 
from shop2020.model.v1.order.impl.DataService import RechargeTransaction, HotspotStore
 27 
from shop2020.model.v1.order.impl.DataService import RechargeTransaction, HotspotStore,\
- 
 
 28 
    WalletForCompany, WalletHistoryForCompany
28 
from shop2020.model.v1.order.impl.model.RechargeOrder import RechargeOrder
 29 
from shop2020.model.v1.order.impl.model.RechargeOrder import RechargeOrder
29 
from shop2020.model.v1.order.impl.RechargeService import checkTransactionStatus, getRefunds
 30 
from shop2020.model.v1.order.impl.RechargeService import checkTransactionStatus, getRefunds
30 
 
 31 
 
31 
 
 32 
 
32 
def main():
 33 
def main():
Line 45... Line 46...
45 
                      action="store_true",
 46 
                      action="store_true",
46 
                      help="")
 47 
                      help="")
47 
    parser.add_option("-c", "--collection", dest="collection",
 48 
    parser.add_option("-c", "--collection", dest="collection",
48 
                      action="store_true",
 49 
                      action="store_true",
49 
                      help="")
 50 
                      help="")
- 
 
 51 
    parser.add_option("-T", "--topup", dest="topup",
- 
 
 52 
                      action="store_true",
- 
 
 53 
                      help="")
50 
    parser.add_option("-t", "--txn-id", dest="txn_id",
 54 
    parser.add_option("-t", "--txn-id", dest="txn_id",
51 
                   type="int",
 55 
                   type="int",
52 
                   help="mark the transaction(recharge order id) TXN_ID as successful",
 56 
                   help="mark the transaction(recharge order id) TXN_ID as successful",
53 
                   metavar="TXN_ID")
 57 
                   metavar="TXN_ID")
54 
 
 58 
 
Line 65... Line 69...
65 
        processAuthorizedTransactions(options.txn_id)
 69 
        processAuthorizedTransactions(options.txn_id)
66 
    if options.collection:
 70 
    if options.collection:
67 
        d = datetime.datetime.now()
 71 
        d = datetime.datetime.now()
68 
        d = d + timedelta(days = -1)
 72 
        d = d + timedelta(days = -1)
69 
        compute_recharge_collection(d)
 73 
        compute_recharge_collection(d)
- 
 
 74 
    if options.topup:
- 
 
 75 
        topup_company_wallet(1,100000)
70 
 
 76 
 
71 
def processRefunds():
 77 
def processRefunds():
72 
    todate = datetime.datetime.now()
 78 
    todate = datetime.datetime.now()
73 
    for i in range(15):
 79 
    for i in range(15):
74 
        orderDate = todate + datetime.timedelta(days= -i)
 80 
        orderDate = todate + datetime.timedelta(days= -i)
Line 203... Line 209...
203 
        print response.status, response.reason
 209 
        print response.status, response.reason
204 
        resp = response.read()
 210 
        resp = response.read()
205 
        conn.close()
 211 
        conn.close()
206 
        print resp
 212 
        print resp
207 
 
 213 
 
- 
 
 214 
def topup_company_wallet(companyId, amount):
- 
 
 215 
    wallet = WalletForCompany.query.filter(WalletForCompany.id == companyId).with_lockmode("update").one()
- 
 
 216 
 
- 
 
 217 
    wh = WalletHistoryForCompany()
- 
 
 218 
    wh.walletId = wallet.id
- 
 
 219 
    wh.openingBal = wallet.amount
- 
 
 220 
    wh.closingBal = wallet.amount +  amount
- 
 
 221 
    wh.amount = amount
- 
 
 222 
    wh.transactionTime = datetime.datetime.now()
- 
 
 223 
    wh.referenceNumber =  get_next_invoice_number(5)
- 
 
 224 
    wh.description = "Wallet Credited"
- 
 
 225
- 
 
 226 
    wallet.amount += amount
- 
 
 227 
    session.commit()
- 
 
 228
208 
if __name__ == '__main__':
 229 
if __name__ == '__main__':
209 
    main()
 230 
    main()
210 

            231