Subversion Repositories SmartDukaan

Rev

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

Rev Author Line No. Line
13569 amit.gupta 1
'''
2
Created on Jan 15, 2015
3
 
4
@author: amit
5
'''
14854 amit.gupta 6
from bs4 import BeautifulSoup
13569 amit.gupta 7
from bson.binary import Binary
13680 amit.gupta 8
from datetime import datetime, date, timedelta
13569 amit.gupta 9
from dtr import main
13576 amit.gupta 10
from dtr.dao import AffiliateInfo, Order, SubOrder
14398 amit.gupta 11
from dtr.main import getBrowserObject, ScrapeException, getStore, ParseException, \
12
    Store as MStore, ungzipResponse, tprint
14415 amit.gupta 13
from dtr.storage import Mongo
14
from dtr.storage.Mongo import getImgSrc
14758 amit.gupta 15
from dtr.utils.utils import fetchResponseUsingProxy
13662 amit.gupta 16
from pprint import pprint
13569 amit.gupta 17
from pymongo import MongoClient
18
import json
19
import pymongo
20
import re
14443 amit.gupta 21
import time
13662 amit.gupta 22
import traceback
13569 amit.gupta 23
import urllib
14854 amit.gupta 24
from urlparse import urlparse, parse_qs
13603 amit.gupta 25
 
13721 amit.gupta 26
USERNAME='profittill2@gmail.com'
13569 amit.gupta 27
PASSWORD='spice@2020'
28
AFFILIATE_URL='http://affiliate.snapdeal.com'
29
POST_URL='https://api-p03.hasoffers.com/v3/Affiliate_Report.json'
30
ORDER_TRACK_URL='https://m.snapdeal.com/orderSummary'
31
CONFIG_URL='http://affiliate.snapdeal.com/publisher/js/config.php'
32
 
13662 amit.gupta 33
class Store(MStore):
13569 amit.gupta 34
 
35
    '''
36
    This is to map order statuses of our system to order statuses of snapdeal.
37
    And our statuses will change accordingly.
38
 
39
    '''
40
    OrderStatusMap = {
14861 amit.gupta 41
                      MStore.ORDER_PLACED : ['in progress', 'pending for verification', 'not available', 'in process', 'processing', 'processed'],
14678 amit.gupta 42
                      MStore.ORDER_DELIVERED : ['delivered'],
14861 amit.gupta 43
                      MStore.ORDER_SHIPPED : ['in transit', 'dispatched'],
14830 amit.gupta 44
                      MStore.ORDER_CANCELLED : ['closed for vendor reallocation', 'cancelled', 'product returned by courier', 'returned', 'n/a']
13569 amit.gupta 45
                      }
13662 amit.gupta 46
 
47
    CONF_CB_AMOUNT = MStore.CONF_CB_DISCOUNTED_PRICE
13569 amit.gupta 48
    def __init__(self,store_id):
49
        super(Store, self).__init__(store_id)
50
 
51
    def getName(self):
52
        return "snapdeal"
53
 
54
    def scrapeAffiliate(self, startDate=None, endDate=None):
55
        br = getBrowserObject()
56
        br.open(AFFILIATE_URL)
57
        br.select_form(nr=0)
58
        br.form['data[User][password]'] = PASSWORD 
59
        br.form['data[User][email]'] = USERNAME
60
        br.submit()
61
        response = br.open(CONFIG_URL)
62
 
13680 amit.gupta 63
        token =  re.findall('"session_token":"(.*?)"', ungzipResponse(response), re.IGNORECASE)[0]
14145 amit.gupta 64
        print token
13569 amit.gupta 65
        allOffers = self._getAllOffers(br, token)
14443 amit.gupta 66
        self._saveToAffiliate(allOffers)
13569 amit.gupta 67
 
13662 amit.gupta 68
    def _setLastSaleDate(self, saleDate):
13680 amit.gupta 69
        self.db.lastSaleDtate.update({'storeId':self.store_id}, {'$set':{'saleDate':saleDate}})
13569 amit.gupta 70
 
13662 amit.gupta 71
 
72
 
73
    def _getLastSaleDate(self,):
74
        lastDaySaleObj = self.db.lastDaySale.find_one({"storeId":self.store_id})
75
        if lastDaySaleObj is None:
76
            return datetime.min
77
 
14854 amit.gupta 78
 
79
 
80
    def _parseB(self, orderId, subTagId, userId, page, orderSuccessUrl):
81
        soup = BeautifulSoup(page)
82
 
83
        orderDetailContainerDivs = soup.body.find("div", {'class':'cardLayoutWrap'}).findAll('div', recursive=False)
84
        orderDetailDiv = orderDetailContainerDivs.pop(0)
85
        paymentDetailDiv  = orderDetailContainerDivs.pop(0)
86
        subOrders = orderDetailContainerDivs
87
 
88
        placedOn = orderDetailDiv.span.text.split(':')[1].strip()
89
        merchantOrder = Order(orderId, userId, subTagId, self.store_id, orderSuccessUrl)
90
        merchantOrder.placedOn = placedOn
91
        merchantOrder.merchantOrderId = parse_qs(urlparse(orderSuccessUrl).query)['order'][0]
92
 
93
        paymentDivs = paymentDetailDiv.findAll('div', recursive=False)
94
        paymentDivs.pop(0)
95
        for orderTr in paymentDivs:
96
            orderTrString = str(orderTr)
97
            if "Total Amount Paid" in orderTrString:
98
                amountPaid = orderTr.div.find('div', {'class':'detailBlock'}).text.strip()
99
                merchantOrder.paidAmount = int(re.findall(r'\d+', amountPaid)[0])
100
            elif "Total Amount" in orderTrString:
101
                merchantOrder.totalAmount = re.findall(r'\d+', orderTrString)[0]
102
            elif "Delivery Charges" in orderTrString:
103
                merchantOrder.deliveryCharges = re.findall(r'\d+', orderTrString)[0]
104
            elif "Discount Applied" in orderTrString:
105
                merchantOrder.discountApplied = re.findall(r'\d+', orderTrString)[0]
106
            elif "Offer Discount" in orderTrString:
107
                merchantOrder.discountApplied = re.findall(r'\d+', orderTrString)[0]
108
 
109
        merchantSubOrders = []
110
        for subOrderElement in subOrders:
111
            subOrder = self.parseSubOrderB(subOrderElement, placedOn)                           
112
            merchantSubOrders.append(subOrder)
113
        merchantOrder.subOrders = merchantSubOrders
114
        return merchantOrder
115
 
13760 amit.gupta 116
    def _parse(self, orderId, subTagId, userId, page, orderSuccessUrl):
13662 amit.gupta 117
 
13760 amit.gupta 118
        #page=page.decode("utf-8")
14854 amit.gupta 119
        soup = BeautifulSoup(page)
13760 amit.gupta 120
        #orderHead = soup.find(name, attrs, recursive, text)
121
        sections = soup.findAll("section")
122
 
123
        #print sections
124
 
125
        order = sections[1]
126
        orderTrs = order.findAll("tr")
127
 
128
        placedOn = str(orderTrs[0].findAll("td")[1].text)
129
 
130
        #Pop two section elements
131
        sections.pop(0) 
132
        sections.pop(0)
133
        subOrders = sections
134
 
135
 
136
        merchantSubOrders = []
137
 
138
        merchantOrder = Order(orderId, userId, subTagId, self.store_id, orderSuccessUrl)
14447 amit.gupta 139
        merchantOrder.placedOn = placedOn
14023 amit.gupta 140
        merchantOrder.merchantOrderId = re.findall(r'\d+', str(soup.find("div", {"class":"deals_heading"})))[1]
13760 amit.gupta 141
        for orderTr in orderTrs:
142
            orderTrString = str(orderTr)
143
            if "Total Amount" in orderTrString:
144
                merchantOrder.totalAmount = re.findall(r'\d+', orderTrString)[0]
145
            elif "Delivery Charges" in orderTrString:
146
                merchantOrder.deliveryCharges = re.findall(r'\d+', orderTrString)[0]
147
            elif "Discount Applied" in orderTrString:
148
                merchantOrder.discountApplied = re.findall(r'\d+', orderTrString)[0]
149
            elif "Paid Amount" in orderTrString:
150
                merchantOrder.paidAmount = re.findall(r'\d+', orderTrString)[0]
151
 
152
        for subOrderElement in subOrders:
13809 amit.gupta 153
            subOrders = self.parseSubOrder(subOrderElement, placedOn)                           
154
            merchantSubOrders.extend(subOrders)   
13760 amit.gupta 155
 
156
        merchantOrder.subOrders = merchantSubOrders
157
        return merchantOrder
158
 
159
    def parseSubOrder(self, subOrderElement, placedOn):
13809 amit.gupta 160
        subOrders = []
13760 amit.gupta 161
        productUrl = str(subOrderElement.find("a")['href'])
162
        subTable = subOrderElement.find("table", {"class":"lrPad"})
163
        subTrs = subTable.findAll("tr")
164
        unitPrice=None
14458 amit.gupta 165
        offerDiscount = 0
13760 amit.gupta 166
        deliveryCharges = None
167
        amountPaid = None
14458 amit.gupta 168
        amount = 0
14473 amit.gupta 169
        sdCash = 0
170
        unitPrice = 0
13760 amit.gupta 171
        for subTr in subTrs:
172
            subTrString = str(subTr)
173
            if "Unit Price" in subTrString:
14473 amit.gupta 174
                unitPrice = int(re.findall(r'\d+', subTrString)[0])
13760 amit.gupta 175
            if "Quantity" in subTrString:
14458 amit.gupta 176
                qty = int(re.findall(r'\d+', subTrString)[0])
13760 amit.gupta 177
            elif "Offer Discount" in subTrString:
14458 amit.gupta 178
                offerDiscount +=   int(re.findall(r'\d+', subTrString)[0])
14861 amit.gupta 179
            elif "SD Cash" in subTrString:
14473 amit.gupta 180
                sdCash =   int(re.findall(r'\d+', subTrString)[0])
13760 amit.gupta 181
            elif "Delivery Charges" in subTrString:
14473 amit.gupta 182
                deliveryCharges =   int(re.findall(r'\d+', subTrString)[0])
13760 amit.gupta 183
            elif "Subtotal" in subTrString:
184
                if int(qty) > 0:
14459 amit.gupta 185
                    amountPaid =   int(re.findall(r'\d+', subTrString)[0])/qty
13760 amit.gupta 186
                else:
14459 amit.gupta 187
                    amountPaid =   0
14458 amit.gupta 188
        if qty>0:
14473 amit.gupta 189
            amount = unitPrice - offerDiscount - sdCash
190
            amount = 0 if amount < 0 else amount
13760 amit.gupta 191
 
14459 amit.gupta 192
        div1 = subOrderElement.find("div", {"class": "blk lrPad subordrs"})
193
        if div1 is None:
13760 amit.gupta 194
            raise ParseException("subOrder", "Could not Parse suborders for Snapdeal")
195
 
14459 amit.gupta 196
        for strDiv in str(div1).split("<div class=\"seperator\"></div>"):
197
            div = BeautifulSoup(strDiv)
13760 amit.gupta 198
            productTitle = str(subOrderElement.find("a").text)
199
            productUrl = "http://m.snapdeal.com/" + productUrl 
200
            subOrder = SubOrder(productTitle, productUrl, placedOn, amountPaid)
201
 
202
            subOrder.amountPaid = amountPaid
203
            subOrder.deliveryCharges = deliveryCharges
204
            subOrder.offerDiscount = offerDiscount
13809 amit.gupta 205
            subOrder.unitPrice = int(unitPrice)
13760 amit.gupta 206
            subOrder.productCode = re.findall(r'\d+$', productUrl)[0]
14415 amit.gupta 207
            subOrder.imgUrl = Mongo.getImgSrc(subOrder.productCode, self.store_id).get('thumbnail')
14458 amit.gupta 208
            cashbackStatus = Store.CB_NA
209
            cashbackAmount = 0
210
            percentage = 0
211
            if amount > 0:
212
                (cashbackAmount, percentage) = self.getCashbackAmount(subOrder.productCode, amount)
14473 amit.gupta 213
                if cashbackAmount > 0:
14476 amit.gupta 214
                    cashbackStatus = Store.CB_PENDING
13760 amit.gupta 215
            subOrder.cashBackStatus = cashbackStatus
216
            subOrder.cashBackAmount = cashbackAmount
14458 amit.gupta 217
            subOrder.cashBackPercentage = percentage
13760 amit.gupta 218
 
219
 
220
            trackAnchor = div.find("a")   
221
            if trackAnchor is not None:
222
                subOrder.tracingkUrl = str(trackAnchor['href'])
223
 
224
            divStr = str(div)
225
            divStr = divStr.replace("\n","").replace("\t", "")
226
 
227
            for line in divStr.split("<br />"):
228
                if "Suborder ID" in line:
229
                    subOrder.merchantSubOrderId = re.findall(r'\d+', line)[0]   
230
                elif "Status" in line:
14402 amit.gupta 231
                    print line
13760 amit.gupta 232
                    subOrder.detailedStatus = re.findall('>(.*?)</span>', line, re.IGNORECASE)[0]
233
                elif "Est. Shipping Date" in line:
234
                    subOrder.estimatedShippingDate = line.split(":")[1].strip()
235
                elif "Est. Delivery Date" in line:
236
                    subOrder.estimatedDeliveryDate = line.split(":")[1].strip()
237
                elif "Courier Name" in line:
238
                    subOrder.courierName = line.split(":")[1].strip()
239
                elif "Tracking No" in line:
240
                    subOrder.trackingNumber = line.split(":")[1].strip()
13809 amit.gupta 241
            subOrders.append(subOrder)
242
        return subOrders
13760 amit.gupta 243
 
14854 amit.gupta 244
    def parseSubOrderB(self, subOrderElement, placedOn):
245
        subOrders = []
246
        prodDivs = subOrderElement.findAll('div', recursive=False)
247
        prodDetailDiv = prodDivs[1].findAll('div', recursive=False)
248
 
249
        offerDiscount = 0
250
        deliveryCharges = None
251
        amountPaid = 0
252
        sdCash = 0
253
        unitPrice = 0
254
 
255
        paymentDivs = prodDivs[2].findAll('div', recursive=False)
256
        for paymentDiv in paymentDivs:
257
            strPaymentDiv = str(paymentDiv)
258
            if "Unit Price" in strPaymentDiv:
259
                unitPrice = int(re.findall(r'\d+', strPaymentDiv)[0])
260
            if "Offer Discount" in strPaymentDiv:
261
                offerDiscount = int(re.findall(r'\d+', strPaymentDiv)[0])
14861 amit.gupta 262
            elif "SD Cash" in strPaymentDiv:
14854 amit.gupta 263
                sdCash =   int(re.findall(r'\d+', strPaymentDiv)[0])
264
            elif "Delivery Charges" in strPaymentDiv:
265
                deliveryCharges =   int(re.findall(r'\d+', strPaymentDiv)[0])
266
            elif "Subtotal" in strPaymentDiv:
267
                amountPaid =   int(re.findall(r'\d+', paymentDiv.find('div', {'class':'itemPriceDetail'}).text)[0])
268
 
269
        amount = unitPrice - offerDiscount - sdCash
270
 
271
        imgDiv = prodDetailDiv[0]
272
        otherDiv = prodDetailDiv[1]
273
        productTitle = otherDiv.find('div',{'class':'orderName'}).text.strip()
274
 
275
        productUrl = imgDiv.a['href']
276
        subOrder = SubOrder(productTitle, productUrl, placedOn, amountPaid)
14863 amit.gupta 277
        subOrder.merchantSubOrderId = prodDivs[0].text.split(':')[1].strip()
14854 amit.gupta 278
        subOrder.detailedStatus = otherDiv.find('div',{'class':'orderStatus'}).span.text.strip()
279
        deliveryStatus = otherDiv.find('div',{'class':'orderDelivery'})
280
        if deliveryStatus is not None:
281
            delString = deliveryStatus.text.strip()
282
            arr = delString.split(':')
283
            if "On" in arr[0]:
284
                subOrder.deliveredOn = arr[1].strip()
285
            elif "Exp. Delivery by" in arr[0]:
286
                subOrder.estimatedDeliveryDate = arr[1].strip()
287
            else:
288
                subOrder.estimatedShippingDate = arr[1].strip()
289
 
290
        subOrder.imgUrl = imgDiv.a.img['src']
291
        subOrder.productCode = re.findall(r'\d+$', productUrl)[0]
292
        subOrder.deliveryCharges = deliveryCharges
293
        subOrder.offerDiscount = offerDiscount
294
        subOrder.unitPrice = int(unitPrice)
295
        cashbackStatus = Store.CB_NA
296
        cashbackAmount = 0
297
        percentage = 0
298
        if amountPaid > 0:
299
            (cashbackAmount, percentage) = self.getCashbackAmount(subOrder.productCode, amount)
300
            if cashbackAmount > 0:
301
                cashbackStatus = Store.CB_PENDING
302
        subOrder.cashBackStatus = cashbackStatus
303
        subOrder.cashBackAmount = cashbackAmount
304
        subOrder.cashBackPercentage = percentage
305
 
306
 
307
        courierDet = subOrderElement.find('div', {'class':'courierDetail'})
308
        if courierDet is not None:
309
            subOrder.courierName = courierDet.span.text.strip()
310
        trackingDet = subOrderElement.find('div', {'class':'trackingNo'})
311
        if trackingDet is not None:
312
            subOrder.trackingUrl = trackingDet.span.a['href']
313
            subOrder.trackingNumber = trackingDet.span.a.text.strip()
314
 
315
        subOrders.append(subOrder)
316
        return subOrder
317
 
13576 amit.gupta 318
    def parseOrderRawHtml(self, orderId, subTagId, userId, rawHtml, orderSuccessUrl):
13760 amit.gupta 319
                    #print merchantOrder
13796 amit.gupta 320
        resp = {}
13582 amit.gupta 321
        try:
322
            br = getBrowserObject()
323
            url = ORDER_TRACK_URL + re.findall('.*(\?.*?)$', orderSuccessUrl,re.IGNORECASE)[0]
13791 amit.gupta 324
            page = br.open(url)
325
            page = ungzipResponse(page)
14854 amit.gupta 326
            try:
327
                merchantOrder = self._parseB(orderId, subTagId, userId, page, orderSuccessUrl)
328
            except:
14861 amit.gupta 329
                traceback.print_exc()
14854 amit.gupta 330
                merchantOrder = self._parse(orderId, subTagId, userId, page, orderSuccessUrl)
331
 
14312 amit.gupta 332
            merchantOrder.orderTrackingUrl = url
333
 
334
            if self._saveToOrder(todict(merchantOrder)):
335
                resp['result'] = 'ORDER_CREATED'
336
            else:
337
                resp['result'] = 'ORDER_ALREADY_CREATED_IGNORED'
338
 
13796 amit.gupta 339
            return resp
13582 amit.gupta 340
        except:
341
            print "Error occurred"
13603 amit.gupta 342
            traceback.print_exc()
14312 amit.gupta 343
            resp['result'] = 'ORDER_NOT_CREATED'
13796 amit.gupta 344
            return resp
13781 amit.gupta 345
 
13569 amit.gupta 346
 
347
        #soup = BeautifulSoup(rawHtml,convertEntities=BeautifulSoup.HTML_ENTITIES)
348
        #soup.find(name, attrs, recursive, text)
13576 amit.gupta 349
 
350
    def _getStatusFromDetailedStatus(self, detailedStatus):
351
        for key, value in Store.OrderStatusMap.iteritems():
14678 amit.gupta 352
            if detailedStatus.lower() in value:
13576 amit.gupta 353
                return key
14675 amit.gupta 354
        print "Detailed Status need to be mapped", detailedStatus, self.store_id
14861 amit.gupta 355
        return None
13569 amit.gupta 356
 
13610 amit.gupta 357
 
13569 amit.gupta 358
    def scrapeStoreOrders(self,):
13760 amit.gupta 359
        #collectionMap = {'palcedOn':1}
13576 amit.gupta 360
        orders = self._getActiveOrders()
361
        for order in orders:
14696 amit.gupta 362
            print "Order", self.store_name, order['orderId']
14398 amit.gupta 363
            try:
364
                url = ORDER_TRACK_URL + re.findall('.*(\?.*?)$', order['orderSuccessUrl'],re.IGNORECASE)[0]
14758 amit.gupta 365
                page = fetchResponseUsingProxy(url)
14398 amit.gupta 366
                #page=page.decode("utf-8")
14861 amit.gupta 367
                soup = BeautifulSoup(page)
368
                try:
369
                    self.tryBParsing(order, soup)
370
                except:
371
                    traceback.print_exc()
372
                    sections = soup.findAll("section")
373
                    orderEl = sections[1]
374
                    orderTrs = orderEl.findAll("tr")
375
 
376
                    placedOn = str(orderTrs[0].findAll("td")[1].text)
377
                    sections.pop(0)
378
                    sections.pop(0)
379
 
380
                    subOrders = sections
381
                    bulk = self.db.merchantOrder.initialize_ordered_bulk_op()
382
                    closed = True
383
                    for subOrderElement in subOrders:
384
                        div1 = subOrderElement.findAll("div", {"class": "blk lrPad subordrs"})
385
                        if len(div1)<=0:
386
                            raise ParseException("subOrder", "Could not Parse suborders for Snapdeal")
387
                        subOrder = None
388
                        breakFlag = False
389
                        for strDiv in str(div1).split("<div class=\"seperator\"></div>"):
390
                            div = BeautifulSoup(strDiv)
391
                            divStr = str(div)
392
                            divStr = divStr.replace("\n","").replace("\t", "")
393
                            updateMap = {}
394
                            for line in divStr.split("<br />"):
395
                                if "Suborder ID" in line:
396
                                    merchantSubOrderId = re.findall(r'\d+', line)[0]
397
                                    #break if suborder is inactive   
398
                                    subOrder =  self._isSubOrderActive(order, merchantSubOrderId)
399
                                    if subOrder is None:
400
                                        subOrders = self.parseSubOrder(subOrderElement, placedOn)
401
                                        self.db.merchantOrder.update({"orderId":order['orderId']},{'$push':{"subOrders":{"$each":todict(subOrders)}}})
402
                                        print "Added new suborders to Order id - ", order['orderId']
403
                                        closed = False
404
                                        breakFlag = True
405
                                        break
406
                                    elif subOrder['closed']:
407
                                        breakFlag = True
408
                                        break
409
                                    else: 
410
                                        findMap = {"orderId": order['orderId'], "subOrders.merchantSubOrderId": merchantSubOrderId}
411
                                elif "Status :" in line:
412
                                    detailedStatus = re.findall('>(.*?)</span>', line, re.IGNORECASE)[0]
413
                                    updateMap["subOrders.$.detailedStatus"] = detailedStatus
414
                                    status = self._getStatusFromDetailedStatus(detailedStatus) 
415
                                    closedStatus = status in [Store.ORDER_DELIVERED, Store.ORDER_CANCELLED]
416
                                    if status is not None:
417
                                        updateMap["subOrders.$.status"] = status
418
                                    if detailedStatus == 'Closed For Vendor Reallocation':
419
                                        #if it is more than 6hours mark closed.
420
                                        closeAt = subOrder.get("closeAt") 
421
                                        if closeAt is None:
422
                                            closeAt = datetime.now() + timedelta(hours=6)
423
                                            updateMap["subOrders.$.closeAt"] = datetime.strftime(closeAt,"%Y-%m-%d %H:%M:%S")
424
                                        else:
425
                                            closeAt = datetime.strptime(closeAt,"%Y-%m-%d %H:%M:%S")
426
                                            if datetime.now() > closeAt:
427
                                                closedStatus = True
428
 
429
 
430
                                    if closedStatus:
431
                                        #if status is closed then change the paybackStatus accordingly
432
                                        updateMap["subOrders.$.closed"] = True
433
                                        if status == Store.ORDER_DELIVERED:
434
                                            if subOrder.get("cashBackStatus") == Store.CB_PENDING:
435
                                                updateMap["subOrders.$.cashBackStatus"] = Store.CB_APPROVED
436
                                        elif status == Store.ORDER_CANCELLED:
437
                                            if subOrder.get("cashBackStatus") == Store.CB_PENDING:
438
                                                updateMap["subOrders.$.cashBackStatus"] = Store.CB_CANCELLED
439
 
14398 amit.gupta 440
                                    else:
14861 amit.gupta 441
                                        closed = False
442
                                elif "Est. Shipping Date" in line:
443
                                    estimatedShippingDate = line.split(":")[1].strip()
444
                                    updateMap["subOrders.$.estimatedShippingDate"] = estimatedShippingDate
445
                                elif "Est. Delivery Date" in line:
446
                                    estimatedDeliveryDate = line.split(":")[1].strip()
447
                                    updateMap["subOrders.$.estimatedDeliveryDate"] = estimatedDeliveryDate
448
                                elif "Courier Name" in line:
449
                                    courierName = line.split(":")[1].strip()
450
                                    updateMap["subOrders.$.courierName"] = courierName
451
                                elif "Tracking No" in line:
452
                                    trackingNumber = line.split(":")[1].strip()
453
                                    updateMap["subOrders.$.trackingNumber"] = trackingNumber
454
 
455
                            if breakFlag:
456
                                continue
457
 
458
                            bulk.find(findMap).update({'$set' : updateMap})
459
                        bulk.find({'orderId': order['orderId']}).update({'$set':{'closed': closed,"parseError":False}})
460
                    result = bulk.execute()
461
                    tprint(result)
14398 amit.gupta 462
            except:
14847 amit.gupta 463
                tprint("Could not update " + str(order['orderId']) + "For store " + self.getName())
14692 amit.gupta 464
                self.db.merchantOrder.update({"orderId":order['orderId']}, {"$set":{"parseError":True}})
14398 amit.gupta 465
                traceback.print_exc()                
13576 amit.gupta 466
 
14854 amit.gupta 467
    def tryBParsing(self, order, soup):
468
        orderDetailContainerDivs = soup.body.find("div", {'class':'cardLayoutWrap'}).findAll('div', recursive=False)
469
        orderDetailDiv = orderDetailContainerDivs.pop(0)
470
        placedOn = orderDetailDiv.span.text.split(':')[1].strip()
471
 
472
        orderDetailContainerDivs.pop(0)
473
 
474
        subOrders = orderDetailContainerDivs
475
        bulk = self.db.merchantOrder.initialize_ordered_bulk_op()
476
        closed = True
477
        for subOrderElement in subOrders:
478
            prodDivs = subOrderElement.findAll('div', recursive=False)
479
            merchantSubOrderId = prodDivs[0].text.split(':')[1].strip()
480
            prodDetailDiv = prodDivs[1].findAll('div', recursive=False)
481
            imgDiv = prodDetailDiv[0]
482
            otherDiv = prodDetailDiv[1]
483
 
484
            subOrder = None
485
            breakFlag = False
14861 amit.gupta 486
            subOrder =  self._isSubOrderActive(order, merchantSubOrderId)
487
            if subOrder is None:
488
                subOrders = self.parseSubOrderB(subOrderElement, placedOn)
489
                self.db.merchantOrder.update({"orderId":order['orderId']},{'$push':{"subOrders":{"$each":todict([subOrders])}}})
490
                print "Added new suborders to Order id - ", order['orderId']
491
                closed = False
492
                continue
493
            elif subOrder['closed']:
494
                continue
495
            else: 
496
                findMap = {"orderId": order['orderId'], "subOrders.merchantSubOrderId": merchantSubOrderId}
497
                updateMap = {}
498
                prodDivs = subOrderElement.findAll('div', recursive=False)
499
                prodDetailDiv = prodDivs[1].findAll('div', recursive=False)
500
                otherDiv = prodDetailDiv[1]
501
                detailedStatus = otherDiv.find('div',{'class':'orderStatus'}).span.text.strip()
502
                updateMap["subOrders.$.detailedStatus"] = detailedStatus
503
                status = self._getStatusFromDetailedStatus(detailedStatus) 
504
                closedStatus = status in [Store.ORDER_DELIVERED, Store.ORDER_CANCELLED]
505
                if status is not None:
14854 amit.gupta 506
                    updateMap["subOrders.$.status"] = status
14861 amit.gupta 507
                if detailedStatus == 'Closed For Vendor Reallocation':
508
                    #if it is more than 6hours mark closed.
509
                    closeAt = subOrder.get("closeAt") 
510
                    if closeAt is None:
511
                        closeAt = datetime.now() + timedelta(hours=6)
512
                        updateMap["subOrders.$.closeAt"] = datetime.strftime(closeAt,"%Y-%m-%d %H:%M:%S")
14854 amit.gupta 513
                    else:
14861 amit.gupta 514
                        closeAt = datetime.strptime(closeAt,"%Y-%m-%d %H:%M:%S")
515
                        if datetime.now() > closeAt:
516
                            closedStatus = True
517
 
518
 
519
                if closedStatus:
520
                    #if status is closed then change the paybackStatus accordingly
521
                    updateMap["subOrders.$.closed"] = True
522
                    if status == Store.ORDER_DELIVERED:
523
                        if subOrder.get("cashBackStatus") == Store.CB_PENDING:
524
                            updateMap["subOrders.$.cashBackStatus"] = Store.CB_APPROVED
525
                    elif status == Store.ORDER_CANCELLED:
526
                        if subOrder.get("cashBackStatus") == Store.CB_PENDING:
527
                            updateMap["subOrders.$.cashBackStatus"] = Store.CB_CANCELLED
528
 
529
                else:
530
                    closed = False
14854 amit.gupta 531
 
14861 amit.gupta 532
                deliveryStatus = otherDiv.find('div',{'class':'orderDelivery'})
533
                if deliveryStatus is not None:
534
                    delString = deliveryStatus.text.strip()
535
                    arr = delString.split(':')    
536
                    if "On" in arr[0]:
537
                        updateMap['subOrders.$.deliveredOn'] = arr[1].strip()
538
                    elif "Exp. Delivery by" in arr[0]:
539
                        updateMap['subOrders.$.estimatedDeliveryDate'] = arr[1].strip()
540
                    else:
541
                        updateMap['subOrders.$.estimatedShippingDate'] = arr[1].strip()
542
                courierDet = subOrderElement.find('div', {'class':'courierDetail'})
543
                if courierDet is not None:
544
                    updateMap['subOrders.$.courierName'] = courierDet.span.text.strip()
545
                trackingDet = subOrderElement.find('div', {'class':'trackingNo'})
546
                if trackingDet is not None:
547
                    updateMap['subOrders.$.trackingUrl'] = trackingDet.span.a['href']
548
                    updateMap['subOrders.$.trackingNumber'] = trackingDet.span.a.text.strip()                    
14854 amit.gupta 549
            bulk.find(findMap).update({'$set' : updateMap})
550
        bulk.find({'orderId': order['orderId']}).update({'$set':{'closed': closed,"parseError":False}})
551
        result = bulk.execute()
552
        tprint(result)
553
 
13576 amit.gupta 554
 
13569 amit.gupta 555
    def _saveToAffiliate(self, offers):
13576 amit.gupta 556
        collection = self.db.snapdealOrderAffiliateInfo
14443 amit.gupta 557
        mcollection = self.db.merchantOrder
558
        for offer in offers:
559
            offer = self.covertToObj(offer)
560
            collection.update({"adId":offer.adId, "saleAmount":offer.saleAmount, "payOut":offer.payOut},{"$set":todict(offer)}, upsert=True)
14446 amit.gupta 561
            mcollection.update({"subTagId":offer.subTagId, "storeId":self.store_id, "subOrders.missingAff":True}, {"$set":{"subOrders.$.missingAff":False}})
13569 amit.gupta 562
 
563
 
564
    def _getAllOffers(self, br, token):
565
        allOffers = []
566
        nextPage = 1  
567
        while True:
568
            data = getPostData(token, nextPage)
569
            response = br.open(POST_URL, data)
13680 amit.gupta 570
            rmap = json.loads(ungzipResponse(response))
13569 amit.gupta 571
            if rmap is not None:
572
                rmap = rmap['response']
14492 amit.gupta 573
                print rmap
13569 amit.gupta 574
                if rmap is not None and len(rmap['errors'])==0:
575
                    allOffers += rmap['data']['data']
576
            nextPage += 1
577
            if rmap['data']['pageCount']<nextPage:
578
                break
579
 
580
        return allOffers
581
 
582
    def covertToObj(self,offer):
583
        offerData = offer['Stat']
584
        offer1 = AffiliateInfo(offerData['affiliate_info1'], self.store_id, offerData['conversion_status'], offerData['ad_id'], 
14576 amit.gupta 585
                              offerData['datetime'], int(float(offerData['payout'])), offer['Offer']['name'], offerData['ip'], int(float(offerData['conversion_sale_amount'])))
14443 amit.gupta 586
        offer1.saleTime = int(time.mktime(datetime.strptime(offer1.saleDate, "%Y-%m-%d %H:%M:%S").timetuple()))
13569 amit.gupta 587
        return offer1
588
def getPostData(token, page = 1, limit= 20, startDate=None, endDate=None):
13680 amit.gupta 589
    endDate=date.today() + timedelta(days=1)
590
    startDate=endDate - timedelta(days=31)
13569 amit.gupta 591
 
592
    parameters = (
593
        ("page",str(page)),
594
        ("limit",str(limit)),
595
        ("fields[]","Stat.offer_id"),
596
        ("fields[]","Stat.datetime"),
597
        ("fields[]","Offer.name"),
598
        ("fields[]","Stat.conversion_status"),
599
        ("fields[]","Stat.conversion_sale_amount"),
600
        ("fields[]","Stat.payout"),
601
        ("fields[]","Stat.ip"),
602
        ("fields[]","Stat.ad_id"),
603
        ("fields[]","Stat.affiliate_info1"),
604
        ("sort[Stat.datetime]","desc"),
605
        ("filters[Stat.date][conditional]","BETWEEN"),
606
        ("filters[Stat.date][values][]",startDate.strftime('%Y-%m-%d')),
607
        ("filters[Stat.date][values][]",endDate.strftime('%Y-%m-%d')),
608
        ("data_start",startDate.strftime('%Y-%m-%d')),
609
        ("data_end",endDate.strftime('%Y-%m-%d')),
610
        ("Method","getConversions"),
611
        ("NetworkId","jasper"),
612
        ("SessionToken",token),
613
    )
614
    #Encode the parameters
615
    return urllib.urlencode(parameters)
616
 
617
def main():
14458 amit.gupta 618
    #print todict([1,2,"3"])
14402 amit.gupta 619
    store = getStore(3)
14861 amit.gupta 620
    #store.parseOrderRawHtml(33222, "32323", 2, "323243", "https://m.snapdeal.com/purchaseMobileComplete?code=1131a4add85aefb40a924d71cbf921bd&order=5822561503")
621
    store.scrapeStoreOrders()
13662 amit.gupta 622
    #store._isSubOrderActive(8, "5970688907")
14653 amit.gupta 623
    #store.scrapeAffiliate()
14758 amit.gupta 624
    #store.scrapeStoreOrders()
13569 amit.gupta 625
 
626
 
14459 amit.gupta 627
 
13576 amit.gupta 628
def todict(obj, classkey=None):
629
    if isinstance(obj, dict):
630
        data = {}
631
        for (k, v) in obj.items():
632
            data[k] = todict(v, classkey)
633
        return data
634
    elif hasattr(obj, "_ast"):
635
        return todict(obj._ast())
636
    elif hasattr(obj, "__iter__"):
637
        return [todict(v, classkey) for v in obj]
638
    elif hasattr(obj, "__dict__"):
639
        data = dict([(key, todict(value, classkey)) 
640
            for key, value in obj.__dict__.iteritems() 
641
            if not callable(value) and not key.startswith('_')])
642
        if classkey is not None and hasattr(obj, "__class__"):
643
            data[classkey] = obj.__class__.__name__
644
        return data
645
    else:
646
        return obj
14239 amit.gupta 647
 
648
if __name__ == '__main__':
649
    main()