WebSVN – SmartDukaan – Diff – /trunk/PyDTR/src/scripts/ProfitMandiPushNotificationSender.py

 import threading
 import time
 import datetime
 import MySQLdb
-from dtr.storage import DataService
-from dtr.storage.DataService import Pushnotifications
 from elixir import *
 from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound
 from sqlalchemy.sql import func
 from sqlalchemy.sql.expression import and_, or_, desc, not_, distinct, cast, \
     between
 import optparse
 import urllib2
 import base64
 import urllib
 import logging
+from dtr.utils.utils import get_mongo_connection
 GCM_URL = "https://android.googleapis.com/gcm/send"
 GOOGLE_API_KEY = "AIzaSyDw1qBnmxtnfR9NqBewryQ-yo3cG2ravGM"
+PENDING_PUSH_NOTIFICATION_URL = "http://45.33.50.227:3001/getPendingNotifications"
+PUSH_NOTIFICATIONS_UPDATE_URL = "http://45.33.50.227:3001/updatePushNotification/?"
 headers = {'content-type':'application/json', "authorization":"key=" + GOOGLE_API_KEY}
 aff_url_headers = {
             'User-agent':'Mozilla/5.0 (X11; Linux i686) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/37.0.2062.120 Safari/537.36',
             'Accept' : 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
             'Accept-Language' : 'en-US,en;q=0.8',
             'Connection':'keep-alive'
+        }
 db = MySQLdb.connect('localhost',"root","shop2020","dtr" )
 cursor = db.cursor()
+userGcmRegIdMap = {}
+campaignsMap = {}
-PUSH_NOTIFICATIONS_SQL = "select p.id, p.user_id, n.*, g.gcm_regid from pushnotifications p join notification_campaigns n on p.notification_campaign_id = n.id left outer join (select * from (select * from gcm_users order by id desc) as X group by user_id) as g on p.user_id = g.user_id where p.type ='pending'"
 ALL_STORES_SQL = "select * from stores"
+GCM_REG_ID_SQL = "select x.user_id, x.gcm_regid from (select * from gcm_users where user_id in %s order by id desc) as x group by x.user_id"
+CAMPAIGNS_SQL = "select * from notification_campaigns where id in %s order by id"
 cursor.execute(ALL_STORES_SQL)
 result_stores = cursor.fetchall()
 domainStoresMap = {}
 for rec in result_stores:
         handleCampaignRequest(self.name, self.recordsList)
         logging.debug('Completed')
 def handleCampaignRequest(threadName, recordsList ):
     for record in recordsList:
+        campaignRecord = campaignsMap.get(record.get('notification_campaign_id'))
-        notificationRecord = NotificationRecord(record[0], record[1], record[2], record[3], record[4], record[5], record[6], record[7], record[9], record[10], record[11], record[12])
+        notificationRecord = NotificationRecord(record.get('_id'), record.get('user_id'), record.get('notification_campaign_id'), campaignRecord[1], campaignRecord[2], campaignRecord[3], campaignRecord[4], campaignRecord[5], campaignRecord[7], campaignRecord[8], campaignRecord[9], userGcmRegIdMap.get(record.get('user_id')))
         if notificationRecord.type=='url':
             parsed_uri = urlparse(notificationRecord.url)
             domain = '{uri.netloc}'.format(uri=parsed_uri)
             logging.debug('Affiliate Domain:-'+str(domain))
             logging.debug('User Id:-'+str(notificationRecord.userId)+' And GCM Reg Id:- '+ str(notificationRecord.gcmRegId))
             logging.debug('User Id:- '+str(notificationRecord.userId)+' Post Data Json :- '+str(post_data_json))
             response = requests.post(GCM_URL, data=post_data_json, headers=headers)
             logging.debug('User Id:- '+str(notificationRecord.userId)+' Response :-'+str(response.text))
             result = json.loads(response.text)
-            pushnotification = Pushnotifications.get_by(id=notificationRecord.pushNotificationId)
             if result["success"]:
-                if pushnotification.message is None or pushnotification.message!="success":
+                update_params = { 'user_id' : notificationRecord.userId,  'notification_campaign_id' : notificationRecord.campaignId, 'type' : 'sent', 'status':1, 'message':'success' }
-                    pushnotification.type = "sent"
+                encoded_update_params = urllib.urlencode(update_params)
-                    pushnotification.status = True
+                updateReq = urllib2.Request(PUSH_NOTIFICATIONS_UPDATE_URL+encoded_update_params)
-                    pushnotification.message = "success"
+                updateResponse = urllib2.urlopen(updateReq)
-                    session.commit()
+                response_str = updateResponse.read()
+                logging.debug('User Id:- '+str(notificationRecord.userId)+' Update Response :-'+str(response_str))
             else:
+                update_params = { 'user_id' : notificationRecord.userId,  'notification_campaign_id' : notificationRecord.campaignId, 'type' : 'sent', 'status':0, 'message':result["results"][0]["error"] }
+                encoded_update_params = urllib.urlencode(update_params)
+                updateReq = urllib2.Request(PUSH_NOTIFICATIONS_UPDATE_URL+encoded_update_params)
-                pushnotification.type = "sent"
+                updateResponse = urllib2.urlopen(updateReq)
-                pushnotification.status = False
+                response_str = updateResponse.read()
-                pushnotification.message = result["results"][0]["error"]
+                logging.debug('User Id:- '+str(notificationRecord.userId)+' Update Response :-'+str(response_str))
                 updateGcmUserSql = "update gcm_users set failurecount=failurecount+1 where gcm_regid='%s'"%(notificationRecord.gcmRegId)
                 logging.debug('Update GCM User Query :-'+str(updateGcmUserSql))
                 try:
                     dtrdb = MySQLdb.connect('localhost',"root","shop2020","dtr" )
     """Yield successive n-sized chunks from l."""
     for i in xrange(0, len(l), n):
         yield l[i:i+n]
 def main():
+    global userGcmRegIdMap
+    global campaignsMap
     parser = optparse.OptionParser()
     parser.add_option("-C", "--chunksize", dest="chunksize",
                       default="100",
                       type="int", help="The requsets a single thread handles",
                       metavar="CHUNKSIZE")
     (options, args) = parser.parse_args()
+    notificationListReq = urllib2.Request(PENDING_PUSH_NOTIFICATION_URL)
+    notificationListResponse = urllib2.urlopen(notificationListReq)
+    notificationListJson = json.loads(notificationListResponse.read())
+    usersList =[]
+    campaignIdList = []
+    for notificationRec in notificationListJson:
+        if notificationRec.get('user_id') not in usersList:
+            usersList.append(notificationRec.get('user_id'))
+        if notificationRec.get('notification_campaign_id') not in campaignIdList:
+            campaignIdList.append(notificationRec.get('notification_campaign_id'))
+    usersList.append(0)
+    campaignIdList.append(0)
-    cursor.execute(PUSH_NOTIFICATIONS_SQL)
+    cursor.execute(GCM_REG_ID_SQL%(str(tuple(usersList))))
     result_data = cursor.fetchall()
+    for dataRec in result_data:
+        userGcmRegIdMap[dataRec[0]] = dataRec[1]
+    cursor.execute(CAMPAIGNS_SQL%(str(tuple(campaignIdList))))
+    campaign_data = cursor.fetchall()
+    for campaignRec in campaign_data:
+        campaignsMap[campaignRec[0]] = campaignRec
     count = 1
-    DataService.initialize(db_hostname="localhost")
     if result_data:
-        campaign_receivers_list = list(chunks(result_data, options.chunksize))
+        campaign_receivers_list = list(chunks(notificationListJson, options.chunksize))
         print len(campaign_receivers_list)
         for sublist in campaign_receivers_list:
             thread = NotificationThread(count, "Thread-"+str(count), sublist)
             thread.start()
             count = count +1

Subversion Repositories SmartDukaan

(root)/trunk/PyDTR/src/scripts/ProfitMandiPushNotificationSender.py – Rev 16386 → 17228