Subversion Repositories SmartDukaan

import time
from datetime import datetime, timedelta

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

PUSH_NOTIFICATIONS_RETRY_SQL ="select  x.user_id, x.id, x.created, x.notification_campaign_id, x.name, TIMESTAMPDIFF(HOUR, x.created, x.expiresat) tdiff, x.sent-IF (y.received is null, 0, y.received) failureCount, x.last_active, x.expiresat from (select p.id,p.user_id, p.created, p.notification_campaign_id, n.name, count(*) sent, ua.last_active, n.expiresat from pushnotifications p join notification_campaigns n on p.notification_campaign_id = n.id join useractive ua on p.user_id = ua.user_id join (select * from (select * from devices order by id desc) dev group by user_id) d on p.user_id = d.user_id where p.notification_campaign_id in (select id from notification_campaigns where TIMESTAMPDIFF(HOUR, created, NOW())<=2 and (name not like 'Batch Credit%' and name not like '%copy%')) and p.type='sent' and p.message = 'success' and d.versioncode >=14 and ua.last_active >= p.created and ua.last_active <= n.expiresat group by p.id) as x left join (select p.id, p.user_id, p.created, p.notification_campaign_id, n.name, count(*) received, ua.last_active, n.expiresat from pushnotifications p join notification_campaigns n on p.notification_campaign_id = n.id join useractive ua on p.user_id = ua.user_id join (select * from (select * from devices order by id desc) dev group by user_id) d on p.user_id = d.user_id where p.notification_campaign_id in (select id from notification_campaigns where TIMESTAMPDIFF(HOUR, created, NOW())<=2 and (name not like 'Batch Credit%' and name not like '%copy%')) and d.versioncode >=14 and p.type='recieved' group by p.id) as y on (x.user_id = y.user_id and x.notification_campaign_id = y.notification_campaign_id) having failureCount =1"

db = MySQLdb.connect('localhost',"root","shop2020","dtr" )
cursor = db.cursor()
DataService.initialize(db_hostname="localhost")

cursor.execute(PUSH_NOTIFICATIONS_RETRY_SQL)
retry_records_data = cursor.fetchall()

campaignUsersMap = {}
campaignNamesMap = {}
campaignTimeDiffMap = {}

if retry_records_data:
    for record in retry_records_data:
        if campaignUsersMap.has_key(long(record[3])):
            users = campaignUsersMap.get(long(record[3]))
            users.append(long(record[0]))
            campaignUsersMap[long(record[3])] = users
        else:
            users = []
            users.append(long(record[0]))
            campaignUsersMap[long(record[3])] = users
            
        if not campaignNamesMap.has_key(long(record[3])):
            name = str(record[4])
            campaignNamesMap[long(record[3])] = name
        
        if not campaignTimeDiffMap.has_key(long(record[3])):
            diff = long(record[5])
            campaignTimeDiffMap[long(record[3])] = diff

print campaignUsersMap

for campaignId, users in campaignUsersMap.iteritems():
    name = campaignNamesMap.get(campaignId)+"- copy"
    created = datetime.now()
    expiresat = created+timedelta(hours = campaignTimeDiffMap.get(campaignId))
    print created, expiresat
    campaignInsertSql = "insert into notification_campaigns(name, title, message, type, url, notification_campaigns.sql, expiresat, status, created) select '%s', title, message, type, url, notification_campaigns.sql, '%s', status, '%s' from notification_campaigns where id = %d"%(name, expiresat.strftime('%Y-%m-%d %H:%M:%S'), created.strftime('%Y-%m-%d %H:%M:%S'), campaignId)
    print campaignInsertSql
    cursor.execute(campaignInsertSql)
    db.commit()
    insertedCampaignId = cursor.lastrowid
    
    for user in users:
        pushnotification = Pushnotifications()
        pushnotification.user_id = user
        pushnotification.notification_campaign_id = insertedCampaignId
        pushnotification.type = 'pending'
        pushnotification.status = False
        pushnotification.created = datetime.now()
        session.commit()
        
db.close()
if session.is_active:
    print "session is active. closing it."
    session.close()