Subversion Repositories SmartDukaan

Rev

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

Rev Author Line No. Line
15081 amit.gupta 1
from bson import json_util
2
from bson.json_util import dumps
15096 amit.gupta 3
from datetime import datetime, timedelta
13582 amit.gupta 4
from dtr import main
15081 amit.gupta 5
from dtr.config import PythonPropertyReader
13629 kshitij.so 6
from dtr.storage import Mongo
15132 amit.gupta 7
from dtr.storage.DataService import Retailers, Users, CallHistory, RetryConfig, \
15254 amit.gupta 8
    RetailerLinks, Activation_Codes, Agents, Agent_Roles, AgentLoginTimings, \
9
    FetchDataHistory, RetailerContacts
15081 amit.gupta 10
from dtr.utils import FetchLivePrices, DealSheet as X_DealSheet, \
11
    UserSpecificDeals
15168 amit.gupta 12
from dtr.utils.utils import getLogger
15081 amit.gupta 13
from elixir import *
15254 amit.gupta 14
from operator import and_
15081 amit.gupta 15
from sqlalchemy.sql.expression import or_
15132 amit.gupta 16
from urllib import urlencode
17
import contextlib
13827 kshitij.so 18
import falcon
15081 amit.gupta 19
import json
15254 amit.gupta 20
import string
15081 amit.gupta 21
import traceback
15132 amit.gupta 22
import urllib
23
import urllib2
24
import uuid
15207 amit.gupta 25
 
26
alphalist = list(string.uppercase)
27
alphalist.remove('O')
28
numList = ['1','2','3','4','5','6','7','8','9']
29
codesys = [alphalist, alphalist, numList, numList, numList]
30
 
31
 
32
def getNextCode(codesys, code=None):
33
    if code is None:
34
        code = []
35
        for charcode in codesys:
36
            code.append(charcode[0])
37
        return string.join(code, '')
38
    carry = True
39
    code = list(code)
40
    lastindex = len(codesys) - 1
41
    while carry:
42
        listChar = codesys[lastindex]
43
        newIndex = (listChar.index(code[lastindex])+1)%len(listChar)
44
        print newIndex
45
        code[lastindex] = listChar[newIndex]
46
        if newIndex != 0:
47
            carry = False
48
        lastindex -= 1
49
        if lastindex ==-1:
50
            raise BaseException("All codes are exhausted")
51
 
52
    return string.join(code, '')
53
 
54
 
55
 
56
 
15081 amit.gupta 57
global RETAILER_DETAIL_CALL_COUNTER
58
RETAILER_DETAIL_CALL_COUNTER = 0
13572 kshitij.so 59
 
15168 amit.gupta 60
lgr = getLogger('/var/log/retailer-acquisition-api.log')
15081 amit.gupta 61
DEALER_RETRY_FACTOR = int(PythonPropertyReader.getConfig('DEALER_RETRY_FACTOR'))
13572 kshitij.so 62
class CategoryDiscountInfo(object):
63
 
64
    def on_get(self, req, resp):
65
 
13629 kshitij.so 66
        result = Mongo.getAllCategoryDiscount()
67
        json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
68
        resp.body = json.dumps(json_docs, encoding='utf-8')
13572 kshitij.so 69
 
70
    def on_post(self, req, resp):
71
        try:
72
            result_json = json.loads(req.stream.read(), encoding='utf-8')
73
        except ValueError:
74
            raise falcon.HTTPError(falcon.HTTP_400,
75
                'Malformed JSON',
76
                'Could not decode the request body. The '
77
                'JSON was incorrect.')
78
 
79
        result = Mongo.addCategoryDiscount(result_json)
80
        resp.body = json.dumps(result, encoding='utf-8')
13969 kshitij.so 81
 
82
    def on_put(self, req, resp, _id):
13970 kshitij.so 83
        try:
84
            result_json = json.loads(req.stream.read(), encoding='utf-8')
85
        except ValueError:
86
            raise falcon.HTTPError(falcon.HTTP_400,
87
                'Malformed JSON',
88
                'Could not decode the request body. The '
89
                'JSON was incorrect.')
90
 
91
        result = Mongo.updateCategoryDiscount(result_json, _id)
92
        resp.body = json.dumps(result, encoding='utf-8')
93
 
13966 kshitij.so 94
 
13969 kshitij.so 95
 
13572 kshitij.so 96
class SkuSchemeDetails(object):
97
 
98
    def on_get(self, req, resp):
13629 kshitij.so 99
 
14070 kshitij.so 100
        offset = req.get_param_as_int("offset")
101
        limit = req.get_param_as_int("limit")
102
 
103
        result = Mongo.getAllSkuWiseSchemeDetails(offset, limit)
13629 kshitij.so 104
        json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
105
        resp.body = json.dumps(json_docs, encoding='utf-8')
13572 kshitij.so 106
 
107
 
108
    def on_post(self, req, resp):
109
 
14552 kshitij.so 110
        multi = req.get_param_as_int("multi")
111
 
13572 kshitij.so 112
        try:
113
            result_json = json.loads(req.stream.read(), encoding='utf-8')
114
        except ValueError:
115
            raise falcon.HTTPError(falcon.HTTP_400,
116
                'Malformed JSON',
117
                'Could not decode the request body. The '
118
                'JSON was incorrect.')
119
 
14552 kshitij.so 120
        result = Mongo.addSchemeDetailsForSku(result_json, multi)
13572 kshitij.so 121
        resp.body = json.dumps(result, encoding='utf-8')
122
 
123
class SkuDiscountInfo():
124
 
125
    def on_get(self, req, resp):
13629 kshitij.so 126
 
13970 kshitij.so 127
        offset = req.get_param_as_int("offset")
128
        limit = req.get_param_as_int("limit")
129
        result = Mongo.getallSkuDiscountInfo(offset,limit)
13629 kshitij.so 130
        json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
131
        resp.body = json.dumps(json_docs, encoding='utf-8')
13572 kshitij.so 132
 
133
 
134
    def on_post(self, req, resp):
135
 
14552 kshitij.so 136
        multi = req.get_param_as_int("multi")
137
 
13572 kshitij.so 138
        try:
139
            result_json = json.loads(req.stream.read(), encoding='utf-8')
140
        except ValueError:
141
            raise falcon.HTTPError(falcon.HTTP_400,
142
                'Malformed JSON',
143
                'Could not decode the request body. The '
144
                'JSON was incorrect.')
145
 
14552 kshitij.so 146
        result = Mongo.addSkuDiscountInfo(result_json, multi)
13572 kshitij.so 147
        resp.body = json.dumps(result, encoding='utf-8')
13970 kshitij.so 148
 
149
    def on_put(self, req, resp, _id):
150
        try:
151
            result_json = json.loads(req.stream.read(), encoding='utf-8')
152
        except ValueError:
153
            raise falcon.HTTPError(falcon.HTTP_400,
154
                'Malformed JSON',
155
                'Could not decode the request body. The '
156
                'JSON was incorrect.')
157
 
158
        result = Mongo.updateSkuDiscount(result_json, _id)
159
        resp.body = json.dumps(result, encoding='utf-8')
13572 kshitij.so 160
 
161
class ExceptionalNlc():
162
 
163
    def on_get(self, req, resp):
13629 kshitij.so 164
 
13970 kshitij.so 165
        offset = req.get_param_as_int("offset")
166
        limit = req.get_param_as_int("limit")
167
 
168
        result = Mongo.getAllExceptionlNlcItems(offset, limit)
13629 kshitij.so 169
        json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
170
        resp.body = json.dumps(json_docs, encoding='utf-8')
13572 kshitij.so 171
 
172
    def on_post(self, req, resp):
173
 
14552 kshitij.so 174
        multi = req.get_param_as_int("multi")
175
 
13572 kshitij.so 176
        try:
177
            result_json = json.loads(req.stream.read(), encoding='utf-8')
178
        except ValueError:
179
            raise falcon.HTTPError(falcon.HTTP_400,
180
                'Malformed JSON',
181
                'Could not decode the request body. The '
182
                'JSON was incorrect.')
183
 
14552 kshitij.so 184
        result = Mongo.addExceptionalNlc(result_json, multi)
13572 kshitij.so 185
        resp.body = json.dumps(result, encoding='utf-8')
13970 kshitij.so 186
 
187
    def on_put(self, req, resp, _id):
188
        try:
189
            result_json = json.loads(req.stream.read(), encoding='utf-8')
190
        except ValueError:
191
            raise falcon.HTTPError(falcon.HTTP_400,
192
                'Malformed JSON',
193
                'Could not decode the request body. The '
194
                'JSON was incorrect.')
195
 
196
        result = Mongo.updateExceptionalNlc(result_json, _id)
197
        resp.body = json.dumps(result, encoding='utf-8')
13572 kshitij.so 198
 
13772 kshitij.so 199
class Deals():
13779 kshitij.so 200
    def on_get(self,req, resp, userId):
201
        categoryId = req.get_param_as_int("categoryId")
202
        offset = req.get_param_as_int("offset")
203
        limit = req.get_param_as_int("limit")
13798 kshitij.so 204
        sort = req.get_param("sort")
13802 kshitij.so 205
        direction = req.get_param_as_int("direction")
14853 kshitij.so 206
        filterData = req.get_param('filterData')
207
        result = Mongo.getNewDeals(int(userId), categoryId, offset, limit, sort, direction, filterData)
13772 kshitij.so 208
        json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
13966 kshitij.so 209
        resp.body = json.dumps(json_docs, encoding='utf-8')
13790 kshitij.so 210
 
211
class MasterData():
212
    def on_get(self,req, resp, skuId):
213
        result = Mongo.getItem(skuId)
13798 kshitij.so 214
        try:
215
            json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
13966 kshitij.so 216
            resp.body = json.dumps(json_docs, encoding='utf-8')
13798 kshitij.so 217
        except:
218
            json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
13966 kshitij.so 219
            resp.body = json.dumps(json_docs, encoding='utf-8')
13790 kshitij.so 220
 
14586 kshitij.so 221
    def on_post(self,req, resp):
222
 
223
        addNew = req.get_param_as_int("addNew")
224
        update = req.get_param_as_int("update")
225
        addToExisting = req.get_param_as_int("addToExisting")
226
        multi = req.get_param_as_int("multi")
227
 
14589 kshitij.so 228
        try:
14592 kshitij.so 229
            result_json = json.loads(req.stream.read(), encoding='utf-8')
230
        except ValueError:
231
            raise falcon.HTTPError(falcon.HTTP_400,
232
                'Malformed JSON',
233
                'Could not decode the request body. The '
234
                'JSON was incorrect.')
235
 
236
        if addNew == 1:
237
            result = Mongo.addNewItem(result_json)
238
        elif update == 1:
239
            result = Mongo.updateMaster(result_json, multi)
240
        elif addToExisting == 1:
241
            result = Mongo.addItemToExistingBundle(result_json)
242
        else:
243
            raise
244
        resp.body = dumps(result)
14586 kshitij.so 245
 
13827 kshitij.so 246
class LiveData():
247
    def on_get(self,req, resp):
13865 kshitij.so 248
        if req.get_param_as_int("id") is not None:
249
            print "****getting only for id"
250
            id = req.get_param_as_int("id")
251
            try:
252
                result = FetchLivePrices.getLatestPriceById(id)
13867 kshitij.so 253
                json_docs = json.dumps(result, default=json_util.default)
13966 kshitij.so 254
                resp.body = json.dumps(json_docs, encoding='utf-8')
13865 kshitij.so 255
            except:
256
                json_docs = json.dumps({}, default=json_util.default)
13966 kshitij.so 257
                resp.body = json.dumps(json_docs, encoding='utf-8')
13834 kshitij.so 258
 
13865 kshitij.so 259
        else:
260
            print "****getting only for skuId"
261
            skuBundleId = req.get_param_as_int("skuBundleId")
262
            source_id = req.get_param_as_int("source_id")
263
            try:
264
                result = FetchLivePrices.getLatestPrice(skuBundleId, source_id)
265
                json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
13966 kshitij.so 266
                resp.body = json.dumps(json_docs, encoding='utf-8')
13865 kshitij.so 267
            except:
268
                json_docs = [json.dumps(doc, default=json_util.default) for doc in [{}]]
13966 kshitij.so 269
                resp.body = json.dumps(json_docs, encoding='utf-8')
13865 kshitij.so 270
 
13834 kshitij.so 271
class CashBack():
272
    def on_get(self,req, resp):
273
        identifier = req.get_param("identifier")
274
        source_id = req.get_param_as_int("source_id")
275
        try:
13838 kshitij.so 276
            result = Mongo.getCashBackDetails(identifier, source_id)
13837 kshitij.so 277
            json_docs = json.dumps(result, default=json_util.default)
13964 kshitij.so 278
            resp.body = json_docs
13834 kshitij.so 279
        except:
13837 kshitij.so 280
            json_docs = json.dumps({}, default=json_util.default)
13963 kshitij.so 281
            resp.body = json_docs
13892 kshitij.so 282
 
14398 amit.gupta 283
class ImgSrc():
284
    def on_get(self,req, resp):
285
        identifier = req.get_param("identifier")
286
        source_id = req.get_param_as_int("source_id")
287
        try:
288
            result = Mongo.getImgSrc(identifier, source_id)
289
            json_docs = json.dumps(result, default=json_util.default)
290
            resp.body = json_docs
291
        except:
292
            json_docs = json.dumps({}, default=json_util.default)
293
            resp.body = json_docs
294
 
13892 kshitij.so 295
class DealSheet():
296
    def on_get(self,req, resp):
13895 kshitij.so 297
        X_DealSheet.sendMail()
13897 kshitij.so 298
        json_docs = json.dumps({'True':'Sheet generated, mail sent.'}, default=json_util.default)
13966 kshitij.so 299
        resp.body = json.dumps(json_docs, encoding='utf-8')
13892 kshitij.so 300
 
13970 kshitij.so 301
class DealerPrice():
302
 
303
    def on_get(self, req, resp):
304
 
305
        offset = req.get_param_as_int("offset")
306
        limit = req.get_param_as_int("limit")
307
        result = Mongo.getAllDealerPrices(offset,limit)
308
        json_docs = [json.dumps(doc, default=json_util.default) for doc in result]
309
        resp.body = json.dumps(json_docs, encoding='utf-8')
310
 
311
    def on_post(self, req, resp):
312
 
14552 kshitij.so 313
        multi = req.get_param_as_int("multi")
314
 
13970 kshitij.so 315
        try:
316
            result_json = json.loads(req.stream.read(), encoding='utf-8')
317
        except ValueError:
318
            raise falcon.HTTPError(falcon.HTTP_400,
319
                'Malformed JSON',
320
                'Could not decode the request body. The '
321
                'JSON was incorrect.')
322
 
14552 kshitij.so 323
        result = Mongo.addSkuDealerPrice(result_json, multi)
13970 kshitij.so 324
        resp.body = json.dumps(result, encoding='utf-8')
325
 
326
    def on_put(self, req, resp, _id):
327
        try:
328
            result_json = json.loads(req.stream.read(), encoding='utf-8')
329
        except ValueError:
330
            raise falcon.HTTPError(falcon.HTTP_400,
331
                'Malformed JSON',
332
                'Could not decode the request body. The '
333
                'JSON was incorrect.')
334
 
335
        result = Mongo.updateSkuDealerPrice(result_json, _id)
336
        resp.body = json.dumps(result, encoding='utf-8')
337
 
338
 
14041 kshitij.so 339
class ResetCache():
340
 
341
    def on_get(self,req, resp, userId):
14044 kshitij.so 342
        result = Mongo.resetCache(userId)
14046 kshitij.so 343
        resp.body = json.dumps(result, encoding='utf-8')
344
 
345
class UserDeals():
346
    def on_get(self,req,resp,userId):
347
        UserSpecificDeals.generateSheet(userId)
348
        json_docs = json.dumps({'True':'Sheet generated, mail sent.'}, default=json_util.default)
349
        resp.body = json.dumps(json_docs, encoding='utf-8')
14075 kshitij.so 350
 
351
class CommonUpdate():
352
 
353
    def on_post(self,req,resp):
14575 kshitij.so 354
 
355
        multi = req.get_param_as_int("multi")
356
 
14075 kshitij.so 357
        try:
358
            result_json = json.loads(req.stream.read(), encoding='utf-8')
359
        except ValueError:
360
            raise falcon.HTTPError(falcon.HTTP_400,
361
                'Malformed JSON',
362
                'Could not decode the request body. The '
363
                'JSON was incorrect.')
364
 
14575 kshitij.so 365
        result = Mongo.updateCollection(result_json, multi)
14075 kshitij.so 366
        resp.body = json.dumps(result, encoding='utf-8')
14106 kshitij.so 367
        resp.content_type = "application/json; charset=utf-8"
14481 kshitij.so 368
 
369
class NegativeDeals():
370
 
371
    def on_get(self, req, resp):
372
 
373
        offset = req.get_param_as_int("offset")
374
        limit = req.get_param_as_int("limit")
375
 
376
        result = Mongo.getAllNegativeDeals(offset, limit)
14483 kshitij.so 377
        resp.body = dumps(result) 
14481 kshitij.so 378
 
379
 
380
    def on_post(self, req, resp):
381
 
14552 kshitij.so 382
        multi = req.get_param_as_int("multi")
383
 
14481 kshitij.so 384
        try:
385
            result_json = json.loads(req.stream.read(), encoding='utf-8')
386
        except ValueError:
387
            raise falcon.HTTPError(falcon.HTTP_400,
388
                'Malformed JSON',
389
                'Could not decode the request body. The '
390
                'JSON was incorrect.')
391
 
14552 kshitij.so 392
        result = Mongo.addNegativeDeals(result_json, multi)
14481 kshitij.so 393
        resp.body = json.dumps(result, encoding='utf-8')
394
 
395
class ManualDeals():
396
 
397
    def on_get(self, req, resp):
398
 
399
        offset = req.get_param_as_int("offset")
400
        limit = req.get_param_as_int("limit")
401
 
402
        result = Mongo.getAllManualDeals(offset, limit)
14483 kshitij.so 403
        resp.body = dumps(result)
14481 kshitij.so 404
 
405
 
406
    def on_post(self, req, resp):
407
 
14552 kshitij.so 408
        multi = req.get_param_as_int("multi")
409
 
14481 kshitij.so 410
        try:
411
            result_json = json.loads(req.stream.read(), encoding='utf-8')
412
        except ValueError:
413
            raise falcon.HTTPError(falcon.HTTP_400,
414
                'Malformed JSON',
415
                'Could not decode the request body. The '
416
                'JSON was incorrect.')
417
 
14552 kshitij.so 418
        result = Mongo.addManualDeal(result_json, multi)
14481 kshitij.so 419
        resp.body = json.dumps(result, encoding='utf-8')
420
 
421
class CommonDelete():
14482 kshitij.so 422
 
14481 kshitij.so 423
    def on_post(self,req,resp):
424
        try:
425
            result_json = json.loads(req.stream.read(), encoding='utf-8')
426
        except ValueError:
427
            raise falcon.HTTPError(falcon.HTTP_400,
428
                'Malformed JSON',
429
                'Could not decode the request body. The '
430
                'JSON was incorrect.')
431
 
432
        result = Mongo.deleteDocument(result_json)
433
        resp.body = json.dumps(result, encoding='utf-8')
434
        resp.content_type = "application/json; charset=utf-8"
14482 kshitij.so 435
 
436
class SearchProduct():
437
 
438
    def on_get(self,req,resp):
439
        offset = req.get_param_as_int("offset")
440
        limit = req.get_param_as_int("limit")
441
        search_term = req.get_param("search")
442
 
443
        result = Mongo.searchMaster(offset, limit, search_term)
14483 kshitij.so 444
        resp.body = dumps(result) 
14482 kshitij.so 445
 
446
 
14495 kshitij.so 447
class FeaturedDeals():
14482 kshitij.so 448
 
14495 kshitij.so 449
    def on_get(self, req, resp):
450
 
451
        offset = req.get_param_as_int("offset")
452
        limit = req.get_param_as_int("limit")
453
 
454
        result = Mongo.getAllFeaturedDeals(offset, limit)
455
        resp.body = dumps(result)
456
 
457
 
458
    def on_post(self, req, resp):
459
 
14552 kshitij.so 460
        multi = req.get_param_as_int("multi")
461
 
14495 kshitij.so 462
        try:
463
            result_json = json.loads(req.stream.read(), encoding='utf-8')
464
        except ValueError:
465
            raise falcon.HTTPError(falcon.HTTP_400,
466
                'Malformed JSON',
467
                'Could not decode the request body. The '
468
                'JSON was incorrect.')
469
 
14552 kshitij.so 470
        result = Mongo.addFeaturedDeal(result_json, multi)
14495 kshitij.so 471
        resp.body = json.dumps(result, encoding='utf-8')
472
 
14497 kshitij.so 473
 
474
class CommonSearch():
14495 kshitij.so 475
 
14497 kshitij.so 476
    def on_get(self,req,resp):
477
        class_name = req.get_param("class")
478
        sku = req.get_param_as_int("sku")
479
        skuBundleId = req.get_param_as_int("skuBundleId")
14499 kshitij.so 480
 
481
        result = Mongo.searchCollection(class_name, sku, skuBundleId)
14497 kshitij.so 482
        resp.body = dumps(result)
14619 kshitij.so 483
 
484
class CricScore():
485
 
486
    def on_get(self,req,resp):
487
 
488
        result = Mongo.getLiveCricScore()
14853 kshitij.so 489
        resp.body = dumps(result)
490
 
491
class Notification():
492
 
493
    def on_post(self, req, resp):
494
 
495
        try:
496
            result_json = json.loads(req.stream.read(), encoding='utf-8')
497
        except ValueError:
498
            raise falcon.HTTPError(falcon.HTTP_400,
499
                'Malformed JSON',
500
                'Could not decode the request body. The '
501
                'JSON was incorrect.')
502
 
503
        result = Mongo.addBundleToNotification(result_json)
504
        resp.body = json.dumps(result, encoding='utf-8')
505
 
506
    def on_get(self, req, resp):
507
 
508
        offset = req.get_param_as_int("offset")
509
        limit = req.get_param_as_int("limit")
510
 
511
        result = Mongo.getAllNotifications(offset, limit)
512
        resp.body = dumps(result)
513
 
14998 kshitij.so 514
class DealBrands():
14853 kshitij.so 515
 
14998 kshitij.so 516
    def on_get(self, req, resp):
517
 
518
        category_id = req.get_param_as_int("category_id")
519
        result = Mongo.getBrandsForFilter(category_id)
14999 kshitij.so 520
        resp.body = dumps(result)
15161 kshitij.so 521
 
522
class DealRank():
14998 kshitij.so 523
 
15161 kshitij.so 524
    def on_get(self, req, resp):
525
        identifier = req.get_param("identifier")
526
        source_id = req.get_param_as_int("source_id")
527
        user_id = req.get_param_as_int("user_id")
528
        result = Mongo.getDealRank(identifier, source_id, user_id)
529
        json_docs = json.dumps(result, default=json_util.default)
530
        resp.body = json_docs
531
 
532
 
15081 amit.gupta 533
 
534
class RetailerDetail():
535
    global RETAILER_DETAIL_CALL_COUNTER
15105 amit.gupta 536
    def getRetryRetailer(self,failback=True):
15081 amit.gupta 537
        retailer = None
538
        status = 'fretry' if self.callType == 'followup' else 'retry'
15239 amit.gupta 539
        retailer = session.query(Retailers).filter_by(status=status).filter(Retailers.next_call_time<=datetime.now()).order_by(Retailers.call_priority).order_by(Retailers.next_call_time).with_lockmode("update").first()
540
        if retailer is not None:
541
            lgr.info( "getRetryRetailer " + str(retailer.id))
542
        else:
543
            if failback:
544
                retailer = self.getNewRetailer(False)
545
                return retailer
546
        if status=='retry':
547
            retailer.status = 'assigned'
548
        else:
549
            retailer.status = 'fassigned'
550
        lgr.info( "getRetryRetailer " + str(retailer.id))
15081 amit.gupta 551
        return retailer
552
 
553
    def getNewRetailer(self,failback=True):
554
        retry = True
555
        retailer = None 
556
        try:
557
            while(retry):
15168 amit.gupta 558
                lgr.info( "Calltype " + self.callType)
15081 amit.gupta 559
                status=self.callType
560
                query = session.query(Retailers).filter(Retailers.status==status)
561
                if status=='fresh':
15172 amit.gupta 562
                    query = query.filter_by(is_or=False, is_std=False).filter(or_(Retailers.agent_id==self.agentId, Retailers.agent_id==None)).order_by(Retailers.is_elavated.desc(), -Retailers.agent_id)
15162 amit.gupta 563
                else:
15171 amit.gupta 564
                    query = query.filter(Retailers.next_call_time<=datetime.now())
15081 amit.gupta 565
                retailer = query.with_lockmode("update").first()
566
                if retailer is not None:
15168 amit.gupta 567
                    lgr.info( "retailer " +str(retailer.id))
15081 amit.gupta 568
                    if status=="fresh":
569
                        userquery = session.query(Users)
570
                        if retailer.contact2 is not None:
571
                            userquery = userquery.filter(Users.mobile_number.in_([retailer.contact1,retailer.contact2]))
572
                        else:
573
                            userquery = userquery.filter_by(mobile_number=retailer.contact1)
574
                        user = userquery.first()
575
                        if user is not None:
576
                            retailer.status = 'alreadyuser'
15168 amit.gupta 577
                            lgr.info( "retailer.status " + retailer.status)
15081 amit.gupta 578
                            session.commit()
579
                            continue
580
                        retailer.status = 'assigned'
581
                    else:
15276 amit.gupta 582
                        if isActivated(retailer.id):
583
                            print "Retailer Already %d activated and marked onboarded"%(retailer.id)
584
                            continue
15081 amit.gupta 585
                        retailer.status = 'fassigned'
586
                    retailer.retry_count = 0
15123 amit.gupta 587
                    retailer.invalid_retry_count = 0
15168 amit.gupta 588
                    lgr.info( "Found Retailer " +  str(retailer.id) + " with status " + status + " assigned to " + str(self.agentId))
15081 amit.gupta 589
 
590
                else:
15168 amit.gupta 591
                    lgr.info( "No fresh/followup retailers found")
15081 amit.gupta 592
                    if failback:
593
                        retailer = self.getRetryRetailer(False)
15104 amit.gupta 594
                        return retailer
15148 amit.gupta 595
                retry=False
15081 amit.gupta 596
        except:
597
            print traceback.print_exc()
598
        return retailer
599
 
15132 amit.gupta 600
    def on_get(self, req, resp, agentId, callType=None, retailerId=None):
15081 amit.gupta 601
        global RETAILER_DETAIL_CALL_COUNTER
602
        RETAILER_DETAIL_CALL_COUNTER += 1
15168 amit.gupta 603
        lgr.info( "RETAILER_DETAIL_CALL_COUNTER " +  str(RETAILER_DETAIL_CALL_COUNTER))
15081 amit.gupta 604
        self.agentId = int(agentId)
605
        self.callType = callType
15139 amit.gupta 606
        if retailerId is not None:
607
            self.retailerId = int(retailerId)
15208 amit.gupta 608
            retailerLink = session.query(RetailerLinks).filter_by(retailer_id=self.retailerId).first()
15132 amit.gupta 609
            if retailerLink is not None:
610
                code = retailerLink.code
611
            else: 
612
                code = self.getCode()
613
                retailerLink = RetailerLinks()
614
                retailerLink.code = code
15139 amit.gupta 615
                retailerLink.agent_id = self.agentId
616
                retailerLink.retailer_id = self.retailerId
15135 amit.gupta 617
 
618
                activationCode=Activation_Codes()
619
                activationCode.code = code
15132 amit.gupta 620
                session.commit()
621
            session.close()
15142 amit.gupta 622
            resp.body =  json.dumps({"result":{"code":code,"link":make_tiny(code)}}, encoding='utf-8')
15132 amit.gupta 623
            return 
15081 amit.gupta 624
        retryFlag = False 
625
        if RETAILER_DETAIL_CALL_COUNTER % DEALER_RETRY_FACTOR ==0:
626
            retryFlag=True
15239 amit.gupta 627
        try:
628
            if retryFlag:
629
                retailer = self.getRetryRetailer()
630
            else:
631
                retailer = self.getNewRetailer()
632
            fetchInfo = FetchDataHistory()
633
            fetchInfo.agent_id = self.agentId
634
            fetchInfo.call_type = self.callType
15240 amit.gupta 635
            agent = session.query(Agents).filter_by(id=self.agentId).first()
15239 amit.gupta 636
            last_disposition = session.query(CallHistory).filter_by(agent_id=self.agentId).order_by(CallHistory.id.desc()).first()
637
            if last_disposition is None or last_disposition.created < agent.last_login:
638
                fetchInfo.last_action = 'login'
639
                fetchInfo.last_action_time = agent.last_login 
640
            else:
641
                fetchInfo.last_action = 'disposition'
642
                fetchInfo.last_action_time = last_disposition.created
643
            fetchInfo.retailer_id = retailer.id
644
            session.commit()
15241 amit.gupta 645
 
646
            if retailer is None:
647
                resp.body = "{}"
648
            else:
649
                resp.body = json.dumps(todict(getRetailerObj(retailer)), encoding='utf-8')
650
 
651
            return
652
 
15239 amit.gupta 653
        finally:
654
            session.close()
15241 amit.gupta 655
 
15081 amit.gupta 656
        if retailer is None:
15157 amit.gupta 657
            resp.body = "{}"
15081 amit.gupta 658
        else:
659
            resp.body = json.dumps(todict(getRetailerObj(retailer)), encoding='utf-8')
660
 
15241 amit.gupta 661
 
15081 amit.gupta 662
    def on_post(self, req, resp, agentId, callType):
15112 amit.gupta 663
        returned = False
15081 amit.gupta 664
        self.agentId = int(agentId)
665
        self.callType = callType
666
        jsonReq = json.loads(req.stream.read(), encoding='utf-8')
15169 amit.gupta 667
        lgr.info( "Request ----\n"  + str(jsonReq))
15091 amit.gupta 668
        self.jsonReq = jsonReq
669
        invalidNumber = self.invalidNumber
670
        callLater = self.callLater
15096 amit.gupta 671
        alreadyUser = self.alReadyUser
15091 amit.gupta 672
        verifiedLinkSent = self.verifiedLinkSent
15109 amit.gupta 673
 
15096 amit.gupta 674
        self.retailerId = int(jsonReq.get('retailerid'))
15254 amit.gupta 675
        self.smsNumber = jsonReq.get('smsnumber') 
15112 amit.gupta 676
        try:
677
            self.retailer = session.query(Retailers).filter_by(id=self.retailerId).first()
678
            self.callDisposition = jsonReq.get('calldispositiontype')
679
            self.callHistory = CallHistory()
680
            self.callHistory.agent_id=self.agentId
681
            self.callHistory.call_disposition = self.callDisposition
682
            self.callHistory.retailer_id=self.retailerId
15115 amit.gupta 683
            self.callHistory.call_type=self.callType
15112 amit.gupta 684
            self.callHistory.duration_sec = int(jsonReq.get("callduration"))
685
            self.callHistory.disposition_description = jsonReq.get('calldispositiondescritption')
15200 manas 686
            self.callHistory.disposition_comments = jsonReq.get('calldispositioncomments')
687
            lgr.info(self.callHistory.disposition_comments)
15112 amit.gupta 688
            self.callHistory.call_time = datetime.strptime(jsonReq.get("calltime"), '%d/%m/%Y %H:%M:%S')
689
            self.callHistory.mobile_number = jsonReq.get('number')
15145 amit.gupta 690
            self.callHistory.sms_verified = int(jsonReq.get("verified"))
15234 amit.gupta 691
            lastFetchData = session.query(FetchDataHistory).filter_by(agent_id=self.agentId).order_by(FetchDataHistory.id.desc()).first()
692
            if lastFetchData is None:
693
                raise
694
            self.callHistory.last_fetch_time= lastFetchData.created  
15112 amit.gupta 695
 
696
            dispositionMap = {  'call_later':callLater,
697
                        'ringing_no_answer':callLater,
698
                        'not_reachable':callLater,
699
                        'switch_off':callLater,
15202 manas 700
                        'not_retailer':invalidNumber,
15112 amit.gupta 701
                        'invalid_no':invalidNumber,
702
                        'wrong_no':invalidNumber,
703
                        'hang_up':invalidNumber,
704
                        'retailer_not_interested':invalidNumber,
15200 manas 705
                        'recharge_retailer':invalidNumber,
706
                        'accessory_retailer':invalidNumber,
707
                        'service_center_retailer':invalidNumber,
15112 amit.gupta 708
                        'alreadyuser':alreadyUser,
709
                        'verified_link_sent':verifiedLinkSent
710
                      }
711
            returned = dispositionMap[jsonReq.get('calldispositiontype')]()
712
        finally:
713
            session.close()
15096 amit.gupta 714
 
15112 amit.gupta 715
        if returned:
716
            resp.body = "{\"result\":\"success\"}"
717
        else:
718
            resp.body = "{\"result\":\"failed\"}"
15081 amit.gupta 719
 
15091 amit.gupta 720
    def invalidNumber(self,):
15108 manas 721
        #self.retailer.status = 'retry' if self.callType == 'fresh' else 'fretry'
722
        if self.callDisposition == 'invalid_no':
723
            self.retailer.status='failed'
724
            self.callHistory.disposition_description = 'Invalid Number'
725
        elif self.callDisposition == 'wrong_no':
15111 manas 726
            self.retailer.status='failed'
727
            self.callHistory.disposition_description = 'Wrong Number'
728
        elif self.callDisposition == 'hang_up':
729
            self.retailer.status='failed'
730
            self.callHistory.disposition_description = 'Hang Up'
731
        elif self.callDisposition == 'retailer_not_interested':
732
            self.retailer.status='failed'
733
            if self.callHistory.disposition_description is None:
734
                self.callHistory.disposition_description = 'NA'
15200 manas 735
            self.callHistory.disposition_description = 'Reason Retailer Not Interested ' + self.callHistory.disposition_description
736
        elif self.callDisposition == 'recharge_retailer':
737
            self.retailer.status='failed'
738
            self.callHistory.disposition_description = 'Recharge related. Not a retailer '
739
        elif self.callDisposition == 'accessory_retailer':
740
            self.retailer.status='failed'
741
            self.callHistory.disposition_description = 'Accessory related. Not a retailer'
742
        elif self.callDisposition == 'service_center_retailer':
743
            self.retailer.status='failed'
744
            self.callHistory.disposition_description = 'Service Center related. Not a retailer'
15202 manas 745
        elif self.callDisposition == 'not_retailer':
746
            self.retailer.status='failed'
747
            self.callHistory.disposition_description = 'Not a retailer'    
15108 manas 748
        session.commit()
749
        return True   
750
 
15132 amit.gupta 751
    def getCode(self,):
15207 amit.gupta 752
        newCode = None
753
        lastLink = session.query(RetailerLinks).order_by(RetailerLinks.id.desc()).with_lockmode("update").first()
754
        if lastLink is not None:
755
            if len(lastLink.code)==len(codesys):
756
                newCode=lastLink.code
757
        return getNextCode(codesys, newCode)
15108 manas 758
 
15254 amit.gupta 759
 
15091 amit.gupta 760
    def callLater(self,):
15100 amit.gupta 761
        self.retailer.status = 'retry' if self.callType == 'fresh' else 'fretry'
762
        self.retailer.call_priority = None
15096 amit.gupta 763
        if self.callDisposition == 'call_later':
15100 amit.gupta 764
            if self.callHistory.disposition_description is not None:
15102 amit.gupta 765
                self.retailer.call_priority = 'user_initiated'
15096 amit.gupta 766
                self.retailer.next_call_time = datetime.strptime(self.callHistory.disposition_description, '%d/%m/%Y %H:%M:%S')
767
                self.callHistory.disposition_description = 'User requested to call on ' + self.callHistory.disposition_description
768
            else:
15102 amit.gupta 769
                self.retailer.call_priority = 'system_initiated'
15096 amit.gupta 770
                self.retailer.next_call_time = self.callHistory.call_time + timedelta(days=1)
15112 amit.gupta 771
                self.callHistory.disposition_description = 'Call scheduled on ' + datetime.strftime(self.retailer.next_call_time, '%d/%m/%Y %H:%M:%S')
772
        else: 
773
            if self.callDisposition == 'ringing_no_answer':
774
                if self.retailer.disposition == 'ringing_no_answer':
775
                    self.retailer.retry_count += 1
776
                else:
777
                    self.retailer.disposition = 'ringing_no_answer'
778
                    self.retailer.retry_count = 1
779
            else:
780
                if self.retailer.disposition == 'ringing_no_answer':
15122 amit.gupta 781
                    pass
15112 amit.gupta 782
                else:
15119 amit.gupta 783
                    self.retailer.disposition = 'not_reachable'
15122 amit.gupta 784
                self.retailer.retry_count += 1
785
                self.retailer.invalid_retry_count += 1
15119 amit.gupta 786
 
15122 amit.gupta 787
            retryConfig = session.query(RetryConfig).filter_by(call_type=self.callType, disposition_type=self.retailer.disposition, retry_count=self.retailer.retry_count).first()
15112 amit.gupta 788
            if retryConfig is not None:
789
                self.retailer.next_call_time = self.callHistory.call_time + timedelta(minutes = retryConfig.minutes_ahead)
15119 amit.gupta 790
                self.callHistory.disposition_description = 'Call scheduled on ' + datetime.strftime(self.retailer.next_call_time, '%d/%m/%Y %H:%M:%S')
15112 amit.gupta 791
            else:
792
                self.retailer.status = 'failed'
793
                self.callHistory.disposition_description = 'Call failed as all attempts exhausted'
15119 amit.gupta 794
 
15101 amit.gupta 795
        session.commit()
796
        return True
15096 amit.gupta 797
 
15100 amit.gupta 798
 
15091 amit.gupta 799
    def alReadyUser(self,):
15112 amit.gupta 800
        self.retailer.status = self.callDisposition
15117 amit.gupta 801
        if self.callHistory.disposition_description is None:
802
            self.callHistory.disposition_description = 'Retailer already user' 
15112 amit.gupta 803
        session.commit()
804
        return True
15091 amit.gupta 805
    def verifiedLinkSent(self,):
15147 amit.gupta 806
        if self.callType == 'fresh':
807
            self.retailer.status = 'followup'
808
            self.retailer.next_call_time = self.callHistory.call_time + timedelta(days=1)
809
            self.callHistory.disposition_description = 'App link sent via ' + self.callHistory.disposition_description+ '. followup on' + datetime.strftime(self.retailer.next_call_time, '%d/%m/%Y %H:%M:%S') 
810
        else:
15224 amit.gupta 811
            self.retailer.status = 'followup'
15147 amit.gupta 812
            self.retailer.next_call_time = self.callHistory.call_time + timedelta(days=7)
15254 amit.gupta 813
            self.callHistory.disposition_description = 'App link sent via' + self.callHistory.disposition_description + '. Followup again on ' + datetime.strftime(self.retailer.next_call_time, '%d/%m/%Y %H:%M:%S')
814
        addContactToRetailer(self.agentId, self.retailerId, self.smsNumber, self.callLater(), 'sms')     
15146 amit.gupta 815
        session.commit()
816
        return True
15254 amit.gupta 817
def isActivated(retailerId):
15276 amit.gupta 818
    retailerLink = session.query(RetailerLinks).filter_by(retailer_id=retailerId).first()
819
    user = session.query(Users).filter_by(referrer=retailerLink.code).first()
820
    if user is None:
821
        mobileNumbers = list(session.query(RetailerContacts.mobile_number).filter_by(retailer_id=retailerId).all())
822
        user = session.query(Users).filter(Users.mobile_number.in_(mobileNumbers)).first()
15254 amit.gupta 823
        if user is None:
15276 amit.gupta 824
            return False
825
        else:
826
            mapped_with = 'contact'
827
    else:
828
        mapped_with = 'code'
829
    retailerLink.mapped_with = mapped_with
830
    retailerLink.user_id = user.id
831
    retailer = session.query(Retailers).filter_by(id=retailerId)
832
    retailer.status = 'onboarded'
833
    session.commit()
834
    session.close()
835
    return True
15254 amit.gupta 836
 
837
class AddContactToRetailer():
838
    def on_post(self,req,resp, agentId):
839
        agentId = int(agentId)
840
        try:
841
            jsonReq = json.loads(req.stream.read(), encoding='utf-8')
842
            retailerId = int(jsonReq.get("retailerid"))
843
            mobile = jsonReq.get("mobile")
844
            callType = jsonReq.get("calltype")
845
            contactType = jsonReq.get("contacttype")
846
            addContactToRetailer(agentId, retailerId, mobile, callType, contactType)
847
            session.commit()
848
        finally:
849
            session.close()
850
 
851
def addContactToRetailer(agentId, retailerId, mobile, callType, contactType):
852
    retailerContacts = session.query(RetailerContacts).filter_by(retailer_id=retailerId).filter_by(mobile_number=mobile).first()
853
    if retailerContacts is None:
854
        retailerContact = RetailerContacts()
15256 amit.gupta 855
        retailerContact.retailer_id = retailerId
15254 amit.gupta 856
        retailerContact.agent_id = agentId
857
        retailerContact.call_type = callType
858
        retailerContact.contact_type = contactType
859
        retailerContact.mobile_number = mobile
860
 
15189 manas 861
class Login():
862
 
863
    def on_get(self, req, resp, agentId, role):
864
        try:
15198 manas 865
            self.agentId = int(agentId)
866
            self.role = role
867
            print str(self.agentId) + self.role;
15199 manas 868
            agents=AgentLoginTimings()
869
            lastLoginTime = session.query(Agents).filter(Agents.id==self.agentId).first()
870
            print 'lastLogintime' + str(lastLoginTime)
871
            agents.loginTime=lastLoginTime.last_login
872
            agents.logoutTime=datetime.now()
873
            agents.role =self.role
874
            agents.agent_id = 2
875
            session.add(agents)    
876
            session.commit()
877
            resp.body =  json.dumps({"result":{"success":"true","message":"Success"}}, encoding='utf-8')
15189 manas 878
        finally:
879
            session.close()        
15112 amit.gupta 880
 
15189 manas 881
    def on_post(self,req,resp):
882
        try:
883
            jsonReq = json.loads(req.stream.read(), encoding='utf-8')
884
            lgr.info( "Request ----\n"  + str(jsonReq))
885
            email=jsonReq.get('email')
886
            password = jsonReq.get('password')
887
            role=jsonReq.get('role')    
888
            checkUser = session.query(Agents.id).filter(and_(Agents.email==email,Agents.password==password)).first()
889
            if checkUser is None:
890
                print checkUser
891
                resp.body =  json.dumps({"result":{"success":"false","message":"Invalid User"}}, encoding='utf-8')
892
            else:
893
                print checkUser[0]
894
                checkRole = session.query(Agent_Roles.id).filter(and_(Agent_Roles.agent_id==checkUser[0],Agent_Roles.role==role)).first()
895
                if checkRole is None:
896
                    resp.body =  json.dumps({"result":{"success":"false","message":"Invalid Role"}}, encoding='utf-8')
897
                else:
15196 manas 898
                    session.query(Agents).filter_by(id = checkUser[0]).update({"last_login":datetime.now()})
15195 manas 899
                    resp.body =  json.dumps({"result":{"success":"true","message":"Valid User","id":checkUser[0]}}, encoding='utf-8')
900
                    #session.query(Agents).filter_by(id = checkUser[0]).    
15189 manas 901
                    session.commit()
902
        finally:
903
            session.close()    
904
 
15195 manas 905
    def test(self,email,password,role):
15189 manas 906
        checkUser = session.query(Agents.id).filter(and_(Agents.email==email,Agents.password==password)).first()
907
        if checkUser is None:
908
            print checkUser
15195 manas 909
 
15189 manas 910
        else:
911
            print checkUser[0]
912
            checkRole = session.query(Agent_Roles.id).filter(and_(Agent_Roles.agent_id==checkUser[0],Agent_Roles.role==role)).first()
913
            if checkRole is None:
15195 manas 914
                pass
15189 manas 915
            else:
15195 manas 916
                agents=AgentLoginTimings()
917
                agents.loginTime=datetime.now()
918
                agents.logoutTime=datetime.now()
919
                agents.role =role
920
                agents.agent_id = 2
921
                #session.query(AgentLoginTimings).filter_by(id = checkUser[0]).update({"last_login":datetime.now()}, synchronize_session=False)
922
                session.add(agents)    
923
                session.commit()
924
                session.close()
925
 
926
                #session.query(Agents).filter(Agents.id==checkUser[0]).update({"last_login":Agents.last_login})
15275 amit.gupta 927
 
928
class RetailerActivation():
929
    def on_get(self, req, resp, userId):
930
        return markDealerActivation(int(userId))
931
 
932
def markDealerActivation(userId):
933
    try:
934
        user = session.query(Users).filter_by(id=userId)
935
        result = False                
936
        mappedWith = 'contact'
937
        if user is not None:
938
            retailerLink = session.query(RetailerLinks).filter_by(code=user.referrer).first()
939
            if retailerLink is None:
940
                retailerContact = session.query(RetailerContacts).filter_by(mobile_number=user.mobile_number).first()
941
                if retailerContact is None:
942
                    retailer = session.query(Retailers).filter(Retailers.status.in_(['followup', 'fretry'])).filter(or_(Retailers.contact1==user.mobile_number,Retailers.contact2==user.mobile_number)).first()
943
                else:
944
                    retailer = session.query(Retailers).filter_by(id = retailerContact.retailer_id).first()
945
            else:
946
                retailer = session.query(Retailers).filter_by(id = retailerLink.retailer_id).first()
947
                mappedWith='code'
948
            if retailer is not None:
949
                retailerLink = session.query(RetailerLinks).filter_by(retailer_id=retailer.id).first()
950
                if retailerLink is not None:
951
                    retailerLink.user_id = user.id
952
                    retailerLink.mapped_with=mappedWith
953
                    retailer.status = 'onboarded'
954
                    result = True
955
                    session.commit()
956
            return result
957
    finally:
958
        session.close()
959
 
15189 manas 960
 
15081 amit.gupta 961
def todict(obj, classkey=None):
962
    if isinstance(obj, dict):
963
        data = {}
964
        for (k, v) in obj.items():
965
            data[k] = todict(v, classkey)
966
        return data
967
    elif hasattr(obj, "_ast"):
968
        return todict(obj._ast())
969
    elif hasattr(obj, "__iter__"):
970
        return [todict(v, classkey) for v in obj]
971
    elif hasattr(obj, "__dict__"):
972
        data = dict([(key, todict(value, classkey)) 
973
            for key, value in obj.__dict__.iteritems() 
974
            if not callable(value) and not key.startswith('_')])
975
        if classkey is not None and hasattr(obj, "__class__"):
976
            data[classkey] = obj.__class__.__name__
977
        return data
978
    else:
979
        return obj
980
 
15276 amit.gupta 981
def getRetailerObj(retailer, otherContacts=None):
15081 amit.gupta 982
    obj = Mock()
983
    obj.title = retailer.title
15276 amit.gupta 984
    if otherContacts:
985
        if retailer.contact1 is not None and retailer.contact1 not in otherContacts:
986
            otherContacts.append(retailer.contact1)
987
        if retailer.contact2 is not None and retailer.contact2 not in otherContacts:
988
            otherContacts.append(retailer.contact2)
989
    obj.address = retailer.address_new if retailer.address_new is not None else retailer.address
15081 amit.gupta 990
    obj.id = retailer.id
15096 amit.gupta 991
    obj.scheduled = (retailer.call_priority is not None)
15081 amit.gupta 992
    return obj
15091 amit.gupta 993
 
15132 amit.gupta 994
def make_tiny(code):
995
    url = 'https://play.google.com/store/apps/details?id=com.saholic.profittill&referrer=utm_source%3D0%26utm_medium%3DCRM%26utm_term%3D001%26utm_campaign%3D' + code
996
    request_url = ('http://tinyurl.com/api-create.php?' + urlencode({'url':url}))
997
    filehandle = urllib2.Request(request_url)
998
    x= urllib2.urlopen(filehandle)
999
    return str(x.read())
15171 amit.gupta 1000
 
1001
 
1002
#def update_pin():
15081 amit.gupta 1003
 
15132 amit.gupta 1004
 
15081 amit.gupta 1005
class Mock(object):
1006
    pass
15189 manas 1007
 
15081 amit.gupta 1008
def main():
15207 amit.gupta 1009
    print getNextCode(codesys, 'ZZ999')
15195 manas 1010
 
15081 amit.gupta 1011
if __name__ == '__main__':
15207 amit.gupta 1012
    main()
15091 amit.gupta 1013