Subversion Repositories SmartDukaan

Rev

Rev 19902 | Rev 20011 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
14305 amit.gupta 1
from bson.objectid import ObjectId
2
from datetime import datetime, timedelta
3
from dtr.config import PythonPropertyReader
19451 manas 4
from dtr.dao import FeaturedDeals, AppTransactions, AppOfferObj, Promotion, NotificationCampaign,CrmRefundWallet
14037 kshitij.so 5
from dtr.storage import DataService
6
from dtr.storage.DataService import price_preferences, brand_preferences, \
16789 amit.gupta 7
    user_actions, Brands, app_offers, approved_app_transactions, user_app_cashbacks, \
19577 manas 8
    user_app_installs, appmasters, notification_campaigns, user_accounts
16304 amit.gupta 9
from dtr.storage.MemCache import MemCache
16789 amit.gupta 10
from dtr.utils.MailSender import Email
11
from dtr.utils.utils import to_java_date, CB_PENDING, CB_APPROVED, CB_INIT, \
19556 manas 12
    to_py_date, CB_REJECTED, SUB_CATEGORY_MAP,CREDIT_TYPE_REFUND,CREDIT_TYPE_OFFER,CREDIT_TYPE_ADJUSTMENT,REFUND_ADJUSTMENT_MAP, todict,REVERSE_SOURCE_MAP,\
19654 kshitij.so 13
    get_mongo_connection_dtr_data, SUB_CATEGORY_HEADER_RANKING
16894 manish.sha 14
from sqlalchemy.sql.expression import func, func, or_, desc, asc, case
14305 amit.gupta 15
from elixir import *
16
from operator import itemgetter
16789 amit.gupta 17
from pymongo.command_cursor import CommandCursor
14305 amit.gupta 18
import pymongo
16398 amit.gupta 19
import random
14322 kshitij.so 20
import re
16789 amit.gupta 21
import time
14791 kshitij.so 22
import traceback
16894 manish.sha 23
from itertools import groupby
17280 manish.sha 24
import urllib
25
import urllib2
26
import json
17367 kshitij.so 27
import collections
19451 manas 28
from string import lower
17367 kshitij.so 29
try:
30
    import ordereddict
31
except:
32
    pass
13572 kshitij.so 33
 
18830 kshitij.so 34
 
35
import bson.son as son
19207 kshitij.so 36
import math
37
 
13572 kshitij.so 38
con = None
17944 kshitij.so 39
es = None
18052 kshitij.so 40
try:
41
    from elasticsearch import Elasticsearch
42
except:
43
    pass
17936 kshitij.so 44
from shop2020.config.client.ConfigClient import ConfigClient
13572 kshitij.so 45
 
14037 kshitij.so 46
DataService.initialize(db_hostname="localhost")
16304 amit.gupta 47
mc = MemCache("127.0.0.1")
17936 kshitij.so 48
config_client = ConfigClient()
49
elastic_search_host = config_client.get_property('elastic_search_host')
50
elastic_search_port = config_client.get_property('elastic_search_port')
14037 kshitij.so 51
 
17936 kshitij.so 52
 
17944 kshitij.so 53
 
17035 kshitij.so 54
SOURCE_MAP = {1:'AMAZON',2:'FLIPKART',3:'SNAPDEAL',4:'SAHOLIC',5:"SHOPCLUES.COM",6:"PAYTM.COM",7:"HOMESHOP18.COM"}
14482 kshitij.so 55
 
15853 kshitij.so 56
COLLECTION_MAP = {
57
                  'ExceptionalNlc':'skuBundleId',
58
                  'SkuDealerPrices':'skuBundleId',
59
                  'SkuDiscountInfo':'skuBundleId',
60
                  'SkuSchemeDetails':'skuBundleId',
17367 kshitij.so 61
                  'DealPoints':'skuBundleId',
62
                  'FeaturedDeals': 'skuBundleId'
15853 kshitij.so 63
                  }
64
 
19337 manish.sha 65
 
17944 kshitij.so 66
def get_elastic_search_connection():
67
    global es
68
    if es is None:
69
        print "Establishing connection with elastic search %s host and port %s"%(elastic_search_host,elastic_search_port)
70
        es = Elasticsearch([{'host': elastic_search_host, 'port': elastic_search_port}])
71
        return es
72
    return es
73
 
74
 
13572 kshitij.so 75
def get_mongo_connection(host='localhost', port=27017):
76
    global con
77
    if con is None:
78
        print "Establishing connection %s host and port %d" %(host,port)
79
        try:
80
            con = pymongo.MongoClient(host, port)
81
        except Exception, e:
82
            print e
83
            return None
84
    return con
85
 
13907 kshitij.so 86
def populateCashBack():
13921 kshitij.so 87
    print "Populating cashback"
88
    cashBackMap = {}
89
    itemCashBackMap = {}
13907 kshitij.so 90
    cashBack = list(get_mongo_connection().Catalog.CategoryCashBack.find())
91
    for row in cashBack:
13970 kshitij.so 92
        temp_map = {}
93
        temp_list = []
13907 kshitij.so 94
        if cashBackMap.has_key(row['source_id']):
95
            arr = cashBackMap.get(row['source_id'])
96
            for val in arr:
97
                temp_list.append(val)
98
            temp_map[row['category_id']] = row
99
            temp_list.append(temp_map)
100
            cashBackMap[row['source_id']] = temp_list 
101
        else:
102
            temp_map[row['category_id']] = row
103
            temp_list.append(temp_map)
104
            cashBackMap[row['source_id']] = temp_list
13921 kshitij.so 105
    itemCashBack = list(get_mongo_connection().Catalog.ItemCashBack.find())
106
    for row in itemCashBack:
107
        if not itemCashBackMap.has_key(row['skuId']):
108
            itemCashBackMap[row['skuId']] = row
14761 kshitij.so 109
    mc.set("item_cash_back", itemCashBackMap, 24 * 60 * 60)
110
    mc.set("category_cash_back", cashBackMap, 24 * 60 * 60)
13907 kshitij.so 111
 
13572 kshitij.so 112
def addCategoryDiscount(data):
13970 kshitij.so 113
    collection = get_mongo_connection().Catalog.CategoryDiscount
13572 kshitij.so 114
    query = []
115
    data['brand'] = data['brand'].strip().upper()
13970 kshitij.so 116
    data['discountType'] = data['discountType'].upper().strip()
13572 kshitij.so 117
    query.append({"brand":data['brand']})
118
    query.append({"category_id":data['category_id']})
119
    r = collection.find({"$and":query})
120
    if r.count() > 0:
13639 kshitij.so 121
        return {0:"Brand & Category info already present."}
13572 kshitij.so 122
    else:
123
        collection.insert(data)
13970 kshitij.so 124
        get_mongo_connection().Catalog.MasterData.update({'brand':data['brand'],'category_id':data['category_id']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
13639 kshitij.so 125
        return {1:"Data added successfully"}
13572 kshitij.so 126
 
13970 kshitij.so 127
def updateCategoryDiscount(data,_id):
128
    try:
129
        collection = get_mongo_connection().Catalog.CategoryDiscount
130
        collection.update({'_id':ObjectId(_id)},{"$set":{'min_discount':data['min_discount'],'max_discount':data['max_discount'],'discountType':data['discountType'].upper().strip()}},upsert=False, multi = False)
131
        get_mongo_connection().Catalog.MasterData.update({'brand':data['brand'],'category_id':data['category_id']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
132
        return {1:"Data updated successfully"}
133
    except:
134
        return {0:"Data not updated."}
135
 
136
 
13572 kshitij.so 137
def getAllCategoryDiscount():
138
    data = []
13970 kshitij.so 139
    collection = get_mongo_connection().Catalog.CategoryDiscount
13572 kshitij.so 140
    cursor = collection.find()
141
    for val in cursor:
142
        data.append(val)
143
    return data
144
 
14553 kshitij.so 145
def __getBundledSkusfromSku(sku):
146
    masterData =  get_mongo_connection().Catalog.MasterData.find_one({"_id":sku},{"skuBundleId":1})
147
    if masterData is not None:
148
        return list(get_mongo_connection().Catalog.MasterData.find({"skuBundleId":masterData.get('skuBundleId')},{'_id':1,'skuBundleId':1}))
149
    else:
150
        return []
13572 kshitij.so 151
 
15853 kshitij.so 152
def addSchemeDetailsForSku(data):
153
    collection = get_mongo_connection().Catalog.SkuSchemeDetails
154
    result = collection.find_one({'skuBundleId':data['skuBundleId']})
155
    if result is None:
14553 kshitij.so 156
        collection.insert(data)
15853 kshitij.so 157
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
14553 kshitij.so 158
        return {1:"Data added successfully"}
15853 kshitij.so 159
    return {1:"BundleId info already present"}
160
 
14069 kshitij.so 161
def getAllSkuWiseSchemeDetails(offset, limit):
13572 kshitij.so 162
    data = []
13970 kshitij.so 163
    collection = get_mongo_connection().Catalog.SkuSchemeDetails
14071 kshitij.so 164
    cursor = collection.find().skip(offset).limit(limit)
13572 kshitij.so 165
    for val in cursor:
15853 kshitij.so 166
        master = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':val['skuBundleId']})
167
        if master is not None:
168
            val['brand'] = master['brand']
169
            val['source_product_name'] = master['source_product_name']
14069 kshitij.so 170
        else:
171
            val['brand'] = ""
172
            val['source_product_name'] = ""
13572 kshitij.so 173
        data.append(val)
174
    return data
175
 
15853 kshitij.so 176
def addSkuDiscountInfo(data):
177
    collection = get_mongo_connection().Catalog.SkuDiscountInfo
178
    cursor = collection.find_one({"skuBundleId":data['skuBundleId']})
179
    if cursor is not None:
180
        return {0:"BundleId information already present."}
13572 kshitij.so 181
    else:
15853 kshitij.so 182
        collection.insert(data)
183
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
13639 kshitij.so 184
        return {1:"Data added successfully"}
13572 kshitij.so 185
 
13970 kshitij.so 186
def getallSkuDiscountInfo(offset, limit):
13572 kshitij.so 187
    data = []
13970 kshitij.so 188
    collection = get_mongo_connection().Catalog.SkuDiscountInfo
189
    cursor = collection.find().skip(offset).limit(limit)
13572 kshitij.so 190
    for val in cursor:
15853 kshitij.so 191
        master = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':val['skuBundleId']})
192
        if master is not None:
193
            val['brand'] = master['brand']
194
            val['source_product_name'] = master['source_product_name']
13970 kshitij.so 195
        else:
196
            val['brand'] = ""
197
            val['source_product_name'] = ""
13572 kshitij.so 198
        data.append(val)
199
    return data
200
 
13970 kshitij.so 201
def updateSkuDiscount(data,_id):
202
    try:
203
        collection = get_mongo_connection().Catalog.SkuDiscountInfo
204
        collection.update({'_id':ObjectId(_id)},{"$set":{'min_discount':data['min_discount'],'max_discount':data['max_discount'],'discountType':data['discountType'].upper().strip()}},upsert=False, multi = False)
15853 kshitij.so 205
        get_mongo_connection().Catalog.MasterData.update({'_id':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
13970 kshitij.so 206
        return {1:"Data updated successfully"}
207
    except:
208
        return {0:"Data not updated."}
209
 
210
 
15853 kshitij.so 211
def addExceptionalNlc(data):
212
    collection = get_mongo_connection().Catalog.ExceptionalNlc
213
    cursor = collection.find_one({"skuBundleId":data['skuBundleId']})
214
    if cursor is not None:
215
        return {0:"BundleId information already present."}
13572 kshitij.so 216
    else:
15853 kshitij.so 217
        collection.insert(data)
218
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
13639 kshitij.so 219
        return {1:"Data added successfully"}
13572 kshitij.so 220
 
13970 kshitij.so 221
def getAllExceptionlNlcItems(offset, limit):
13572 kshitij.so 222
    data = []
13970 kshitij.so 223
    collection = get_mongo_connection().Catalog.ExceptionalNlc
14071 kshitij.so 224
    cursor = collection.find().skip(offset).limit(limit)
13572 kshitij.so 225
    for val in cursor:
15853 kshitij.so 226
        master = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':val['skuBundleId']})
227
        if master is not None:
228
            val['brand'] = master['brand']
229
            val['source_product_name'] = master['source_product_name']
13970 kshitij.so 230
        else:
231
            val['brand'] = ""
232
            val['source_product_name'] = ""
13572 kshitij.so 233
        data.append(val)
234
    return data
235
 
14005 amit.gupta 236
def getMerchantOrdersByUser(userId, page=1, window=50, searchMap={}):
237
    if searchMap is None:
238
        searchMap = {}
13603 amit.gupta 239
    if page==None:
240
        page = 1
241
 
242
    if window==None:
243
        window = 50
244
    result = {}
13582 amit.gupta 245
    skip = (page-1)*window
14005 amit.gupta 246
 
14353 amit.gupta 247
    if userId is not None:
248
        searchMap['userId'] = userId
13603 amit.gupta 249
    collection = get_mongo_connection().Dtr.merchantOrder
14609 amit.gupta 250
    cursor = collection.find(searchMap).sort("orderId",-1)
13603 amit.gupta 251
    total_count = cursor.count()
252
    pages = total_count/window + (0 if total_count%window==0 else 1)  
253
    print "total_count", total_count
254
    if total_count > skip:
13999 amit.gupta 255
        cursor = cursor.skip(skip).limit(window)
13603 amit.gupta 256
        orders = []
257
        for order in cursor:
14002 amit.gupta 258
            del(order["_id"])
259
            orders.append(order)
13603 amit.gupta 260
        result['data'] = orders
261
        result['window'] = window
262
        result['totalCount'] = total_count 
263
        result['currCount'] = cursor.count()
264
        result['totalPages'] = pages
265
        result['currPage'] = page    
266
        return result
267
    else:
268
        return result
13630 kshitij.so 269
 
13927 amit.gupta 270
def getRefunds(userId, page=1, window=10):
271
    if page==None:
272
        page = 1
273
 
274
    if window==None:
13995 amit.gupta 275
        window = 10
13927 amit.gupta 276
    result = {}
277
    skip = (page-1)*window
16737 amit.gupta 278
    con = get_mongo_connection()
279
    collection = con.Dtr.refund
280
    user = con.Dtr.user
16754 amit.gupta 281
    credited = user.find_one({"userId":userId})
282
    if credited:
283
        credited = credited.get("credited")
284
    else:
285
        credited=0
286
 
13927 amit.gupta 287
    cursor = collection.find({"userId":userId})
288
    total_count = cursor.count()
289
    pages = total_count/window + (0 if total_count%window==0 else 1)  
290
    print "total_count", total_count
291
    if total_count > skip:
14668 amit.gupta 292
        cursor = cursor.skip(skip).limit(window).sort([('batch',-1)])
13927 amit.gupta 293
        refunds = []
294
        for refund in cursor:
295
            del(refund["_id"])
19511 manas 296
            if (refund["type"]) == CREDIT_TYPE_REFUND:
19493 manas 297
                batchDetails = fetchCrmRefundByBatchId(refund["batch"])
298
                referenceNumber = batchDetails.get('reference_no')                
19511 manas 299
                refund["description"] = "Refunded against Saholic order #%s"%(referenceNumber)
300
            elif (refund["type"]) == CREDIT_TYPE_ADJUSTMENT:
19493 manas 301
                batchDetails = fetchCrmRefundByBatchId(refund["batch"])
302
                referenceNumber = batchDetails.get('reference_no')
19517 manas 303
                referenceStore = REVERSE_SOURCE_MAP.get(batchDetails.get('store'))
19784 manas 304
                refund["description"] = "Adjusted against %s order #%s"%(referenceStore,referenceNumber)
305
            elif (refund["type"]) == CREDIT_TYPE_OFFER:
306
                refund["description"] = "April Accessories Cashback Offer"                
19493 manas 307
            else:
308
                refund["description"] = ""
13927 amit.gupta 309
            refunds.append(refund)
16754 amit.gupta 310
        result['credited'] = credited
13927 amit.gupta 311
        result['data'] = refunds
312
        result['window'] = window
313
        result['totalCount'] = total_count 
314
        result['currCount'] = cursor.count()
315
        result['totalPages'] = pages
316
        result['currPage'] = page    
317
        return result
318
    else:
319
        return result
320
 
321
def getPendingRefunds(userId):
13991 amit.gupta 322
    print type(userId)
13927 amit.gupta 323
    result = get_mongo_connection().Dtr.merchantOrder\
324
        .aggregate([
16398 amit.gupta 325
                    {'$match':{'subOrders.cashBackStatus':CB_APPROVED, 'userId':userId}},
13927 amit.gupta 326
                    {'$unwind':"$subOrders"},
16398 amit.gupta 327
                    {'$match':{'subOrders.cashBackStatus':CB_APPROVED}},
13927 amit.gupta 328
                    { 
329
                     '$group':{
330
                               '_id':None,
331
                               'amount': { '$sum':'$subOrders.cashBackAmount'},
332
                               }
333
                     }
13987 amit.gupta 334
                ])['result']
335
 
336
    if len(result)>0:
337
        result = result[0]        
338
        result.pop("_id")
339
    else:
340
        result={}
341
        result['amount'] = 0.0
14305 amit.gupta 342
    result['nextCredit'] = datetime.strftime(next_weekday(datetime.now(), int(PythonPropertyReader.getConfig('CREDIT_DAY_OF_WEEK'))),"%Y-%m-%d %H:%M:%S")
13927 amit.gupta 343
    return result
14037 kshitij.so 344
 
14671 amit.gupta 345
def getPendingCashbacks(userId):
346
    result = get_mongo_connection().Dtr.merchantOrder\
347
        .aggregate([
16398 amit.gupta 348
                    {'$match':{'subOrders.cashBackStatus':CB_PENDING, 'userId':userId}},
14671 amit.gupta 349
                    {'$unwind':"$subOrders"},
16398 amit.gupta 350
                    {'$match':{'subOrders.cashBackStatus':CB_PENDING}},
14671 amit.gupta 351
                    { 
352
                     '$group':{
353
                               '_id':None,
354
                               'amount': { '$sum':'$subOrders.cashBackAmount'},
355
                               }
356
                     }
357
                ])['result']
358
 
359
    if len(result)>0:
360
        result = result[0]        
361
        result.pop("_id")
362
    else:
363
        result={}
364
        result['amount'] = 0.0
365
    return result
366
 
17369 kshitij.so 367
def __constructDummyDealObject(skuBundleId):
368
    temp =[]
19193 kshitij.so 369
    deals = get_mongo_connection().Catalog.Deals.find({"skuBundleId":skuBundleId,"$or":[{"showDeal":1}, {"prepaidDeal":1}]},{'_id':1,'category_id':1,'brand':1,'totalPoints':1,'bestSellerPoints':1,'nlcPoints':1,'rank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1,'subCategoryId':1,'netPriceAfterCashBack':1})
17369 kshitij.so 370
    for d in deals:
371
        d['persPoints'] = 0
19575 kshitij.so 372
        d['dealRankPoints'] = 0
17369 kshitij.so 373
        temp.append(d)
374
    return temp
375
 
376
 
377
 
14037 kshitij.so 378
def __populateCache(userId):
17369 kshitij.so 379
    try:
380
        if mc.get("featured_deals") is None:
381
            __populateFeaturedDeals()
382
        featuredDeals = mc.get("featured_deals")
383
 
384
    except Exception as e:
385
        print traceback.print_exc()
17469 kshitij.so 386
        featuredDeals = {3:{},5:{},6:{}}
17369 kshitij.so 387
 
19559 kshitij.so 388
 
389
    #Not taking into account accessories
390
    featuredDeals.pop(6,None)
19597 kshitij.so 391
    print featuredDeals
19559 kshitij.so 392
 
17369 kshitij.so 393
    fd_bundles = []
394
 
395
    for catMap in featuredDeals.itervalues():
396
        for fd_map in catMap.itervalues():
17541 kshitij.so 397
            if not fd_map.get('skuBundleId') in fd_bundles:
398
                fd_bundles.append(fd_map.get('skuBundleId'))
17369 kshitij.so 399
 
14037 kshitij.so 400
    print "Populating memcache for userId",userId
401
    outer_query = []
16067 kshitij.so 402
    outer_query.append({ "$or": [ { "showDeal": 1} , { "prepaidDeal": 1 } ] })
14037 kshitij.so 403
    query = {}
16170 kshitij.so 404
    query['$gt'] = -100
14037 kshitij.so 405
    outer_query.append({'totalPoints':query})
19559 kshitij.so 406
    outer_query.append({'category_id':{"$in":[3,5]}})
14037 kshitij.so 407
    brandPrefMap = {}
408
    pricePrefMap = {}
409
    actionsMap = {}
19596 kshitij.so 410
    try:
411
        brand_p = session.query(price_preferences).filter_by(user_id=userId).all()
412
        for x in brand_p:
413
            pricePrefMap[x.category_id] = [x.min_price,x.max_price]
414
        for x in session.query(brand_preferences).filter_by(user_id=userId).all():
415
            temp_map = {}
416
            if brandPrefMap.has_key((x.brand).strip().upper()):
417
                val = brandPrefMap.get((x.brand).strip().upper())
418
                temp_map[x.category_id] = 1 if x.status == 'show' else 0
419
                val.append(temp_map)
420
            else:
421
                temp = []
422
                temp_map[x.category_id] = 1 if x.status == 'show' else 0
423
                temp.append(temp_map)
424
                brandPrefMap[(x.brand).strip().upper()] = temp
425
 
426
        for x in session.query(user_actions).filter_by(user_id=userId).all():
427
            actionsMap[x.store_product_id] = 1 if x.action == 'like' else 0
428
    except:
429
        pass
430
    finally:
431
        session.close()
19575 kshitij.so 432
 
19193 kshitij.so 433
    all_deals = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query},{'_id':1,'category_id':1,'brand':1,'totalPoints':1,'bestSellerPoints':1,'nlcPoints':1,'rank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1, 'subCategoryId':1,'netPriceAfterCashBack':1}).sort([('totalPoints',pymongo.DESCENDING),('bestSellerPoints',pymongo.DESCENDING),('nlcPoints',pymongo.DESCENDING),('rank',pymongo.DESCENDING)]))
14037 kshitij.so 434
    mobile_deals = []
435
    tablet_deals = []
19559 kshitij.so 436
    #accessories_deals = []
14037 kshitij.so 437
    for deal in all_deals:
19135 kshitij.so 438
 
17369 kshitij.so 439
        try:
440
            fd_bundles.remove(deal.get('skuBundleId'))
441
        except:
442
            pass
443
 
14037 kshitij.so 444
        if actionsMap.get(deal['_id']) == 0:
445
            fav_weight =.25
446
        elif actionsMap.get(deal['_id']) == 1:
447
            fav_weight = 1.5
448
        else:
449
            fav_weight = 1
450
 
451
        if brandPrefMap.get(deal['brand'].strip().upper()) is not None:
17469 kshitij.so 452
 
14037 kshitij.so 453
            brand_weight = 1
454
            for brandInfo in brandPrefMap.get(deal['brand'].strip().upper()):
455
                if brandInfo.get(deal['category_id']) is not None:
456
                    if brandInfo.get(deal['category_id']) == 1:
14055 kshitij.so 457
                        brand_weight = 2.0
14037 kshitij.so 458
        else:
459
            brand_weight = 1
460
 
461
        if pricePrefMap.get(deal['category_id']) is not None:
462
 
463
            if deal['available_price'] >= pricePrefMap.get(deal['category_id'])[0] and deal['available_price'] <= pricePrefMap.get(deal['category_id'])[1]:
464
                asp_weight = 1.5
465
            elif  deal['available_price'] >= pricePrefMap.get(deal['category_id'])[0] - 0.5 * pricePrefMap.get(deal['category_id'])[0] and deal['available_price'] <= pricePrefMap.get(deal['category_id'])[1] + 0.5 * pricePrefMap.get(deal['category_id'])[1]:
466
                asp_weight = 1.2
467
            else:
468
                asp_weight = 1
469
        else:
470
            asp_weight = 1
471
 
472
        persPoints = deal['totalPoints'] * fav_weight * brand_weight * asp_weight
473
        deal['persPoints'] = persPoints
474
 
475
        if deal['category_id'] ==3:
476
            mobile_deals.append(deal)
477
        elif deal['category_id'] ==5:
478
            tablet_deals.append(deal)
19559 kshitij.so 479
        #elif deal['category_id'] ==6:
480
        #    accessories_deals.append(deal)
14037 kshitij.so 481
        else:
482
            continue
483
 
17367 kshitij.so 484
    mobile_deals = sorted(mobile_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
485
    tablet_deals = sorted(tablet_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
19559 kshitij.so 486
    #accessories_deals = sorted(accessories_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
17367 kshitij.so 487
 
488
 
17369 kshitij.so 489
    if len(fd_bundles) > 0:
490
        print "There is still bundle in feature deals with points less than -100.Lets add info"
491
        for skuBundleId in fd_bundles:
492
            dummyDealObjList = __constructDummyDealObject(skuBundleId)
493
            for dummyDealObj in dummyDealObjList:
494
                if dummyDealObj['category_id'] ==3:
495
                    mobile_deals.append(dummyDealObj)
496
                elif dummyDealObj['category_id'] ==5:
497
                    tablet_deals.append(dummyDealObj)
19559 kshitij.so 498
                #elif dummyDealObj['category_id'] ==6:
499
                #    accessories_deals.append(dummyDealObj)
17369 kshitij.so 500
                else:
501
                    pass
502
 
17469 kshitij.so 503
    for super_category, categoryFeaturedDeals in featuredDeals.iteritems():
504
        for val in categoryFeaturedDeals.itervalues():
505
            dummyDealObjList = __constructDummyDealObject(val.get('skuBundleId'))
506
            for dummyDealObj in dummyDealObjList:
507
                if super_category == dummyDealObj['category_id']:
508
                    continue  
509
 
510
                if super_category ==3:
511
                    mobile_deals.append(dummyDealObj)
512
                elif super_category ==5:
513
                    tablet_deals.append(dummyDealObj)
19559 kshitij.so 514
                #elif super_category ==6:
515
                #    accessories_deals.append(dummyDealObj)
17469 kshitij.so 516
                else:
517
                    pass
518
 
17367 kshitij.so 519
    sortedMapMobiles = {}
520
    rankMapMobiles = {}
521
    rankMobiles = 0
522
 
523
    blockedRanksMobiles = []
524
    blockedSkuBundlesMobiles = []
525
    blockedInfoMobiles = {}
526
 
527
    featuredDealsMobiles = featuredDeals.get(3)
528
 
529
    for k, v in featuredDealsMobiles.iteritems():
17469 kshitij.so 530
        blockedRanksMobiles.append(k-1)
531
        blockedSkuBundlesMobiles.append(v.get('skuBundleId'))
17367 kshitij.so 532
 
533
 
534
    for sorted_deal in mobile_deals:
17369 kshitij.so 535
 
17367 kshitij.so 536
        while(True):
537
            if rankMobiles in blockedRanksMobiles:
538
                rankMobiles = rankMobiles +1
539
            else:
540
                break
541
 
542
        if sorted_deal['skuBundleId'] in blockedSkuBundlesMobiles:
543
            if blockedInfoMobiles.has_key(sorted_deal['skuBundleId']):
544
                blockedInfoMobiles.get(sorted_deal['skuBundleId']).append(sorted_deal)
545
            else:
546
                blockedInfoMobiles[sorted_deal['skuBundleId']] = [sorted_deal]
547
            continue
548
 
549
        if sortedMapMobiles.get(sorted_deal['skuBundleId']) is None:
550
            sortedMapMobiles[sorted_deal['skuBundleId']] = {rankMobiles:[sorted_deal]}
551
            rankMapMobiles[rankMobiles] = (sortedMapMobiles[sorted_deal['skuBundleId']].values())[0]
552
            rankMobiles = rankMobiles +1
553
        else:
554
            for temp_list in sortedMapMobiles.get(sorted_deal['skuBundleId']).itervalues():
555
                temp_list.append(sorted_deal)
556
            rankMapMobiles[(sortedMapMobiles.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
557
 
558
 
559
    for rank in blockedRanksMobiles:
17369 kshitij.so 560
        if blockedInfoMobiles.get((featuredDealsMobiles.get(rank+1)).get('skuBundleId')) is not None:
561
            rankMapMobiles[rank] = blockedInfoMobiles.get((featuredDealsMobiles.get(rank+1)).get('skuBundleId'))
17367 kshitij.so 562
 
563
 
564
    sortedMapTablets = {}
565
    rankMapTablets = {}
566
    rankTablets = 0
567
 
568
    blockedRanksTablets = []
569
    blockedSkuBundlesTablets = []
570
    blockedInfoTablets = {}
571
 
572
    featuredDealsTablets = featuredDeals.get(5)
573
 
574
 
575
    for k, v in featuredDealsTablets.iteritems():
17469 kshitij.so 576
        blockedRanksTablets.append(k-1)
577
        blockedSkuBundlesTablets.append(v.get('skuBundleId'))
578
 
17367 kshitij.so 579
 
580
    for sorted_deal in tablet_deals:
581
        while(True):
582
            if rankTablets in blockedRanksTablets:
583
                rankTablets = rankTablets +1
584
            else:
585
                break
586
 
587
        if sorted_deal['skuBundleId'] in blockedSkuBundlesTablets:
588
            if blockedInfoTablets.has_key(sorted_deal['skuBundleId']):
589
                blockedInfoTablets.get(sorted_deal['skuBundleId']).append(sorted_deal)
590
            else:
591
                blockedInfoTablets[sorted_deal['skuBundleId']] = [sorted_deal]
592
            continue
593
 
594
        if sortedMapTablets.get(sorted_deal['skuBundleId']) is None:
595
            sortedMapTablets[sorted_deal['skuBundleId']] = {rankTablets:[sorted_deal]}
596
            rankMapTablets[rankTablets] = (sortedMapTablets[sorted_deal['skuBundleId']].values())[0]
597
            rankTablets = rankTablets +1
598
        else:
599
            for temp_list in sortedMapTablets.get(sorted_deal['skuBundleId']).itervalues():
600
                temp_list.append(sorted_deal)
601
            rankMapTablets[(sortedMapTablets.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
602
 
603
    for rank in blockedRanksTablets:
17369 kshitij.so 604
        if blockedInfoTablets.get((featuredDealsTablets.get(rank+1)).get('skuBundleId')) is not None:
605
            rankMapTablets[rank] = blockedInfoTablets.get((featuredDealsTablets.get(rank+1)).get('skuBundleId'))
17469 kshitij.so 606
 
17367 kshitij.so 607
 
19559 kshitij.so 608
#     sortedMapAccessories = {}
609
#     rankMapAccessories = {}
610
#     rankAccessories = 0
611
#     
612
#     blockedRanksAccessories = []
613
#     blockedSkuBundlesAccessories = []
614
#     blockedInfoAccessories = {}
615
#     
616
#     featuredDealsAccessories = featuredDeals.get(6)
617
#     
618
#     for k, v in featuredDealsAccessories.iteritems():
619
#         blockedRanksAccessories.append(k-1)
620
#         blockedSkuBundlesAccessories.append(v.get('skuBundleId'))
621
#     
622
#     
623
#     for sorted_deal in accessories_deals:
624
#        
625
#         while(True):
626
#             if rankAccessories in blockedRanksAccessories:
627
#                 rankAccessories = rankAccessories +1
628
#             else:
629
#                 break
630
#              
631
#         if sorted_deal['skuBundleId'] in blockedSkuBundlesAccessories:
632
#             if blockedInfoAccessories.has_key(sorted_deal['skuBundleId']):
633
#                 blockedInfoAccessories.get(sorted_deal['skuBundleId']).append(sorted_deal)
634
#             else:
635
#                 blockedInfoAccessories[sorted_deal['skuBundleId']] = [sorted_deal]
636
#             continue
637
#          
638
#         if sortedMapAccessories.get(sorted_deal['skuBundleId']) is None:
639
#             sortedMapAccessories[sorted_deal['skuBundleId']] = {rankAccessories:[sorted_deal]}
640
#             rankMapAccessories[rankAccessories] = (sortedMapAccessories[sorted_deal['skuBundleId']].values())[0]
641
#             rankAccessories = rankAccessories +1
642
#         else:
643
#             for temp_list in sortedMapAccessories.get(sorted_deal['skuBundleId']).itervalues():
644
#                 temp_list.append(sorted_deal)
645
#             rankMapAccessories[(sortedMapAccessories.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
646
#     
647
#             
648
#     for rank in blockedRanksAccessories:
649
#         if blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId')) is not None:
650
#             rankMapAccessories[rank] = blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId'))
651
#     
652
    specialOffer = None
653
    promoOffer = get_mongo_connection().Catalog.PromoOffer.find_one({'user_id':userId})
654
    try:
655
        if promoOffer is not None:
656
            offer = get_mongo_connection().Catalog.Promotions.find_one({'_id':promoOffer.get('offer_id')})
657
            if offer is None:
658
                raise
659
            promo = Promotion(offer.get('_id'),offer.get('offer_name'),offer.get('offer_description') , offer.get('categories_applicable'), offer.get('sub_categories_not_applicable'),
660
                     offer.get('startDate'), offer.get('endDate'), promoOffer.get('target1'), promoOffer.get('target1_cash_back_percetage'), promoOffer.get('target2'), promoOffer.get('target2_cash_back_percetage'), promoOffer.get('maxCashBack'),offer.get('url'),
661
                     promoOffer.get('pending_order_value'), promoOffer.get('delivered_order_value'), promoOffer.get('last_run_timestamp'))
662
 
663
            specialOffer = promo
664
    except:
665
        traceback.print_exc()
666
 
667
 
668
    mem_cache_val = {3:mobile_deals, 5:tablet_deals,"3_rankMap": rankMapMobiles,"5_rankMap": rankMapTablets, "specialOffer":specialOffer }
669
    mc.set(str(userId), mem_cache_val)
670
 
671
def __populateCacheForAccessories(userId):
672
    try:
673
        if mc.get("featured_deals") is None:
674
            __populateFeaturedDeals()
675
        featuredDeals = mc.get("featured_deals")
676
 
677
    except Exception as e:
678
        print traceback.print_exc()
679
        featuredDeals = {3:{},5:{},6:{}}
680
 
681
 
682
    #Not taking into account mobiles, tablets
683
    featuredDeals.pop(3,None)
684
    featuredDeals.pop(5,None)
685
 
20009 kshitij.so 686
    print "featuredDeals are ",featuredDeals
687
 
19559 kshitij.so 688
    fd_bundles = []
689
 
690
    for catMap in featuredDeals.itervalues():
691
        for fd_map in catMap.itervalues():
692
            if not fd_map.get('skuBundleId') in fd_bundles:
693
                fd_bundles.append(fd_map.get('skuBundleId'))
694
 
695
    print "Populating accessory memcache for userId",userId
696
    outer_query = []
697
    outer_query.append({ "$or": [ { "showDeal": 1} , { "prepaidDeal": 1 } ] })
698
    query = {}
19570 kshitij.so 699
    query['$gt'] = 0
19575 kshitij.so 700
    outer_query.append({'dealRankPoints':query})
19559 kshitij.so 701
    outer_query.append({'category_id':{"$in":[6]}})
702
 
19581 kshitij.so 703
    accessories_deals = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query},{'_id':1,'category_id':1,'brand':1,'bestSellerPoints':1,'nlcPoints':1,'totalPoints':1,'dealRankPoints':1,'rank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1, 'subCategoryId':1,'netPriceAfterCashBack':1}).sort([('dealRankPoints',pymongo.DESCENDING),('bestSellerPoints',pymongo.DESCENDING)]))
19559 kshitij.so 704
 
705
    actionsMap = {}
19580 kshitij.so 706
    actionRank = 3
19596 kshitij.so 707
    try:
19599 kshitij.so 708
        favourite_data = session.query(user_actions).filter_by(user_id=userId).order_by(asc(user_actions.created)).all()
709
        favourite_ids = [x.store_product_id for x in favourite_data]
710
        master_ids = list(get_mongo_connection().Catalog.MasterData.find({'_id':{"$in":favourite_ids},'category_id':6},{'_id':1}))
711
        accessory_ids = [x['_id'] for x in master_ids]
712
        for x in favourite_data:
713
            if x.store_product_id not in accessory_ids:
714
                continue
19596 kshitij.so 715
            action = 1 if x.action == 'like' else 0
716
            if action==1:
717
                actionsMap[x.store_product_id] = {'action':action,'rank':actionRank,'dealRankPoints':None}
718
                actionRank = actionRank+3
719
            else:
720
                actionsMap[x.store_product_id] = {'action':action,'rank':None,'dealRankPoints':None}
721
    except:
19614 kshitij.so 722
        traceback.print_exc()
19599 kshitij.so 723
        print "Exception in actionsMap creation"
19596 kshitij.so 724
    finally:    
725
        session.close()
19559 kshitij.so 726
 
19583 kshitij.so 727
    pseudo_count = 1
728
    inserted_count = 0
729
    try:
730
        if actionsMap:
731
            for k,v in actionsMap.iteritems():
732
                tmp_deal = get_mongo_connection().Catalog.Deals.find_one({'_id':k})
19599 kshitij.so 733
                if tmp_deal is None:
19580 kshitij.so 734
                    v['rank'] = None
735
                    continue
19599 kshitij.so 736
                if tmp_deal['category_id']!=6 and (tmp_deal['showDeal']!=1 or tmp_deal['prepaidDeal']!=1) :
737
                    v['rank'] = None
738
                    continue
19583 kshitij.so 739
                if v['action'] ==0:
740
                    leastDealRankPoints = list(get_mongo_connection().Catalog.Deals.find({'subCategoryId':tmp_deal.get('subCategoryId')}).sort([('dealRankPoints',pymongo.ASCENDING)]).limit(1))
741
                    if len(leastDealRankPoints) == 0 or leastDealRankPoints[0].get('dealRankPoints') is None:
742
                        v['rank'] = None
743
                        continue
744
                    v['dealRankPoints'] = leastDealRankPoints[0]['dealRankPoints'] - 1
745
                    v['rank'] = 999999
746
                else:
747
                    v['dealRankPoints'] = tmp_deal.get('dealRankPoints')
748
    except:
749
        print "Exception in action map"
750
        traceback.print_exc()         
751
 
19559 kshitij.so 752
    for deal in accessories_deals:
753
 
754
        try:
755
            fd_bundles.remove(deal.get('skuBundleId'))
756
        except:
757
            pass
19583 kshitij.so 758
        try:
759
            if actionsMap and actionsMap.get(deal['_id'])!=None:
760
                if actionsMap.get(deal['_id']).get('action') ==1 and actionsMap.get(deal['_id']).get('rank')!=None:
761
                    actionRank = actionsMap.get(deal['_id']).get('rank')
762
                    if actionRank < pseudo_count:
763
                        deal['dealRankPoints'] = accessories_deals[actionRank-1]['dealRankPoints'] +.5 + inserted_count
764
                        actionsMap.get(deal['_id'])['dealRankPoints'] = deal['dealRankPoints'] 
765
                        inserted_count = inserted_count+1
19580 kshitij.so 766
                elif actionsMap.get(deal['_id']).get('action') ==0 and actionsMap.get(deal['_id']).get('rank')!=None:
767
                    deal['dealRankPoints'] = actionsMap.get(deal['_id']).get('dealRankPoints')
19583 kshitij.so 768
                    actionsMap.get(deal['_id'])['dealRankPoints'] = deal['dealRankPoints']
769
                else:
770
                    pass
771
            pseudo_count = pseudo_count+1
772
        except:
773
            print "Exception while arranging dealRankPoints"
774
            traceback.print_exc()
19580 kshitij.so 775
 
776
    accessories_deals = sorted(accessories_deals, key = lambda x: (x['dealRankPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
19575 kshitij.so 777
 
19559 kshitij.so 778
    if len(fd_bundles) > 0:
779
        print "There is still bundle in feature deals.Lets add info"
780
        for skuBundleId in fd_bundles:
781
            dummyDealObjList = __constructDummyDealObject(skuBundleId)
782
            for dummyDealObj in dummyDealObjList:
783
                if dummyDealObj['category_id'] ==6:
784
                    accessories_deals.append(dummyDealObj)
785
 
786
    for super_category, categoryFeaturedDeals in featuredDeals.iteritems():
787
        for val in categoryFeaturedDeals.itervalues():
788
            dummyDealObjList = __constructDummyDealObject(val.get('skuBundleId'))
789
            for dummyDealObj in dummyDealObjList:
790
                if super_category == dummyDealObj['category_id']:
791
                    continue  
792
                if super_category ==6:
793
                    accessories_deals.append(dummyDealObj)
794
 
795
 
17469 kshitij.so 796
    sortedMapAccessories = {}
797
    rankMapAccessories = {}
798
    rankAccessories = 0
19559 kshitij.so 799
 
17469 kshitij.so 800
    blockedRanksAccessories = []
801
    blockedSkuBundlesAccessories = []
802
    blockedInfoAccessories = {}
19559 kshitij.so 803
 
17469 kshitij.so 804
    featuredDealsAccessories = featuredDeals.get(6)
19559 kshitij.so 805
 
17469 kshitij.so 806
    for k, v in featuredDealsAccessories.iteritems():
807
        blockedRanksAccessories.append(k-1)
808
        blockedSkuBundlesAccessories.append(v.get('skuBundleId'))
19559 kshitij.so 809
 
810
 
17469 kshitij.so 811
    for sorted_deal in accessories_deals:
19559 kshitij.so 812
 
17469 kshitij.so 813
        while(True):
814
            if rankAccessories in blockedRanksAccessories:
815
                rankAccessories = rankAccessories +1
816
            else:
817
                break
19559 kshitij.so 818
 
17469 kshitij.so 819
        if sorted_deal['skuBundleId'] in blockedSkuBundlesAccessories:
820
            if blockedInfoAccessories.has_key(sorted_deal['skuBundleId']):
821
                blockedInfoAccessories.get(sorted_deal['skuBundleId']).append(sorted_deal)
822
            else:
823
                blockedInfoAccessories[sorted_deal['skuBundleId']] = [sorted_deal]
824
            continue
19559 kshitij.so 825
 
17469 kshitij.so 826
        if sortedMapAccessories.get(sorted_deal['skuBundleId']) is None:
827
            sortedMapAccessories[sorted_deal['skuBundleId']] = {rankAccessories:[sorted_deal]}
828
            rankMapAccessories[rankAccessories] = (sortedMapAccessories[sorted_deal['skuBundleId']].values())[0]
829
            rankAccessories = rankAccessories +1
830
        else:
831
            for temp_list in sortedMapAccessories.get(sorted_deal['skuBundleId']).itervalues():
832
                temp_list.append(sorted_deal)
833
            rankMapAccessories[(sortedMapAccessories.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
19559 kshitij.so 834
 
835
 
17469 kshitij.so 836
    for rank in blockedRanksAccessories:
837
        if blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId')) is not None:
838
            rankMapAccessories[rank] = blockedInfoAccessories.get((featuredDealsAccessories.get(rank+1)).get('skuBundleId'))
19559 kshitij.so 839
 
19288 kshitij.so 840
    specialOffer = None
841
    promoOffer = get_mongo_connection().Catalog.PromoOffer.find_one({'user_id':userId})
842
    try:
843
        if promoOffer is not None:
844
            offer = get_mongo_connection().Catalog.Promotions.find_one({'_id':promoOffer.get('offer_id')})
845
            if offer is None:
846
                raise
19352 kshitij.so 847
            promo = Promotion(offer.get('_id'),offer.get('offer_name'),offer.get('offer_description') , offer.get('categories_applicable'), offer.get('sub_categories_not_applicable'),
19331 kshitij.so 848
                     offer.get('startDate'), offer.get('endDate'), promoOffer.get('target1'), promoOffer.get('target1_cash_back_percetage'), promoOffer.get('target2'), promoOffer.get('target2_cash_back_percetage'), promoOffer.get('maxCashBack'),offer.get('url'),
19350 kshitij.so 849
                     promoOffer.get('pending_order_value'), promoOffer.get('delivered_order_value'), promoOffer.get('last_run_timestamp'))
17469 kshitij.so 850
 
19288 kshitij.so 851
            specialOffer = promo
852
    except:
19559 kshitij.so 853
        traceback.print_exc()
19288 kshitij.so 854
 
855
 
19559 kshitij.so 856
    mem_cache_val = {6:accessories_deals,"6_rankMap": rankMapAccessories, "specialOffer":specialOffer }
857
    mc.set(str(userId)+'_acc', mem_cache_val)
858
 
14037 kshitij.so 859
 
860
 
14531 kshitij.so 861
def __populateFeaturedDeals():
17367 kshitij.so 862
    print "Populating featured deals....."
863
    featuredDealsMobiles = {}
864
    featuredDealsTablets = {}
17469 kshitij.so 865
    featuredDealsAccessories = {}
17618 kshitij.so 866
    activeFeaturedDeals = get_mongo_connection().Catalog.FeaturedDeals.find({'startDate':{'$lte':to_java_date(datetime.now())},'endDate':{'$gte':to_java_date(datetime.now())}})
14531 kshitij.so 867
    for activeFeaturedDeal in activeFeaturedDeals:
17367 kshitij.so 868
        deals = get_mongo_connection().Catalog.Deals.find({"skuBundleId":activeFeaturedDeal.get('skuBundleId'),"$or":[{"showDeal":1}, {"prepaidDeal":1}]})
869
        for deal in deals:
870
            if deal.get('available_price') <= activeFeaturedDeal.get('thresholdPrice'):
17469 kshitij.so 871
                for category_id, rank  in activeFeaturedDeal['otherInfo'].iteritems():
872
                    f_deal = {'skuBundleId':activeFeaturedDeal.get('skuBundleId'), 'thresholdPrice':activeFeaturedDeal.get('thresholdPrice'), 'rank':rank,'category_id':int(category_id),'totalPoints':activeFeaturedDeal.get('totalPoints')}
873
                    if f_deal['category_id'] == 3:
874
                        featuredDealsMobiles[rank] = f_deal
875
                    elif f_deal['category_id']==5:
876
                        featuredDealsTablets[rank] = f_deal
877
                    elif f_deal['category_id']==6:
878
                        featuredDealsAccessories[rank] = f_deal
17367 kshitij.so 879
                break
17469 kshitij.so 880
 
881
#                f_deal = {'skuBundleId':activeFeaturedDeal.get('skuBundleId'), 'thresholdPrice':activeFeaturedDeal.get('thresholdPrice'), 'rank':activeFeaturedDeal.get('rank'),'category_id':activeFeaturedDeal.get('category_id'),'totalPoints':activeFeaturedDeal.get('totalPoints')}
882
#                 if activeFeaturedDeal.get('category_id') == 3:
883
#                     featuredDealsMobiles[activeFeaturedDeal.get('rank')] = f_deal
884
#                 else:
885
#                     featuredDealsTablets[activeFeaturedDeal.get('rank')] = f_deal
886
#                 break
887
 
17367 kshitij.so 888
 
17469 kshitij.so 889
    mc.set("featured_deals", {3:featuredDealsMobiles,5:featuredDealsTablets,6:featuredDealsAccessories}, 600)
14037 kshitij.so 890
 
17618 kshitij.so 891
def __populateFeaturedDealsForDealObject():
892
    print "Caching deal objects...."
893
    featuredDealObjectMobiles = {}
894
    featuredDealObjectTablets = {}
895
    featuredDealObjectAccessories = {}
896
    dealObjects = get_mongo_connection().Catalog.FeaturedDealsDealObject.find({'startDate':{'$lte':to_java_date(datetime.now())},'endDate':{'$gte':to_java_date(datetime.now())}})
897
    for dealObject in dealObjects:
898
        d_object = list(get_mongo_connection().Catalog.DealObject.find({'_id':dealObject['_id']}))
899
        if len(d_object) == 0:
900
            continue
901
        for category_id, rank  in dealObject['otherInfo'].iteritems():
902
            d_object[0]['dealObject'] = 1
19288 kshitij.so 903
            f_deal = {'rank':rank,'category_id':int(category_id),'object':d_object[0],'offer_id':dealObject.get('offer_id')}
17618 kshitij.so 904
            if f_deal['category_id'] == 3:
905
                featuredDealObjectMobiles[rank] = f_deal
906
            elif f_deal['category_id']==5:
907
                featuredDealObjectTablets[rank] = f_deal
908
            elif f_deal['category_id']==6:
909
                featuredDealObjectAccessories[rank] = f_deal
910
    mc.set("featured_deals_deal_object", {3:featuredDealObjectMobiles,5:featuredDealObjectTablets,6:featuredDealObjectAccessories}, 600)
17619 kshitij.so 911
    print mc.get('featured_deals_deal_object')
17618 kshitij.so 912
 
14791 kshitij.so 913
def getNewDeals(userId, category_id, offset, limit, sort, direction, filterData=None):
17469 kshitij.so 914
    if mc.get("category_cash_back") is None or not bool(mc.get("category_cash_back")):
13921 kshitij.so 915
        populateCashBack()
14037 kshitij.so 916
 
16079 kshitij.so 917
    dealsListMap = []
14037 kshitij.so 918
    user_specific_deals = mc.get(str(userId))
919
    if user_specific_deals is None:
920
        __populateCache(userId)
921
        user_specific_deals = mc.get(str(userId))
14038 kshitij.so 922
    else:
923
        print "Getting user deals from cache"
14037 kshitij.so 924
    category_specific_deals = user_specific_deals.get(category_id)
925
 
17367 kshitij.so 926
    insert_featured_deals = False
14037 kshitij.so 927
    if sort is None or direction is None:
17367 kshitij.so 928
        insert_featured_deals = True
929
        rankMap = user_specific_deals.get(str(category_id)+"_rankMap")
14037 kshitij.so 930
    else:
931
        if sort == "bestSellerPoints":
932
            sorted_deals = sorted(category_specific_deals, key = lambda x: (x['bestSellerPoints'], x['rank'], x['nlcPoints']),reverse=True)
933
        else:
934
            if direction == -1:
935
                rev = True
936
            else:
937
                rev = False
938
            sorted_deals = sorted(category_specific_deals, key = lambda x: (x['available_price']),reverse=rev)
939
 
14791 kshitij.so 940
 
17367 kshitij.so 941
    filtered_deals = []
17668 kshitij.so 942
    dataExist = True
14791 kshitij.so 943
    if filterData is not None:
944
        try:
17669 kshitij.so 945
            filtered_deals = filterDeals(category_specific_deals, filterData)
946
            if len(filtered_deals)==0:
947
                dataExist = False
14791 kshitij.so 948
        except:
16067 kshitij.so 949
            traceback.print_exc()
17668 kshitij.so 950
    if dataExist:
14791 kshitij.so 951
 
17668 kshitij.so 952
        if not insert_featured_deals:
953
            sortedMap = {}
954
            rankMap = {}
955
            rank = 0
956
            for sorted_deal in sorted_deals:
957
 
958
                if len(filtered_deals) > 0 and  sorted_deal['skuBundleId'] not in filtered_deals:
959
                    continue
960
 
961
                if sortedMap.get(sorted_deal['skuBundleId']) is None:
962
                    sortedMap[sorted_deal['skuBundleId']] = {rank:[sorted_deal]}
963
                    rankMap[rank] = (sortedMap[sorted_deal['skuBundleId']].values())[0]
964
                    rank = rank +1
965
                else:
966
                    for temp_list in sortedMap.get(sorted_deal['skuBundleId']).itervalues():
967
                        temp_list.append(sorted_deal)
968
                    rankMap[(sortedMap.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
969
        else:
970
            filterMap = {}
971
            rank = 0
972
            if len(filtered_deals) > 0:
973
                try:
974
                    od = collections.OrderedDict(sorted(rankMap.items()))
975
                except:
976
                    od = ordereddict.OrderedDict(sorted(rankMap.items()))
977
                for k, v in od.iteritems():
978
                    if v[0].get('skuBundleId') in filtered_deals:
979
                        filterMap[rank] = v
980
                        rank =rank+1
981
                rankMap = filterMap
982
 
983
        rankCounter = offset
984
        if filterData is None:
985
            if mc.get("featured_deals_deal_object") is None or not bool(mc.get("featured_deals_deal_object")):
986
                try:
987
                    __populateFeaturedDealsForDealObject()
988
                    fd_dealObject = mc.get("featured_deals_deal_object")
989
                except:
990
                    fd_dealObject = {3:{},5:{},6:{}}
17367 kshitij.so 991
            else:
17668 kshitij.so 992
                fd_dealObject = mc.get("featured_deals_deal_object")
993
 
17367 kshitij.so 994
    else:
17668 kshitij.so 995
        rankMap = {}
17618 kshitij.so 996
 
19288 kshitij.so 997
 
998
    specialOffer = user_specific_deals.get('specialOffer')
999
 
17627 kshitij.so 1000
    for dealList in [rankMap.get(k, []) for k in range(offset, offset+limit)]:
17629 kshitij.so 1001
        if filterData is None:
17627 kshitij.so 1002
            while(True):
1003
 
1004
                rankCounter = rankCounter+1
1005
                if (fd_dealObject.get(category_id).get(rankCounter)) is not None:
19288 kshitij.so 1006
                    if (fd_dealObject.get(category_id).get(rankCounter).get('offer_id') !=0):
1007
                        #Check whether user can see offer
19518 manas 1008
                        if specialOffer is None or specialOffer.endDate < to_java_date(datetime.now()):
19288 kshitij.so 1009
                            continue
19334 kshitij.so 1010
                    deal_object_temp = []
1011
                    deal_object_temp.append(fd_dealObject.get(category_id).get(rankCounter).get('object'))
1012
                    dealsListMap.append(deal_object_temp)
17627 kshitij.so 1013
                else:
1014
                    break
17620 kshitij.so 1015
 
16079 kshitij.so 1016
        temp = []
17367 kshitij.so 1017
        if dealList is None or len(dealList)==0:
1018
            continue
16067 kshitij.so 1019
        for d in dealList:
1020
            item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))
16320 kshitij.so 1021
            if len(item) ==0:
1022
                continue
16079 kshitij.so 1023
            item[0]['persPoints'] = d['persPoints']
1024
            if d['dealType'] == 1 and d['source_id'] ==1:
16511 kshitij.so 1025
                try:
1026
                    manualDeal = list(get_mongo_connection().Catalog.ManualDeals.find({'sku':d['_id'],'startDate':{'$lte':to_java_date(datetime.now())},'endDate':{'$gte':to_java_date(datetime.now())}}))
1027
                    item[0]['marketPlaceUrl'] = manualDeal[0]['dealUrl'].strip()
1028
                except:
1029
                    pass
16079 kshitij.so 1030
            elif d['source_id'] ==3:
1031
                item[0]['marketPlaceUrl'] = item[0]['marketPlaceUrl']+'?supc='+item[0].get('identifier')
1032
            else:
1033
                pass 
1034
            try:
1035
                cashBack = getCashBack(item[0]['_id'], item[0]['source_id'], item[0]['category_id'])
1036
                if not cashBack or cashBack.get('cash_back_status')!=1:
14037 kshitij.so 1037
                    item[0]['cash_back_type'] = 0
1038
                    item[0]['cash_back'] = 0
16079 kshitij.so 1039
                else:
17180 kshitij.so 1040
                    if cashBack.get('maxCashBack') is not None:
1041
 
1042
                        if cashBack.get('cash_back_type') ==1 and (float(cashBack.get('cash_back'))*item[0]['available_price'])/100 > cashBack.get('maxCashBack'):
1043
                            cashBack['cash_back_type'] = 2
1044
                            cashBack['cash_back'] = cashBack['maxCashBack']
1045
                        elif cashBack.get('cash_back_type') ==2 and cashBack.get('cash_back') > cashBack.get('maxCashBack'):
1046
                            cashBack['cash_back'] = cashBack['maxCashBack']
1047
 
16079 kshitij.so 1048
                    item[0]['cash_back_type'] = int(cashBack['cash_back_type'])
1049
                    item[0]['cash_back'] = cashBack['cash_back']
1050
            except:
1051
                print "Error in adding cashback to deals"
1052
                item[0]['cash_back_type'] = 0
1053
                item[0]['cash_back'] = 0
16252 kshitij.so 1054
            try:
1055
                item[0]['dp'] = d['dp']
1056
                item[0]['showDp'] = d['showDp']
1057
            except:
1058
                item[0]['dp'] = 0.0
1059
                item[0]['showDp'] = 0
16483 kshitij.so 1060
            try:
1061
                item[0]['thumbnail'] = item[0]['thumbnail'].strip()
1062
            except:
17115 manish.sha 1063
                pass
17603 kshitij.so 1064
 
1065
            try:
1066
                if item[0]['showVideo']==0:
1067
                    item[0]['videoLink'] =""
1068
            except:
1069
                item[0]['videoLink'] =""
17937 kshitij.so 1070
            try:
1071
                if item[0]['quantity'] >1:
1072
                    ppq = float(item[0]['available_price'])/item[0]['quantity']
1073
 
1074
                    if ppq == round(ppq):
1075
                        item[0]['ppq'] = int(ppq)
1076
                    else:
1077
                        item[0]['ppq'] = round(ppq,2)
1078
                else:
1079
                    item[0]['ppq'] = 0
1080
            except:
1081
                item[0]['ppq'] = 0
17627 kshitij.so 1082
            if filterData is None:
17679 kshitij.so 1083
                if item[0]['category_id']!=6:
1084
                    try:
18830 kshitij.so 1085
                        item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
1086
                        item[0]['filterTextBrand'] = (item[0]['brand'])
17679 kshitij.so 1087
                    except:
18830 kshitij.so 1088
                        pass
17679 kshitij.so 1089
                else:
1090
                    try:
17756 kshitij.so 1091
                        if item[0]['subCategory']=='' or item[0]['subCategory'] is None or item[0]['subCategoryId']==0:
17679 kshitij.so 1092
                            raise
18830 kshitij.so 1093
                        item[0]['filterLinkSubCategory'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['subCategory'])+'&filter=subcategory&subcategories='+str(int(item[0]['subCategoryId']))
1094
                        item[0]['filterTextSubCategory'] = (item[0]['subCategory'])
1095
                        item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
1096
                        item[0]['filterTextBrand'] = (item[0]['brand'])
17679 kshitij.so 1097
                    except:
18830 kshitij.so 1098
                        pass
17771 kshitij.so 1099
            try:
1100
                if item[0]['source_id']==4:
1101
                    item_availability_info = mc.get("item_availability_"+str(item[0]['identifier']).strip())
1102
                    if item_availability_info is None or len(item_availability_info)==0:
1103
                        item[0]['availabilityInfo'] = [{}]
1104
                    else:
1105
                        item[0]['availabilityInfo'] = item_availability_info
1106
            except:
1107
                item[0]['availabilityInfo'] = [{}]
19193 kshitij.so 1108
 
1109
            if item[0]['source_id']==4:
1110
                for av_info in item[0]['availabilityInfo']:
1111
                    if not av_info:
1112
                        continue
1113
                    av_info['cash_back'] = item[0]['cash_back']
1114
                    av_info['cash_back_type'] = item[0]['cash_back_type']
1115
                    netPriceAfterCashBack = av_info['sellingPrice'] 
19199 kshitij.so 1116
                    if av_info['cash_back_type'] ==1:
19207 kshitij.so 1117
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(float(av_info['sellingPrice'])*av_info['cash_back']/100)
19199 kshitij.so 1118
                    elif av_info['cash_back_type'] ==2:
19207 kshitij.so 1119
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(av_info['cash_back'])
19193 kshitij.so 1120
                    else:
1121
                        pass
1122
                    av_info['netPriceAfterCashBack'] = netPriceAfterCashBack
1123
 
1124
            #item[0]['netPriceAfterCashBack'] = d['netPriceAfterCashBack']
1125
 
19896 kshitij.so 1126
            if item[0]['source_id'] == SOURCE_MAP.get("PAYTM.COM"):
19193 kshitij.so 1127
                if item[0]['codAvailable'] ==1:
1128
                    paytmPrice = item[0]['available_price']
1129
                else:
1130
                    paytmPrice = item[0]['gross_price']
1131
 
1132
                if item[0]['cash_back_type'] ==1:
19207 kshitij.so 1133
                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(float(paytmPrice)*item[0]['cash_back']/100)
19193 kshitij.so 1134
                elif item[0]['cash_back_type'] ==2:
19207 kshitij.so 1135
                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(item[0]['cash_back'])
19193 kshitij.so 1136
                else:
1137
                    item[0]['netPriceAfterCashBack'] = paytmPrice
1138
 
1139
            else:
1140
                if item[0]['cash_back_type'] ==1:
19207 kshitij.so 1141
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(float(item[0]['available_price'])*item[0]['cash_back']/100)
19193 kshitij.so 1142
                elif item[0]['cash_back_type'] ==2:
19207 kshitij.so 1143
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(item[0]['cash_back'])
19193 kshitij.so 1144
                else:
1145
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price']
1146
 
19559 kshitij.so 1147
            temp.append(item[0])
1148
        if len(temp) > 1:
1149
            temp = sorted(temp, key = lambda x: (x['available_price']),reverse=False)
1150
        dealsListMap.append(temp)
1151
    return dealsListMap
1152
 
1153
 
1154
def getAccesoryDeals(userId, category_id, offset, limit, sort, direction, filterData=None):
1155
    if mc.get("category_cash_back") is None or not bool(mc.get("category_cash_back")):
1156
        populateCashBack()
1157
 
1158
    dealsListMap = []
1159
    user_specific_deals = mc.get(str(userId)+'_acc')
1160
    if user_specific_deals is None:
1161
        __populateCacheForAccessories(userId)
1162
        user_specific_deals = mc.get(str(userId)+'_acc')
1163
    else:
1164
        print "Getting user deals from accessory cache"
1165
    category_specific_deals = user_specific_deals.get(category_id)
1166
 
1167
    insert_featured_deals = False
1168
    if sort is None or direction is None:
1169
        insert_featured_deals = True
1170
        rankMap = user_specific_deals.get(str(category_id)+"_rankMap")
1171
    else:
1172
        if sort == "bestSellerPoints":
19597 kshitij.so 1173
            sorted_deals = sorted(category_specific_deals, key = lambda x: (x['bestSellerPoints'], x['dealRankPoints'], x['nlcPoints']),reverse=True)
19559 kshitij.so 1174
        else:
1175
            if direction == -1:
1176
                rev = True
1177
            else:
1178
                rev = False
1179
            sorted_deals = sorted(category_specific_deals, key = lambda x: (x['available_price']),reverse=rev)
1180
 
1181
 
1182
    filtered_deals = []
1183
    dataExist = True
1184
    if filterData is not None:
1185
        try:
1186
            filtered_deals = filterDeals(category_specific_deals, filterData)
1187
            if len(filtered_deals)==0:
1188
                dataExist = False
1189
        except:
1190
            traceback.print_exc()
1191
    if dataExist:
1192
 
1193
        if not insert_featured_deals:
1194
            sortedMap = {}
1195
            rankMap = {}
1196
            rank = 0
1197
            for sorted_deal in sorted_deals:
1198
 
1199
                if len(filtered_deals) > 0 and  sorted_deal['skuBundleId'] not in filtered_deals:
1200
                    continue
1201
 
1202
                if sortedMap.get(sorted_deal['skuBundleId']) is None:
1203
                    sortedMap[sorted_deal['skuBundleId']] = {rank:[sorted_deal]}
1204
                    rankMap[rank] = (sortedMap[sorted_deal['skuBundleId']].values())[0]
1205
                    rank = rank +1
1206
                else:
1207
                    for temp_list in sortedMap.get(sorted_deal['skuBundleId']).itervalues():
1208
                        temp_list.append(sorted_deal)
1209
                    rankMap[(sortedMap.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
1210
        else:
1211
            filterMap = {}
1212
            rank = 0
1213
            if len(filtered_deals) > 0:
1214
                try:
1215
                    od = collections.OrderedDict(sorted(rankMap.items()))
1216
                except:
1217
                    od = ordereddict.OrderedDict(sorted(rankMap.items()))
1218
                for k, v in od.iteritems():
1219
                    if v[0].get('skuBundleId') in filtered_deals:
1220
                        filterMap[rank] = v
1221
                        rank =rank+1
1222
                rankMap = filterMap
1223
 
1224
        rankCounter = offset
1225
        if filterData is None:
1226
            if mc.get("featured_deals_deal_object") is None or not bool(mc.get("featured_deals_deal_object")):
1227
                try:
1228
                    __populateFeaturedDealsForDealObject()
1229
                    fd_dealObject = mc.get("featured_deals_deal_object")
1230
                except:
1231
                    fd_dealObject = {3:{},5:{},6:{}}
1232
            else:
1233
                fd_dealObject = mc.get("featured_deals_deal_object")
1234
 
1235
    else:
1236
        rankMap = {}
1237
 
1238
 
1239
    specialOffer = user_specific_deals.get('specialOffer')
1240
 
1241
    for dealList in [rankMap.get(k, []) for k in range(offset, offset+limit)]:
1242
        if filterData is None:
1243
            while(True):
1244
 
1245
                rankCounter = rankCounter+1
1246
                if (fd_dealObject.get(category_id).get(rankCounter)) is not None:
1247
                    if (fd_dealObject.get(category_id).get(rankCounter).get('offer_id') !=0):
1248
                        #Check whether user can see offer
1249
                        if specialOffer is None or specialOffer.endDate < to_java_date(datetime.now()):
1250
                            continue
1251
                    deal_object_temp = []
1252
                    deal_object_temp.append(fd_dealObject.get(category_id).get(rankCounter).get('object'))
1253
                    dealsListMap.append(deal_object_temp)
1254
                else:
1255
                    break
1256
 
1257
        temp = []
1258
        if dealList is None or len(dealList)==0:
1259
            continue
1260
        for d in dealList:
1261
            item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))
1262
            if len(item) ==0:
1263
                continue
19575 kshitij.so 1264
            item[0]['dealRankPoints'] = d.get('dealRankPoints')
19559 kshitij.so 1265
            if d['dealType'] == 1 and d['source_id'] ==1:
1266
                try:
1267
                    manualDeal = list(get_mongo_connection().Catalog.ManualDeals.find({'sku':d['_id'],'startDate':{'$lte':to_java_date(datetime.now())},'endDate':{'$gte':to_java_date(datetime.now())}}))
1268
                    item[0]['marketPlaceUrl'] = manualDeal[0]['dealUrl'].strip()
1269
                except:
1270
                    pass
1271
            elif d['source_id'] ==3:
1272
                item[0]['marketPlaceUrl'] = item[0]['marketPlaceUrl']+'?supc='+item[0].get('identifier')
1273
            else:
1274
                pass 
1275
            try:
1276
                cashBack = getCashBack(item[0]['_id'], item[0]['source_id'], item[0]['category_id'])
1277
                if not cashBack or cashBack.get('cash_back_status')!=1:
1278
                    item[0]['cash_back_type'] = 0
1279
                    item[0]['cash_back'] = 0
1280
                else:
1281
                    if cashBack.get('maxCashBack') is not None:
1282
 
1283
                        if cashBack.get('cash_back_type') ==1 and (float(cashBack.get('cash_back'))*item[0]['available_price'])/100 > cashBack.get('maxCashBack'):
1284
                            cashBack['cash_back_type'] = 2
1285
                            cashBack['cash_back'] = cashBack['maxCashBack']
1286
                        elif cashBack.get('cash_back_type') ==2 and cashBack.get('cash_back') > cashBack.get('maxCashBack'):
1287
                            cashBack['cash_back'] = cashBack['maxCashBack']
1288
 
1289
                    item[0]['cash_back_type'] = int(cashBack['cash_back_type'])
1290
                    item[0]['cash_back'] = cashBack['cash_back']
1291
            except:
1292
                print "Error in adding cashback to deals"
1293
                item[0]['cash_back_type'] = 0
1294
                item[0]['cash_back'] = 0
1295
            try:
1296
                item[0]['dp'] = d['dp']
1297
                item[0]['showDp'] = d['showDp']
1298
            except:
1299
                item[0]['dp'] = 0.0
1300
                item[0]['showDp'] = 0
1301
            try:
1302
                item[0]['thumbnail'] = item[0]['thumbnail'].strip()
1303
            except:
1304
                pass
1305
 
1306
            try:
1307
                if item[0]['showVideo']==0:
1308
                    item[0]['videoLink'] =""
1309
            except:
1310
                item[0]['videoLink'] =""
1311
            try:
1312
                if item[0]['quantity'] >1:
1313
                    ppq = float(item[0]['available_price'])/item[0]['quantity']
1314
 
1315
                    if ppq == round(ppq):
1316
                        item[0]['ppq'] = int(ppq)
1317
                    else:
1318
                        item[0]['ppq'] = round(ppq,2)
1319
                else:
1320
                    item[0]['ppq'] = 0
1321
            except:
1322
                item[0]['ppq'] = 0
1323
            if filterData is None:
1324
                if item[0]['category_id']!=6:
1325
                    try:
1326
                        item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
1327
                        item[0]['filterTextBrand'] = (item[0]['brand'])
1328
                    except:
1329
                        pass
1330
                else:
1331
                    try:
1332
                        if item[0]['subCategory']=='' or item[0]['subCategory'] is None or item[0]['subCategoryId']==0:
1333
                            raise
1334
                        item[0]['filterLinkSubCategory'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['subCategory'])+'&filter=subcategory&subcategories='+str(int(item[0]['subCategoryId']))
1335
                        item[0]['filterTextSubCategory'] = (item[0]['subCategory'])
1336
                        item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
1337
                        item[0]['filterTextBrand'] = (item[0]['brand'])
1338
                    except:
1339
                        pass
1340
            try:
1341
                if item[0]['source_id']==4:
1342
                    item_availability_info = mc.get("item_availability_"+str(item[0]['identifier']).strip())
1343
                    if item_availability_info is None or len(item_availability_info)==0:
1344
                        item[0]['availabilityInfo'] = [{}]
1345
                    else:
1346
                        item[0]['availabilityInfo'] = item_availability_info
1347
            except:
1348
                item[0]['availabilityInfo'] = [{}]
1349
 
1350
            if item[0]['source_id']==4:
1351
                for av_info in item[0]['availabilityInfo']:
1352
                    if not av_info:
1353
                        continue
1354
                    av_info['cash_back'] = item[0]['cash_back']
1355
                    av_info['cash_back_type'] = item[0]['cash_back_type']
1356
                    netPriceAfterCashBack = av_info['sellingPrice'] 
1357
                    if av_info['cash_back_type'] ==1:
1358
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(float(av_info['sellingPrice'])*av_info['cash_back']/100)
1359
                    elif av_info['cash_back_type'] ==2:
1360
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(av_info['cash_back'])
1361
                    else:
1362
                        pass
1363
                    av_info['netPriceAfterCashBack'] = netPriceAfterCashBack
1364
 
1365
            #item[0]['netPriceAfterCashBack'] = d['netPriceAfterCashBack']
1366
 
19896 kshitij.so 1367
            if item[0]['source_id'] == SOURCE_MAP.get("PAYTM.COM"):
19559 kshitij.so 1368
                if item[0]['codAvailable'] ==1:
1369
                    paytmPrice = item[0]['available_price']
1370
                else:
1371
                    paytmPrice = item[0]['gross_price']
1372
 
1373
                if item[0]['cash_back_type'] ==1:
1374
                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(float(paytmPrice)*item[0]['cash_back']/100)
1375
                elif item[0]['cash_back_type'] ==2:
1376
                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(item[0]['cash_back'])
1377
                else:
1378
                    item[0]['netPriceAfterCashBack'] = paytmPrice
1379
 
1380
            else:
1381
                if item[0]['cash_back_type'] ==1:
1382
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(float(item[0]['available_price'])*item[0]['cash_back']/100)
1383
                elif item[0]['cash_back_type'] ==2:
1384
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(item[0]['cash_back'])
1385
                else:
1386
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price']
1387
 
19320 kshitij.so 1388
#             if specialOffer is not None and item[0]['source_id']==4:
1389
#                 if item[0]['category_id'] in specialOffer.categories_applicable and item[0]['subCategoryId'] not in specialOffer.sub_categories_not_applicable:
1390
#                     item[0]['specialOffer'] = specialOffer.__dict__
1391
#             else:
1392
#                 item[0]['specialOffer'] = {}
19288 kshitij.so 1393
 
16079 kshitij.so 1394
            temp.append(item[0])
16125 kshitij.so 1395
        if len(temp) > 1:
16126 kshitij.so 1396
            temp = sorted(temp, key = lambda x: (x['available_price']),reverse=False)
16079 kshitij.so 1397
        dealsListMap.append(temp)
1398
    return dealsListMap
14037 kshitij.so 1399
 
14791 kshitij.so 1400
def filterDeals(deals, filterData):
20009 kshitij.so 1401
    print "Filter data ",filterData
1402
    brandsToFilter = []
1403
    subCategoryToFilter = []
1404
    priceToFilter = []
14791 kshitij.so 1405
    filterArray = filterData.split('|')
1406
    for data in filterArray:
1407
        try:
1408
            filter, info = data.split(':')
1409
        except Exception as ex:
1410
            traceback.print_exc()
1411
            continue
17657 kshitij.so 1412
        if filter == 'brandFilter':
20009 kshitij.so 1413
            brandsToFilter = info.split('^')
1414
        if filter == 'subCategoryFilter':
1415
            subCategoryToFilter = info.split('^')
1416
        if filter == 'priceFilter':
1417
            priceToFilter = info.split('^')
1418
 
1419
    if brandsToFilter:
1420
        temp = []    
1421
        for deal in deals:
1422
            if str(int(deal['brand_id'])) in brandsToFilter:
1423
                temp.append(deal)
1424
        deals = temp
14791 kshitij.so 1425
 
20009 kshitij.so 1426
    if subCategoryToFilter:
1427
        temp = []
1428
        for deal in deals:
1429
            if str(int(deal['subCategoryId'])) in subCategoryToFilter:
1430
                temp.append(deal)
1431
        deals = temp
1432
 
1433
    if priceToFilter and len(priceToFilter) ==2:
1434
        temp = []
1435
        for deal in deals:
1436
            if deal.get('available_price') > float(priceToFilter[0]) and deal.get('available_price') <= float(priceToFilter[0]):
1437
                temp.append(deal)
1438
        deals = temp
1439
 
1440
    return [trend['skuBundleId'] for trend in temp]
14791 kshitij.so 1441
 
14037 kshitij.so 1442
def getDeals(userId, category_id, offset, limit, sort, direction):
17469 kshitij.so 1443
    if mc.get("category_cash_back") is None or not bool(mc.get("category_cash_back")):
14037 kshitij.so 1444
        populateCashBack()
1445
    rank = 1
13771 kshitij.so 1446
    deals = {}
13910 kshitij.so 1447
    outer_query = []
1448
    outer_query.append({"showDeal":1})
1449
    query = {}
1450
    query['$gt'] = 0
1451
    outer_query.append({'totalPoints':query})
1452
    if category_id in (3,5):
1453
        outer_query.append({'category_id':category_id})
13803 kshitij.so 1454
    if sort is None or direction is None:
1455
        direct = -1
13910 kshitij.so 1456
        print outer_query
1457
        data = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query}).sort([('totalPoints',direct),('bestSellerPoints',direct),('nlcPoints',direct),('rank',direct)]).skip(offset).limit(limit))
13795 kshitij.so 1458
    else:
13910 kshitij.so 1459
        print outer_query
13803 kshitij.so 1460
        direct = direction
13910 kshitij.so 1461
        if sort == "bestSellerPoints":
1462
            print "yes,sorting by bestSellerPoints"
1463
            data = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query}).sort([('bestSellerPoints',direct),('rank',direct),('nlcPoints',direct)]).skip(offset).limit(limit))
1464
        else:
1465
            data = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query}).sort([(sort,direct)]).skip(offset).limit(limit))
13771 kshitij.so 1466
    for d in data:
1467
        item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))
1468
        if not deals.has_key(item[0]['identifier']):
13921 kshitij.so 1469
            item[0]['dealRank'] = rank
1470
            try:
14761 kshitij.so 1471
                cashBack = getCashBack(item[0]['_id'], item[0]['source_id'], item[0]['category_id'])
13921 kshitij.so 1472
                if not cashBack or cashBack.get('cash_back_status')!=1:
13928 kshitij.so 1473
                    item[0]['cash_back_type'] = 0
1474
                    item[0]['cash_back'] = 0
13921 kshitij.so 1475
                else:
14766 kshitij.so 1476
                    item[0]['cash_back_type'] = int(cashBack['cash_back_type'])
13928 kshitij.so 1477
                    item[0]['cash_back'] = cashBack['cash_back']
13921 kshitij.so 1478
            except:
1479
                print "Error in adding cashback to deals"
13928 kshitij.so 1480
                item[0]['cash_back_type'] = 0
1481
                item[0]['cash_back'] = 0
13771 kshitij.so 1482
            deals[item[0]['identifier']] = item[0]
13921 kshitij.so 1483
 
13771 kshitij.so 1484
            rank +=1
13785 kshitij.so 1485
    return sorted(deals.values(), key=itemgetter('dealRank'))
1486
 
16222 kshitij.so 1487
def getItem(skuId,showDp=None):
16224 kshitij.so 1488
    temp = []
17469 kshitij.so 1489
    if mc.get("category_cash_back") is None or not bool(mc.get("category_cash_back")):
14113 kshitij.so 1490
        populateCashBack()
13795 kshitij.so 1491
    try:
16222 kshitij.so 1492
        skuData = get_mongo_connection().Catalog.MasterData.find_one({'_id':int(skuId)})
1493
        if skuData is None:
1494
            raise
1495
        try:
1496
            cashBack = getCashBack(skuData['_id'], skuData['source_id'], skuData['category_id'])
1497
            if not cashBack or cashBack.get('cash_back_status')!=1:
1498
                skuData['cash_back_type'] = 0
1499
                skuData['cash_back'] = 0
1500
            else:
1501
                skuData['cash_back_type'] = cashBack['cash_back_type']
1502
                skuData['cash_back'] = cashBack['cash_back']
1503
        except:
1504
            print "Error in adding cashback to deals"
1505
            skuData['cash_back_type'] = 0
1506
            skuData['cash_back'] = 0
1507
        skuData['in_stock'] = int(skuData['in_stock'])
1508
        skuData['is_shortage'] = int(skuData['is_shortage'])
1509
        skuData['category_id'] = int(skuData['category_id'])
17607 kshitij.so 1510
        skuData['subCategoryId'] = int(skuData['subCategoryId'])
16222 kshitij.so 1511
        skuData['status'] = int(skuData['status'])
16483 kshitij.so 1512
        try:
1513
            skuData['thumbnail'] = skuData['thumbnail'].strip()
1514
        except:
1515
            pass 
16222 kshitij.so 1516
        if showDp is not None:
1517
            dealerPrice = get_mongo_connection().Catalog.SkuDealerPrices.find_one({'skuBundleId':skuData['skuBundleId']})
1518
            skuData['dp'] = 0.0
1519
            skuData['showDp'] = 0
1520
            if dealerPrice is not None:
1521
                skuData['dp'] = dealerPrice['dp']
1522
                skuData['showDp'] = dealerPrice['showDp']
16224 kshitij.so 1523
        temp.append(skuData)
1524
        return temp
13795 kshitij.so 1525
    except:
16224 kshitij.so 1526
        return []
13836 kshitij.so 1527
 
14761 kshitij.so 1528
def getCashBack(skuId, source_id, category_id):
17469 kshitij.so 1529
    if mc.get("category_cash_back") is None or not bool(mc.get("category_cash_back")):
14761 kshitij.so 1530
        populateCashBack()
1531
    itemCashBackMap = mc.get("item_cash_back")
13921 kshitij.so 1532
    itemCashBack = itemCashBackMap.get(skuId)
1533
    if itemCashBack is not None:
1534
        return itemCashBack
14761 kshitij.so 1535
    cashBackMap = mc.get("category_cash_back")
13921 kshitij.so 1536
    sourceCashBack = cashBackMap.get(source_id)
1537
    if sourceCashBack is not None and len(sourceCashBack) > 0:
1538
        for cashBack in sourceCashBack:
1539
            if cashBack.get(category_id) is None:
1540
                continue
1541
            else:
1542
                return cashBack.get(category_id)
1543
    else:
1544
        return {}
16560 amit.gupta 1545
 
1546
def getBundleBySourceSku(source_id, identifier):
17035 kshitij.so 1547
    if source_id in (1,2,4,5,6,7):
16560 amit.gupta 1548
        skuData = list(get_mongo_connection().Catalog.MasterData.find({'identifier':identifier.strip(), 'source_id':source_id}))
1549
    elif source_id == 3:
1550
        skuData = list(get_mongo_connection().Catalog.MasterData.find({'secondaryIdentifier':identifier.strip(), 'source_id':source_id}))
1551
    else:
1552
        return {}
1553
 
1554
    if not skuData:
1555
        return {}
1556
    else:
16564 amit.gupta 1557
        bundleId = skuData[0]["skuBundleId"]
17035 kshitij.so 1558
        itemIds = list(get_mongo_connection().Catalog.MasterData.find({'skuBundleId':bundleId, 'in_stock':1, 'source_id':{"$in":[1,2,3,5,6,7]}}, 
16580 amit.gupta 1559
                       {"source_id":1, "available_price":1, "marketPlaceUrl":1, "gross_price":1, "codAvailable":1, "source_product_name":1, "offer":1, "coupon":1}))
16579 amit.gupta 1560
        return {'products':itemIds}
16560 amit.gupta 1561
 
1562
 
13921 kshitij.so 1563
 
15129 kshitij.so 1564
def getDealRank(identifier, source_id, userId):
17035 kshitij.so 1565
    if source_id in (1,2,4,5,6,7):
15129 kshitij.so 1566
        skuData = list(get_mongo_connection().Catalog.MasterData.find({'identifier':identifier.strip(), 'source_id':source_id}))
1567
    elif source_id == 3:
1568
        skuData = list(get_mongo_connection().Catalog.MasterData.find({'secondaryIdentifier':identifier.strip(), 'source_id':source_id}))
1569
    else:
16282 amit.gupta 1570
        return {'rank':0, 'description':'Source not valid','maxNlc':None, 'minNlc':None,'status':None,'dp':None}
15349 kshitij.so 1571
    if len(skuData) == 0:
16282 amit.gupta 1572
        return {'rank':0, 'description':'No matching product identifier found','maxNlc':None, 'minNlc':None,'status':None,'dp':None}
19580 kshitij.so 1573
    category_id = skuData[0]['category_id']
1574
    if category_id != 6:
15129 kshitij.so 1575
        user_specific_deals = mc.get(str(userId))
19580 kshitij.so 1576
        if user_specific_deals is None:
1577
            __populateCache(userId)
1578
            user_specific_deals = mc.get(str(userId))
1579
        else:
1580
            print "Getting user deals from cache mobiles/tablets"
15129 kshitij.so 1581
    else:
19580 kshitij.so 1582
        user_specific_deals = mc.get(str(userId)+'_acc')
1583
        if user_specific_deals is None:
1584
            __populateCacheForAccessories(userId)
1585
            user_specific_deals = mc.get(str(userId)+'_acc')
1586
        else:
1587
            print "Getting user deals from accessory cache"
1588
 
15129 kshitij.so 1589
    category_specific_deals = user_specific_deals.get(category_id)
16280 kshitij.so 1590
 
1591
    dealsData = get_mongo_connection().Catalog.Deals.find_one({'skuBundleId':skuData[0]['skuBundleId']})
1592
    if dealsData is None:
1593
        dealsData = {}
19580 kshitij.so 1594
        return {'rank':0, 'description':'Deal deleted from system','maxNlc':None, 'minNlc':None,'status':None,'dp':None}
16280 kshitij.so 1595
 
15129 kshitij.so 1596
    if category_specific_deals is None or len(category_specific_deals) ==0:
16280 kshitij.so 1597
        return {'rank':0,'description':'Category specific deals is empty','maxNlc':dealsData.get('maxNlc'),'minNlc':dealsData.get('minNlc'),'status':dealsData.get('status'),'dp':dealsData.get('dp')}
19580 kshitij.so 1598
 
1599
    if category_id!=6:
1600
        sorted_deals = sorted(category_specific_deals, key = lambda x: (x['persPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints'], x['rank']),reverse=True)
1601
    else:
1602
        sorted_deals = sorted(category_specific_deals, key = lambda x: (x['dealRankPoints'],x['totalPoints'],x['bestSellerPoints'], x['nlcPoints']),reverse=True)
1603
 
16168 kshitij.so 1604
    sortedMap = {}
1605
    rankMap = {}
1606
    rank = 0
15129 kshitij.so 1607
    for sorted_deal in sorted_deals:
16168 kshitij.so 1608
        if sortedMap.get(sorted_deal['skuBundleId']) is None:
1609
            sortedMap[sorted_deal['skuBundleId']] = {rank:[sorted_deal]}
1610
            rankMap[rank] = (sortedMap[sorted_deal['skuBundleId']].values())[0]
1611
            rank = rank +1
1612
        else:
1613
            for temp_list in sortedMap.get(sorted_deal['skuBundleId']).itervalues():
1614
                temp_list.append(sorted_deal)
1615
            rankMap[(sortedMap.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
1616
 
16187 kshitij.so 1617
    for dealRank ,dealList in rankMap.iteritems():
16168 kshitij.so 1618
        for d in dealList:
16186 kshitij.so 1619
            if d['skuBundleId'] == skuData[0]['skuBundleId']:
16280 kshitij.so 1620
                return {'rank':dealRank+1,'description':'Rank found','maxNlc':dealsData.get('maxNlc'),'minNlc':dealsData.get('minNlc'),'status':dealsData.get('status'),'dp':dealsData.get('dp')}
16168 kshitij.so 1621
 
16280 kshitij.so 1622
    return {'rank':0,'description':'Rank not found','maxNlc':dealsData.get('maxNlc'),'minNlc':dealsData.get('minNlc'),'status':dealsData.get('status'),'dp':dealsData.get('dp')}
15129 kshitij.so 1623
 
1624
 
13836 kshitij.so 1625
def getCashBackDetails(identifier, source_id):
17469 kshitij.so 1626
    if mc.get("category_cash_back") is None or not bool(mc.get("category_cash_back")):
13921 kshitij.so 1627
        populateCashBack()
13771 kshitij.so 1628
 
17035 kshitij.so 1629
    if source_id in (1,2,4,5,6,7):
13839 kshitij.so 1630
        skuData = list(get_mongo_connection().Catalog.MasterData.find({'identifier':identifier.strip(), 'source_id':source_id}))
13840 kshitij.so 1631
    elif source_id == 3:
13839 kshitij.so 1632
        skuData = list(get_mongo_connection().Catalog.MasterData.find({'secondaryIdentifier':identifier.strip(), 'source_id':source_id}))
13840 kshitij.so 1633
    else:
1634
        return {}
13836 kshitij.so 1635
    if len(skuData) > 0:
14761 kshitij.so 1636
        itemCashBackMap = mc.get("item_cash_back")
13921 kshitij.so 1637
        itemCashBack = itemCashBackMap.get(skuData[0]['_id'])
1638
        if itemCashBack is not None:
1639
            return itemCashBack
14761 kshitij.so 1640
        cashBackMap = mc.get("category_cash_back")
13921 kshitij.so 1641
        sourceCashBack = cashBackMap.get(source_id)
1642
        if sourceCashBack is not None and len(sourceCashBack) > 0:
1643
            for cashBack in sourceCashBack:
1644
                if cashBack.get(skuData[0]['category_id']) is None:
1645
                    continue
1646
                else:
1647
                    return cashBack.get(skuData[0]['category_id'])
13836 kshitij.so 1648
        else:
1649
            return {} 
1650
    else:
1651
        return {}
13986 amit.gupta 1652
 
14398 amit.gupta 1653
def getImgSrc(identifier, source_id):
1654
    skuData = None
17035 kshitij.so 1655
    if source_id in (1,2,4,5,6,7):
14414 amit.gupta 1656
        skuData = get_mongo_connection().Catalog.MasterData.find_one({'identifier':identifier.strip(), 'source_id':source_id})
14398 amit.gupta 1657
    elif source_id == 3:
14414 amit.gupta 1658
        skuData = get_mongo_connection().Catalog.MasterData.find_one({'secondaryIdentifier':identifier.strip(), 'source_id':source_id})
14398 amit.gupta 1659
    if skuData is None:
1660
        return {}
1661
    else:
16483 kshitij.so 1662
        try:
1663
            return {'thumbnail':skuData.get('thumbnail').strip()}
1664
        except:
1665
            return {'thumbnail':skuData.get('thumbnail')}
13986 amit.gupta 1666
 
1667
def next_weekday(d, weekday):
1668
    days_ahead = weekday - d.weekday()
1669
    if days_ahead <= 0: # Target day already happened this week
1670
        days_ahead += 7
1671
    return d + timedelta(days_ahead)
1672
 
13771 kshitij.so 1673
 
13970 kshitij.so 1674
def getAllDealerPrices(offset, limit):
1675
    data = []
1676
    collection = get_mongo_connection().Catalog.SkuDealerPrices
1677
    cursor = collection.find().skip(offset).limit(limit)
1678
    for val in cursor:
15853 kshitij.so 1679
        master = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':val['skuBundleId']})
1680
        if master is not None:
1681
            val['brand'] = master['brand']
1682
            val['source_product_name'] = master['source_product_name']
13970 kshitij.so 1683
        else:
1684
            val['brand'] = ""
1685
            val['source_product_name'] = ""
16231 kshitij.so 1686
        val['showDp'] = int(val['showDp'])
13970 kshitij.so 1687
        data.append(val)
1688
    return data
1689
 
15853 kshitij.so 1690
def addSkuDealerPrice(data):
1691
    collection = get_mongo_connection().Catalog.SkuDealerPrices
1692
    cursor = collection.find_one({"skuBundleId":data['skuBundleId']})
1693
    if cursor is not None:
1694
        return {0:"BundleId information already present."}
13970 kshitij.so 1695
    else:
15853 kshitij.so 1696
        collection.insert(data)
1697
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
13970 kshitij.so 1698
        return {1:"Data added successfully"}
1699
 
1700
def updateSkuDealerPrice(data, _id):
1701
    try:
1702
        collection = get_mongo_connection().Catalog.SkuDealerPrices
1703
        collection.update({'_id':ObjectId(_id)},{"$set":{'dp':data['dp']}},upsert=False, multi = False)
15853 kshitij.so 1704
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
13970 kshitij.so 1705
        return {1:"Data updated successfully"}
1706
    except:
1707
        return {0:"Data not updated."}
1708
 
1709
def updateExceptionalNlc(data, _id):
1710
    try:
1711
        collection = get_mongo_connection().Catalog.ExceptionalNlc
1712
        collection.update({'_id':ObjectId(_id)},{"$set":{'maxNlc':data['maxNlc'], 'minNlc':data['minNlc'], 'overrideNlc':data['overrideNlc']}},upsert=False, multi = False)
15853 kshitij.so 1713
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
13970 kshitij.so 1714
        return {1:"Data updated successfully"}
1715
    except:
1716
        return {0:"Data not updated."}
1717
 
14041 kshitij.so 1718
def resetCache(userId):
14043 kshitij.so 1719
    try:
1720
        mc.delete(userId)
19559 kshitij.so 1721
        mc.delete(userId+'_acc')
14043 kshitij.so 1722
        return {1:'Cache cleared.'}
1723
    except:
1724
        return {0:'Unable to clear cache.'}
1725
 
15853 kshitij.so 1726
def updateCollection(data):
1727
    try:
1728
        collection = get_mongo_connection().Catalog[data['class']]
1729
        class_name = data.pop('class')
1730
        _id = data.pop('oid')
1731
        result = collection.update({'_id':ObjectId(_id)},{"$set":data},upsert=False, multi = False)
1732
        if class_name != "Notifications":
1733
            record = list(collection.find({'_id':ObjectId(_id)}))
1734
            if class_name !="CategoryDiscount":
1735
                if record[0].has_key('sku'):
1736
                    field = '_id'
1737
                    val = record[0]['sku']
14852 kshitij.so 1738
                else:
15853 kshitij.so 1739
                    field = 'skuBundleId'
1740
                    val = record[0]['skuBundleId']
1741
                get_mongo_connection().Catalog.MasterData.update({field:val},{"$set":{'updatedOn':to_java_date(datetime.now())}},upsert=False, multi = True)
1742
            else:
1743
                get_mongo_connection().Catalog.MasterData.update({'brand':re.compile(record[0]['brand'], re.IGNORECASE),'category_id':record[0]['category_id']}, \
1744
                                                                 {"$set":{'updatedOn':to_java_date(datetime.now())}},upsert=False,multi=True)
16233 kshitij.so 1745
        if class_name =='SkuDealerPrices':
16237 kshitij.so 1746
            get_mongo_connection().Catalog.Deals.update({'skuBundleId':val},{"$set":data},upsert=False, multi=True)
15853 kshitij.so 1747
        return {1:"Data updated successfully"}
1748
    except Exception as e:
1749
        print e
1750
        return {0:"Data not updated."}
14575 kshitij.so 1751
 
14076 kshitij.so 1752
 
14553 kshitij.so 1753
def addNegativeDeals(data, multi):
1754
    if multi !=1: 
1755
        collection = get_mongo_connection().Catalog.NegativeDeals
1756
        cursor = collection.find({"sku":data['sku']})
1757
        if cursor.count() > 0:
1758
            return {0:"Sku information already present."}
1759
        else:
1760
            collection.insert(data)
1761
            return {1:"Data added successfully"}
14481 kshitij.so 1762
    else:
14553 kshitij.so 1763
        skuIds = __getBundledSkusfromSku(data['sku'])
1764
        for sku in skuIds:
1765
            data['sku'] = sku.get('_id')
1766
            collection = get_mongo_connection().Catalog.NegativeDeals
1767
            cursor = collection.find({"sku":data['sku']})
1768
            if cursor.count() > 0:
1769
                continue
1770
            else:
14558 kshitij.so 1771
                data.pop('_id',None)
14553 kshitij.so 1772
                collection.insert(data)
14481 kshitij.so 1773
        return {1:"Data added successfully"}
1774
 
1775
def getAllNegativeDeals(offset, limit):
1776
    data = []
1777
    collection = get_mongo_connection().Catalog.NegativeDeals
1778
    cursor = collection.find().skip(offset).limit(limit)
1779
    for val in cursor:
1780
        master = list(get_mongo_connection().Catalog.MasterData.find({'_id':val['sku']}))
1781
        if len(master) > 0:
1782
            val['brand'] = master[0]['brand']
1783
            val['source_product_name'] = master[0]['source_product_name']
1784
            val['skuBundleId'] = master[0]['skuBundleId']
1785
        else:
1786
            val['brand'] = ""
1787
            val['source_product_name'] = ""
1788
            val['skuBundleId'] = ""
1789
        data.append(val)
1790
    return data
1791
 
1792
def getAllManualDeals(offset, limit):
1793
    data = []
1794
    collection = get_mongo_connection().Catalog.ManualDeals
15090 kshitij.so 1795
    cursor = collection.find().skip(offset).limit(limit)
14481 kshitij.so 1796
    for val in cursor:
1797
        master = list(get_mongo_connection().Catalog.MasterData.find({'_id':val['sku']}))
1798
        if len(master) > 0:
1799
            val['brand'] = master[0]['brand']
1800
            val['source_product_name'] = master[0]['source_product_name']
1801
            val['skuBundleId'] = master[0]['skuBundleId']
1802
        else:
1803
            val['brand'] = ""
1804
            val['source_product_name'] = ""
1805
            val['skuBundleId'] = ""
1806
        data.append(val)
1807
    return data
14076 kshitij.so 1808
 
14553 kshitij.so 1809
def addManualDeal(data, multi):
1810
    if multi !=1:
1811
        collection = get_mongo_connection().Catalog.ManualDeals
15090 kshitij.so 1812
        cursor = collection.find({'sku':data['sku']})
14553 kshitij.so 1813
        if cursor.count() > 0:
1814
            return {0:"Sku information already present."}
1815
        else:
1816
            collection.insert(data)
1817
            get_mongo_connection().Catalog.MasterData.update({'_id':data['sku']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=False)
1818
            return {1:"Data added successfully"}
14481 kshitij.so 1819
    else:
14553 kshitij.so 1820
        skuIds = __getBundledSkusfromSku(data['sku'])
1821
        for sku in skuIds:
1822
            data['sku'] = sku.get('_id')
1823
            collection = get_mongo_connection().Catalog.ManualDeals
15090 kshitij.so 1824
            cursor = collection.find({'sku':data['sku']})
14553 kshitij.so 1825
            if cursor.count() > 0:
1826
                continue
1827
            else:
14558 kshitij.so 1828
                data.pop('_id',None)
14553 kshitij.so 1829
                collection.insert(data)
1830
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':sku.get('skuBundleId')},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
14481 kshitij.so 1831
        return {1:"Data added successfully"}
14553 kshitij.so 1832
 
14481 kshitij.so 1833
def deleteDocument(data):
15853 kshitij.so 1834
    print "inside detete document"
14481 kshitij.so 1835
    print data
1836
    try:
1837
        collection = get_mongo_connection().Catalog[data['class']]
1838
        class_name = data.pop('class')
1839
        _id = data.pop('oid')
1840
        record = list(collection.find({'_id':ObjectId(_id)}))
1841
        collection.remove({'_id':ObjectId(_id)})
15075 kshitij.so 1842
        if class_name != "Notifications":
1843
            if class_name !="CategoryDiscount":
15853 kshitij.so 1844
                print record[0]
1845
                if record[0].has_key('sku'):
1846
                    field = '_id'
1847
                    val = record[0]['sku']
1848
                else:
1849
                    field = 'skuBundleId'
1850
                    val = record[0]['skuBundleId']
1851
                print "Updating master"
1852
                print field
1853
                print val
1854
                get_mongo_connection().Catalog.MasterData.update({field:val},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
15075 kshitij.so 1855
            else:
1856
                get_mongo_connection().Catalog.MasterData.update({'brand':re.compile(record[0]['brand'], re.IGNORECASE),'category_id':record[0]['category_id']}, \
1857
                                                                 {"$set":{'updatedOn':to_java_date(datetime.now())}},upsert=False,multi=True)
14481 kshitij.so 1858
        return {1:"Document deleted successfully"}
1859
    except Exception as e:
1860
        print e
1861
        return {0:"Document not deleted."}
13970 kshitij.so 1862
 
14482 kshitij.so 1863
def searchMaster(offset, limit, search_term):
1864
    data = []
14531 kshitij.so 1865
    if search_term is not None:
14551 kshitij.so 1866
        terms = search_term.split(' ')
1867
        outer_query = []
1868
        for term in terms:
1869
            outer_query.append({"source_product_name":re.compile(term, re.IGNORECASE)})
14531 kshitij.so 1870
        try:
14551 kshitij.so 1871
            collection = get_mongo_connection().Catalog.MasterData.find({"$and":outer_query,'source_id':{'$in':SOURCE_MAP.keys()}}).skip(offset).limit(limit)
14531 kshitij.so 1872
            for record in collection:
1873
                data.append(record)
1874
        except:
1875
            pass
1876
    else:
1877
        collection = get_mongo_connection().Catalog.MasterData.find({'source_id':{'$in':SOURCE_MAP.keys()}}).skip(offset).limit(limit)
14482 kshitij.so 1878
        for record in collection:
1879
            data.append(record)
1880
    return data
14481 kshitij.so 1881
 
14495 kshitij.so 1882
def getAllFeaturedDeals(offset, limit):
1883
    data = []
1884
    collection = get_mongo_connection().Catalog.FeaturedDeals
17561 kshitij.so 1885
    cursor = collection.find({}).skip(offset).limit(limit)
14495 kshitij.so 1886
    for val in cursor:
17367 kshitij.so 1887
        master = list(get_mongo_connection().Catalog.MasterData.find({'skuBundleId':val['skuBundleId']}))
14495 kshitij.so 1888
        if len(master) > 0:
1889
            val['brand'] = master[0]['brand']
1890
            val['source_product_name'] = master[0]['source_product_name']
1891
            val['skuBundleId'] = master[0]['skuBundleId']
1892
        else:
1893
            val['brand'] = ""
1894
            val['source_product_name'] = ""
1895
            val['skuBundleId'] = ""
1896
        data.append(val)
1897
    return data
1898
 
14553 kshitij.so 1899
def addFeaturedDeal(data, multi):
17367 kshitij.so 1900
    collection = get_mongo_connection().Catalog.FeaturedDeals
1901
    cursor = collection.find({'skuBundleId':data['skuBundleId']})
1902
    master = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':data['skuBundleId']})
1903
    if master is None:
1904
        return {0:"BundleId is wrong"}
17469 kshitij.so 1905
    otherInfoObj = data['otherInfo']
1906
    toPop = []
1907
    for x,y in otherInfoObj.iteritems():
1908
        if y==0:
1909
            toPop.append(x)
1910
 
1911
    for popItem in toPop:
1912
        otherInfoObj.pop(popItem)
1913
    if len(otherInfoObj.keys())==0:
1914
        return {0:"No rank info to add"}
1915
 
1916
    for x,y in otherInfoObj.iteritems():
1917
        exist = get_mongo_connection().Catalog.FeaturedDeals.find({'otherInfo.'+x:y})
1918
        if exist.count() >0:
1919
            return {0:"Rank already assigned bundleId %s"%(exist[0]['skuBundleId'])}
17367 kshitij.so 1920
    if cursor.count() > 0:
1921
        return {0:"SkuBundleId information already present."}
14495 kshitij.so 1922
    else:
17367 kshitij.so 1923
        collection.insert(data)
14495 kshitij.so 1924
        return {1:"Data added successfully"}
1925
 
14499 kshitij.so 1926
def searchCollection(class_name, sku, skuBundleId):
14497 kshitij.so 1927
    data = []
1928
    collection = get_mongo_connection().Catalog[class_name]
15076 kshitij.so 1929
    if class_name == "Notifications":
1930
        cursor = collection.find({'skuBundleId':skuBundleId})
1931
        for val in cursor:
1932
            master = list(get_mongo_connection().Catalog.MasterData.find({'skuBundleId':val['skuBundleId']}))
1933
            if len(master) > 0:
1934
                val['brand'] = master[0]['brand']
1935
                val['model_name'] = master[0]['model_name']
1936
                val['skuBundleId'] = master[0]['skuBundleId']
1937
            else:
1938
                val['brand'] = ""
1939
                val['model_name'] = ""
1940
                val['skuBundleId'] = val['skuBundleId']
1941
            data.append(val)
15095 kshitij.so 1942
        return data
15853 kshitij.so 1943
    master = None
14499 kshitij.so 1944
    if sku is not None:
15853 kshitij.so 1945
        if COLLECTION_MAP.has_key(class_name):
1946
            master = get_mongo_connection().Catalog.MasterData.find_one({'_id':sku})
1947
            cursor = collection.find({'skuBundleId':master['skuBundleId']})
1948
        else:
1949
            cursor = collection.find({'sku':sku})
14499 kshitij.so 1950
        for val in cursor:
15853 kshitij.so 1951
            if master is None:
1952
                master = get_mongo_connection().Catalog.MasterData.find_one({'_id':val['sku']})
1953
            if master is not None:
1954
                val['brand'] = master['brand']
1955
                val['source_product_name'] = master['source_product_name']
1956
                val['skuBundleId'] = master['skuBundleId']
14499 kshitij.so 1957
            else:
1958
                val['brand'] = ""
1959
                val['source_product_name'] = ""
1960
                val['skuBundleId'] = ""
1961
            data.append(val)
1962
        return data
1963
    else:
15853 kshitij.so 1964
        if not COLLECTION_MAP.has_key(class_name):
1965
            skuIds = get_mongo_connection().Catalog.MasterData.find({'skuBundleId':skuBundleId}).distinct('_id')
1966
            for sku in skuIds:
1967
                cursor = collection.find({'sku':sku})
1968
                for val in cursor:
1969
                    master = list(get_mongo_connection().Catalog.MasterData.find({'_id':val['sku']}))
1970
                    if len(master) > 0:
1971
                        val['brand'] = master[0]['brand']
1972
                        val['source_product_name'] = master[0]['source_product_name']
1973
                        val['skuBundleId'] = master[0]['skuBundleId']
1974
                    else:
1975
                        val['brand'] = ""
1976
                        val['source_product_name'] = ""
1977
                        val['skuBundleId'] = ""
1978
                    data.append(val)
1979
            return data
1980
        else:
1981
            cursor = collection.find({'skuBundleId':skuBundleId})
14499 kshitij.so 1982
            for val in cursor:
15853 kshitij.so 1983
                master = list(get_mongo_connection().Catalog.MasterData.find({'skuBundleId':val['skuBundleId']}))
14499 kshitij.so 1984
                if len(master) > 0:
1985
                    val['brand'] = master[0]['brand']
1986
                    val['source_product_name'] = master[0]['source_product_name']
1987
                else:
1988
                    val['brand'] = ""
1989
                    val['source_product_name'] = ""
1990
                data.append(val)
15853 kshitij.so 1991
            return data
14495 kshitij.so 1992
 
15074 kshitij.so 1993
def __getBrandIdForBrand(brandName, category_id):
1994
    brandInfo = Brands.query.filter(Brands.category_id==category_id).filter(Brands.name == brandName).all()
1995
    if brandInfo is None or len(brandInfo)!=1:
1996
        raise
1997
    else:
1998
        return brandInfo[0].id
1999
 
14588 kshitij.so 2000
def addNewItem(data):
14594 kshitij.so 2001
    try:
2002
        data['updatedOn'] = to_java_date(datetime.now())
2003
        data['addedOn'] = to_java_date(datetime.now())
2004
        data['priceUpdatedOn'] = to_java_date(datetime.now())
2005
        max_id = list(get_mongo_connection().Catalog.MasterData.find().sort([('_id',pymongo.DESCENDING)]).limit(1))
2006
        max_bundle = list(get_mongo_connection().Catalog.MasterData.find().sort([('skuBundleId',pymongo.DESCENDING)]).limit(1))
2007
        data['_id'] = max_id[0]['_id'] + 1
2008
        data['skuBundleId'] = max_bundle[0]['skuBundleId'] + 1
2009
        data['identifier'] = str(data['identifier'])
2010
        data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
17115 manish.sha 2011
        data['model_name'] = str(data['model_name'])
15074 kshitij.so 2012
        data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
17746 kshitij.so 2013
        data['shippingCost'] = float(data['shippingCost'])
19559 kshitij.so 2014
        data['internalRank'] = int(data['internalRank'])
17603 kshitij.so 2015
        if data['category_id'] ==6 and data['subCategoryId'] ==0:
17606 kshitij.so 2016
            return {0:'Sub-Category not selected'}
17608 kshitij.so 2017
        elif data['category_id'] !=6 and data['subCategoryId'] !=0:
2018
            return {0:'Sub-Category not valid'}
2019
        else:
2020
            pass
17604 kshitij.so 2021
        data['subCategory'] = SUB_CATEGORY_MAP.get(data['subCategoryId'])
14594 kshitij.so 2022
        get_mongo_connection().Catalog.MasterData.insert(data)
2023
        return {1:'Data added successfully'}
2024
    except Exception as e:
2025
        print e
2026
        return {0:'Unable to add data.'}
15130 kshitij.so 2027
    finally:
2028
        session.close()
14588 kshitij.so 2029
 
2030
def addItemToExistingBundle(data):
2031
    try:
2032
        data['updatedOn'] = to_java_date(datetime.now())
2033
        data['addedOn'] = to_java_date(datetime.now())
2034
        data['priceUpdatedOn'] = to_java_date(datetime.now())
14593 kshitij.so 2035
        max_id = list(get_mongo_connection().Catalog.MasterData.find().sort([('_id',pymongo.DESCENDING)]).limit(1))
14590 kshitij.so 2036
        data['_id'] = max_id[0]['_id'] + 1
14594 kshitij.so 2037
        data['identifier'] = str(data['identifier'])
2038
        data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
17115 manish.sha 2039
        data['model_name'] = str(data['model_name'])
15074 kshitij.so 2040
        data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
17746 kshitij.so 2041
        data['shippingCost'] = float(data['shippingCost'])
19559 kshitij.so 2042
        data['internalRank'] = int(data['internalRank'])
17603 kshitij.so 2043
        if data['category_id'] ==6 and data['subCategoryId'] ==0:
17606 kshitij.so 2044
            return {0:'Sub-Category not selected'}
17608 kshitij.so 2045
        elif data['category_id'] !=6 and data['subCategoryId'] !=0:
2046
            return {0:'Sub-Category not valid'}
2047
        else:
2048
            pass
17604 kshitij.so 2049
        data['subCategory'] = SUB_CATEGORY_MAP.get(data['subCategoryId'])
14588 kshitij.so 2050
        get_mongo_connection().Catalog.MasterData.insert(data)
2051
        return {1:'Data added successfully.'}
14594 kshitij.so 2052
    except Exception as e:
2053
        print e
14588 kshitij.so 2054
        return {0:'Unable to add data.'}
15130 kshitij.so 2055
    finally:
2056
        session.close()
14588 kshitij.so 2057
 
2058
def updateMaster(data, multi):
15130 kshitij.so 2059
    try:
2060
        print data
2061
        if multi != 1:
2062
            _id = data.pop('_id')
2063
            skuBundleId = data.pop('skuBundleId')
2064
            data['updatedOn'] = to_java_date(datetime.now())
2065
            data['identifier'] = str(data['identifier'])
2066
            data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
17115 manish.sha 2067
            data['model_name'] = str(data['model_name'])
17746 kshitij.so 2068
            data['shippingCost'] = float(data['shippingCost'])
19559 kshitij.so 2069
            data['internalRank'] = int(data['internalRank'])
15130 kshitij.so 2070
            data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
17603 kshitij.so 2071
            if data['category_id'] ==6 and data['subCategoryId'] ==0:
17606 kshitij.so 2072
                return {0:'Sub-Category not selected'}
17608 kshitij.so 2073
            elif data['category_id'] !=6 and data['subCategoryId'] !=0:
2074
                return {0:'Sub-Category not valid'}
2075
            else:
2076
                pass
17604 kshitij.so 2077
            data['subCategory'] = SUB_CATEGORY_MAP.get(data['subCategoryId'])
15130 kshitij.so 2078
            get_mongo_connection().Catalog.MasterData.update({'_id':_id},{"$set":data},upsert=False)
19559 kshitij.so 2079
            get_mongo_connection().Catalog.MasterData.update({'skuBundleId':skuBundleId},{"$set":{'internalRank':data['internalRank']}},upsert=False, multi=True)
15130 kshitij.so 2080
            return {1:'Data updated successfully.'}
2081
        else:
2082
            _id = data.pop('_id')
2083
            skuBundleId = data.pop('skuBundleId')
2084
            data['updatedOn'] = to_java_date(datetime.now())
2085
            data['identifier'] = str(data['identifier'])
2086
            data['secondaryIdentifier'] = str(data['secondaryIdentifier'])
2087
            data['brand_id'] = __getBrandIdForBrand(data['brand'], data['category_id'])
17115 manish.sha 2088
            data['model_name'] = str(data['model_name'])
17746 kshitij.so 2089
            data['shippingCost'] = float(data['shippingCost'])
19559 kshitij.so 2090
            data['internalRank'] = int(data['internalRank'])
17603 kshitij.so 2091
            if data['category_id'] ==6 and data['subCategoryId'] ==0:
17606 kshitij.so 2092
                return {0:'Sub-Category not selected'}
17608 kshitij.so 2093
            elif data['category_id'] !=6 and data['subCategoryId'] !=0:
2094
                return {0:'Sub-Category not valid'}
2095
            else:
2096
                pass
17604 kshitij.so 2097
            data['subCategory'] = SUB_CATEGORY_MAP.get(data['subCategoryId'])
15130 kshitij.so 2098
            get_mongo_connection().Catalog.MasterData.update({'_id':_id},{"$set":data},upsert=False)
2099
            similarItems = get_mongo_connection().Catalog.MasterData.find({'skuBundleId':skuBundleId})
2100
            for item in similarItems:
2101
                if item['_id'] == _id:
2102
                    continue
2103
                item['updatedOn'] = to_java_date(datetime.now())
2104
                item['thumbnail'] = data['thumbnail']
2105
                item['category'] = data['category']
2106
                item['category_id'] = data['category_id']
2107
                item['is_shortage'] = data['is_shortage']
2108
                item['mrp'] = data['mrp']
2109
                item['status'] = data['status']
2110
                item['maxPrice'] = data['maxPrice']
2111
                item['brand_id'] = data['brand_id']
17603 kshitij.so 2112
                item['subCategoryId'] = data['subCategoryId']
2113
                item['subCategory'] = data['subCategory']
19559 kshitij.so 2114
                item['internalRank'] = data['internalRank']
15130 kshitij.so 2115
                similar_item_id = item.pop('_id')
2116
                get_mongo_connection().Catalog.MasterData.update({'_id':similar_item_id},{"$set":item},upsert=False)
2117
            return {1:'Data updated successfully.'}
17603 kshitij.so 2118
    except:
2119
        return {0:'Unable to update data'}
15130 kshitij.so 2120
    finally:
2121
        session.close()
14619 kshitij.so 2122
 
2123
def getLiveCricScore():
2124
    return mc.get('liveScore')
14852 kshitij.so 2125
 
2126
def addBundleToNotification(data):
2127
    try:
15069 kshitij.so 2128
        collection = get_mongo_connection().Catalog.Notifications
14852 kshitij.so 2129
        cursor = collection.find({'skuBundleId':data['skuBundleId']})
2130
        if cursor.count() > 0:
2131
            return {0:"SkuBundleId information already present."}
2132
        else:
2133
            collection.insert(data)
2134
            return {1:'Data updated successfully.'}
2135
    except:
2136
        return {0:'Unable to add data.'}
2137
 
2138
def getAllNotifications(offset, limit):
2139
    data = []
15069 kshitij.so 2140
    collection = get_mongo_connection().Catalog.Notifications
16450 kshitij.so 2141
    cursor = collection.find().sort([('endDate',pymongo.DESCENDING)]).skip(offset).limit(limit)
14852 kshitij.so 2142
    for val in cursor:
15072 kshitij.so 2143
        master = list(get_mongo_connection().Catalog.MasterData.find({'skuBundleId':val['skuBundleId']}))
14852 kshitij.so 2144
        if len(master) > 0:
2145
            val['brand'] = master[0]['brand']
15071 kshitij.so 2146
            val['model_name'] = master[0]['model_name']
14852 kshitij.so 2147
            val['skuBundleId'] = master[0]['skuBundleId']
2148
        else:
2149
            val['brand'] = ""
15071 kshitij.so 2150
            val['model_name'] = ""
2151
            val['skuBundleId'] = val['skuBundleId']
14852 kshitij.so 2152
        data.append(val)
2153
    return data
14997 kshitij.so 2154
 
2155
def getBrandsForFilter(category_id):
2156
    if mc.get("brandFilter") is None:
15095 kshitij.so 2157
        print "Populating brand data for category_id %d" %(category_id)
17469 kshitij.so 2158
        tabData, mobData, accData = [], [], []
14997 kshitij.so 2159
        mobileDeals = get_mongo_connection().Catalog.Deals.aggregate([
16170 kshitij.so 2160
                                                                      {"$match":{"category_id":3,"showDeal":1,"totalPoints":{"$gt":-100}}
14997 kshitij.so 2161
                                                                    },
2162
                                                                 {"$group" : 
2163
                                                                  {'_id':{'brand_id':'$brand_id','brand':'$brand'},'count':{'$sum':1}}
2164
                                                                  }
2165
                                                                ])
14588 kshitij.so 2166
 
14997 kshitij.so 2167
        tabletDeals = get_mongo_connection().Catalog.Deals.aggregate([
16170 kshitij.so 2168
                                                                      {"$match":{"category_id":5,"showDeal":1,"totalPoints":{"$gt":-100}}
14997 kshitij.so 2169
                                                                    },
2170
                                                                 {"$group" : 
2171
                                                                  {'_id':{'brand_id':'$brand_id','brand':'$brand'},'count':{'$sum':1}}
2172
                                                                  }
2173
                                                                ])
2174
 
17469 kshitij.so 2175
        accessoriesDeals = get_mongo_connection().Catalog.Deals.aggregate([
19597 kshitij.so 2176
                                                                      {"$match":{"category_id":6,"showDeal":1,"dealRankPoints":{"$gt":0}}
17469 kshitij.so 2177
                                                                    },
2178
                                                                 {"$group" : 
2179
                                                                  {'_id':{'brand_id':'$brand_id','brand':'$brand'},'count':{'$sum':1}}
2180
                                                                  }
2181
                                                                ])
2182
 
14997 kshitij.so 2183
        #print mobileDeals
2184
        #print "==========Mobile data ends=========="
2185
 
2186
        #print tabletDeals
2187
        #print "==========Tablet data ends=========="
2188
 
2189
        #print allDeals
2190
        #print "==========All deal data ends========="
2191
 
2192
        for mobileDeal in mobileDeals['result']:
2193
            if mobileDeal.get('_id').get('brand_id') != 0:
2194
                tempMap = {}
2195
                tempMap['brand'] = mobileDeal.get('_id').get('brand')
2196
                tempMap['brand_id'] = mobileDeal.get('_id').get('brand_id')
2197
                tempMap['count'] = mobileDeal.get('count')
2198
                mobData.append(tempMap)
2199
 
2200
        for tabletDeal in tabletDeals['result']:
2201
            if tabletDeal.get('_id').get('brand_id') != 0:
2202
                tempMap = {}
2203
                tempMap['brand'] = tabletDeal.get('_id').get('brand')
2204
                tempMap['brand_id'] = tabletDeal.get('_id').get('brand_id')
2205
                tempMap['count'] = tabletDeal.get('count')
2206
                tabData.append(tempMap)
2207
 
17469 kshitij.so 2208
        for accessoryDeal in accessoriesDeals['result']:
2209
            if accessoryDeal.get('_id').get('brand_id') != 0:
2210
                tempMap = {}
2211
                tempMap['brand'] = accessoryDeal.get('_id').get('brand')
2212
                tempMap['brand_id'] = accessoryDeal.get('_id').get('brand_id')
2213
                tempMap['count'] = accessoryDeal.get('count')
2214
                accData.append(tempMap)
14997 kshitij.so 2215
 
19597 kshitij.so 2216
        mc.set("brandFilter",{3:mobData, 5:tabData, 6:accData}, 600)  
14997 kshitij.so 2217
 
15062 kshitij.so 2218
    return sorted(mc.get("brandFilter").get(category_id), key = lambda x: (-x['count'], x['brand']))
15375 kshitij.so 2219
 
15459 kshitij.so 2220
def getStaticDeals(offset, limit, category_id, direction):
15375 kshitij.so 2221
    user_specific_deals = mc.get("staticDeals")
2222
    if user_specific_deals is None:
2223
        __populateStaticDeals()
2224
        user_specific_deals = mc.get("staticDeals")
15459 kshitij.so 2225
    rev = False
2226
    if direction is None or direction == -1:
2227
        rev=True
2228
    return sorted((user_specific_deals.get(category_id))[offset:offset+limit],reverse=rev)
15375 kshitij.so 2229
 
2230
def __populateStaticDeals():
2231
    print "Populating memcache for static deals"
2232
    outer_query = []
2233
    outer_query.append({"showDeal":1})
2234
    query = {}
16170 kshitij.so 2235
    query['$gt'] = -100
15375 kshitij.so 2236
    outer_query.append({'totalPoints':query})
2237
    all_deals = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query},{'_id':1,'category_id':1,'brand':1,'totalPoints':1,'bestSellerPoints':1,'nlcPoints':1,'rank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1}).sort([('totalPoints',pymongo.DESCENDING),('bestSellerPoints',pymongo.DESCENDING),('nlcPoints',pymongo.DESCENDING),('rank',pymongo.DESCENDING)]))
2238
    mobile_deals = []
2239
    tablet_deals = []
2240
    for deal in all_deals:
2241
        item = get_mongo_connection().Catalog.MasterData.find({'_id':deal['_id']})
2242
        if deal['category_id'] ==3:
2243
            mobile_deals.append(getItemObjForStaticDeals(item[0]))
2244
        elif deal['category_id'] ==5:
2245
            tablet_deals.append(getItemObjForStaticDeals(item[0]))
2246
        else:
2247
            continue
2248
 
2249
    random.shuffle(mobile_deals,random.random)
2250
    random.shuffle(tablet_deals,random.random)
2251
 
2252
    mem_cache_val = {3:mobile_deals, 5:tablet_deals}
2253
    mc.set("staticDeals", mem_cache_val, 3600)
2254
 
17733 kshitij.so 2255
def getItemObjForStaticDeals(item,transform=None):
2256
    mpu = str(item.get('marketPlaceUrl'))
2257
    if transform:
2258
        if mpu.find("snapdeal")!=-1:
2259
            mpu = "http://m.snapdeal.com"+mpu[mpu.find("/product"):]
2260
        elif mpu.find("homeshop18")!=-1:
2261
            mpu = "http://m.homeshop18.com/product.mobi?productId="+item['identifier']
17735 kshitij.so 2262
        elif mpu.find("saholic.com")!=-1:
17746 kshitij.so 2263
            mpu = "http://m."+mpu[mpu.find("saholic.com"):]
17735 kshitij.so 2264
        elif mpu.find("shopclues.com")!=-1:
17733 kshitij.so 2265
            mpu = "http://m."+mpu[mpu.find("shopclues.com"):]
2266
        else:
2267
            pass
2268
 
2269
    return {'marketPlaceUrl':mpu,'available_price':int(item.get('available_price')),'source_product_name':item.get('source_product_name'),'thumbnail':item.get('thumbnail'),'source_id':int(item.get('source_id'))}
16300 manas 2270
 
2271
def getItemByMerchantIdentifier(identifier, source_id):
2272
    skuData = None
17035 kshitij.so 2273
    if source_id in (1,2,4,5,6,7):
16300 manas 2274
        skuData = get_mongo_connection().Catalog.MasterData.find_one({'identifier':identifier.strip(), 'source_id':source_id})
2275
    elif source_id == 3:
2276
        skuData = get_mongo_connection().Catalog.MasterData.find_one({'secondaryIdentifier':identifier.strip(), 'source_id':source_id})
2277
    if skuData is None:
2278
        return {}
2279
    else:
2280
        return skuData
15375 kshitij.so 2281
 
17936 kshitij.so 2282
def getDealsForNotification(skuBundleIds,skipSplitting=False):
2283
    if not skipSplitting:
2284
        bundles= [int(skuBundleId) for skuBundleId in skuBundleIds.split(',')]
2285
        print bundles
2286
    else:
2287
        bundles = skuBundleIds
16406 kshitij.so 2288
    dealsList = []
16364 kshitij.so 2289
    dealsListMap = []
16406 kshitij.so 2290
    for bundleId in bundles:
2291
        outer_query = []
2292
        outer_query.append({ "$or": [ { "showDeal": 1} , { "prepaidDeal": 1 } ] })
2293
        outer_query.append({'skuBundleId':bundleId})
2294
        deals = get_mongo_connection().Catalog.Deals.find({"$and":outer_query})
2295
        for deal in deals:
2296
            dealsList.append(deal)
16364 kshitij.so 2297
    sortedMap = {}
2298
    rankMap = {}
2299
    rank = 0
16406 kshitij.so 2300
    for sorted_deal in dealsList:
16364 kshitij.so 2301
        if sortedMap.get(sorted_deal['skuBundleId']) is None:
2302
            sortedMap[sorted_deal['skuBundleId']] = {rank:[sorted_deal]}
2303
            rankMap[rank] = (sortedMap[sorted_deal['skuBundleId']].values())[0]
2304
            rank = rank +1
2305
        else:
2306
            for temp_list in sortedMap.get(sorted_deal['skuBundleId']).itervalues():
2307
                temp_list.append(sorted_deal)
2308
            rankMap[(sortedMap.get(sorted_deal['skuBundleId']).keys())[0]] = temp_list
2309
 
16367 kshitij.so 2310
    for dealList in [rankMap.get(k, []) for k in range(0, len(bundles))]:
16364 kshitij.so 2311
        temp = []
2312
        for d in dealList:
2313
            item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))
2314
            if len(item) ==0:
2315
                continue
2316
            if d['dealType'] == 1 and d['source_id'] ==1:
2317
                item[0]['marketPlaceUrl'] = "http://www.amazon.in/dp/%s"%(item[0]['identifier'].strip())
2318
            elif d['source_id'] ==3:
2319
                item[0]['marketPlaceUrl'] = item[0]['marketPlaceUrl']+'?supc='+item[0].get('identifier')
2320
            else:
2321
                pass 
2322
            try:
2323
                cashBack = getCashBack(item[0]['_id'], item[0]['source_id'], item[0]['category_id'])
2324
                if not cashBack or cashBack.get('cash_back_status')!=1:
2325
                    item[0]['cash_back_type'] = 0
2326
                    item[0]['cash_back'] = 0
2327
                else:
2328
                    item[0]['cash_back_type'] = int(cashBack['cash_back_type'])
2329
                    item[0]['cash_back'] = cashBack['cash_back']
2330
            except:
2331
                print "Error in adding cashback to deals"
2332
                item[0]['cash_back_type'] = 0
2333
                item[0]['cash_back'] = 0
2334
            try:
2335
                item[0]['dp'] = d['dp']
2336
                item[0]['showDp'] = d['showDp']
2337
            except:
2338
                item[0]['dp'] = 0.0
2339
                item[0]['showDp'] = 0
17727 kshitij.so 2340
 
2341
            try:
2342
                if item[0]['showVideo']==0:
2343
                    item[0]['videoLink'] =""
2344
            except:
2345
                item[0]['videoLink'] =""
2346
            try:
2347
                if item[0]['quantity'] >1:
2348
                    ppq = float(item[0]['available_price'])/item[0]['quantity']
2349
 
2350
                    if ppq == round(ppq):
2351
                        item[0]['ppq'] = int(ppq)
2352
                    else:
2353
                        item[0]['ppq'] = round(ppq,2)
2354
                else:
2355
                    item[0]['ppq'] = 0
2356
            except:
2357
                item[0]['ppq'] = 0
2358
            if item[0]['category_id']!=6:
2359
                try:
18830 kshitij.so 2360
                    item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
2361
                    item[0]['filterTextBrand'] = (item[0]['brand'])
17727 kshitij.so 2362
                except:
18830 kshitij.so 2363
                    pass
17727 kshitij.so 2364
            else:
2365
                try:
18051 kshitij.so 2366
                    if item[0]['subCategory']=='' or item[0]['subCategory'] is None or item[0]['subCategoryId']==0:
17727 kshitij.so 2367
                        raise
18830 kshitij.so 2368
                    item[0]['filterLinkSubCategory'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['subCategory'])+'&filter=subcategory&subcategories='+str(int(item[0]['subCategoryId']))
2369
                    item[0]['filterTextSubCategory'] = (item[0]['subCategory'])
2370
                    item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
2371
                    item[0]['filterTextBrand'] = (item[0]['brand'])
17727 kshitij.so 2372
                except:
18830 kshitij.so 2373
                    pass
17944 kshitij.so 2374
            try:
2375
                if item[0]['source_id']==4:
2376
                    item_availability_info = mc.get("item_availability_"+str(item[0]['identifier']).strip())
2377
                    if item_availability_info is None or len(item_availability_info)==0:
2378
                        item[0]['availabilityInfo'] = [{}]
2379
                    else:
2380
                        item[0]['availabilityInfo'] = item_availability_info
2381
            except:
2382
                item[0]['availabilityInfo'] = [{}]
19135 kshitij.so 2383
 
19193 kshitij.so 2384
            if item[0]['source_id']==4:
2385
                for av_info in item[0]['availabilityInfo']:
2386
                    if not av_info:
2387
                        continue
2388
                    av_info['cash_back'] = item[0]['cash_back']
2389
                    av_info['cash_back_type'] = item[0]['cash_back_type']
2390
                    netPriceAfterCashBack = av_info['sellingPrice'] 
19199 kshitij.so 2391
                    if av_info['cash_back_type'] ==1:
19207 kshitij.so 2392
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(float(av_info['sellingPrice'])*av_info['cash_back']/100)
19199 kshitij.so 2393
                    elif av_info['cash_back_type'] ==2:
19207 kshitij.so 2394
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(av_info['cash_back'])
19193 kshitij.so 2395
                    else:
2396
                        pass
2397
                    av_info['netPriceAfterCashBack'] = netPriceAfterCashBack
2398
 
2399
            if item[0]['source_id'] != SOURCE_MAP.get("PAYTM.COM"):
2400
                if item[0]['codAvailable'] ==1:
2401
                    paytmPrice = item[0]['available_price']
2402
                else:
2403
                    paytmPrice = item[0]['gross_price']
2404
 
2405
                if item[0]['cash_back_type'] ==1:
19207 kshitij.so 2406
                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(float(paytmPrice)*item[0]['cash_back']/100)
19193 kshitij.so 2407
                elif item[0]['cash_back_type'] ==2:
19207 kshitij.so 2408
                    item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(item[0]['cash_back'])
19193 kshitij.so 2409
                else:
2410
                    item[0]['netPriceAfterCashBack'] = paytmPrice
2411
 
2412
            else:
2413
                if item[0]['cash_back_type'] ==1:
19207 kshitij.so 2414
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(float(item[0]['available_price'])*item[0]['cash_back']/100)
19193 kshitij.so 2415
                elif item[0]['cash_back_type'] ==2:
19207 kshitij.so 2416
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(item[0]['cash_back'])
19193 kshitij.so 2417
                else:
2418
                    item[0]['netPriceAfterCashBack'] = item[0]['available_price']
2419
 
2420
 
19135 kshitij.so 2421
            item[0]['totalPoints'] = d['totalPoints']
17727 kshitij.so 2422
 
16364 kshitij.so 2423
            temp.append(item[0])
2424
        if len(temp) > 1:
2425
            temp = sorted(temp, key = lambda x: (x['available_price']),reverse=False)
2426
        dealsListMap.append(temp)
2427
    return dealsListMap
2428
 
16373 manas 2429
def getSkuBrandData(sku):
2430
    if sku is None:
2431
        return {}
2432
    else:
2433
        orders = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':sku})
2434
        if orders is None:
2435
            return {}
2436
        return orders
16488 kshitij.so 2437
 
2438
def addDealPoints(data):
2439
    collection = get_mongo_connection().Catalog.DealPoints
2440
    cursor = collection.find({'skuBundleId':data['skuBundleId']})
2441
    if cursor.count() > 0:
2442
        return {0:"Sku information already present."}
2443
    else:
2444
        collection.insert(data)
2445
        get_mongo_connection().Catalog.MasterData.update({'skuBundleId':data['skuBundleId']},{"$set":{'updatedOn':to_java_date(datetime.now())}},multi=True)
2446
        return {1:"Data added successfully"}
2447
 
2448
def getAllBundlesWithDealPoints(offset, limit):
2449
    data = []
2450
    collection = get_mongo_connection().Catalog.DealPoints
2451
    cursor = collection.find().sort([('endDate',pymongo.DESCENDING)]).skip(offset).limit(limit)
2452
    for val in cursor:
2453
        master = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':val.get('skuBundleId')})
2454
        if master is not None:
2455
            val['brand'] = master['brand']
2456
            val['source_product_name'] = master['product_name']
2457
        else:
2458
            val['brand'] = ""
2459
            val['source_product_name'] = ""
2460
        data.append(val)
2461
    return data
16546 kshitij.so 2462
 
2463
def generateRedirectUrl(retailer_id, app_id):
2464
    try:
16663 manish.sha 2465
        if retailer_id <= 0:
2466
            return {'url':"","message":"Retailer Id not valid"}
16546 kshitij.so 2467
        d_app_offer = app_offers.get_by(id=app_id)
16605 manish.sha 2468
    except:
2469
        traceback.print_exc()
16546 kshitij.so 2470
    finally:
2471
        session.close()
2472
    if d_app_offer is None:
2473
        return {'url':"","message":"App id doesn't exist"}
16364 kshitij.so 2474
 
16915 manish.sha 2475
    final_user_payout = 0
2476
    if d_app_offer.override_payout:
2477
        final_user_payout = d_app_offer.overriden_payout
2478
    else:
2479
        final_user_payout = d_app_offer.user_payout
16546 kshitij.so 2480
 
16915 manish.sha 2481
    appTransactions = AppTransactions(app_id, retailer_id, to_java_date(datetime.now()), None, 1, CB_INIT, 1, CB_INIT, None, None, d_app_offer.offer_price, d_app_offer.overriden_payout, d_app_offer.override_payout, d_app_offer.user_payout, final_user_payout)
2482
 
16546 kshitij.so 2483
    get_mongo_connection().AppOrder.AppTransaction.insert(appTransactions.__dict__)
2484
    embedd = str((appTransactions.__dict__).get('_id'))
16548 kshitij.so 2485
    try:
2486
        index = d_app_offer.link.index(".freeb.co.in")
2487
    except:
16605 manish.sha 2488
        traceback.print_exc()
16548 kshitij.so 2489
        return {'url':"","message":"Substring not found"}
2490
    redirect_url = d_app_offer.link[0:index]+"."+embedd+d_app_offer.link[index:]
16546 kshitij.so 2491
    get_mongo_connection().AppOrder.AppTransaction.update({'_id':ObjectId(embedd)},{"$set":{'redirect_url':redirect_url}})
16548 kshitij.so 2492
    return {'url':redirect_url,"message":"Success"}
16556 kshitij.so 2493
 
2494
def addPayout(payout, transaction_id):
2495
    try:
16764 manish.sha 2496
        print 'Got Response for Transaction Id:- '+ str(transaction_id)+ ' and Actual Payout:- '+str(payout)
16556 kshitij.so 2497
        transaction = list(get_mongo_connection().AppOrder.AppTransaction.find({'_id':ObjectId(transaction_id)}))
2498
        if len(transaction) > 0:
2499
            if (transaction[0])['payout_status'] ==1:
16753 manish.sha 2500
                get_mongo_connection().AppOrder.AppTransaction.update({'_id':ObjectId(transaction_id)},{"$set":{'payout_amount':float(payout), 'payout_description': CB_APPROVED,'payout_status':2, 'cashback_status': 2, 'cash_back_description': CB_APPROVED, 'payout_time':to_java_date(datetime.now())}})
16782 manish.sha 2501
 
2502
                approvedAppTransaction = approved_app_transactions.get_by(transaction_id=transaction_id)
2503
                if approvedAppTransaction is None:
2504
                    approvedAppTransaction = approved_app_transactions()
16919 manish.sha 2505
                    approvedAppTransaction.app_id = transaction[0].get('app_id')
16782 manish.sha 2506
                    approvedAppTransaction.cash_back_description = CB_APPROVED
16919 manish.sha 2507
                    approvedAppTransaction.retailer_id = transaction[0].get('retailer_id')
16782 manish.sha 2508
                    approvedAppTransaction.transaction_id = transaction_id
16919 manish.sha 2509
                    approvedAppTransaction.transaction_time = to_py_date(long(transaction[0].get('transaction_time')))
2510
                    approvedAppTransaction.redirect_url = transaction[0].get('redirect_url')
16782 manish.sha 2511
                    approvedAppTransaction.payout_status = 2
2512
                    approvedAppTransaction.payout_description = CB_APPROVED
2513
                    approvedAppTransaction.cashback_status = 2 
2514
                    approvedAppTransaction.payout_amount = long(payout)
2515
                    approvedAppTransaction.payout_time = datetime.now()
16919 manish.sha 2516
                    approvedAppTransaction.offer_price = long(transaction[0].get('offer_price')) 
2517
                    approvedAppTransaction.overridenCashBack = transaction[0].get('overridenCashBack')
2518
                    approvedAppTransaction.isCashBackOverriden = transaction[0].get('isCashBackOverriden')
2519
                    approvedAppTransaction.user_payout = transaction[0].get('user_payout')
16782 manish.sha 2520
                    approvedAppTransaction.cashBackConsidered = False
16915 manish.sha 2521
                    if transaction[0].get('final_user_payout') is not None:
2522
                        approvedAppTransaction.final_user_payout = transaction[0].get('final_user_payout')
2523
                    else:
16919 manish.sha 2524
                        if transaction[0].get('isCashBackOverriden'):
2525
                            approvedAppTransaction.final_user_payout = transaction[0].get('overridenCashBack')
16915 manish.sha 2526
                        else:
16919 manish.sha 2527
                            approvedAppTransaction.final_user_payout = transaction[0].get('user_payout')
16782 manish.sha 2528
                    session.commit()
2529
 
2530
                    updateResult = _updateApprovedCashbackToUser(approvedAppTransaction.id)
16631 manish.sha 2531
                else:
16783 manish.sha 2532
                    approvedAppTransaction.payout_amount = long(payout)
2533
                    approvedAppTransaction.payout_time = datetime.now()
2534
                    if approvedAppTransaction.cashBackConsidered == False:
2535
                        updateResult = _updateApprovedCashbackToUser(approvedAppTransaction.id)
2536
                    session.commit()
16782 manish.sha 2537
 
2538
                    '''
2539
                    try:
2540
                        if int(transaction[0]['offer_price']) != payout or int(transaction[0]['user_payout']) < payout:
2541
                            _sendAlertForAppPayouts(int(transaction[0]['app_id']), int(transaction[0]['offer_price']), int(transaction[0]['user_payout']), payout)
2542
                    except:
2543
                        print traceback.print_exc()
2544
                    '''
16556 kshitij.so 2545
                return {'status':'ok','message':'Payout updated'}
2546
            elif (transaction[0])['payout_status'] ==2:
2547
                return {'status':'ok','message':'Payout already processed'}
2548
            else:
2549
                return {'status':'fail','message':'Something is wrong'}
2550
        else:
2551
            return {'status':'fail','message':'transaction_id not found'}
2552
    except:
16657 manish.sha 2553
        print traceback.print_exc()
16631 manish.sha 2554
        try:
16753 manish.sha 2555
            get_mongo_connection().AppOrder.AppTransaction.update({'_id':ObjectId(transaction_id)},{"$set":{'payout_amount':None, 'payout_description': CB_INIT,'payout_status':1, 'cashback_status': 1, 'cash_back_description': CB_INIT, 'payout_time':None}})
16631 manish.sha 2556
        except:
2557
            print 'Data Inconsistency in Cashback Process'
16636 manish.sha 2558
            print traceback.print_exc()
16559 kshitij.so 2559
        return  {'status':'fail','message':'Something is wrong'}
16885 kshitij.so 2560
    finally:
2561
        session.close()
16556 kshitij.so 2562
 
16631 manish.sha 2563
def _updateApprovedCashbackToUser(approvedAppTransactionId):
2564
    approvedAppTransaction = approved_app_transactions.get_by(id=approvedAppTransactionId)
16915 manish.sha 2565
    if approvedAppTransaction.retailer_id < 0:
2566
        return False
2567
 
16631 manish.sha 2568
    currentMonth = datetime.today().month
2569
    currentDay = datetime.today().day
2570
    currentYear = datetime.today().year
2571
    fortNight = (currentMonth - 1)*2 + (currentDay/15)
2572
    if currentDay == 30 or currentDay ==31:
2573
        fortNight = fortNight-1
2574
    userAppCashbackObj = user_app_cashbacks.query.filter(user_app_cashbacks.yearVal==currentYear).filter(user_app_cashbacks.user_id==approvedAppTransaction.retailer_id).filter(user_app_cashbacks.fortnightOfYear==fortNight).filter(user_app_cashbacks.status=='Approved').first()
2575
    userAppInstallObj = user_app_installs.query.filter(user_app_installs.user_id==approvedAppTransaction.retailer_id).filter(user_app_installs.app_id==approvedAppTransaction.app_id).filter(user_app_installs.transaction_date==datetime(currentYear, currentMonth, currentDay, 0, 0, 0, 0).date()).first()
2576
    try:
2577
        if userAppCashbackObj is None:
2578
            userAppCashbackObj = user_app_cashbacks()
2579
            userAppCashbackObj.user_id = approvedAppTransaction.retailer_id
2580
            userAppCashbackObj.status = 'Approved'
16915 manish.sha 2581
            userAppCashbackObj.amount = approvedAppTransaction.final_user_payout
16631 manish.sha 2582
            userAppCashbackObj.fortnightOfYear = fortNight
2583
            userAppCashbackObj.yearVal = currentYear
2584
        else:
16915 manish.sha 2585
            userAppCashbackObj.amount = userAppCashbackObj.amount + approvedAppTransaction.final_user_payout
16631 manish.sha 2586
 
2587
        if userAppInstallObj is None:
2588
            app_offer = app_offers.get_by(id=approvedAppTransaction.app_id)
2589
            userAppInstallObj = user_app_installs()
2590
            userAppInstallObj.user_id = approvedAppTransaction.retailer_id
2591
            userAppInstallObj.fortnightOfYear = fortNight
2592
            userAppInstallObj.transaction_date = datetime(currentYear, currentMonth, currentDay, 0, 0, 0, 0).date()
2593
            userAppInstallObj.app_id = approvedAppTransaction.app_id
2594
            userAppInstallObj.app_name = app_offer.app_name
2595
            userAppInstallObj.installCount = 1
16915 manish.sha 2596
            userAppInstallObj.payoutAmount = approvedAppTransaction.final_user_payout
16631 manish.sha 2597
        else:
2598
            userAppInstallObj.installCount = userAppInstallObj.installCount + 1
16915 manish.sha 2599
            userAppInstallObj.payoutAmount = userAppInstallObj.payoutAmount +approvedAppTransaction.final_user_payout
16631 manish.sha 2600
 
2601
        approvedAppTransaction.cashBackConsidered = True
2602
        session.commit()
2603
        return True
2604
    except:
2605
        session.rollback()
16636 manish.sha 2606
        print traceback.print_exc()
16631 manish.sha 2607
        return False
16693 manish.sha 2608
 
2609
def _sendAlertForAppPayouts(app_id, offerPrice, userPayout, payout):
2610
    m = Email('localhost')
2611
    mFrom = "dtr@shop2020.in"
2612
    m.setFrom(mFrom)
2613
    mTo = ['rajneesh.arora@saholic.com','kshitij.sood@saholic.com','chaitnaya.vats@saholic.com','manoj.kumar@saholic.com','manish.sharma@shop2020.in']
2614
    for receipient in mTo:
2615
        m.addRecipient(receipient)
2616
    appOffer = app_offers.get_by(id=app_id)
2617
    if offerPrice != payout:
2618
        m.setSubject("Mismatch in Offer Price and Actual Pay Out for App Id:- "+str(appOffer.id)+" App Name:- "+str(appOffer.app_name))
2619
        message1 = "Offer Price:- "+ str(appOffer.offer_price) + "\n" + "Actual Pay Out:- "+ str(payout)
2620
        m.setTextBody(message1)
2621
    if userPayout > payout:
2622
        m.setSubject("User Pay Out greater than actual PayOut for App Id:- "+str(appOffer.id)+" App Name:- "+str(appOffer.app_name))
2623
        message2 = "User Pay Out Price:- "+ str(appOffer.user_payout) + "\n" + "Actual Pay Out:- "+ str(payout)
2624
        m.setTextBody(message2)
2625
    m.send()
16631 manish.sha 2626
 
16789 amit.gupta 2627
def rejectCashback(orderId, subOrderId):
16833 amit.gupta 2628
    cashBack = get_mongo_connection().Dtr.merchantOrder.update({"orderId":orderId, "subOrders.merchantSubOrderId":subOrderId}, 
16789 amit.gupta 2629
                                                               {"$set":{"subOrders.$.cashBackStatus":CB_REJECTED}})
16837 amit.gupta 2630
    print cashBack
16836 amit.gupta 2631
    return cashBack.get("nModified")
16894 manish.sha 2632
 
2633
def getAppOffers(retailer_id):
2634
    if not bool(mc.get("cached_app_offers_"+str(retailer_id))):
2635
        populateAppOffers(retailer_id)
2636
    return mc.get("cached_app_offers_"+str(retailer_id)) 
2637
 
2638
def populateAppOffers(retailerId):
2639
    retailerId = int(retailerId)
2640
    #offset = req.get_param_as_int("offset")
2641
    #limit = req.get_param_as_int("limit")
16941 manish.sha 2642
    nonPriortizedOffers = session.query(app_offers.id).join((appmasters,appmasters.id==app_offers.appmaster_id)).filter(app_offers.affiliate_id==retailerId).filter(appmasters.showApp==True).filter(app_offers.show==True).filter(app_offers.offer_active==True).filter(app_offers.offer_price>0).filter(appmasters.rank==0).order_by(desc('user_payout')).all()
2643
    priortizeders = session.query(app_offers.id, appmasters.rank).join((appmasters,appmasters.id==app_offers.appmaster_id)).filter(app_offers.affiliate_id==retailerId).filter(appmasters.showApp==True).filter(app_offers.show==True).filter(app_offers.offer_active==True).filter(app_offers.offer_price>0).filter(appmasters.rank>0).order_by(asc(appmasters.rank)).all()
16894 manish.sha 2644
    #session.query(app_offers.id,app_offers.appmaster_id, app_offers.app_name, app_offers.affiliate_offer_id, app_offers.image_url, app_offers.downloads, app_offers.link, app_offers.offer_price, app_offers.offerCategory, app_offers.package_name, app_offers.promoImage, app_offers.ratings, case([(app_offers.override_payout == True, app_offers.overriden_payout)], else_=app_offers.user_payout).label('user_payout'), case([(appmasters.shortDescription != None, appmasters.shortDescription)], else_=None).label('shortDescription'), case([(appmasters.longDescription != None, appmasters.longDescription)], else_=None).label('longDescription'), appmasters.customerOneLiner, appmasters.retailerOneLiner,app_offers.priority, app_offers.offerCondition, app_offers.location).join((appmasters,appmasters.id==app_offers.appmaster_id)).filter(app_offers.affiliate_id==retailerId).filter(appmasters.showApp==True).filter(app_offers.show==True).filter(app_offers.offer_active==True).filter(app_offers.offer_price>0).order_by(case([(app_offers.priority.in_(tuple(nullCheckList)), 1)], else_=0), asc(app_offers.priority),desc('user_payout')).all()
2645
 
2646
    appOffersMap = {}
2647
    priorityList = []
2648
    priorityMap = {}
2649
    for offer in priortizeders:
16941 manish.sha 2650
        if priorityMap.has_key(long(offer[1])):
2651
            offersVal = priorityMap.get(long(offer[1]))
2652
            offersVal.append(offer)
2653
            priorityMap[long(offer[1])]=offersVal
2654
        else:
2655
            offersVal = []
2656
            offersVal.append(offer)
2657
            priorityMap[long(offer[1])]=offersVal
2658
 
16894 manish.sha 2659
    priorityList = sorted(priorityMap)
2660
 
2661
    maxCount = len(nonPriortizedOffers) + len(priortizeders)
2662
 
16909 manish.sha 2663
    if priorityList is not None and len(priorityList)>0:
2664
        if maxCount < max(priorityList):
2665
            maxCount = max(priorityList)
16894 manish.sha 2666
 
2667
    count = 1
2668
    blockedPlaces = []
2669
    availablePlaces = []
2670
    while count <= maxCount:
2671
        if count in priorityList:
2672
            blockedPlaces.append(count)
2673
        else:
2674
            availablePlaces.append(count)
2675
        count = count +1
16941 manish.sha 2676
    print 'Blocked Places', blockedPlaces
16894 manish.sha 2677
    for val in blockedPlaces:
16941 manish.sha 2678
        key = val
2679
        for offerObj in priorityMap.get(val):
2680
            while appOffersMap.has_key(key):
2681
                key = key+1
2682
            appOffersMap[key] = long(offerObj[0])
2683
            if key in availablePlaces:
2684
                availablePlaces.remove(key)
2685
 
2686
    print 'Available Places', availablePlaces
2687
 
16894 manish.sha 2688
    i=0
2689
    for val in availablePlaces:
16899 manish.sha 2690
        if i<len(nonPriortizedOffers):
16902 manish.sha 2691
            appOffersMap[val]= long(nonPriortizedOffers[i][0])
16899 manish.sha 2692
        else:
2693
            break
2694
        i=i+1
16901 manish.sha 2695
    print 'AppOffersMap', appOffersMap
16894 manish.sha 2696
 
2697
    finalAppOffersMap = {}
2698
    if len(appOffersMap) > 0:
2699
        for key in sorted(appOffersMap):               
2700
            finalAppOffersMap[key] = appOffersMap[key]
2701
 
2702
        mc.set("cached_app_offers_"+str(retailerId), finalAppOffersMap, 120)
2703
    else:
2704
        print 'No Offers'
2705
        mc.set("cached_app_offers_"+str(retailerId), {}, 10)              
17280 manish.sha 2706
 
19556 manas 2707
def sendNotification(userIds, campaignName, title, message,notificationtype, url, expiresat='2999-01-01', sendsms=False, smstext=None):
2708
    max_id = list(get_mongo_connection_dtr_data().User.notificationcampaigns.find().sort([('_id',pymongo.DESCENDING)]).limit(1))
19337 manish.sha 2709
    if len(max_id) ==0:
2710
        max_id = 0
2711
    else:
2712
        max_id = max_id[0]['_id']
2713
    max_id = max_id +1
19362 manish.sha 2714
    campaign = NotificationCampaign(max_id, campaignName, title, message, "SELECT User.id from users User where User.id="+str(userIds[0]), url, to_java_date(datetime.now()), to_java_date(datetime.now()+timedelta(days=730)), notificationtype, 'active', sendsms, smstext, 0, 0, 'BATCH_CREDIT')
19556 manas 2715
    get_mongo_connection_dtr_data().User.notificationcampaigns.insert(campaign.__dict__)
17302 kshitij.so 2716
 
2717
def getDummyDeals(categoryId, offset, limit):
2718
    outer_query = []
2719
    outer_query.append({"showDeal":1})
2720
    outer_query.append({"category_id":categoryId})
2721
    query = {}
2722
    query['$gt'] = -100
2723
    outer_query.append({'totalPoints':query})
2724
    all_deals = list(get_mongo_connection().Catalog.Deals.find({"$and":outer_query},{'_id':1}).sort([('totalPoints',pymongo.DESCENDING),('bestSellerPoints',pymongo.DESCENDING),('nlcPoints',pymongo.DESCENDING),('rank',pymongo.DESCENDING)]).skip(offset).limit(limit))
2725
    returnObj = []
2726
    for deal in all_deals:
17306 kshitij.so 2727
        item = list(get_mongo_connection().Catalog.MasterData.find({'_id':deal['_id']}))
17734 kshitij.so 2728
        returnObj.append(getItemObjForStaticDeals(item[0],transform=True))
17302 kshitij.so 2729
    return returnObj
17367 kshitij.so 2730
 
17469 kshitij.so 2731
def searchDummyDeals(search_term , limit, offset):
17367 kshitij.so 2732
    data = []
2733
    uniqueMap = {}
2734
    if search_term is not None:
2735
        terms = search_term.split(' ')
2736
        outer_query = []
2737
        for term in terms:
2738
            outer_query.append({"source_product_name":re.compile(term, re.IGNORECASE)})
2739
        try:
2740
            collection = get_mongo_connection().Catalog.MasterData.find({"$and":outer_query,'source_id':{'$in':SOURCE_MAP.keys()}})
2741
            for record in collection:
2742
                data.append(record)
2743
        except:
2744
            pass
2745
    else:
2746
        collection = get_mongo_connection().Catalog.MasterData.find({'source_id':{'$in':SOURCE_MAP.keys()}})
2747
        for record in collection:
2748
            data.append(record)
2749
    for x in data:
2750
        if not uniqueMap.has_key(x['skuBundleId']):
17469 kshitij.so 2751
            uniqueMap[x['skuBundleId']] = {'source_product_name':x['source_product_name'],'skuBundleId':x['skuBundleId'],'thumbnail':x['thumbnail']}
17367 kshitij.so 2752
 
17469 kshitij.so 2753
    if (offset + limit) > len(uniqueMap.values()):
2754
        limit = len(uniqueMap.values()) - offset - 1
2755
 
2756
    count = 0
2757
#     for x in uniqueMap.values():
2758
#         print count,
2759
#         print '\t',
2760
#         print x
2761
#         count = count+1
17555 kshitij.so 2762
 
17469 kshitij.so 2763
    print offset
2764
    print limit
2765
 
2766
    return uniqueMap.values()[offset:limit+offset]
17367 kshitij.so 2767
 
17469 kshitij.so 2768
 
2769
def getDummyPricing(skuBundleId):
2770
    collection = get_mongo_connection().Catalog.MasterData.find({'source_id':{'$in':SOURCE_MAP.keys()},'skuBundleId':skuBundleId,'in_stock':1})
2771
    print collection.count()
2772
    cheapest = 99999999
2773
    cheapestDetails= None
2774
    returnMap = {}
2775
 
2776
    for d in collection:
2777
        if d['available_price'] < cheapest:
2778
            cheapestDetails = d
2779
            cheapest = int(d['available_price'])
2780
        if returnMap.has_key(d['source_id']):
17733 kshitij.so 2781
            d = getItemObjForStaticDeals(d,transform=True)
17469 kshitij.so 2782
            d['toShowStore'] = 0
2783
            returnMap[d['source_id']].append(d)
2784
        else:
2785
            temp = []
17733 kshitij.so 2786
            d = getItemObjForStaticDeals(d,transform=True)
17469 kshitij.so 2787
            d['toShowStore'] = 1
2788
            temp.append(d)
2789
            returnMap[d['source_id']] = temp
2790
    returnMap['cheapest'] = cheapestDetails
2791
    for y in returnMap.itervalues():
2792
        if isinstance(y, (list, tuple)):
2793
            (y[len(y)-1])['lastElement'] = 1
2794
        else:
2795
            pass
2796
    if returnMap['cheapest'] ==None:
2797
        dum = get_mongo_connection().Catalog.MasterData.find_one({'skuBundleId':skuBundleId,'source_id':{"$in":SOURCE_MAP.keys()}})
2798
        returnMap['cheapest'] = {'thumbnail':dum['thumbnail'],'source_product_name':dum['source_product_name']}
2799
    return returnMap
17555 kshitij.so 2800
 
17603 kshitij.so 2801
 
17555 kshitij.so 2802
def addDealObject(data):
17570 kshitij.so 2803
    if data.get('img_url') is None or data.get('img_url')=="" or data.get('name') is None or data.get('name')=="":
2804
        return {0:'Invalid data'}   
17555 kshitij.so 2805
    try:
2806
        max_id = list(get_mongo_connection().Catalog.DealObject.find().sort([('_id',pymongo.DESCENDING)]).limit(1))
2807
        if len(max_id) ==0:
2808
            max_id = 0
17559 kshitij.so 2809
        else:
2810
            max_id = max_id[0]['_id']
2811
        data['_id'] = max_id +1
17555 kshitij.so 2812
        get_mongo_connection().Catalog.DealObject.insert(data)
2813
        return {1:'Data added successfully'}
2814
    except:
2815
        return {0:'Error in adding data'}
2816
 
17561 kshitij.so 2817
def getAllDealObjects(offset, limit):
2818
    data = []
2819
    collection = get_mongo_connection().Catalog.DealObject
2820
    cursor = collection.find({}).skip(offset).limit(limit)
2821
    for val in cursor:
17564 kshitij.so 2822
        data.append({'_id':val['_id'],'name':val['name']})
17561 kshitij.so 2823
    return data
17564 kshitij.so 2824
 
2825
def deleteDealObject(id):
2826
    try:
2827
        get_mongo_connection().Catalog.DealObject.remove({'_id':id})
17616 kshitij.so 2828
        get_mongo_connection().Catalog.FeaturedDealsDealObject.remove({'_id':id})
17564 kshitij.so 2829
        return {1:"Deal Object deleted"}
2830
    except:
2831
        return {0:"Error in deleting object"}
17566 kshitij.so 2832
 
2833
def getDealObjectById(id):
2834
    try:
2835
        data = get_mongo_connection().Catalog.DealObject.find({'_id':id})
2836
        return data
2837
    except:
17570 kshitij.so 2838
        return {}
17561 kshitij.so 2839
 
17570 kshitij.so 2840
def updateDealObject(data):
2841
    try:
2842
        _id = data['_id']
2843
        data.pop('_id')
2844
        get_mongo_connection().Catalog.DealObject.update({'_id':_id},{"$set":data},upsert=False)
2845
        return {1:"Data updated successfully"}
2846
    except:
2847
        return {0:"Error in updating data"}
2848
 
17610 kshitij.so 2849
def getAllFeaturedDealsForDealObject(offset, limit):
2850
    data = []
2851
    collection = get_mongo_connection().Catalog.FeaturedDealsDealObject
2852
    cursor = collection.find({}).skip(offset).limit(limit)
2853
    for val in cursor:
2854
        data.append(val)
2855
    return data
2856
 
2857
def addFeaturedDealForDealObject(data):
19280 kshitij.so 2858
    print data
2859
 
17610 kshitij.so 2860
    collection = get_mongo_connection().Catalog.FeaturedDealsDealObject
2861
    cursor = collection.find({'_id':data['_id']})
2862
    master = get_mongo_connection().Catalog.DealObject.find_one({'_id':data['_id']})
2863
    if master is None:
2864
        return {0:"Deal Object id wrong"}
2865
    otherInfoObj = data['otherInfo']
2866
    toPop = []
2867
    for x,y in otherInfoObj.iteritems():
19280 kshitij.so 2868
        if y==0 or y=="":
17610 kshitij.so 2869
            toPop.append(x)
2870
 
2871
    for popItem in toPop:
2872
        otherInfoObj.pop(popItem)
2873
    if len(otherInfoObj.keys())==0:
2874
        return {0:"No rank info to add"}
2875
 
2876
    for x,y in otherInfoObj.iteritems():
2877
        exist = get_mongo_connection().Catalog.FeaturedDealsDealObject.find({'otherInfo.'+x:y})
2878
        if exist.count() >0:
2879
            return {0:"Rank already assigned Deal Object Id %s"%(exist[0]['_id'])}
2880
    if cursor.count() > 0:
2881
        return {0:"Deal Object Id information already present."}
2882
    else:
2883
        collection.insert(data)
2884
        return {1:"Data added successfully"}
2885
 
17616 kshitij.so 2886
def deleteFeaturedDealObject(id):
2887
    try:
2888
        get_mongo_connection().Catalog.FeaturedDealsDealObject.remove({'_id':id})
2889
        return {1:"Deal Object deleted"}
2890
    except:
2891
        return {0:"Error in deleting object"}
17654 kshitij.so 2892
 
2893
def getSubCategoryForFilter(category_id):
2894
    if mc.get("subCategoryFilter") is None:
2895
        print "Populating subcategory data for category_id %d" %(category_id)
2896
        tabData, mobData, accData = [], [], []
2897
 
2898
        accessoriesDeals = get_mongo_connection().Catalog.Deals.aggregate([
19597 kshitij.so 2899
                                                                      {"$match":{"category_id":6,"showDeal":1,"dealRankPoints":{"$gt":0}}
17654 kshitij.so 2900
                                                                    },
2901
                                                                 {"$group" : 
2902
                                                                  {'_id':{'subCategoryId':'$subCategoryId','subCategory':'$subCategory'},'count':{'$sum':1}}
2903
                                                                  }
2904
                                                                ])
2905
 
2906
        for accessoryDeal in accessoriesDeals['result']:
2907
            if accessoryDeal.get('_id').get('subCategoryId') != 0:
2908
                tempMap = {}
2909
                tempMap['subCategory'] = accessoryDeal.get('_id').get('subCategory')
2910
                tempMap['subCategoryId'] = accessoryDeal.get('_id').get('subCategoryId')
2911
                tempMap['count'] = accessoryDeal.get('count')
2912
                accData.append(tempMap)
2913
 
2914
 
2915
        mc.set("subCategoryFilter",{3:mobData, 5:tabData, 6:accData}, 600)  
2916
 
2917
    return sorted(mc.get("subCategoryFilter").get(category_id), key = lambda x: (-x['count'], x['subCategory']))
17610 kshitij.so 2918
 
17727 kshitij.so 2919
def dummyLogin(data):
17731 kshitij.so 2920
    exist = list(get_mongo_connection().Catalog.DummyUser.find({'email':data['email'],'password':data['password']}))
17727 kshitij.so 2921
    if len(exist) > 0:
17728 kshitij.so 2922
        return {'user_id':exist[0]['_id'],'fullName':exist[0]['fullName']}
17727 kshitij.so 2923
    else:
2924
        return {'user_id':0}
17616 kshitij.so 2925
 
17727 kshitij.so 2926
def dummyRegister(data):
2927
    try:
2928
        exist = list(get_mongo_connection().Catalog.DummyUser.find({'email':data['email']}))
2929
        if len(exist) > 0:
2930
            return {'msg':"Email already registered",'user_id':0}
2931
        else:
2932
            max_id = list(get_mongo_connection().Catalog.DummyUser.find().sort([('_id',pymongo.DESCENDING)]).limit(1))
2933
            if len(max_id) ==0:
2934
                max_id = 0
2935
            else:
2936
                max_id = max_id[0]['_id']
2937
            data['_id'] = max_id +1
2938
            get_mongo_connection().Catalog.DummyUser.insert(data)
17728 kshitij.so 2939
            return {'msg':"Registration successful",'user_id':data['_id'],'fullName':data['fullName']}
17727 kshitij.so 2940
    except:
2941
        return {'msg':"Error in registration.",'user_id':0}
2942
 
17731 kshitij.so 2943
def getDummyUser(user_id):
2944
    exist = list(get_mongo_connection().Catalog.DummyUser.find({'_id':user_id}))
2945
    if len(exist) > 0:
2946
        return exist[0]
2947
    else:
2948
        return {'user_id':0,'fullName':"",'email':"","contactNumber":""}
2949
 
2950
def updateDummyUser(data):
2951
    try:
17732 kshitij.so 2952
        get_mongo_connection().Catalog.DummyUser.update({'_id':data['user_id']},{"$set":data},upsert=False)
17731 kshitij.so 2953
        return {'msg':"Data updated successfuly",'response':1}
2954
    except:
2955
        return {'msg':"Unable to update data",'response':0}
17936 kshitij.so 2956
 
18089 kshitij.so 2957
def getDealsForSearchText(subCategoryIds, searchTerm, offset, limit):
2958
    subCategories= [int(subCategoryId) for subCategoryId in subCategoryIds.split('^')]
18106 kshitij.so 2959
    print "SubCategories are ",subCategories
17936 kshitij.so 2960
    payload = {
18106 kshitij.so 2961
    "query": {
17936 kshitij.so 2962
        "bool": {
18106 kshitij.so 2963
            "must": [{
2964
                    "match": {
2965
                        "title": {
2966
                            "query": searchTerm,
2967
                            "operator": "and"
2968
                        }
17936 kshitij.so 2969
                    }
18106 kshitij.so 2970
                },
2971
 
2972
                {
2973
                    "in": {
2974
                        "subCategoryId": subCategories
2975
                    }
17936 kshitij.so 2976
                }
18106 kshitij.so 2977
 
2978
            ]
17936 kshitij.so 2979
        }
19135 kshitij.so 2980
    },
19580 kshitij.so 2981
    "sort": { "dealRankPoints": { "order": "desc" }}
18106 kshitij.so 2982
}
17944 kshitij.so 2983
    result =  get_elastic_search_connection().search(index="my_index", body=payload, from_=offset, size=limit)
17936 kshitij.so 2984
    totalCount= result['hits']['total']
2985
    temp = []
2986
    bundles = []
2987
    if len(result['hits']['hits']) > 0:
2988
        for x in result['hits']['hits']:
2989
            tempMap = {}
2990
            tempMap = {'skuBundleId':x['_source']['id'],'title':x['_source']['title']}
2991
            temp.append(tempMap)
2992
            bundles.append(x['_source']['id'])
18271 kshitij.so 2993
    #return temp
17936 kshitij.so 2994
    return getDealsForNotification(bundles, skipSplitting=True)
17731 kshitij.so 2995
 
18089 kshitij.so 2996
def getCountForSearchText(subCategoryIds, searchTerm):
2997
    subCategories= [int(subCategoryId) for subCategoryId in subCategoryIds.split('^')]
17936 kshitij.so 2998
    payload = {
18106 kshitij.so 2999
    "query": {
17936 kshitij.so 3000
        "bool": {
18106 kshitij.so 3001
            "must": [{
3002
                    "match": {
3003
                        "title": {
3004
                            "query": searchTerm,
3005
                            "operator": "and"
3006
                        }
17936 kshitij.so 3007
                    }
18106 kshitij.so 3008
                },
3009
 
3010
                {
3011
                    "in": {
3012
                        "subCategoryId": subCategories
3013
                    }
17936 kshitij.so 3014
                }
18106 kshitij.so 3015
 
3016
            ]
17936 kshitij.so 3017
        }
3018
    }
18106 kshitij.so 3019
}
17731 kshitij.so 3020
 
17944 kshitij.so 3021
    result =  get_elastic_search_connection().search(index="my_index", body=payload)
17936 kshitij.so 3022
    totalCount= result['hits']['total']
3023
    return {'searchTerm':searchTerm,'count':totalCount}
18271 kshitij.so 3024
 
3025
def getDealById(id):
3026
    item = list(get_mongo_connection().Catalog.MasterData.find({'_id':int(id)}))
3027
    d = get_mongo_connection().Catalog.Deals.find_one({'_id':int(id)})
19193 kshitij.so 3028
    if len(item) ==0:
3029
        return {}
18271 kshitij.so 3030
    if d is None:
3031
        d = {'showDp':0,'dp':0}
3032
    try:
3033
        cashBack = getCashBack(item[0]['_id'], item[0]['source_id'], item[0]['category_id'])
3034
        if not cashBack or cashBack.get('cash_back_status')!=1:
3035
            item[0]['cash_back_type'] = 0
3036
            item[0]['cash_back'] = 0
3037
        else:
3038
            if cashBack.get('maxCashBack') is not None:
3039
 
3040
                if cashBack.get('cash_back_type') ==1 and (float(cashBack.get('cash_back'))*item[0]['available_price'])/100 > cashBack.get('maxCashBack'):
3041
                    cashBack['cash_back_type'] = 2
3042
                    cashBack['cash_back'] = cashBack['maxCashBack']
3043
                elif cashBack.get('cash_back_type') ==2 and cashBack.get('cash_back') > cashBack.get('maxCashBack'):
3044
                    cashBack['cash_back'] = cashBack['maxCashBack']
17936 kshitij.so 3045
 
18271 kshitij.so 3046
            item[0]['cash_back_type'] = int(cashBack['cash_back_type'])
3047
            item[0]['cash_back'] = cashBack['cash_back']
3048
    except:
3049
        print "Error in adding cashback to deals"
3050
        item[0]['cash_back_type'] = 0
3051
        item[0]['cash_back'] = 0
3052
    try:
3053
        item[0]['dp'] = d['dp']
3054
        item[0]['showDp'] = d['showDp']
3055
    except:
3056
        item[0]['dp'] = 0.0
3057
        item[0]['showDp'] = 0
3058
    try:
3059
        item[0]['thumbnail'] = item[0]['thumbnail'].strip()
3060
    except:
3061
        pass
3062
 
3063
    try:
3064
        if item[0]['showVideo']==0:
3065
            item[0]['videoLink'] =""
3066
    except:
3067
        item[0]['videoLink'] =""
3068
    try:
3069
        if item[0]['quantity'] >1:
3070
            ppq = float(item[0]['available_price'])/item[0]['quantity']
3071
 
3072
            if ppq == round(ppq):
3073
                item[0]['ppq'] = int(ppq)
3074
            else:
3075
                item[0]['ppq'] = round(ppq,2)
3076
        else:
3077
            item[0]['ppq'] = 0
3078
    except:
3079
        item[0]['ppq'] = 0
3080
    if item[0]['category_id']!=6:
3081
        try:
18830 kshitij.so 3082
            item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
3083
            item[0]['filterTextBrand'] = (item[0]['brand'])
18271 kshitij.so 3084
        except:
18830 kshitij.so 3085
            pass
18271 kshitij.so 3086
    else:
3087
        try:
3088
            if item[0]['subCategory']=='' or item[0]['subCategory'] is None or item[0]['subCategoryId']==0:
3089
                raise
18830 kshitij.so 3090
            item[0]['filterLinkSubCategory'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['subCategory'])+'&filter=subcategory&subcategories='+str(int(item[0]['subCategoryId']))
3091
            item[0]['filterTextSubCategory'] = (item[0]['subCategory'])
3092
            item[0]['filterLinkBrand'] = '/category/'+str(int(item[0]['category_id']))+'?searchFor='+urllib.quote(item[0]['brand'])+'&filter=brand&brands='+str(int(item[0]['brand_id']))
3093
            item[0]['filterTextBrand'] = (item[0]['brand'])
18271 kshitij.so 3094
        except:
18830 kshitij.so 3095
            pass
18271 kshitij.so 3096
    try:
3097
        if item[0]['source_id']==4:
3098
            item_availability_info = mc.get("item_availability_"+str(item[0]['identifier']).strip())
3099
            if item_availability_info is None or len(item_availability_info)==0:
3100
                item[0]['availabilityInfo'] = [{}]
3101
            else:
3102
                item[0]['availabilityInfo'] = item_availability_info
3103
    except:
3104
        item[0]['availabilityInfo'] = [{}]
19193 kshitij.so 3105
 
3106
    if item[0]['source_id']==4:
3107
        for av_info in item[0]['availabilityInfo']:
3108
            if not av_info:
3109
                continue
3110
            av_info['cash_back'] = item[0]['cash_back']
3111
            av_info['cash_back_type'] = item[0]['cash_back_type']
3112
            netPriceAfterCashBack = av_info['sellingPrice'] 
19199 kshitij.so 3113
            if av_info['cash_back_type'] ==1:
19207 kshitij.so 3114
                netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(float(av_info['sellingPrice'])*av_info['cash_back']/100)
19199 kshitij.so 3115
            elif av_info['cash_back_type'] ==2:
19207 kshitij.so 3116
                netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(av_info['cash_back'])
19193 kshitij.so 3117
            else:
3118
                pass
3119
            av_info['netPriceAfterCashBack'] = netPriceAfterCashBack
3120
 
3121
    if item[0]['source_id'] != SOURCE_MAP.get("PAYTM.COM"):
3122
        if item[0]['codAvailable'] ==1:
3123
            paytmPrice = item[0]['available_price']
3124
        else:
3125
            paytmPrice = item[0]['gross_price']
3126
 
3127
        if item[0]['cash_back_type'] ==1:
19207 kshitij.so 3128
            item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(float(paytmPrice)*item[0]['cash_back']/100)
19193 kshitij.so 3129
        elif item[0]['cash_back_type'] ==2:
19207 kshitij.so 3130
            item[0]['netPriceAfterCashBack'] = paytmPrice -  math.floor(item[0]['cash_back'])
19193 kshitij.so 3131
        else:
3132
            item[0]['netPriceAfterCashBack'] = paytmPrice
3133
 
3134
    else:
3135
        if item[0]['cash_back_type'] ==1:
19207 kshitij.so 3136
            item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(float(item[0]['available_price'])*item[0]['cash_back']/100)
19193 kshitij.so 3137
        elif item[0]['cash_back_type'] ==2:
19207 kshitij.so 3138
            item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(item[0]['cash_back'])
19193 kshitij.so 3139
        else:
3140
            item[0]['netPriceAfterCashBack'] = item[0]['available_price']
3141
 
18271 kshitij.so 3142
    return item[0]
3143
 
19288 kshitij.so 3144
def getOfferForUser(userId):
3145
    specialOffer = {}
3146
    promoOffer = get_mongo_connection().Catalog.PromoOffer.find_one({'user_id':userId})
3147
    try:
3148
        if promoOffer is not None:
3149
            offer = get_mongo_connection().Catalog.Promotions.find_one({'_id':promoOffer.get('offer_id')})
3150
            if offer is None:
3151
                raise
19355 kshitij.so 3152
            promo = Promotion(offer.get('_id'),offer.get('offer_name'),offer.get('offer_description') , offer.get('categories_applicable'), offer.get('sub_categories_not_applicable'),
19331 kshitij.so 3153
                     offer.get('startDate'), offer.get('endDate'), promoOffer.get('target1'), promoOffer.get('target1_cash_back_percetage'), promoOffer.get('target2'), promoOffer.get('target2_cash_back_percetage'), promoOffer.get('maxCashBack'), offer.get('url'),
19350 kshitij.so 3154
                     promoOffer.get('pending_order_value'), promoOffer.get('delivered_order_value'), promoOffer.get('last_run_timestamp'))
19354 kshitij.so 3155
 
19355 kshitij.so 3156
            promo.offer_description = "Get upto %d%% cashback"%(promo.target2_cash_back_percetage)
3157
 
19288 kshitij.so 3158
            specialOffer = promo.__dict__
3159
    except:
3160
        pass
3161
    return specialOffer
3162
 
19389 kshitij.so 3163
def getBrandSubCategoryInfo(category_id, id_list, type):
19402 kshitij.so 3164
    try:
3165
        id_list_mapping = {}
3166
        if type == "brandInfo":
3167
            brands = [int(brand_id) for brand_id in id_list.split('^')]
3168
            for brand in brands:
3169
                tm_brand = get_mongo_connection().Catalog.Deals.find_one({'brand_id':brand},{'brand':1})
3170
                id_list_mapping[brand] = tm_brand.get('brand')
3171
        elif type == "subCategoryInfo":
3172
            subCategories = [int(subCategoryId) for subCategoryId in id_list.split('^')]
3173
            for subCat in subCategories:
3174
                id_list_mapping[subCat] = SUB_CATEGORY_MAP.get(subCat)
3175
        else:
3176
            return {}
19389 kshitij.so 3177
 
19402 kshitij.so 3178
        if mc.get("brandSubCategoryInfo") is None:
3179
            collection =  get_mongo_connection().Catalog.Deals
3180
            result = collection.aggregate( 
3181
                    [
3182
                        {'$match':{'category_id':category_id,'showDeal':1}},
3183
                        {"$group": { "_id": { "subCategoryId": "$subCategoryId", "brand_id": "$brand_id", "brand":"$brand","subCategory":"$subCategory" }, "count": {"$sum": 1} } }
3184
                    ]
3185
                );
3186
            #from pprint import pprint
3187
            #pprint(result)
3188
 
3189
            subCategoryMap = {}
3190
            brandMap = {}
3191
 
3192
            for data in result['result']:
3193
                result_set = data['_id']
3194
                count = data['count']
3195
                subCategoryId = result_set['subCategoryId']
3196
                brand_id = result_set['brand_id']
3197
                subCategory = result_set['subCategory']
3198
                brand = result_set['brand']
3199
 
3200
                if subCategoryMap.has_key(subCategoryId):
3201
                    subCategoryMap.get(subCategoryId).append({'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand})
3202
                else:
3203
                    subCategoryMap[subCategoryId] = [{'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand}]
3204
 
3205
                if brandMap.has_key(brand_id):
3206
                    brandMap.get(brand_id).append({'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand})
3207
                else:
3208
                    brandMap[brand_id] = [{'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand}]
3209
 
3210
            mc.set("brandSubCategoryInfo",{'subCategoryMap':subCategoryMap, 'brandMap': brandMap}, 600)
19389 kshitij.so 3211
 
19402 kshitij.so 3212
 
3213
        returnMap = {}    
3214
        brandSubCategoryInfo = mc.get("brandSubCategoryInfo")
3215
        if brandSubCategoryInfo is None:
3216
            return {'idListInfo':id_list_mapping,'result':[]}
3217
 
3218
        if type == "subCategoryInfo":
3219
            if brandSubCategoryInfo.get('subCategoryMap') is None:
3220
                return {'idListInfo':id_list_mapping,'result':[]}
3221
            subCategories = [int(subCategoryId) for subCategoryId in id_list.split('^')]
3222
            subCategoryMap = brandSubCategoryInfo.get('subCategoryMap')
3223
            for subCategory in subCategories:
3224
                tempList = subCategoryMap.get(subCategory)
3225
                print tempList
3226
                if tempList is None:
3227
                    continue
3228
                for v in tempList:
3229
                    if returnMap.has_key(v['brand_id']):
3230
                        existing = returnMap.get(v['brand_id'])
3231
                        existing['count'] = existing['count'] + v['count'] 
3232
                    else:
3233
                        returnMap[v['brand_id']] = {'brand_id':v['brand_id'],'brand':v['brand'],'count':v['count']}
19389 kshitij.so 3234
 
19402 kshitij.so 3235
 
3236
        elif type == "brandInfo":
3237
            if brandSubCategoryInfo.get('brandMap') is None:
3238
                return {'idListInfo':id_list_mapping,'result':[]}
3239
            brands = [int(brand_id) for brand_id in id_list.split('^')]
3240
            brandMap = brandSubCategoryInfo.get('brandMap')
3241
            for brand_id in brands:
3242
                tempList = brandMap.get(brand_id)
3243
                print tempList
3244
                if tempList is None:
3245
                    continue
3246
                for v in tempList:
3247
                    if returnMap.has_key(v['subCategoryId']):
3248
                        existing = returnMap.get(v['subCategoryId'])
3249
                        existing['count'] = existing['count'] + v['count'] 
3250
                    else:
3251
                        returnMap[v['subCategoryId']] = {'subCategoryId':v['subCategoryId'],'subCategory':v['subCategory'],'count':v['count']}
19389 kshitij.so 3252
 
19402 kshitij.so 3253
        return {'idListInfo':id_list_mapping,'result':returnMap.values()}
3254
    except:
3255
        traceback.print_exc()
3256
        return {}
19389 kshitij.so 3257
 
19451 manas 3258
def refundAmountInWallet(jsonReq):
3259
    userId = int(jsonReq.get('user_id'))
3260
    emailId = jsonReq.get('email_id')
3261
    amount = int(jsonReq.get('amount'))
3262
    store = jsonReq.get('store')
3263
    refundType = jsonReq.get('type')
3264
    mobileNumber = jsonReq.get('mobile')
3265
    referenceId = jsonReq.get('reference_no')
3266
    referenceDescription = jsonReq.get('reference_description')
19514 manas 3267
    if not store:
19456 manas 3268
        store = None
19514 manas 3269
    else:
19517 manas 3270
        store = store
19451 manas 3271
    if lower(refundType)=='refund':
19511 manas 3272
        refundType = CREDIT_TYPE_REFUND
19451 manas 3273
    elif lower(refundType)=='adjustment':
19511 manas 3274
        refundType = CREDIT_TYPE_ADJUSTMENT
19451 manas 3275
    collection = get_mongo_connection().Dtr.crmrefundwallet
19511 manas 3276
    offer1 = CrmRefundWallet(userId,emailId,mobileNumber,amount,refundType,store,referenceId,referenceDescription,REFUND_ADJUSTMENT_MAP.get(0),to_java_date(datetime.now()))    
19451 manas 3277
    dict1 = todict(offer1)
3278
    collection.save(dict1)
3279
    return True
3280
 
19497 manas 3281
def updateCrmWalletStatus(status,_id,userId,batchId,approvedBy):
19482 manas 3282
    if batchId is None:
3283
        get_mongo_connection().Dtr.crmrefundwallet.update({'_id':ObjectId(_id),},{"$set":{"status":status,"update_timestamp":to_java_date(datetime.now())}})
3284
    else:
19497 manas 3285
        get_mongo_connection().Dtr.crmrefundwallet.update({'_id':ObjectId(_id),},{"$set":{"status":status,"update_timestamp":to_java_date(datetime.now()),"batchId":batchId,"approved_by":approvedBy}})
19463 manas 3286
 
19451 manas 3287
def fetchCrmRefundUsers(status,offset,limit):
19529 manas 3288
    if status is not None:
3289
        cursor = list(get_mongo_connection().Dtr.crmrefundwallet.find({"status":status}).skip(offset).limit(limit).sort("created_timestamp",-1))
3290
    else:
3291
        cursor = list(get_mongo_connection().Dtr.crmrefundwallet.find().skip(offset).limit(limit).sort("created_timestamp",-1))
19451 manas 3292
    return cursor
3293
 
19485 manas 3294
def fetchCrmRefundByBatchId(batchId):
19493 manas 3295
    cursor = get_mongo_connection().Dtr.crmrefundwallet.find_one({"batchId":int(batchId)})
19485 manas 3296
    return cursor
3297
 
19577 manas 3298
def userLookUpForSaholicId(userId):
3299
    result = None
3300
    try:
3301
        response = session.query(user_accounts.account_key).filter_by(user_id=userId).filter(user_accounts.account_type=='saholic').first()
3302
        if response is None:
3303
            result = 'User Not present in profitmandi'
3304
        else:
3305
            result = response[0]
3306
    except:
3307
        result = 'User Not present in profitmandi'
3308
        traceback.print_exc()
3309
    finally:
3310
        session.close()
3311
    return result
19453 manas 3312
 
19880 naman 3313
 
3314
def getAllItemCasback(itemcashback_id,offset,limit):
3315
    itemcashbackcon = get_mongo_connection().Catalog.ItemCashBack
3316
    if itemcashback_id is None: 
3317
        response = list(itemcashbackcon.find().skip(offset).limit(limit))
3318
    else:
3319
        response = itemcashbackcon.find({'_id' : ObjectId(itemcashback_id)})
3320
    return response
3321
 
3322
def itemCashbackAdd(item_skuId,cb,cb_description,cb_type,cb_status):
3323
    itemcashbackcon = get_mongo_connection().Catalog.ItemCashBack
3324
    response = list(itemcashbackcon.find({'sku':item_skuId}))
3325
 
3326
    if len(response) ==  0:
3327
        itemcashbackcon.insert({'sku':item_skuId,'cash_back_description':cb_description,'cash_back':cb,'cash_back_type':cb_type,'cash_back_status': cb_status})
3328
        return True     
3329
 
3330
 
19654 kshitij.so 3331
def getHeaderLinks(category_id):
3332
    result = {}
3333
    if category_id == 6:
3334
        rank = 1
3335
        for subCat in SUB_CATEGORY_HEADER_RANKING:
3336
            result[rank] = {'subCategoryId':subCat,'subCategory':SUB_CATEGORY_MAP.get(subCat)}
3337
            rank = rank + 1
3338
    return result.values()
3339
 
19896 kshitij.so 3340
def getDealsByType(category_id, offset, limit, type):
3341
    dealListMap = []
3342
    if type == 'exclusive':
3343
        exclusive_deals = mc.get("exclusive_deals_"+str(category_id))
3344
        if exclusive_deals is None:
3345
            print "Populating exclusive_deals for ",category_id
3346
            exclusive_deals = list(get_mongo_connection().Catalog.Deals.find({'category_id': category_id,'showDeal': 1, 'source_id': 4, 'totalPoints': {'$gt': -100}},{'_id':1,'category_id':1,'brand':1,'totalPoints':1,'bestSellerPoints':1,'nlcPoints':1,'rank':1,'available_price':1,'dealType':1,'source_id':1,'brand_id':1,'skuBundleId':1,'dp':1, 'showDp':1, 'gross_price':1, 'subCategoryId':1,'netPriceAfterCashBack':1}).sort([('totalPoints',pymongo.DESCENDING),('bestSellerPoints',pymongo.DESCENDING),('nlcPoints',pymongo.DESCENDING),('rank',pymongo.DESCENDING)]))
3347
            mc.set("exclusive_deals_"+str(category_id), exclusive_deals)
3348
 
19902 kshitij.so 3349
        for d in exclusive_deals[offset:limit+offset]:
19896 kshitij.so 3350
            item = list(get_mongo_connection().Catalog.MasterData.find({'_id':d['_id']}))
3351
            if len(item) ==0:
3352
                continue 
3353
            try:
3354
                cashBack = getCashBack(item[0]['_id'], item[0]['source_id'], item[0]['category_id'])
3355
                if not cashBack or cashBack.get('cash_back_status')!=1:
3356
                    item[0]['cash_back_type'] = 0
3357
                    item[0]['cash_back'] = 0
3358
                else:
3359
                    if cashBack.get('maxCashBack') is not None:
3360
 
3361
                        if cashBack.get('cash_back_type') ==1 and (float(cashBack.get('cash_back'))*item[0]['available_price'])/100 > cashBack.get('maxCashBack'):
3362
                            cashBack['cash_back_type'] = 2
3363
                            cashBack['cash_back'] = cashBack['maxCashBack']
3364
                        elif cashBack.get('cash_back_type') ==2 and cashBack.get('cash_back') > cashBack.get('maxCashBack'):
3365
                            cashBack['cash_back'] = cashBack['maxCashBack']
3366
 
3367
                    item[0]['cash_back_type'] = int(cashBack['cash_back_type'])
3368
                    item[0]['cash_back'] = cashBack['cash_back']
3369
            except:
3370
                print "Error in adding cashback to deals"
3371
                item[0]['cash_back_type'] = 0
3372
                item[0]['cash_back'] = 0
3373
            try:
3374
                item[0]['dp'] = d['dp']
3375
                item[0]['showDp'] = d['showDp']
3376
            except:
3377
                item[0]['dp'] = 0.0
3378
                item[0]['showDp'] = 0
3379
            try:
3380
                item[0]['thumbnail'] = item[0]['thumbnail'].strip()
3381
            except:
3382
                pass
3383
 
3384
            try:
3385
                if item[0]['showVideo']==0:
3386
                    item[0]['videoLink'] =""
3387
            except:
3388
                item[0]['videoLink'] =""
3389
            try:
3390
                if item[0]['quantity'] >1:
3391
                    ppq = float(item[0]['available_price'])/item[0]['quantity']
3392
 
3393
                    if ppq == round(ppq):
3394
                        item[0]['ppq'] = int(ppq)
3395
                    else:
3396
                        item[0]['ppq'] = round(ppq,2)
3397
                else:
3398
                    item[0]['ppq'] = 0
3399
            except:
3400
                item[0]['ppq'] = 0
3401
            try:
3402
                if item[0]['source_id']==4:
3403
                    item_availability_info = mc.get("item_availability_"+str(item[0]['identifier']).strip())
3404
                    if item_availability_info is None or len(item_availability_info)==0:
3405
                        item[0]['availabilityInfo'] = [{}]
3406
                    else:
3407
                        item[0]['availabilityInfo'] = item_availability_info
3408
            except:
3409
                item[0]['availabilityInfo'] = [{}]
3410
 
3411
            if item[0]['source_id']==4:
3412
                for av_info in item[0]['availabilityInfo']:
3413
                    if not av_info:
3414
                        continue
3415
                    av_info['cash_back'] = item[0]['cash_back']
3416
                    av_info['cash_back_type'] = item[0]['cash_back_type']
3417
                    netPriceAfterCashBack = av_info['sellingPrice'] 
3418
                    if av_info['cash_back_type'] ==1:
3419
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(float(av_info['sellingPrice'])*av_info['cash_back']/100)
3420
                    elif av_info['cash_back_type'] ==2:
3421
                        netPriceAfterCashBack = av_info['sellingPrice'] - math.floor(av_info['cash_back'])
3422
                    else:
3423
                        pass
3424
                    av_info['netPriceAfterCashBack'] = netPriceAfterCashBack
3425
 
3426
            if item[0]['cash_back_type'] ==1:
3427
                item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(float(item[0]['available_price'])*item[0]['cash_back']/100)
3428
            elif item[0]['cash_back_type'] ==2:
3429
                item[0]['netPriceAfterCashBack'] = item[0]['available_price'] -  math.floor(item[0]['cash_back'])
3430
            else:
3431
                item[0]['netPriceAfterCashBack'] = item[0]['available_price']         
3432
            dealListMap.append([item[0]])
3433
    return dealListMap
3434
 
3435
 
3436
 
13572 kshitij.so 3437
def main():
20009 kshitij.so 3438
    print getAccesoryDeals(47, 6, 0, 20, None, None, None)
17944 kshitij.so 3439
 
13572 kshitij.so 3440
if __name__=='__main__':
19512 kshitij.so 3441
    main()
3442
 
3443