WebSVN – SmartDukaan – Diff – //trunk/PyDTR/src/dtr/main.py

             'Accept-Language' : 'en-US,en;q=0.8',
             'Accept-Charset' : 'ISO-8859-1,utf-8;q=0.7,*;q=0.3'
+        }
 CASHBACK_URL = 'http://api.profittill.com/cashbacks/index/%s/%s'
 USER_LOOKUP_URL = 'http://api.profittill.com/user_account/saholic/%s'
-WALLET_CREDIT_URL = 'http://www.shop2020.in:8080/mobileapi/wallet!batchUpdate'
+WALLET_CREDIT_URL = 'http://localhost:8080/mobileapi/wallet!batchUpdate'
 def getStore(source_id):
     #module = sourceMap[source_id]
     store = Store(source_id)
     try:
     ORDER_SHIPPED = 'Shipped' #Lets see if we can make use of it
     ORDER_CANCELLED = 'Cancelled'
     CB_INIT = 'Waiting Confirmation'
     CB_PENDING = 'Pending'
+    CB_CREDIT_IN_PROCESS = 'Credited to wallet'
     CB_CREDITED = 'Credited to wallet'
     CB_NA = 'Not Applicable'
     CB_APPROVED = 'Approved'
     CB_CANCELLED = 'Cancelled'
         pass
     def _updateOrdersPayBackStatus(self, searchMap, updateMap):
         searchMap['subOrders.missingAff'] = False
         updateMap['subOrders.$.missingAff'] = True
-        self.db.merchantOrder.update(searchMap, { '$set': updateMap })
+        self.db.merchantOrder.update(searchMap, { '$set': updateMap }, multi=True)
     def _getActiveOrders(self, searchMap={}, collectionMap={}):
         collection = self.db.merchantOrder
         searchMap = dict(searchMap.items()+ {"closed": False, "storeId" : self.store_id}.items())
         collectionMap =  dict(collectionMap.items() + {"orderSuccessUrl":1, "orderId":1,"subOrders":1, "placedOn":1}.items())
             if merchantSubOrderId == subOrder.get("merchantSubOrderId"):
                 return subOrder
         return None
 def settlePayBack():
+        client.Dtr.merchantOrder.update({'subOrders.cashBackStatus':Store.CB_APPROVED},{'subOrders.$.cashBackStatus':Store.CB_CREDIT_IN_PROCESS}, multi=True)
         result = client.Dtr.merchantOrder\
             .aggregate([
-                        {'$match':{'subOrders.cashBackStatus':Store.CB_APPROVED}},
+                        {'$match':{'subOrders.cashBackStatus':Store.CB_CREDIT_IN_PROCESS}},
                         {'$unwind':"$subOrders"},
+                        {
                          '$group':{
                                    '_id':'$userId',
                                    'amount': { '$sum':'$subOrders.cashBackAmount'},
+                         }
                     ])['result']
         userAmountMap = {}
         for res in result:
             userAmountMap[res['_id']] = res['amount']
+        datetimeNow = datetime.now()
-        batchId = int(time.mktime(datetime.now().timetuple()))
+        batchId = int(time.mktime(datetimeNow.timetuple()))
-        batchUpdateMap = {'batchId':batchId}
+        if refundToWallet(batchId, userAmountMap):
-        batchUpdateMap['userAmount'] = result
+            client.Dtr.merchantOrder.update({'subOrders.cashBackStatus':Store.CB_CREDIT_IN_PROCESS},{'subOrders.$.cashBackStatus':Store.CB_CREDITED, 'subOrders.$.batchId':batchId}, multi=True)
-        #for
-        #get batch id
+            for key, value in userAmountMap:
-        print batchId
-        if __refundToWallet(batchUpdateMap):
+                client.Dtr.refund.insert({"userId": key, "batch":batchId, "userAmount":value, "timestamp":datetime.strftime(datetimeNow,"%Y-%m-%d %H:%M:%S")})
-            client.Dtr.merchantOrder.update({'subOrders.cashBackStatus':Store.CB_APPROVED},{'subOrders.$.cashBackStatus':Store.CB_CREDITED}, multi=True)
+                client.Dtr.user.update({"userId":key}, {'$inc': { "credited": value}}, upsert=True)
         else:
-            tprint("Error Occurred while running batch")
+            tprint("Error Occurred while running batch.. Rolling Back")
+            client.Dtr.merchantOrder.update({'subOrders.cashBackStatus':Store.CB_CREDIT_IN_PROCESS},{'subOrders.$.cashBackStatus':Store.CB_APPROVED}, multi=True)
+def refundToWallet(batchId, userAmountMap):
+    batchUpdateMap = {}
+    try :
+        saholicUserAmountMap = {}
+        for key, value in userAmountMap.iteritems():
+            userLookupRequest = urllib2.Request(USER_LOOKUP_URL %(key), headers=headers)
+            try:
+                response = urllib2.urlopen(userLookupRequest).read()
+                saholicUserId = json.loads(response)['account_id']
+                saholicUserAmountMap[saholicUserId] = value
+            except:
+                tprint("Could not fetch saholic id for user : " + key)
+                continue
+        batchUpdateMap['userAmount'] = json.dumps(saholicUserAmountMap)
+        batchUpdateMap['batchId'] = batchId
+        request = urllib2.Request(WALLET_CREDIT_URL, headers=headers)
+        data = urllib.urlencode(batchUpdateMap)
+        response = urllib2.urlopen(request, data)
+        return json.loads(response.read())['response']['credited']
+    except:
+        traceback.print_exc()
+        tprint("Could not batch refund")
+        return False
+    #return json.loads(response.read())['credited']
 def main():
     #store = getStore(3)
     #print store.getCashbackAmount('864683341', 100)
     #data = urllib.urlencode({'orderId':6000, 'amount':200})
     #request = urllib2.Request(WALLET_CREDIT_URL % (483649), headers=headers)
     #response = urllib2.urlopen(request, data)
     #print response.read()
-    settlePayBack()
+    #settlePayBack()
+    batchId =1
+    userAmount = {483649:3.0}
+    refundToWallet(batchId, userAmount)
 if __name__ == '__main__':
     main()
 ###
 #Settlement process is suposed to be a batch and run weekly
 #It is should be running on first hour of mondays. As cron should
 # Maintain a batch id.
-def __refundToWallet(self, batchUpdateMap):
-    result = False
-    try :
-        userAmountMap = batchUpdateMap['userAmount']
-        saholicUserAmountMap = {}
-        for key, value in userAmountMap:
-            userLookupRequest = urllib2.Request(USER_LOOKUP_URL %(key), headers=headers)
-            response = urllib2.urlopen(userLookupRequest).read()
-            saholicUserId = json.loads(response)['account_id']
-            saholicUserAmountMap[saholicUserId] = value
-        batchUpdateMap['userAmount'] = saholicUserAmountMap
-        request = urllib2.Request(WALLET_CREDIT_URL, headers=headers)
-        data = urllib.urlencode(batchUpdateMap)
-        response = urllib2.urlopen(request, data)
-    except:
-        traceback.print_exc()
-        return False
-    return json.loads(response.read())['credited']
 def getBrowserObject():
     import cookielib
     br = mechanize.Browser(factory=mechanize.RobustFactory())
     cj = cookielib.LWPCookieJar()
         html = gz.read()
         gz.close()
         return html
 def tprint(*msg):
-    print datetime.now(), "-", msg
     print datetime.now(), "-", msg

Subversion Repositories SmartDukaan

(root)//trunk/PyDTR/src/dtr/main.py – Rev 13869 → 13927