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