Subversion Repositories SmartDukaan

Rev

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

Rev Author Line No. Line
104 ashish 1
'''
2
Created on 29-Mar-2010
3
 
2536 chandransh 4
@author: Chandranshu
104 ashish 5
'''
2536 chandransh 6
 
4369 rajveer 7
from datetime import date, timedelta
6424 rajveer 8
from math import ceil
2536 chandransh 9
from elixir import *
5944 mandeep.dh 10
from random import randrange
3499 mandeep.dh 11
from reportlab.lib.units import inch
2536 chandransh 12
from reportlab.pdfgen.canvas import Canvas
13
from reportlab.rl_config import defaultPageSize
5944 mandeep.dh 14
from shop2020.clients.AlertClient import AlertClient
3499 mandeep.dh 15
from shop2020.clients.CatalogClient import CatalogClient
16
from shop2020.clients.HelperClient import HelperClient
5944 mandeep.dh 17
from shop2020.clients.InventoryClient import InventoryClient
3499 mandeep.dh 18
from shop2020.clients.LogisticsClient import LogisticsClient
2747 chandransh 19
from shop2020.clients.PaymentClient import PaymentClient
3499 mandeep.dh 20
from shop2020.clients.UserClient import UserClient
21
from shop2020.clients.WarehouseClient import WarehouseClient
22
from shop2020.model.v1 import order
6048 rajveer 23
from shop2020.model.v1.order.impl import DataService, RechargeService
3499 mandeep.dh 24
from shop2020.model.v1.order.impl.DataService import Transaction, LineItem, \
4435 mandeep.dh 25
    Order, BatchNoGenerator, InvoiceIDGenerator, TransactionRequiringExtraProcessing, \
5348 anupam.sin 26
    OrderInventory, Alert, PaymentSettlement, EBSSettlementSummary, \
6389 rajveer 27
    CodVerificationAgent, Attribute, RechargeVoucherTracker, EmiScheme,\
28
    MiscCharges
6000 mandeep.dh 29
from shop2020.model.v1.order.impl.model.MobileRechargeOrder import \
30
    MobileRechargeOrder
2687 chandransh 31
from shop2020.model.v1.order.impl.model.ReturnOrder import ReturnOrder
5944 mandeep.dh 32
from shop2020.thriftpy.alert.ttypes import MonitoredEntity, EntityType
5527 anupam.sin 33
from shop2020.thriftpy.logistics.ttypes import DeliveryType, PickUpType
6000 mandeep.dh 34
from shop2020.thriftpy.model.v1.catalog.ttypes import ItemType
5944 mandeep.dh 35
from shop2020.thriftpy.model.v1.inventory.ttypes import BillingType, \
36
    InventoryServiceException, WarehouseType, InventoryType
4435 mandeep.dh 37
from shop2020.thriftpy.model.v1.order.ttypes import TransactionServiceException, \
38
    TransactionStatus, OrderStatus, DelayReason, ExtraTransactionProcessingType, \
6031 rajveer 39
    HotspotAction, TimeoutSummary, OrderStatusGroups, OrderType,\
6389 rajveer 40
    RechargeOrderStatus, RechargeType, RechargeStatistics, DeviceNumberInfo,\
41
    EmiChargeType
5944 mandeep.dh 42
from shop2020.thriftpy.model.v1.user.ttypes import VoucherType
4421 mandeep.dh 43
from shop2020.thriftpy.payments.ttypes import PaymentException
5110 mandeep.dh 44
from shop2020.thriftpy.warehouse.ttypes import ScanType, \
45
    WarehouseServiceException
6000 mandeep.dh 46
from shop2020.utils.EmailAttachmentSender import mail, get_attachment_part
855 rajveer 47
from shop2020.utils.Utils import to_py_date, to_java_date
4008 mandeep.dh 48
from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound
3499 mandeep.dh 49
from sqlalchemy.sql import func
6000 mandeep.dh 50
from sqlalchemy.sql.expression import and_, or_, desc, not_, distinct
3499 mandeep.dh 51
from string import Template
52
from textwrap import dedent
53
import datetime
5944 mandeep.dh 54
import httplib
3499 mandeep.dh 55
import logging
56
import os
4421 mandeep.dh 57
import sys
4435 mandeep.dh 58
import time
4421 mandeep.dh 59
import traceback
5806 rajveer 60
import urllib
6307 anupam.sin 61
 
62
 
63
import time
64
 
65
 
5966 rajveer 66
from shop2020.config.client.ConfigClient import ConfigClient
6177 anupam.sin 67
from shop2020.model.v1.order.impl.model.RechargePlan import RechargePlan
6031 rajveer 68
from shop2020.model.v1.order.impl.model.RechargeOrder import RechargeOrder
69
from shop2020.model.v1.order.impl.model.UserWallet import UserWallet
70
from shop2020.model.v1.order.impl.model.UserWalletHistory import UserWalletHistory
6048 rajveer 71
from shop2020.model.v1.order.impl.model.ServiceProvider import ServiceProvider
72
from shop2020.model.v1.order.impl.model.DTHRechargeOrder import DTHRechargeOrder
6289 anupam.sin 73
from shop2020.model.v1.order.impl.model.TelecomCircle import TelecomCircle
6307 anupam.sin 74
from shop2020.model.v1.order.impl.model.ServiceAvailability import ServiceAvailability
6289 anupam.sin 75
from shop2020.model.v1.order.impl.model.RechargeDenomination import RechargeDenomination
6281 rajveer 76
from shop2020.clients.PromotionClient import PromotionClient
6305 rajveer 77
from decimal import Decimal
6396 amit.gupta 78
from shop2020.model.v1.order.impl.Convertors import to_t_emi_scheme
6507 anupam.sin 79
from shop2020.model.v1.order.impl.model.BaseOrder import BaseOrder
5944 mandeep.dh 80
 
3499 mandeep.dh 81
logging.basicConfig(level=logging.DEBUG)
104 ashish 82
 
5966 rajveer 83
sourceId = int(ConfigClient().get_property("sourceid"))
84
mail_user = 'cnc.center@shop2020.in'
85
mail_password = '5h0p2o2o'
86
help_user = 'help@shop2020.in'
87
help_password = '5h0p2o2o'
6029 rajveer 88
source_url = 'www.saholic.com'
89
source_name = 'Saholic'
3499 mandeep.dh 90
 
4763 rajveer 91
PREPAID_SHIPPING_CUTOFF_TIME = 15
4019 chandransh 92
COD_SHIPPING_CUTOFF_TIME = 12
3499 mandeep.dh 93
 
2819 chandransh 94
PAGE_HEIGHT=defaultPageSize[1]
95
PAGE_WIDTH=defaultPageSize[0]
96
 
4763 rajveer 97
billedOrdersColorMap = {}
98
 
3499 mandeep.dh 99
ORDER_STATUS_TO_USER_TRUST_LEVEL_DELTA_DICT = {
4873 mandeep.dh 100
        OrderStatus.RTO_RESHIPPED    :  3,
101
        OrderStatus.RTO_IN_TRANSIT   : -5,
102
        OrderStatus.DELIVERY_SUCCESS :  1,
103
        OrderStatus.DOA_CERT_INVALID : -5
3499 mandeep.dh 104
}
105
 
5783 rajveer 106
delhi_pincodes = ['110001','110002','110003','110004','110005','110006','110007','110008','110009','110010','110011','110012','110013','110014','110015',\
107
                '110016','110017','110018','110019','110020','110021','110022','110023','110024','110025','110026','110027','110028','110029','110030',\
108
                '110031','110032','110033','110034','110035','110037','110038','110041','110042','110044','110045','110046','110047','110048','110049',\
109
                '110051','110052','110053','110054','110055','110056','110057','110058','110059','110060','110061','110062','110063','110064','110065',\
110
                '110066','110067','110068','110070','110071','110074','110075','110076','110078','110081','110082','110083','110084','110085','110086',\
111
                '110087','110088','110089','110091','110092','110093','110094','110095','110096','110101','110103','110104','110105','110106','110107',\
112
                '110108','110109','110110','110112','110113','110114','110115','110116','110117','110118','110119','110120','110122','110124','110125',\
113
                '110301','110302','110501','110502','110503','110504','110505','110510','110511','110512','110601','110602','110603','110604','110605',\
114
                '110606','110607','110608','110609']
115
 
104 ashish 116
def get_new_transaction():
117
    transaction = Transaction()
118
    transaction.createdOn = datetime.datetime.now()
738 chandransh 119
    transaction.status = TransactionStatus.INIT
104 ashish 120
    transaction.status_message = "New transaction"
483 rajveer 121
    return transaction
122
 
123
def create_order(t_order):
124
    order = Order()
125
    if t_order.warehouse_id:
126
        order.warehouse_id = t_order.warehouse_id
509 rajveer 127
    if t_order.logistics_provider_id:    
483 rajveer 128
        order.logistics_provider_id = t_order.logistics_provider_id
129
        order.airwaybill_no = t_order.airwaybill_no
130
        order.tracking_id = t_order.tracking_id
4009 chandransh 131
    if t_order.expected_shipping_time:
4019 chandransh 132
        order.expected_shipping_time = to_py_date(t_order.expected_shipping_time).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4102 chandransh 133
        order.promised_shipping_time = order.expected_shipping_time
4009 chandransh 134
    if t_order.expected_delivery_time:
483 rajveer 135
        order.expected_delivery_time = to_py_date(t_order.expected_delivery_time)
4102 chandransh 136
        order.promised_delivery_time = order.expected_delivery_time
483 rajveer 137
    if t_order.customer_id:    
138
        order.customer_id = t_order.customer_id
139
        order.customer_name = t_order.customer_name
860 rajveer 140
        order.customer_city = t_order.customer_city
141
        order.customer_state = t_order.customer_state
483 rajveer 142
        order.customer_mobilenumber = t_order.customer_mobilenumber
143
        order.customer_pincode = t_order.customer_pincode
738 chandransh 144
        order.customer_address1 = t_order.customer_address1
145
        order.customer_address2 = t_order.customer_address2
483 rajveer 146
        order.customer_email = t_order.customer_email
147
    #new status and status description to be added    
716 rajveer 148
    order.status = t_order.status  
483 rajveer 149
    order.statusDescription = t_order.statusDescription
150
    order.total_amount = t_order.total_amount
6318 rajveer 151
    order.gvAmount = t_order.gvAmount
483 rajveer 152
    order.total_weight = t_order.total_weight
153
    if t_order.created_timestamp:
154
        order.created_timestamp = to_py_date(t_order.created_timestamp)
155
    else:
685 chandransh 156
        order.created_timestamp = datetime.datetime.now()
5555 rajveer 157
    order.pickupStoreId = t_order.pickupStoreId
483 rajveer 158
    return order
159
 
160
def create_transaction(t_transaction):
161
    t_orders = t_transaction.orders
162
    if not t_orders:
163
        raise TransactionServiceException(101, "Orders missing from the transaction")
164
    transaction = get_new_transaction()
165
    transaction.customer_id = t_transaction.customer_id 
166
    transaction.shopping_cart_id = t_transaction.shoppingCartid
2219 varun.gupt 167
    transaction.coupon_code = t_transaction.coupon_code
2815 vikas 168
    transaction.session_source = t_transaction.sessionSource
169
    transaction.session_start_time = to_py_date(t_transaction.sessionStartTime)
3858 vikas 170
    transaction.first_source = t_transaction.firstSource
171
    transaction.first_source_start_time = to_py_date(t_transaction.firstSourceTime)
483 rajveer 172
 
6389 rajveer 173
    totalAmount = 0
483 rajveer 174
    for t_order in t_orders:
175
        order = create_order(t_order)
6389 rajveer 176
        totalAmount = totalAmount + order.total_amount
483 rajveer 177
        order.transaction = transaction 
178
        for line_item in t_order.lineitems:
179
            litem = LineItem()
716 rajveer 180
            litem.item_id = line_item.item_id
963 chandransh 181
            litem.productGroup = line_item.productGroup
483 rajveer 182
            litem.brand = line_item.brand
183
            if line_item.model_number:
184
                litem.model_number = line_item.model_number
185
            if line_item.model_name:    
186
                litem.model_name = line_item.model_name
908 chandransh 187
            if line_item.color:
188
                litem.color = line_item.color
483 rajveer 189
            if line_item.extra_info:
190
                litem.extra_info = line_item.extra_info
191
            litem.quantity = line_item.quantity
192
            litem.unit_price = line_item.unit_price
193
            litem.unit_weight = line_item.unit_weight
194
            litem.total_price = line_item.total_price
195
            litem.total_weight = line_item.total_weight
4283 anupam.sin 196
            '''litem.transfer_price = line_item.transfer_price'''
4172 rajveer 197
            litem.dealText = line_item.dealText
4295 varun.gupt 198
            litem.warranty_expiry_timestamp = to_py_date(line_item.warrantry_expiry_timestamp)
1976 varun.gupt 199
 
483 rajveer 200
            """
201
            if line_item.addedOn:
202
                litem.added_on = to_py_date(line_item.addedOn)
203
            else:
204
                litem.added_on = datetime.datetime.now()
205
            """
206
            litem.order = order
207
    session.commit()
6389 rajveer 208
 
209
    generateOtherCharges(transaction, t_transaction.emiSchemeId, totalAmount)
210
 
483 rajveer 211
    return transaction.id
212
 
6389 rajveer 213
def generateOtherCharges(transaction, emiSchemeId, totalAmount):
214
    emiScheme = EmiScheme.get_by(id=emiSchemeId)
215
    if not emiScheme:
216
        return
217
 
218
    if totalAmount < emiScheme.minAmount:
219
        return
220
 
221
    emiCharge = 0
222
    if emiScheme.chargeType == EmiChargeType.FIXED:
6424 rajveer 223
        emiCharge = ceil(emiScheme.chargeValue)
6389 rajveer 224
    if emiScheme.chargeType == EmiChargeType.PERCENTAGE:
6424 rajveer 225
        emiCharge = ceil(totalAmount*(emiScheme.chargeValue)/100)
6389 rajveer 226
 
227
    miscCharge = MiscCharges()
228
    miscCharge.transaction = transaction
229
    miscCharge.chargeType = 1 ##EMI charges
230
    miscCharge.chargeAmount = emiCharge
231
    session.commit()
232
 
483 rajveer 233
def get_transaction(transaction_id):
234
    transaction = Transaction.get_by(id=transaction_id)
235
    if not transaction:
236
        raise TransactionServiceException(108, "no such transaction")
237
 
238
    return transaction
239
 
240
def get_transactions_for_customer(customer_id, from_date, to_date, status):
241
    if not customer_id:
242
        raise TransactionServiceException(101, "bad customer id")
243
    query = Transaction.query.filter(Transaction.customer_id == customer_id)
1493 vikas 244
    if status is not None:
483 rajveer 245
        query = query.filter(Transaction.status == status)
246
    if from_date:
247
        query = query.filter(Transaction.createdOn >from_date)
248
    if to_date:
249
        query = query.filter(Transaction.createdOn < to_date)
250
    return query.all()
251
 
252
def get_transactions_for_shopping_cart_id(shopping_cart_id):
253
    transactions = Transaction.query.filter_by(shopping_cart_id=shopping_cart_id).all()
254
    return transactions
255
 
256
def get_transaction_status(transaction_id):
257
    if not transaction_id:
258
        raise TransactionServiceException(101, "bad transaction id")
259
 
260
    transaction = get_transaction(transaction_id)
261
 
262
    return transaction.status
263
 
5527 anupam.sin 264
def change_transaction_status(transaction_id, new_status, description, pickUp, orderType):
483 rajveer 265
    transaction = get_transaction(transaction_id)
266
    transaction.status = new_status
267
    transaction.status_message = description
5783 rajveer 268
    ## Assign runner in case of delhi pincodes
5784 rajveer 269
    if transaction.orders[0].pickupStoreId:
270
        logistics_client = LogisticsClient().get_client() 
271
        store = logistics_client.getPickupStore(transaction.orders[0].pickupStoreId)
272
        if delhi_pincodes.__contains__(store.pin):
5793 rajveer 273
            pickUp = PickUpType.RUNNER  
5783 rajveer 274
 
690 chandransh 275
    if new_status == TransactionStatus.FAILED:
276
        for order in transaction.orders:
277
            order.status = OrderStatus.PAYMENT_FAILED
2339 chandransh 278
            order.statusDescription = "Payment Failed"
4247 rajveer 279
    elif new_status == TransactionStatus.AUTHORIZED or new_status == TransactionStatus.FLAGGED:
690 chandransh 280
        for order in transaction.orders:
4247 rajveer 281
            if new_status == TransactionStatus.AUTHORIZED:
282
                order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
283
                order.statusDescription = "Submitted to warehouse"
284
            elif new_status == TransactionStatus.FLAGGED:
285
                order.status = OrderStatus.PAYMENT_FLAGGED
286
                order.statusDescription = "Payment flagged by gateway"
3092 chandransh 287
            order.cod = False
5527 anupam.sin 288
            order.orderType = orderType
743 rajveer 289
            #After we got payment success, we will set logistics info also 
290
            logistics_client = LogisticsClient().get_client()
291
            #FIXME line item is only one now. If multiple will come, need to fix.
871 chandransh 292
            item_id = order.lineitems[0].item_id
5767 rajveer 293
            logistics_info = logistics_client.getLogisticsInfo(order.customer_pincode, item_id, DeliveryType.PREPAID, pickUp)
743 rajveer 294
 
1730 ankur.sing 295
            current_time = datetime.datetime.now()
296
            logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
4009 chandransh 297
            logistics_info.shippingTime = adjust_delivery_time(current_time, logistics_info.shippingTime)
5110 mandeep.dh 298
 
743 rajveer 299
            order.warehouse_id = logistics_info.warehouseId
5110 mandeep.dh 300
            order.fulfilmentWarehouseId = logistics_info.fulfilmentWarehouseId
743 rajveer 301
            order.logistics_provider_id = logistics_info.providerId
302
            order.airwaybill_no = logistics_info.airway_billno
303
            order.tracking_id = order.airwaybill_no
4019 chandransh 304
            order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4102 chandransh 305
            order.promised_shipping_time = order.expected_shipping_time
5165 rajveer 306
            order.expected_delivery_time = (current_time + datetime.timedelta(days=logistics_info.deliveryTime)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
3986 chandransh 307
            order.promised_delivery_time = order.expected_delivery_time 
871 chandransh 308
 
5944 mandeep.dh 309
            inventory_client = InventoryClient().get_client()
5966 rajveer 310
            inventory_client.reserveItemInWarehouse(item_id, logistics_info.fulfilmentWarehouseId, sourceId, order.id, to_java_date(order.created_timestamp), to_java_date(order.promised_shipping_time), order.lineitems[0].quantity)
1351 varun.gupt 311
 
4316 anupam.sin 312
            try:
5944 mandeep.dh 313
                item_pricing = inventory_client.getItemPricing(item_id, -1)
4316 anupam.sin 314
                order.lineitems[0].transfer_price = item_pricing.transferPrice
315
            except:
316
                print "Not able to get transfer price. Skipping"
5944 mandeep.dh 317
 
318
            catalog_client = CatalogClient().get_client() 
5508 rajveer 319
            voucherAmount = catalog_client.getVoucherAmount(item_id, VoucherType.SPICEDECK_MOBILE)
320
            if voucherAmount:
5514 rajveer 321
                __create_recharge_voucher_tracker(order, voucherAmount, VoucherType.SPICEDECK_MOBILE)
5508 rajveer 322
 
3064 chandransh 323
    elif new_status == TransactionStatus.COD_IN_PROCESS:
324
        for order in transaction.orders:
4662 rajveer 325
            order.status = OrderStatus.COD_VERIFICATION_PENDING
3064 chandransh 326
            order.statusDescription = "Verification Pending"
327
            order.cod = True
5527 anupam.sin 328
            order.orderType = orderType
3064 chandransh 329
            #After we got payment success, we will set logistics info also 
330
            logistics_client = LogisticsClient().get_client()
331
            #FIXME line item is only one now. If multiple will come, need to fix.
332
            item_id = order.lineitems[0].item_id
5555 rajveer 333
            if order.pickupStoreId:
5885 mandeep.dh 334
                # Need to send prepaid awb number for store pick up orders
335
                logistics_info = logistics_client.getLogisticsEstimation(item_id, order.customer_pincode, DeliveryType.COD)
336
                prepaid_logistics_info = logistics_client.getLogisticsInfo(order.customer_pincode, item_id, DeliveryType.PREPAID, pickUp)
337
                logistics_info.airway_billno = prepaid_logistics_info.airway_billno
338
                logistics_info.providerId = prepaid_logistics_info.providerId
5555 rajveer 339
            else:
5767 rajveer 340
                logistics_info = logistics_client.getLogisticsInfo(order.customer_pincode, item_id, DeliveryType.COD, pickUp)
5885 mandeep.dh 341
 
3064 chandransh 342
            current_time = datetime.datetime.now()
343
            logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
4009 chandransh 344
            logistics_info.shippingTime = adjust_delivery_time(current_time, logistics_info.shippingTime)
3064 chandransh 345
 
346
            order.warehouse_id = logistics_info.warehouseId
5110 mandeep.dh 347
            order.fulfilmentWarehouseId = logistics_info.fulfilmentWarehouseId
3064 chandransh 348
            order.logistics_provider_id = logistics_info.providerId
349
            order.airwaybill_no = logistics_info.airway_billno
350
            order.tracking_id = order.airwaybill_no
4019 chandransh 351
            order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=COD_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4102 chandransh 352
            order.promised_shipping_time = order.expected_shipping_time
5165 rajveer 353
            order.expected_delivery_time = (current_time + datetime.timedelta(days=logistics_info.deliveryTime)).replace(hour=COD_SHIPPING_CUTOFF_TIME, minute=0, second=0)
3986 chandransh 354
            order.promised_delivery_time = order.expected_delivery_time
3064 chandransh 355
 
5944 mandeep.dh 356
            inventory_client = InventoryClient().get_client()
5966 rajveer 357
            inventory_client.reserveItemInWarehouse(item_id, logistics_info.fulfilmentWarehouseId, sourceId, order.id, to_java_date(order.created_timestamp), to_java_date(order.promised_shipping_time), order.lineitems[0].quantity)
3064 chandransh 358
 
4316 anupam.sin 359
            try:
5944 mandeep.dh 360
                item_pricing = inventory_client.getItemPricing(item_id, -1)
4316 anupam.sin 361
                order.lineitems[0].transfer_price = item_pricing.transferPrice        
362
            except:
363
                print "Not able to get transfer price. Skipping"
5508 rajveer 364
 
5944 mandeep.dh 365
            catalog_client = CatalogClient().get_client()
5508 rajveer 366
            voucherAmount = catalog_client.getVoucherAmount(item_id, VoucherType.SPICEDECK_MOBILE)
367
            if voucherAmount:
5514 rajveer 368
                __create_recharge_voucher_tracker(order, voucherAmount, VoucherType.SPICEDECK_MOBILE)
483 rajveer 369
    session.commit()
3499 mandeep.dh 370
 
3530 mandeep.dh 371
    try:
372
        if new_status == TransactionStatus.COD_IN_PROCESS:
4008 mandeep.dh 373
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
374
            transaction_requiring_extra_processing.category = 'COD_VERIFICATION'
375
            transaction_requiring_extra_processing.transaction_id = transaction_id
376
            session.commit()
4247 rajveer 377
        elif new_status == TransactionStatus.FLAGGED:
378
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
379
            transaction_requiring_extra_processing.category = 'PAYMENT_FLAGGED'
380
            transaction_requiring_extra_processing.transaction_id = transaction_id
381
            session.commit()
3530 mandeep.dh 382
    except Exception as e:
4065 mandeep.dh 383
        print "Error inserting transaction Id: " + str(transaction_id) + " due to " + str(e)
3499 mandeep.dh 384
 
483 rajveer 385
    return True
386
 
5514 rajveer 387
def __create_recharge_voucher_tracker(order, amount, voucherType):
5508 rajveer 388
    rvt = RechargeVoucherTracker()
389
    rvt.order = order
390
    rvt.amount = amount
391
    rvt.voucherIssued = False
5514 rajveer 392
    rvt.voucherType = voucherType
5508 rajveer 393
 
3994 chandransh 394
def __get_holidays(start_time=0L, to_time=-1L):
395
    '''
396
    Get the list of all public holidays since start time including the
397
    day of start time if it's a holiday
398
    '''
399
    midnightTime = start_time.replace(hour=0, minute=0, second=0, microsecond=0)
1730 ankur.sing 400
    fromDate = to_java_date(midnightTime)
401
    logistics_client = LogisticsClient().get_client()
3994 chandransh 402
    #TODO: This list should be cached after the first call.
403
    #Calling logistics server is also unnecessary.
404
    holidays = logistics_client.getHolidays(fromDate, to_time)
405
    holidays = [to_py_date(holiday).date() for holiday in holidays]
406
    return holidays
407
 
408
def adjust_delivery_time(start_time, delivery_days):
409
    '''
410
    Returns the actual no. of days which will pass while 'delivery_days'
411
    no. of business days will pass since 'order_time'. 
412
    '''
413
    start_date = start_time.date()
414
    end_date = start_date + datetime.timedelta(days = delivery_days)
415
    holidays = __get_holidays(start_time)
1730 ankur.sing 416
 
3994 chandransh 417
    while start_date <= end_date:
418
        if start_date.weekday() == 6 or start_date in holidays:
1730 ankur.sing 419
            delivery_days = delivery_days + 1
3994 chandransh 420
            end_date = end_date + datetime.timedelta(days = 1)
4088 chandransh 421
        start_date = start_date + datetime.timedelta(days = 1)
3994 chandransh 422
 
1730 ankur.sing 423
    return delivery_days
424
 
1398 varun.gupt 425
def enqueue_transaction_info_email(transaction_id):
1262 varun.gupt 426
    transaction = get_transaction(transaction_id)
5793 rajveer 427
    customer_name = "" 
1262 varun.gupt 428
    html_header = """
429
<html>
430
<body>
431
<div>
432
    <p>
5793 rajveer 433
        Hello $customer_name,<br /><br />
1742 rajveer 434
        Thanks for placing order with us. Following are the details of your order:
1262 varun.gupt 435
    </p>
436
    <div>
437
        Order Date: $order_date
1569 varun.gupt 438
    </div>
1262 varun.gupt 439
    <div>
440
        <table>
5879 rajveer 441
        <tr><td colspan="6"><hr /></td></tr>
442
        <tr><td colspan="6" align="left"><b>Order Details</b></td></tr>
443
        <tr><td colspan="6"><hr /></td></tr>
1262 varun.gupt 444
        <tr>
2050 varun.gupt 445
            <th width="100">Order No.</th>
446
            <th>Product</th>
5879 rajveer 447
            <th width="100">Estimated Delivery Date</th>
2050 varun.gupt 448
            <th width="100">Quantity</th>
449
            <th width="100">Unit Price</th>
450
            <th width="100">Amount</th>
1262 varun.gupt 451
        </tr>
452
    """
1398 varun.gupt 453
    user_email = ""
1262 varun.gupt 454
    total_amount = 0.0
455
    html_table = ""
1569 varun.gupt 456
    order_date = datetime.datetime.now()
5676 rajveer 457
 
458
    orders = transaction.orders
459
    html_pickup_text = ""
460
    random_text = ""
5697 rajveer 461
    if orders[0].pickupStoreId:
5793 rajveer 462
        random_text = str(randrange(100000, 999999)) 
5676 rajveer 463
 
464
    for order in orders:
5793 rajveer 465
        customer_name = order.customer_name
1262 varun.gupt 466
        order_date = order.created_timestamp
1398 varun.gupt 467
        user_email = order.customer_email
1569 varun.gupt 468
        html_tr = "<tr><td align='center'>" + str(order.id) + "</td>"
5676 rajveer 469
 
5879 rajveer 470
        sdt = datetime.datetime.strptime(str(orders[0].expected_delivery_time), "%Y-%m-%d %H:%M:%S")
471
        store_delivery_date = sdt.strftime("%d %B %Y")
472
 
5676 rajveer 473
        if order.pickupStoreId:
474
            attribute = Attribute()
475
            attribute.orderId = order.id
476
            attribute.name = "SECRET_CODE"
477
            attribute.value = random_text
478
            session.commit()
1262 varun.gupt 479
 
480
        for lineitem in order.lineitems:
2050 varun.gupt 481
            lineitem_total_price = round(lineitem.total_price, 2)
482
 
1276 chandransh 483
            html_tr += "<td>" + str(lineitem) + "</td>"
5879 rajveer 484
            html_tr += "<td align='center'>" + store_delivery_date + "</td>"
2050 varun.gupt 485
            html_tr += "<td align='center'>" + ("%.0f" % lineitem.quantity) + "</td>"
486
            html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem.unit_price) + "</td>"
487
            html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem_total_price) + "</td></tr>"
488
            total_amount += lineitem_total_price
1262 varun.gupt 489
            html_table += html_tr
6422 rajveer 490
 
491
    emi_amount = 0
492
    miscCharges = get_misc_charges(transaction_id)
493
    if miscCharges.get(1):
494
        emi_amount = miscCharges.get(1)
495
 
1262 varun.gupt 496
    html_footer = """
5879 rajveer 497
        <tr><td colspan=6>&nbsp;</td></tr>
498
        <tr><td colspan=6><hr /></td></tr>
6422 rajveer 499
        """
500
    if emi_amount > 0:
501
        html_footer = html_footer + "<tr>\
502
                        <td colspan=5>EMI Processing Charges</td>\
503
                        <td> Rs. $emi_amount</td>\
504
                    </tr>"
505
 
506
    html_footer = html_footer + "<tr>\
507
                <td colspan=5>Total Amount</td>\
508
                <td> Rs. $total_amount</td>\
509
                </tr>\
510
            </table>\
511
        </div>"
5793 rajveer 512
 
6422 rajveer 513
 
5793 rajveer 514
    store_address = ""
515
    if orders[0].pickupStoreId:
516
        store = LogisticsClient().get_client().getPickupStore(orders[0].pickupStoreId)
517
        store_address = __get_store_address_html(store)
518
 
519
        html_pickup_text = html_pickup_text + """
520
        <h3>Steps to Pickup In Store</h3>
5879 rajveer 521
        1) You will receive a confirmation email from us when your product(s) reaches the store. Once you receive the same, please visit the following store selected by you.<br>
5793 rajveer 522
        <br><i>$store_address</i><br>
523
        """
524
 
525
        if orders[0].cod:
526
            html_pickup_text = html_pickup_text + """
6318 rajveer 527
            2) Pay Rs. $net_amount, present a valid Photo Identity Proof* and use your Secret Code <b><i>$secret_code</i></b> to claim your Product(s).<br><br>
5793 rajveer 528
            *Valid Photo Identity Proof are Driving License, Passport, PAN Card, Voter ID, Company Issued ID Card, College issued Student ID Card Or any other Government approved Photo Id.<br>
529
            <br>
6029 rajveer 530
            In case you have any questions please visit <a href="$source_url/static/buy-online-and-pickup-in-store">Details on Pickup in Store</a>. If you still have more questions feel free to contact us 24/7 via <a href="$source_url/contact-us">$source_url/contact-us</a>
5793 rajveer 531
            </br><hr />
532
            """
533
        else:
534
            html_pickup_text = html_pickup_text + """
5879 rajveer 535
            2) Present a valid Photo Identity Proof* and use your Secret Code <b><i>$secret_code</i></b> to claim your Product(s).<br><br>
5793 rajveer 536
            *Valid Photo Identity Proof are Driving License, Passport, PAN Card, Voter ID, Company Issued ID Card, College issued Student ID Card Or any other Government approved Photo Id.<br>
537
            <br>
6029 rajveer 538
            In case you have any questions please visit <a href="$source_url/static/buy-online-and-pickup-in-store">Details on Pickup in Store</a>. If you still have more questions feel free to contact us 24/7 via <a href="$source_url/contact-us">$source_url/contact-us</a>
5793 rajveer 539
            </br><hr />
540
            """
541
 
5676 rajveer 542
    html_footer = html_footer + html_pickup_text + """
1262 varun.gupt 543
    <p>
544
        Best Wishes,<br />
6029 rajveer 545
        $source_name Team
1262 varun.gupt 546
    </p>
547
</div>
548
</body>
549
</html>
550
    """
1398 varun.gupt 551
 
1569 varun.gupt 552
    dt = datetime.datetime.strptime(str(order_date), "%Y-%m-%d %H:%M:%S")
1565 varun.gupt 553
    formated_order_date = dt.strftime("%A, %d. %B %Y %I:%M%p")
5793 rajveer 554
 
555
    sdt = datetime.datetime.strptime(str(orders[0].expected_delivery_time), "%Y-%m-%d %H:%M:%S")
556
    store_delivery_date = sdt.strftime("%A, %d %B %Y")
557
 
6029 rajveer 558
    email_header = Template(html_header).substitute(dict(order_date = formated_order_date, customer_name = customer_name, source_url = source_url))
6422 rajveer 559
    email_footer = Template(html_footer).substitute(dict(total_amount = "%.2f" % total_amount, emi_amount = "%.2f" % emi_amount, net_amount = "%.2f" % (total_amount-order.gvAmount), secret_code = random_text, store_address = store_address, store_delivery_date = store_delivery_date, source_url = source_url, source_name = source_name))
1565 varun.gupt 560
 
1351 varun.gupt 561
    try:
562
        helper_client = HelperClient().get_client()
6280 rajveer 563
        helper_client.saveUserEmailForSending([user_email], "", source_name + " - Order Details", email_header + html_table + email_footer, str(transaction_id), "TransactionInfo", [], [])
5866 rajveer 564
 
1351 varun.gupt 565
        return True
566
    except Exception as e:
567
        print e
568
        return False
1262 varun.gupt 569
 
5802 rajveer 570
def enqueue_received_at_store_email(order):
571
    html_header = """
572
    <html>
573
    <body>
574
    <div>
575
        <p>
576
            Hello $customer_name,<br /><br />
577
            Your Product has reached the Store selected by you and is available for pickup. Please pick up your product on or before $order_expiry_date. Following are the details of your order and steps to pickup the same:
578
        </p>
579
        <div>
580
            Order Date: $order_date
581
        </div>
582
        <div>
583
            <table>
584
            <tr><td colspan="5"><hr /></td></tr>
585
            <tr><td colspan="5" align="left"><b>Order Details</b></td></tr>
586
            <tr><td colspan="5"><hr /></td></tr>
587
            <tr>
588
                <th width="100">Order No.</th>
589
                <th>Product</th>
590
                <th width="100">Quantity</th>
591
                <th width="100">Unit Price</th>
592
                <th width="100">Amount</th>
593
            </tr>
594
    """
595
    total_amount = 0.0
596
    html_table = ""
597
 
598
    html_pickup_text = ""
599
    random_text = __get_secret_code(order.id)
600
 
601
    customer_name = order.customer_name
602
    order_date = order.created_timestamp
603
    user_email = order.customer_email
604
    html_tr = "<tr><td align='center'>" + str(order.id) + "</td>"
605
 
606
    for lineitem in order.lineitems:
607
        lineitem_total_price = round(lineitem.total_price, 2)
608
 
609
        html_tr += "<td>" + str(lineitem) + "</td>"
610
        html_tr += "<td align='center'>" + ("%.0f" % lineitem.quantity) + "</td>"
611
        html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem.unit_price) + "</td>"
612
        html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem_total_price) + "</td></tr>"
613
        total_amount += lineitem_total_price
614
        html_table += html_tr
615
 
616
    html_footer = """
617
        <tr><td colspan=5>&nbsp;</td></tr>
618
        <tr><td colspan=5><hr /></td></tr>
619
        <tr>
620
            <td colspan=4>Total Amount</td>
621
            <td> Rs. $total_amount</td>
622
        </tr>
623
        </table>
624
    </div>
625
    """
626
 
627
    store_address = ""
628
    if order.pickupStoreId:
629
        store = LogisticsClient().get_client().getPickupStore(order.pickupStoreId)
630
        store_address = __get_store_address_html(store)
631
 
632
        html_pickup_text = html_pickup_text + """
633
        <h3>Steps to Pickup In Store</h3>
634
        1) Please visit the following store selected by you on or before $order_expiry_date<br>
635
        <br><i>$store_address</i><br>
636
        """
637
 
638
        if order.cod:
639
            html_pickup_text = html_pickup_text + """
6318 rajveer 640
            2) Pay Rs. $net_amount, present a valid Photo Identity Proof* and quote your Secret Code <b><i>$secret_code</i></b> to claim your Product.<br><br>
5802 rajveer 641
            *Valid Photo Identity Proof are Driving License, Passport, PAN Card, Voter ID, Company Issued ID Card, College issued Student ID Card Or any other Government approved Photo Id.<br>
642
            <br>
6029 rajveer 643
            In case you have any questions please visit <a href="$source_url/static/buy-online-and-pickup-in-store">Details on Pickup in Store</a>. If you still have more questions feel free to contact us 24/7 via <a href="$source_url/contact-us">$source_url/contact-us</a>
5802 rajveer 644
            </br><hr />
645
            """
646
        else:
647
            html_pickup_text = html_pickup_text + """
648
            2) Present a valid Photo Identity Proof* and use your Secret Code <b><i>$secret_code</i></b> to claim your Product(s).<br><br>
649
            *Valid Photo Identity Proof are Driving License, Passport, PAN Card, Voter ID, Company Issued ID Card, College issued Student ID Card Or any other Government approved Photo Id.<br>
650
            <br>
6029 rajveer 651
            In case you have any questions please visit <a href="$source_url/static/buy-online-and-pickup-in-store">Details on Pickup in Store</a>. If you still have more questions feel free to contact us 24/7 via <a href="$source_url/contact-us">$source_url/contact-us</a>
5802 rajveer 652
            </br><hr />
653
            """
654
 
655
        html_footer = html_footer + html_pickup_text + """
656
        <p>
657
            Best Wishes,<br />
6029 rajveer 658
            $source_name Team
5802 rajveer 659
        </p>
660
        </div>
661
        </body>
662
        </html>
663
        """
664
 
665
    subject = "Your order for " + str(order.lineitems[0]) + " is ready for Pickup in Store"
666
    dt = datetime.datetime.strptime(str(order_date), "%Y-%m-%d %H:%M:%S")
667
    formated_order_date = dt.strftime("%A, %d. %B %Y %I:%M%p")
668
 
669
    oetd = datetime.datetime.now() + datetime.timedelta(days=4)
670
    formated_order_expiry_date = oetd.strftime("%A, %d %B %Y")
671
 
672
 
6029 rajveer 673
    email_header = Template(html_header).substitute(dict(order_date = formated_order_date, customer_name = customer_name, order_expiry_date = formated_order_expiry_date, source_url = source_url, source_name = source_name))
6318 rajveer 674
    email_footer = Template(html_footer).substitute(dict(total_amount = "%.2f" % total_amount, net_amount = "%.2f" % (total_amount-order.netAmount), secret_code = random_text, store_address = store_address, order_expiry_date = formated_order_expiry_date, source_url = source_url, source_name = source_name))
5802 rajveer 675
 
676
    try:
5812 rajveer 677
        helper_client = HelperClient(host_key = "helper_service_server_host_prod").get_client()
6280 rajveer 678
        helper_client.saveUserEmailForSending([user_email], "", subject, email_header + html_table + email_footer, str(order.id), "ReceivedAtStore", [], [])
5802 rajveer 679
        return True
680
    except Exception as e:
681
        print e
682
        return False
683
 
5966 rajveer 684
def enqueue_delivery_success_mail(order):
5807 rajveer 685
    html_header = """
686
    <html>
687
    <body>
688
    <div>
689
        <p>
690
            Dear $customer_name,<br /><br />
691
    """
692
 
693
    if order.pickupStoreId:
694
        pickup_text = "Pickup Date: " + order.delivery_timestamp.strftime("%A, %d. %B %Y %I:%M%p")
695
        thank_you_html = """
696
        Thank you for shopping with us and picking up your order in Store. Please do join us at our <a href="http://www.facebook.com/mysaholic">Facebook Page</a>, and share your experience with other shoppers.<br><br>
697
 
6029 rajveer 698
        Following are the details of your order for which the pickup has happened. If you have not picked up this product please <a href="$source_url/contact-us" target="_blank">contact us</a> immediately.<br>
5807 rajveer 699
        """
700
    else:
701
        pickup_text = ""
702
        thank_you_html = """
703
        Thank you for shopping with us. Please do join us at our <a href="http://www.facebook.com/mysaholic">Facebook Page</a>, and share your experience with other shoppers. Following are the details of your order.<br>
704
        """ 
705
 
706
    html_header = html_header + thank_you_html + """        
707
        </p>
708
        <div>
709
            Order Date: $order_date <br>
710
            $pickup_text
711
        </div>
712
        <div>
713
            <table>
714
            <tr><td colspan="5"><hr /></td></tr>
715
            <tr><td colspan="5" align="left"><b>Order Details</b></td></tr>
716
            <tr><td colspan="5"><hr /></td></tr>
717
            <tr>
718
                <th width="100">Order No.</th>
719
                <th>Product</th>
720
                <th width="100">Quantity</th>
721
                <th width="100">Unit Price</th>
722
                <th width="100">Amount</th>
723
            </tr>
724
    """
725
    total_amount = 0.0
726
    html_table = ""
5802 rajveer 727
 
5807 rajveer 728
    customer_name = order.customer_name
729
    order_date = order.created_timestamp
730
    user_email = order.customer_email
731
    html_tr = "<tr><td align='center'>" + str(order.id) + "</td>"
732
 
733
    for lineitem in order.lineitems:
734
        lineitem_total_price = round(lineitem.total_price, 2)
735
 
736
        html_tr += "<td>" + str(lineitem) + "</td>"
737
        html_tr += "<td align='center'>" + ("%.0f" % lineitem.quantity) + "</td>"
738
        html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem.unit_price) + "</td>"
739
        html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem_total_price) + "</td></tr>"
740
        total_amount += lineitem_total_price
741
        html_table += html_tr
742
 
743
    html_footer = """
744
        <tr><td colspan=5>&nbsp;</td></tr>
745
        <tr><td colspan=5><hr /></td></tr>
746
        <tr>
747
            <td colspan=4>Total Amount</td>
748
            <td> Rs. $total_amount</td>
749
        </tr>
750
        </table>
751
    </div>
752
    <p>
753
        Best Wishes,<br />
6029 rajveer 754
        $source_name Team
5807 rajveer 755
    </p>
756
    </div>
757
    </body>
758
    </html>
759
    """
760
 
761
    subject = "Your order for " + str(order.lineitems[0]) + " is now complete"
762
    dt = datetime.datetime.strptime(str(order_date), "%Y-%m-%d %H:%M:%S")
763
    formated_order_date = dt.strftime("%A, %d. %B %Y %I:%M%p")
764
 
765
 
6029 rajveer 766
    email_header = Template(html_header).substitute(dict(order_date = formated_order_date, customer_name = customer_name, pickup_text = pickup_text, source_url = source_url))
767
    email_footer = Template(html_footer).substitute(dict(total_amount = "%.2f" % total_amount, source_name = source_name))
5807 rajveer 768
 
769
    try:
5812 rajveer 770
        helper_client = HelperClient(host_key = "helper_service_server_host_prod").get_client()
6280 rajveer 771
        helper_client.saveUserEmailForSending([user_email], "", subject, email_header + html_table + email_footer, str(order.id), "DeliverySuccess", [], [])
5807 rajveer 772
        return True
773
    except Exception as e:
774
        print e
775
        return False
776
 
5802 rajveer 777
def __get_secret_code(order_id):
5855 rajveer 778
    return get_order_attribute_value(order_id, 'SECRET_CODE')
5806 rajveer 779
 
5833 rajveer 780
def get_order_attribute_value(order_id, attribute_name):
5808 rajveer 781
    existingAttribute = Attribute.query.filter(Attribute.orderId == order_id).filter(Attribute.name == attribute_name).first()
782
    if existingAttribute:
783
        return existingAttribute.value
5802 rajveer 784
    return ""
5806 rajveer 785
 
6019 rajveer 786
def change_jacket_number(order_id, jacket_number):
787
    order = get_order(order_id)
788
    if order.status == OrderStatus.BILLED:
789
        order.jacket_number = jacket_number
6047 rajveer 790
        session.commit()
6019 rajveer 791
        return True
792
    return False
793
 
794
def mark_order_as_rto_in_transit(order_id):
795
    order = get_order(order_id)
6032 rajveer 796
    if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.DELIVERED_AT_STORE, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE, OrderStatus.DELIVERY_SUCCESS]:
6019 rajveer 797
        return False
798
    order.status = OrderStatus.RTO_IN_TRANSIT
799
    order.delivery_timestamp = datetime.datetime.now()
800
    order.statusDescription = "Order Returned to Origin:"
801
    update_trust_level(order)
6032 rajveer 802
    session.commit()
6019 rajveer 803
    return True
804
 
483 rajveer 805
def get_order(order_id):
806
    order = Order.get_by(id=order_id)
807
    if not order:
808
        raise TransactionServiceException(108, "no such order")
809
 
810
    return order
811
 
4999 phani.kuma 812
def get_order_list(order_ids):
813
    orders = Order.query.filter(Order.id.in_(tuple(order_ids)))
814
    return orders
815
 
5386 phani.kuma 816
def get_order_list_for_vendor(order_ids, vendor_id):
817
    if vendor_id == -1:
818
        return get_order_list(order_ids)
819
    else:
820
        orders = Order.query.filter_by(vendorId = vendor_id).filter(Order.id.in_(tuple(order_ids)))
821
        return orders
822
 
3014 chandransh 823
def get_orders_for_customer(customer_id, from_date, to_date, statuses):
483 rajveer 824
    if not customer_id:
825
        raise TransactionServiceException(101, "bad customer id")
826
    query = Order.query.filter(Order.customer_id == customer_id)
3014 chandransh 827
    if statuses:
828
        query = query.filter(Order.status.in_(tuple(statuses)))
483 rajveer 829
    if from_date:
830
        query = query.filter(Order.created_timestamp >from_date)
831
    if to_date:
832
        query = query.filter(Order.created_timestamp < to_date)
833
    return query.all()
834
 
1528 ankur.sing 835
def get_order_for_customer(order_id, customer_id):
836
    if not customer_id:
837
        raise TransactionServiceException(101, "bad customer id")
838
    order = get_order(order_id)
839
    if order.customer_id != customer_id:
840
        raise TransactionServiceException(108, "order: " + str(order_id) + " does not belong to customer: " + str(customer_id))
841
    return order
842
 
4801 anupam.sin 843
def get_all_orders(statuses, from_date, to_date, warehouse_id):
844
    if not len(statuses):
845
        raise TransactionServiceException(101, "Status list is empty")
846
    query = Order.query.filter(Order.status.in_(statuses))
847
    if warehouse_id:
848
        query = query.filter(Order.warehouse_id == warehouse_id)
849
        #raise TransactionServiceException(101, "bad warehouse id")
483 rajveer 850
    if from_date:
851
        query = query.filter(Order.created_timestamp >from_date)
852
    if to_date:
853
        query = query.filter(Order.created_timestamp < to_date)
854
    return query.all()
855
 
4133 chandransh 856
def get_orders_in_batch(statuses=[], offset=0, limit=0, warehouse_id=None):
4296 rajveer 857
    query = Order.query
858
    if warehouse_id:
859
        query = query.filter(Order.warehouse_id == warehouse_id)
4133 chandransh 860
    if statuses:
861
        query = query.filter(Order.status.in_(statuses))
862
    query = query.offset(offset)
863
    if limit:
864
        query = query.limit(limit)
865
    return query.all()
866
 
867
def get_order_count(statuses=[], warehouse_id=None):
868
    if not warehouse_id:
869
        raise TransactionServiceException(101, "bad warehouse id")
870
    query = session.query(func.count(Order.id)).filter(Order.warehouse_id == warehouse_id)
871
    if statuses:
872
        query = query.filter(Order.status.in_(statuses))
873
    return query.scalar()
874
 
1382 varun.gupt 875
def get_returnable_orders_for_customer(customer_id, limit = None):
876
    if not customer_id:
877
        raise TransactionServiceException(101, "bad customer id")
878
 
879
    query = Order.query.filter(Order.customer_id == customer_id)
880
    query = query.filter(Order.status != OrderStatus.PAYMENT_FAILED)
881
    query = query.filter(Order.status != OrderStatus.REJECTED)
882
    query = query.filter(Order.status > OrderStatus.SHIPPED_FROM_WH)
883
    query = query.filter(or_(Order.status < OrderStatus.DELIVERY_SUCCESS, Order.delivery_timestamp < datetime.datetime.now() + datetime.timedelta(hours = 48)))
884
    orders = query.all()
885
    order_ids = [order.id for order in orders]
886
    return order_ids
887
 
888
def get_cancellable_orders_for_customer(customer_id, limit = None):
889
    if not customer_id:
890
        raise TransactionServiceException(101, "bad customer id")
4815 phani.kuma 891
    cancellableOrderStatuses = OrderStatusGroups().codCancellable
1382 varun.gupt 892
    query = Order.query.filter(Order.customer_id == customer_id)
4815 phani.kuma 893
    query = query.filter(Order.status.in_(cancellableOrderStatuses))
1382 varun.gupt 894
    orders = query.all()
895
    return [order.id for order in orders]
896
 
995 varun.gupt 897
def get_orders_by_billing_date(status, start_billing_date, end_billing_date, warehouse_id):
898
    if not warehouse_id:
899
        raise TransactionServiceException(101, "bad warehouse id")
900
 
901
    query = Order.query.filter(Order.warehouse_id == warehouse_id)
902
 
903
    if status:
904
        query = query.filter(Order.status == status)
905
 
906
    if start_billing_date:
907
        query = query.filter(Order.billing_timestamp >= start_billing_date)
908
 
909
    if end_billing_date:
910
        query = query.filter(Order.billing_timestamp <= end_billing_date)
911
 
912
    return query.all()
913
 
3451 chandransh 914
def get_orders_by_shipping_date(from_shipping_date, to_shipping_date, provider_id, warehouse_id, cod):
915
    query = Order.query.filter(Order.cod == cod)
3427 chandransh 916
 
917
    if warehouse_id and warehouse_id != -1:
918
        query = query.filter(Order.warehouse_id == warehouse_id)
919
 
920
    if provider_id and provider_id != -1:
921
        query = query.filter(Order.logistics_provider_id == provider_id)
922
 
923
    if from_shipping_date:
924
        query = query.filter(Order.shipping_timestamp >= from_shipping_date)
925
 
926
    if to_shipping_date:
927
        query = query.filter(Order.shipping_timestamp <= to_shipping_date)
928
 
929
    query = query.order_by(Order.airwaybill_no)
930
 
931
    return query.all()
932
 
1528 ankur.sing 933
def get_orders_for_transaction(transaction_id, customer_id):
934
    orders = Order.query.filter_by(transaction_id=transaction_id, customer_id=customer_id).all()
935
 
1208 chandransh 936
    if not orders:
937
        raise TransactionServiceException(101, "No order for the transaction")
938
    return orders
939
 
1405 ankur.sing 940
def get_undelivered_orders(provider_id, warehouse_id):
941
    query = Order.query
942
    if provider_id != -1:
943
        query = query.filter_by(Order.logistics_provider_id == provider_id)
944
    if warehouse_id != -1:
945
        query = query.filter(Order.warehouse_id == warehouse_id)
946
    query = query.filter(and_(Order.status != OrderStatus.DELIVERY_SUCCESS, Order.status != OrderStatus.FAILED, \
947
                              Order.status > OrderStatus.SHIPPED_FROM_WH))
1479 ankur.sing 948
    #Get last midnight time instead of current time as courier company would have sent the 
949
    #delivery records updated at least till midnight and not by the current time.
950
    time_format = "%Y-%m-%d %H:%M:%S.%f"
951
    t = time.strptime(str(datetime.datetime.now()), time_format)
952
    midnightTime = datetime.datetime(*t[:3])
953
    query = query.filter(Order.expected_delivery_time < midnightTime)
1405 ankur.sing 954
    orders = query.all()
955
    return orders
956
 
1208 chandransh 957
def get_line_items_for_order(order_id):
958
    query = LineItem.query.filter(LineItem.order_id == order_id)
959
    return query.all()
960
 
483 rajveer 961
def change_order_status(self, orderId, status, description):
962
    order = get_order(orderId)
963
    order.status = status
964
    order.statusDescription = description
965
    session.commit()
3578 mandeep.dh 966
    update_trust_level(order)
967
    return True
3499 mandeep.dh 968
 
3578 mandeep.dh 969
def update_trust_level(order):
3530 mandeep.dh 970
    try:
3578 mandeep.dh 971
        trust_level_delta = 0
972
        if order.cod == True:
973
            trust_level_delta = ORDER_STATUS_TO_USER_TRUST_LEVEL_DELTA_DICT.get(order.status, 0)
974
 
3530 mandeep.dh 975
        if (trust_level_delta != 0):
976
            user_client = UserClient().get_client()
977
            user_client.increaseTrustLevel(order.customer_id, trust_level_delta)
978
    except Exception as e:
979
        print e
3499 mandeep.dh 980
 
483 rajveer 981
    return True
3064 chandransh 982
 
983
def verify_order(self, orderId):
984
    logging.info("Verifying order no: " + str(orderId))
985
    order = get_order(orderId)
4662 rajveer 986
    if order.status == OrderStatus.COD_VERIFICATION_PENDING:
3064 chandransh 987
        order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
988
        order.statusDescription = "Submitted for processing"
989
        order.verification_timestamp = datetime.datetime.now()
990
        session.commit()
991
        logging.info("Successfully verified order no: " + str(orderId))
992
        return True
993
    else:
994
        logging.warning("Verify called for order no." + str(orderId) +" which is not in verification pending state");
995
        return False
996
 
4285 rajveer 997
def accept_order(orderId):
3064 chandransh 998
    logging.info("Accepting order no: " + str(orderId))
921 rajveer 999
    order = get_order(orderId)
5110 mandeep.dh 1000
    if order.status in [OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT, OrderStatus.CAPTURE_IN_PROCESS]:
3064 chandransh 1001
        if not order.cod:
1002
            if order.transaction.status == TransactionStatus.AUTHORIZED:
4421 mandeep.dh 1003
                __capture_txn(order)
3010 chandransh 1004
        order.status = OrderStatus.ACCEPTED
1005
        order.statusDescription = "Order Accepted"
1006
        order.accepted_timestamp = datetime.datetime.now()
5530 mandeep.dh 1007
        __amend_fulfilment_warehouse(order)
3010 chandransh 1008
        session.commit()
3064 chandransh 1009
        logging.info("Successfully accepted the order no.:" + str(orderId))
3010 chandransh 1010
        return True
1011
    else:
3064 chandransh 1012
        logging.warning("Accept called for the unacceptable order: " + str(orderId))
3010 chandransh 1013
        return False
921 rajveer 1014
 
4421 mandeep.dh 1015
def __capture_txn(order):
1016
    txnId = order.transaction_id
3010 chandransh 1017
    logging.info("Capturing payment for merchant txn:" + str(txnId))
4421 mandeep.dh 1018
    try:
1019
        payment_client = PaymentClient().get_client()
1020
        capture_result = payment_client.capturePayment(txnId)
1021
        if capture_result:
1022
            logging.info("Successfully captured payment for merchant txn:" + str(txnId))
5527 anupam.sin 1023
            change_transaction_status(txnId, TransactionStatus.IN_PROCESS, "Payment received", PickUpType.COURIER, order.orderType)
4421 mandeep.dh 1024
        else:
1025
            raise TransactionServiceException(115, "Payment capture failed.")
1026
    except PaymentException as e:
1027
        if e.error_code == 106:
1028
            order.status = OrderStatus.CAPTURE_IN_PROCESS
1029
            session.commit()
1030
            raise TransactionServiceException(122, "Unable to capture payment due to connection issue.")
1031
        else:
1032
            raise TransactionServiceException(115, "Payment capture failed.")
3010 chandransh 1033
 
5530 mandeep.dh 1034
def __amend_fulfilment_warehouse(order):
1035
    itemId = order.lineitems[0].item_id
5944 mandeep.dh 1036
    inventoryClient = InventoryClient().get_client()
1037
    inventory = inventoryClient.getItemInventoryByItemId(itemId)
5530 mandeep.dh 1038
    if not (inventory.availability.has_key(order.fulfilmentWarehouseId) and inventory.availability[order.fulfilmentWarehouseId]):
5944 mandeep.dh 1039
        for warehouse in inventoryClient.getWarehouses(WarehouseType.OURS, InventoryType.GOOD, 0, order.warehouse_id, 0):
5558 mandeep.dh 1040
            if inventory.availability.has_key(warehouse.id) and inventory.availability[warehouse.id] >= inventory.reserved[warehouse.id] + order.lineitems[0].quantity:
5966 rajveer 1041
                inventoryClient.reduceReservationCount(itemId, order.fulfilmentWarehouseId, sourceId, order.id, order.lineitems[0].quantity)
1042
                inventoryClient.reserveItemInWarehouse(itemId, warehouse.id, sourceId, order.id, to_java_date(order.created_timestamp), to_java_date(order.promised_shipping_time), order.lineitems[0].quantity)
5637 mandeep.dh 1043
                order.fulfilmentWarehouseId = warehouse.id
5530 mandeep.dh 1044
                break
1045
 
5411 rajveer 1046
def add_billing_details(orderId, invoice_number, serialNumbers, itemNumbers, billedBy, jacketNumber, billingType, fulfilmentWarehouseId, authorize):
4285 rajveer 1047
    if jacketNumber is None or jacketNumber <= 0:
2783 chandransh 1048
        raise TransactionServiceException(110, "Invalid jacket number")
1049
 
4285 rajveer 1050
    if billedBy is None or billedBy.strip() == "":
2783 chandransh 1051
        raise TransactionServiceException(110, "Invalid Biller")
1052
 
1149 chandransh 1053
    order = Order.get_by(id=orderId)
1054
    if not order:
2783 chandransh 1055
        raise TransactionServiceException(101, "No order found for the given order id" + str(orderId))
5110 mandeep.dh 1056
 
5966 rajveer 1057
 
5110 mandeep.dh 1058
    lineitem = order.lineitems[0]
1059
    item_id = lineitem.item_id
5966 rajveer 1060
 
6007 rajveer 1061
#    inventoryClient = InventoryClient().get_client()
1062
#    if not inventoryClient.isOrderBillable(item_id, order.fulfilmentWarehouseId, sourceId, orderId):
1063
#        raise TransactionServiceException(101, "There are other orders to be billed before this order id " + str(orderId))
5966 rajveer 1064
 
5110 mandeep.dh 1065
    catalog_client = CatalogClient().get_client()
1066
    item = catalog_client.getItem(item_id)
4283 anupam.sin 1067
    if order.status == OrderStatus.ACCEPTED:
4285 rajveer 1068
        order.jacket_number = jacketNumber
1069
        if billingType == BillingType.OURS:
5411 rajveer 1070
            if itemNumbers:
1071
                lineitem.item_number = itemNumbers[0]    
4283 anupam.sin 1072
 
5411 rajveer 1073
            if ItemType.SERIALIZED == item.type and not (serialNumbers and serialNumbers[0]):
1074
                raise TransactionServiceException(110, "No Serial Number supplied")
1075
            if serialNumbers:
1076
                lineitem.serial_number = serialNumbers[0]
1077
 
5527 anupam.sin 1078
            order.invoice_number = get_next_invoice_number(order.orderType)
4283 anupam.sin 1079
        else:
1080
            order.invoice_number = invoice_number
4763 rajveer 1081
            # In case of hotspot billing, we should try to verify that color billed in ETP is same as color in order.
1082
            if billedOrdersColorMap.has_key(orderId):
1083
                etpColor = billedOrdersColorMap.get(orderId)
1084
                ourColor = order.lineitems[0].color
4781 rajveer 1085
                if ourColor and etpColor != ourColor and not authorize:
4763 rajveer 1086
                    raise TransactionServiceException(777, "ETP color: " + etpColor + " and Order color: " + ourColor)
1087
                else:
1088
                    billedOrdersColorMap.pop(orderId)
4283 anupam.sin 1089
 
2842 chandransh 1090
        order.status = OrderStatus.BILLED
1091
        order.statusDescription = "Order Billed"
1092
        order.billing_timestamp = datetime.datetime.now()
4285 rajveer 1093
        order.billed_by = billedBy
4435 mandeep.dh 1094
 
5436 mandeep.dh 1095
        # Letting the billing process fail in cases where we are unable to 
4435 mandeep.dh 1096
        # fill in transfer price
1097
        try:
5944 mandeep.dh 1098
            inventory_client = InventoryClient().get_client()
1099
            warehouse = inventory_client.getWarehouse(fulfilmentWarehouseId)
1100
            item_pricing = inventory_client.getItemPricing(item_id, warehouse.vendor.id)
5110 mandeep.dh 1101
            lineitem.transfer_price = item_pricing.transferPrice
6075 amit.gupta 1102
            lineitem.vatRate = catalog_client.getVatPercentageForItem(lineitem.item_id, lineitem.unit_price)
5110 mandeep.dh 1103
            order.vendorId = warehouse.vendor.id
4435 mandeep.dh 1104
        except InventoryServiceException as e:
1105
            print sys.exc_info()[0]
1106
            print e.message
5110 mandeep.dh 1107
            raise TransactionServiceException(110, 'Transfer price missing for itemId: ' + str(item_id) + ' and vendor: ' + str(warehouse.vendor.id))
4435 mandeep.dh 1108
 
5110 mandeep.dh 1109
        # For OUR warehouse, we need to scan out items for every billed order
1110
        if billingType == BillingType.OURS:
5530 mandeep.dh 1111
            for index, serialNumber in enumerate(serialNumbers):
5411 rajveer 1112
                try:
1113
                    warehouse_client = WarehouseClient().get_client()
5436 mandeep.dh 1114
                    # Fetching GOOD w/h corresponding to the virtual one here
5835 mandeep.dh 1115
                    if not warehouse.billingWarehouseId:
5950 mandeep.dh 1116
                        warehouse = inventory_client.getWarehouses(None, InventoryType.GOOD, warehouse.vendor.id, order.warehouse_id, 0)[0]
5436 mandeep.dh 1117
 
5411 rajveer 1118
                    if ItemType.SERIALIZED == item.type:
1119
                        inventoryItem = warehouse_client.getInventoryItem(serialNumber)
1120
                    else:
5661 mandeep.dh 1121
                        inventoryItem = warehouse_client.getNonSeralizedInventoryItem(itemNumbers[index], item_id, warehouse.id)
5411 rajveer 1122
 
1123
                    if (inventoryItem.itemId != item_id):
1124
                        catalog_client = CatalogClient().get_client()
1125
                        item = catalog_client.getItem(inventoryItem.itemId)
1126
                        scanItemString = " ".join([str(item.brand), str(item.modelName), str(item.modelNumber), str(item.color)])
1127
                        lineItemString = " ".join([str(lineitem.brand), str(lineitem.model_name), str(lineitem.model_number), str(lineitem.color)])
1128
                        raise TransactionServiceException(110, 'Trying to scan ' + scanItemString + ' instead of ' + lineItemString)
1129
 
1130
                    if ItemType.SERIALIZED == item.type:
5661 mandeep.dh 1131
                        inventoryItem = warehouse_client.scanSerializedItemForOrder(serialNumber, ScanType.SALE, orderId, warehouse.id, 1, order.warehouse_id)
5411 rajveer 1132
                    else:
5661 mandeep.dh 1133
                        inventoryItem = warehouse_client.scanForOrder(inventoryItem, ScanType.SALE, 1, order.id, warehouse.id, order.warehouse_id)
5411 rajveer 1134
 
1135
                    lineitem.transfer_price = inventoryItem.unitPrice
1136
                    order.vendorId = inventoryItem.supplierId
5636 amar.kumar 1137
                    catalog_client = CatalogClient().get_client()
5411 rajveer 1138
                except WarehouseServiceException as e:
1139
                    print sys.exc_info()[0]
1140
                    print 'Could not scan out orders due to: ' + e.message
1141
                    raise TransactionServiceException(110, e.message)
5123 mandeep.dh 1142
        else:
1143
            if not warehouse.isAvailabilityMonitored:
5944 mandeep.dh 1144
                inventory_client = InventoryClient().get_client()
1145
                inventory_client.addInventory(item.id, warehouse.id, -1 * lineitem.quantity)
4365 mandeep.dh 1146
 
5110 mandeep.dh 1147
        __update_inventory_reservation(order)
5661 mandeep.dh 1148
        order.fulfilmentWarehouseId = warehouse.id
4541 mandeep.dh 1149
        session.commit()
4283 anupam.sin 1150
        return True
1151
    else:
1152
        return False
1149 chandransh 1153
 
4763 rajveer 1154
def add_invoice_number(orderId, invoiceNumber, color):
4579 rajveer 1155
    order = Order.get_by(id=orderId)
1156
    if not order:
1157
        raise TransactionServiceException(101, "No order found for the given order id" + str(orderId))
1158
 
1159
    if order.status == OrderStatus.ACCEPTED:
1160
        order.invoice_number = invoiceNumber
4767 rajveer 1161
        if color:
1162
            billedOrdersColorMap[orderId] = color;
4579 rajveer 1163
    session.commit()
1164
 
1220 chandransh 1165
def batch_orders(warehouseId):
1166
    if not warehouseId:
1167
        raise TransactionServiceException(101, "bad warehouse id")
1225 chandransh 1168
 
1169
    batchno_generator = BatchNoGenerator()
1170
    session.commit()
1220 chandransh 1171
    query = Order.query.filter_by(warehouse_id=warehouseId)
4592 mandeep.dh 1172
    query = query.filter(Order.status == OrderStatus.LOW_INV_PO_RAISED)
1220 chandransh 1173
    query = query.order_by(Order.created_timestamp)
1174
    pending_orders = query.all()
1225 chandransh 1175
 
1220 chandransh 1176
    serial_no = 1
1177
    for order in pending_orders:
1276 chandransh 1178
        order.batchNo = batchno_generator.id
1220 chandransh 1179
        order.serialNo = serial_no
1180
        serial_no += 1
1181
    session.commit() 
1182
    pending_orders = query.all()
1183
    return pending_orders
1184
 
1185
def order_outofstock(orderId):
1911 chandransh 1186
    '''
1187
    Mark order as out of stock
1188
    '''
1220 chandransh 1189
    order = get_order(orderId)
1190
    order.status = OrderStatus.INVENTORY_LOW
1191
    order.statusDescription = "Low Inventory"
1192
    order.outofstock_timestamp = datetime.datetime.now()
1193
    session.commit()
1194
    return True
1195
 
4789 rajveer 1196
def mark_orders_as_shipped_from_warehouse(warehouseId, providerId, cod, orderIds):
4410 rajveer 1197
    try:
1198
        current_timestamp = datetime.datetime.now()
5822 rajveer 1199
        orders = Order.query.filter(Order.id.in_(orderIds)).filter_by(warehouse_id = warehouseId, logistics_provider_id = providerId, status = OrderStatus.BILLED).all()
4410 rajveer 1200
        for order in orders:
5822 rajveer 1201
            order.status = OrderStatus.SHIPPED_FROM_WH
1202
            order.statusDescription = "Order shipped from warehouse"
1203
            order.shipping_timestamp = current_timestamp
1204
            try:
1205
                print "Trying to schedule alert"
1206
                monitoredEntity = MonitoredEntity();
1207
                monitoredEntity.entityType=EntityType.COURIER;
1208
                monitoredEntity.eventType=OrderStatus.SHIPPED_FROM_WH;
1209
                monitoredEntity.entityIdentifier="orderId = " + str(order.id);
1210
                adjustedDeliveryDays = adjust_delivery_time(datetime.datetime.now(), 1);
1211
                sec = datetime.datetime.now()
1212
                #Warn alert time is taken as 16 hrs
1213
                warn_time = (sec + timedelta(days=(adjustedDeliveryDays-1)) + timedelta(hours=16))
1214
                #Critical alert time is taken as 26 hrs                    
1215
                critical_time = (sec + timedelta(days=(adjustedDeliveryDays-1)) + timedelta(hours=26))
1216
                monitoredEntity.warnExpiryTime = int(warn_time.strftime("%s"))*1000
1217
                monitoredEntity.criticalExpiryTime = int(critical_time.strftime("%s"))*1000
1218
                monitoredEntity.description="warehouseId = " + str(warehouseId) + " providerId = " + str(providerId) + " orderId = "  + str(order.id);
1219
                alert_client = AlertClient().get_client()
1220
                alert_client.scheduleAlert(monitoredEntity)
1221
            except Exception as e:
1222
                print "Exception in scheduling alert in ShippedFromWarehouse method"
1223
                print e
759 chandransh 1224
        session.commit()
5832 rajveer 1225
 
4749 varun.gupt 1226
 
1227
        try:
5832 rajveer 1228
            for order in orders:
1229
                if order.pickupStoreId:
1230
                    send_mails_to_bdms(order)
1231
                else:
1232
                    enqueue_shipping_confirmation_email(order)
4749 varun.gupt 1233
        except Exception as e:
1234
            print e
5832 rajveer 1235
 
759 chandransh 1236
        return True
1237
    except:
1238
        return False
4410 rajveer 1239
 
5832 rajveer 1240
def send_mails_to_bdms(order):
5816 rajveer 1241
    logistics_client = LogisticsClient().get_client()
5832 rajveer 1242
    store = logistics_client.getPickupStore(order.pickupStoreId)
1243
    provider = logistics_client.getProvider(order.logistics_provider_id)
6029 rajveer 1244
    subject = source_name + " order " + str(order.id) +" shipped to store" 
5832 rajveer 1245
    raw_message = '''
1246
    Dear Sir,
5816 rajveer 1247
 
5832 rajveer 1248
    Order mentioned below is being shipped to your store. Customer will pick it from your store. 
1249
 
1250
    Store Id: %(store_id)s
1251
    Order Id: %(order_id)s 
1252
    Product: %(product)s
1253
    Delivery By: %(provider)s
5844 mandeep.dh 1254
    Payment Mode: %(payMode)s
1255
 
1256
    %(note)s
1257
 
5832 rajveer 1258
    Thanks and Regards
1259
    Sandeep Sachdeva
1260
    '''
1261
 
5844 mandeep.dh 1262
    payMode = 'Prepaid'
1263
    note = ''
1264
    if order.cod:
1265
        payMode = 'COD'
6318 rajveer 1266
        note = 'Note: Please collect Rs. ' + str(order.total_amount-order.gvAmount) + ' from customer'
5844 mandeep.dh 1267
 
5832 rajveer 1268
    message = dedent(raw_message) % { 'order_id' : str(order.id),
1269
                                     'store_id' : str(store.hotspotId),
1270
                                  'provider' : provider.name,
5844 mandeep.dh 1271
                                  'product' : str(order.lineitems[0]),
1272
                                  'payMode' : payMode,
1273
                                  'note'    : note
5832 rajveer 1274
                                  }
5894 rajveer 1275
    to_addresses = [store.email, store.bdmEmail, 'sandeep.sachdeva@shop2020.in', 'suraj.sharma@shop2020.in', 'manoj.kumar@shop2020.in']
1276
    cc = ['khushal.bhatia@spiceretail.co.in']
1277
    bcc = ['rajneesh.arora@spiceretail.co.in']
5966 rajveer 1278
    mail(mail_user, mail_password, to_addresses, subject, message, [], cc, bcc)
5816 rajveer 1279
 
5713 rajveer 1280
def mark_orders_as_returned_from_store(providerId, orderIds, awbs):
5676 rajveer 1281
    try:
1282
        current_timestamp = datetime.datetime.now()
5824 rajveer 1283
        orders = Order.query.filter(Order.id.in_(orderIds)).filter_by(logistics_provider_id = providerId, status = OrderStatus.RET_PICKUP_REQUEST_RAISED).all()
5676 rajveer 1284
        for order in orders:
5824 rajveer 1285
            order.status = OrderStatus.RET_RETURN_IN_TRANSIT
1286
            order.statusDescription = "Order returned from store"
1287
            order.doa_pickup_timestamp = current_timestamp
1288
            #order.airwaybill_no = awbs[orderIds.index(order.id)]
1289
            order.tracking_id = awbs[orderIds.index(order.id)]
5676 rajveer 1290
        session.commit()
1291
 
1292
        return True
1293
    except:
1294
        return False    
1295
 
1296
def set_order_attributes(orderId, attributes):
5871 rajveer 1297
    for attribute in attributes :
1298
        existingAttribute = Attribute.query.filter(Attribute.orderId == orderId).filter(Attribute.name == attribute.name).first()
1299
        if existingAttribute:
1300
            existingAttribute.value = attribute.value
1301
        else :                
1302
            attr = Attribute()
1303
            attr.orderId = orderId
1304
            attr.name = attribute.name
1305
            attr.value = attribute.value
5791 anupam.sin 1306
    session.commit()
5724 rajveer 1307
 
1113 chandransh 1308
def mark_orders_as_picked_up(provider_id, pickup_details):
1309
    for awb, pickup_timestamp in pickup_details.iteritems():
1135 chandransh 1310
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
2684 chandransh 1311
        if order == None or order.status != OrderStatus.SHIPPED_FROM_WH:
2220 chandransh 1312
            #raise TransactionServiceException(102, "No order found for the awb: " + awb)
1313
            continue
1113 chandransh 1314
        order.status = OrderStatus.SHIPPED_TO_LOGST
1315
        order.statusDescription = "Order picked up by Courier Company"
1246 chandransh 1316
        order.pickup_timestamp = pickup_timestamp
5636 amar.kumar 1317
 
1318
        try:
1319
            monitoredEntity = MonitoredEntity();
1320
            monitoredEntity.entityType=EntityType.COURIER;
1321
            monitoredEntity.eventType=OrderStatus.SHIPPED_TO_LOGST
1322
            monitoredEntity.entityIdentifier="orderId = " + str(order.id);
5744 amar.kumar 1323
            adjustedDeliveryDays = adjust_delivery_time(datetime.datetime.now(), 1);
1324
            sec = datetime.datetime.now()
1325
            #Warn alert time is taken as 16 hrs
5821 amit.gupta 1326
            warn_time = (sec + timedelta(days=(adjustedDeliveryDays-1)) + timedelta(hours=15))
5744 amar.kumar 1327
            #Critical alert time is taken as 26 hrs                    
5821 amit.gupta 1328
            critical_time = (sec + timedelta(days=(adjustedDeliveryDays-1)) + timedelta(hours=20))
5744 amar.kumar 1329
            monitoredEntity.warnExpiryTime = int(warn_time.strftime("%s"))*1000
1330
            monitoredEntity.criticalExpiryTime = int(critical_time.strftime("%s"))*1000
5819 amit.gupta 1331
            monitoredEntity.description="providerId = " + str(provider_id)
5636 amar.kumar 1332
            alert_client = AlertClient().get_client()
1333
            alert_client.updateMonitoredObject(monitoredEntity)
1334
        except Exception as e:
5801 amar.kumar 1335
            print "Exception in updating alert in MarkOrdersAsPickedUp method"
5636 amar.kumar 1336
            print e
1113 chandransh 1337
    session.commit()
1217 chandransh 1338
 
4910 phani.kuma 1339
def get_orders_not_picked_up(provider_id):
1132 chandransh 1340
    current_time = datetime.datetime.now()
1341
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
4741 phani.kuma 1342
    orders_not_picked_up = Order.query.filter_by(logistics_provider_id = provider_id).filter_by(status=OrderStatus.SHIPPED_FROM_WH).filter(Order.shipping_timestamp <= to_datetime).all()
1113 chandransh 1343
    return orders_not_picked_up
1344
 
1132 chandransh 1345
def mark_orders_as_delivered(provider_id, delivered_orders):
1346
    for awb, detail in delivered_orders.iteritems():
1135 chandransh 1347
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
4910 phani.kuma 1348
        if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE]:
1911 chandransh 1349
            continue
1350
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
5555 rajveer 1351
        if order.pickupStoreId:
1352
            order.status = OrderStatus.DELIVERED_AT_STORE
1353
            order.statusDescription = "Order delivered At Store"
5762 amar.kumar 1354
            try:
1355
                monitoredEntity = MonitoredEntity();
1356
                monitoredEntity.entityType=EntityType.COURIER;
1357
                monitoredEntity.eventType=OrderStatus.RECEIVED_AT_STORE;
1358
                monitoredEntity.entityIdentifier="orderId = " + str(order.id);
1359
                #adjustedDeliveryDays = adjust_delivery_time(datetime.datetime.now(), 1);
1360
                sec = datetime.datetime.now()
1361
                #Critical alert time is taken as 26 hrs                    
1362
                critical_time = (sec + timedelta(hours=4))
1363
                monitoredEntity.criticalExpiryTime = int(critical_time.strftime("%s"))*1000
1364
                monitoredEntity.description="deliveryTimeAtStore = " + str(order.delivery_timestamp) + " orderId = "  + str(order.id);
1365
                alert_client = AlertClient().get_client()
1366
                alert_client.updateMonitoredObject(monitoredEntity)
1367
            except Exception as e:
1368
                print e
5555 rajveer 1369
        else:
1370
            order.status = OrderStatus.DELIVERY_SUCCESS
1371
            order.statusDescription = "Order delivered"
5762 amar.kumar 1372
            update_trust_level(order)
5966 rajveer 1373
            enqueue_delivery_success_mail(order)
5762 amar.kumar 1374
            try:
1375
                alert_client = AlertClient().get_client()
1376
                alert_client.endMonitoringEntity(EntityType.COURIER, "orderId = " + str(order.id))
1377
            except Exception as e:
5801 amar.kumar 1378
                print "Exception in scheduling alert in ShippedFromWarehouse method"
5762 amar.kumar 1379
                print e
4910 phani.kuma 1380
        timestamp, receiver = detail.split('|')
1381
        if order.first_dlvyatmp_timestamp is None:
1382
            order.first_dlvyatmp_timestamp = timestamp
1383
        order.delivery_timestamp = timestamp
1384
        order.receiver = receiver
5636 amar.kumar 1385
 
1132 chandransh 1386
    session.commit()
1387
 
4712 rajveer 1388
def mark_order_as_delivered(orderId, deliveryTimestamp, receiver):
1389
    order = Order.get_by(id=orderId)
5555 rajveer 1390
    if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE, OrderStatus.DOA_PICKUP_REQUEST_RAISED, OrderStatus.DOA_PICKUP_CONFIRMED, OrderStatus.RET_REQUEST_RECEIVED, OrderStatus.RET_PICKUP_CONFIRMED, OrderStatus.RET_REQUEST_AUTHORIZED, OrderStatus.RET_PICKUP_REQUEST_RAISED, OrderStatus.RTO_IN_TRANSIT, OrderStatus.BILLED, OrderStatus.RECEIVED_AT_STORE]:
4712 rajveer 1391
        raise TransactionServiceException(101, "Either wrond order id or invalid state " + str(orderId))
5402 rajveer 1392
 
1393
    # Provider is 4 is for self pickup and hardcoded. We should figure out a way to not to hard code.   
1394
    if order.status == OrderStatus.BILLED and order.logistics_provider_id != 4:
5807 rajveer 1395
        raise TransactionServiceException(101, "Order is not marked for self pickup: " + str(orderId))
5402 rajveer 1396
 
5555 rajveer 1397
    if order.status in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE, OrderStatus.RTO_IN_TRANSIT, OrderStatus.BILLED, OrderStatus.RECEIVED_AT_STORE]:
4712 rajveer 1398
        order.delivery_timestamp = deliveryTimestamp
1399
        order.receiver = receiver
1400
    order.status = OrderStatus.DELIVERY_SUCCESS
1401
    order.statusDescription = "Order delivered"
1402
    update_trust_level(order)
1403
    session.commit()
5713 rajveer 1404
 
5812 rajveer 1405
    if order.pickupStoreId and order.cod:
5806 rajveer 1406
        __push_collection_to_hotspot(order, "SALE")
1407
 
5966 rajveer 1408
    enqueue_delivery_success_mail(order)
5713 rajveer 1409
#    if order.pickupStoreId:
1410
#        payment_client = PaymentClient().get_client()
1411
#        payment_client.createRefund(order.id, order.transaction.id, order.total_amount)
1412
#        payment_client = PaymentClient().get_client()
1413
#        payment_client.partiallyCapturePayment(order.transaction.id, order.total_amount, xferBy, xferTxnId, now())
1414
#        order.cod_reconciliation_timestamp = datetime.datetime.now()
1415
#        session.commit()
1416
 
5636 amar.kumar 1417
    try:
1418
        alert_client = AlertClient().get_client()
5744 amar.kumar 1419
        alert_client.endMonitoringEntity(EntityType.COURIER, "orderId = " + str(order.id))
5636 amar.kumar 1420
    except Exception as e:
5801 amar.kumar 1421
        print "Exception in ending alert in MarkOrderAsDelivered method"
5636 amar.kumar 1422
        print e
1423
 
4712 rajveer 1424
 
5555 rajveer 1425
def mark_order_as_received_at_store(orderId, deliveryTimestamp):
1426
    order = Order.get_by(id=orderId)
1427
    if order != None or order.status in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE, OrderStatus.DELIVERED_AT_STORE]:
1428
        order.status = OrderStatus.RECEIVED_AT_STORE
1429
        order.statusDescription = "Order received at store"
1430
        order.delivery_timestamp = deliveryTimestamp
1431
    session.commit()
5713 rajveer 1432
    try:
5762 amar.kumar 1433
        alert_client = AlertClient().get_client()
1434
        alert_client.endMonitoringEntity(EntityType.COURIER, "orderId = " + str(order.id))
1435
    except Exception as e:
1436
        print e
1437
 
1438
    try:
5713 rajveer 1439
        enqueue_received_at_store_email(order)
1440
    except Exception as e:
1441
        print e
1442
 
1443
 
5555 rajveer 1444
 
4910 phani.kuma 1445
def mark_orders_as_rto(provider_id, returned_orders):
1135 chandransh 1446
    for awb, detail in returned_orders.iteritems():
1447
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
5713 rajveer 1448
        if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.DELIVERED_AT_STORE, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE]:
2684 chandransh 1449
            continue
1450
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
4484 rajveer 1451
        order.status = OrderStatus.RTO_IN_TRANSIT
1135 chandransh 1452
        order.delivery_timestamp, reason = detail.split('|')
1453
        order.statusDescription = "Order Returned to Origin:" + reason
3578 mandeep.dh 1454
        update_trust_level(order)
1911 chandransh 1455
    session.commit()
5636 amar.kumar 1456
    try:
1457
        alert_client = AlertClient().get_client()
5801 amar.kumar 1458
        alert_client.endMonitoringEntity(EntityType.COURIER, "orderId = " + str(order.id))
5636 amar.kumar 1459
    except Exception as e:
5801 amar.kumar 1460
        print "Exception in ending alert in MarkOrderAsRTO method"
5636 amar.kumar 1461
        print e
1135 chandransh 1462
 
4910 phani.kuma 1463
def get_rto_orders(provider_id):
1464
    current_time = datetime.datetime.now()
1465
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
1466
    rto_orders = Order.query.filter_by(logistics_provider_id = provider_id).filter_by(status=OrderStatus.RTO_IN_TRANSIT).filter(Order.delivery_timestamp <= to_datetime).all()
1467
    return rto_orders
1468
 
5555 rajveer 1469
def get_receive_pending_orders(storeId):
1470
    return Order.query.filter_by(pickupStoreId = storeId).filter(Order.status.in_((OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE, OrderStatus.DELIVERED_AT_STORE))).all()
1471
 
1472
def get_received_at_store_orders(storeId):
5676 rajveer 1473
    return Order.query.filter_by(pickupStoreId = storeId).filter(Order.status.in_((OrderStatus.RECEIVED_AT_STORE, OrderStatus.RET_PICKUP_REQUEST_RAISED))).all()
5555 rajveer 1474
 
5713 rajveer 1475
def get_orders_collection_at_store(storeId, fromDate, toDate, onlyCod):
1476
    query = Order.query.filter(Order.status == OrderStatus.DELIVERY_SUCCESS)
1477
    if storeId:
1478
        query = query.filter_by(pickupStoreId = storeId)
5720 rajveer 1479
    else:
1480
        query = query.filter(Order.pickupStoreId != 0)
5713 rajveer 1481
    if onlyCod:
1482
        query = query.filter_by(cod = onlyCod)
1483
    if fromDate:
1484
        query = query.filter(Order.delivery_timestamp >= fromDate)
1485
    if toDate:
1486
        query = query.filter(Order.delivery_timestamp <= toDate)
1487
    return query.all()
1488
 
1246 chandransh 1489
def update_non_delivery_reason(provider_id, undelivered_orders):
1490
    for awb, reason in undelivered_orders.iteritems():
1491
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
4920 phani.kuma 1492
        if order == None or order.status == OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE:
2684 chandransh 1493
            continue
1494
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
1246 chandransh 1495
        order.statusDescription = reason
1496
    session.commit()
4910 phani.kuma 1497
 
1498
def get_non_delivered_orders_by_courier(provider_id):
4581 phani.kuma 1499
    current_time = datetime.datetime.now()
1500
    today_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
1501
    upto_datetime = today_datetime - datetime.timedelta(4)
4910 phani.kuma 1502
    orders_not_delivered = Order.query.filter_by(logistics_provider_id = provider_id).filter(Order.status.in_((OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE))).filter(or_(Order.shipping_timestamp <= upto_datetime, Order.pickup_timestamp <= upto_datetime)).all()
4581 phani.kuma 1503
    return orders_not_delivered
1246 chandransh 1504
 
4910 phani.kuma 1505
def mark_orders_as_local_connected(provider_id, local_connected_orders):
1506
    for awb, timestamp in local_connected_orders.iteritems():
1507
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
1508
        if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST]:
1509
            #raise TransactionServiceException(102, "No order found for the awb: " + awb)
1510
            continue
1511
        order.status = OrderStatus.SHIPPED_TO_DESTINATION_CITY
1512
        order.statusDescription = "Left Out of Origin City"
5636 amar.kumar 1513
        current_time = datetime.datetime.now()
1514
        order.local_connected_timestamp = datetime.datetime(current_time.year, current_time.month, current_time.day)
1515
        try:
1516
            monitoredEntity = MonitoredEntity();
1517
            monitoredEntity.entityType=EntityType.COURIER;
1518
            monitoredEntity.eventType=OrderStatus.SHIPPED_TO_DESTINATION_CITY
1519
            monitoredEntity.entityIdentifier="orderId = " + str(order.id);
5744 amar.kumar 1520
            #Warn alert time is taken as 23 hrs less than promised delivery time
5821 amit.gupta 1521
            warn_time = (order.expected_delivery_time-timedelta(hours = 15))
5744 amar.kumar 1522
            #Critical alert time is taken as 13 hrs less than promised delivery time                    
5821 amit.gupta 1523
            critical_time = (order.expected_delivery_time-timedelta(hours = 10))
5801 amar.kumar 1524
            monitoredEntity.warnExpiryTime = int(warn_time.strftime("%s"))*1000
1525
            monitoredEntity.criticalExpiryTime = int(critical_time.strftime("%s"))*1000
5636 amar.kumar 1526
            monitoredEntity.description="providerId = " + str(provider_id) + " destination city " + order.customer_city 
1527
            alert_client = AlertClient().get_client()
1528
            alert_client.updateMonitoredObject(monitoredEntity)
1529
        except Exception as e:
5801 amar.kumar 1530
            print "Exception in updating alert in MarkOrderAsLocalConnected method"
5636 amar.kumar 1531
            print e
1532
        session.commit()
4910 phani.kuma 1533
 
1534
def get_orders_not_local_connected(provider_id):
1535
    current_time = datetime.datetime.now()
1536
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
1537
    orders_pending_local_connection = Order.query.filter_by(logistics_provider_id = provider_id).filter(Order.status.in_((OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST))).filter(or_(Order.shipping_timestamp <= to_datetime, Order.pickup_timestamp <= to_datetime)).all()
1538
    return orders_pending_local_connection
1539
 
1540
def mark_orders_as_destinationCityReached(provider_id, destination_city_reached_orders):
1541
    for awb, timestamp in destination_city_reached_orders.iteritems():
1542
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
1543
        if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY]:
1544
            continue
1545
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
1546
        order.status = OrderStatus.REACHED_DESTINATION_CITY
1547
        order.statusDescription = "Reached Destination City"
1548
        order.reached_destination_timestamp = timestamp
5636 amar.kumar 1549
        try:
1550
            monitoredEntity = MonitoredEntity();
1551
            monitoredEntity.entityType=EntityType.COURIER;
1552
            monitoredEntity.eventType=OrderStatus.REACHED_DESTINATION_CITY
1553
            monitoredEntity.entityIdentifier="orderId = " + str(order.id);
5764 amar.kumar 1554
            warn_time = (order.expected_delivery_time)
5801 amar.kumar 1555
            critical_time = (order.expected_delivery_time + timedelta(hours=12))
1556
            monitoredEntity.warnExpiryTime = int(warn_time.strftime("%s"))*1000
1557
            monitoredEntity.criticalExpiryTime = int(critical_time.strftime("%s"))*1000
5636 amar.kumar 1558
            monitoredEntity.description="providerId = " + str(provider_id) + " destination city " + order.customer_city 
1559
            alert_client = AlertClient().get_client()
1560
            alert_client.updateMonitoredObject(monitoredEntity)
1561
        except Exception as e:
5801 amar.kumar 1562
            print "Exception in updating alert in MarkOrderAsDestCityReached method"
5636 amar.kumar 1563
            print e
4910 phani.kuma 1564
    session.commit()
1565
 
1566
def mark_orders_as_firstDeliveryAttempted(provider_id, first_atdl_orders):
1567
    for awb, detail in first_atdl_orders.iteritems():
1568
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
1569
        if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY]:
1570
            continue
1571
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
1572
        order.status = OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE
1573
        order.first_dlvyatmp_timestamp, reason = detail.split('|')
1574
        order.statusDescription = reason
5636 amar.kumar 1575
        try:
1576
            monitoredEntity = MonitoredEntity();
1577
            monitoredEntity.entityType=EntityType.COURIER;
1578
            monitoredEntity.eventType=OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE
1579
            monitoredEntity.entityIdentifier="orderId = " + str(order.id);
5744 amar.kumar 1580
            adjustedDeliveryDaysfor1day = adjust_delivery_time(datetime.datetime.now(), 1);
1581
            adjustedDeliveryDaysfor2days = adjust_delivery_time(datetime.datetime.now(), 2);
1582
            sec = datetime.datetime.now()
1583
            #Warn alert time is taken as 16 hrs
1584
            warn_time = (sec + timedelta(days=(adjustedDeliveryDaysfor1day-1)) + timedelta(hours=24))
1585
            #Critical alert time is taken as 26 hrs                    
1586
            critical_time = (sec + timedelta(days=(adjustedDeliveryDaysfor2days-1)) + timedelta(hours=36))
1587
            monitoredEntity.warnExpiryTime = int(warn_time.strftime("%s"))*1000
1588
            monitoredEntity.criticalExpiryTime = int(critical_time.strftime("%s"))*1000
5636 amar.kumar 1589
            monitoredEntity.description="providerId = " + str(provider_id) + " first attempt timestamp" + order.first_dlvyatmp_timestamp
1590
            alert_client = AlertClient().get_client()
1591
            alert_client.updateMonitoredObject(monitoredEntity)
1592
        except Exception as e:
5801 amar.kumar 1593
            print "Exception in updating alert in MarkOrdersAsFirstDeliveryAttempted method"
5636 amar.kumar 1594
            print e
4910 phani.kuma 1595
    session.commit()
1596
 
4783 phani.kuma 1597
def get_orders_not_met_expected_delivery_date():
1598
    current_time = datetime.datetime.now()
1599
    today_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
4910 phani.kuma 1600
    orders_not_delivered = Order.query.filter(Order.status.in_((OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY))).filter(Order.expected_delivery_time <= today_datetime).all()
4783 phani.kuma 1601
    return orders_not_delivered
1602
 
4444 rajveer 1603
def get_alerts(type, warehouseId, status, timestamp):
1604
    query = Alert.query.filter(Alert.warehouseId == warehouseId)
1605
    if type != -1 :
1606
        query = query.filter(Alert.type == type)
1607
    if status != -1 :
1608
        query = query.filter(Alert.status == status)
4394 rajveer 1609
    if timestamp:
1610
        query = query.filter(Alert.timestamp >= to_py_date(timestamp))
4412 rajveer 1611
    query = query.order_by(desc(Alert.timestamp))
483 rajveer 1612
    return query.all()
1613
 
4447 rajveer 1614
def add_alert(type, warehouseId, description):
4394 rajveer 1615
    alert = Alert()
1616
    alert.type = type
1617
    alert.status = 1
1618
    alert.description = description
1619
    alert.timestamp = datetime.datetime.now()
4447 rajveer 1620
    alert.warehouseId = warehouseId
4444 rajveer 1621
    session.commit()
766 rajveer 1622
 
4444 rajveer 1623
def mark_alerts_as_seen(warehouseId):
1624
    query = Alert.query.filter(Alert.warehouseId == warehouseId).filter(Alert.status == 1)
1625
    alerts = query.all()
1626
    for alert in alerts:
1627
        alert.status = 0    
1628
    session.commit()
1629
 
1596 ankur.sing 1630
def get_valid_order_count():
2851 chandransh 1631
    '''
1632
    Returns the number of orders which we processed. The reshipped orders are not
1633
    counted since there'll always be a new compensating order present.
1634
    '''
4682 rajveer 1635
    return Order.query.filter(Order.status >= OrderStatus.SUBMITTED_FOR_PROCESSING).filter(not_(Order.status.in_((OrderStatus.RTO_RESHIPPED, OrderStatus.DOA_INVALID_RESHIPPED, OrderStatus.DOA_VALID_RESHIPPED, OrderStatus.COD_VERIFICATION_FAILED)))).count()
1596 ankur.sing 1636
 
1627 ankur.sing 1637
def get_cust_count_with_successful_txn():
2851 chandransh 1638
    '''
1639
    Returns the number of distinct customers who have done successful transactions.
1640
    It uses the sqlalchemy func module to construct an efficient query.
1641
    '''
1642
    return session.query(func.count(distinct(Order.customer_id))).filter(Order.status >= OrderStatus.SUBMITTED_FOR_PROCESSING).scalar()
1627 ankur.sing 1643
 
1730 ankur.sing 1644
def get_valid_orders_amount_range():
2851 chandransh 1645
    '''
1646
    Returns a list containing the minimum and maximum amount across all
1647
    orders placed with us. No need to consider reshippped orders for min/max
1648
    amount of orders.
1649
    '''
1650
    return session.query(func.min(Order.total_amount), func.max(Order.total_amount)).filter(Order.status >= OrderStatus.SUBMITTED_FOR_PROCESSING).one()
1627 ankur.sing 1651
 
5874 rajveer 1652
def get_valid_orders(limit, onlyStore):
2851 chandransh 1653
    '''
1654
    Returns the last limit number of valid orders placed with us. Ignored the
1655
    limit if it's passed as 0. Raises an excpetion if the supplied limit is 
1656
    less than 0. 
1657
    '''
1658
    if limit < 0:
1659
        raise TransactionServiceException(101, "Invalid argument limit " + limit)
4662 rajveer 1660
    query = Order.query.filter(Order.status >= OrderStatus.COD_VERIFICATION_PENDING)
5874 rajveer 1661
    if onlyStore:
1662
        query = query.filter(Order.pickupStoreId != 0)
1886 ankur.sing 1663
    query = query.order_by(desc(Order.created_timestamp))
2851 chandransh 1664
    if limit != 0:
1886 ankur.sing 1665
        query = query.limit(limit)
1666
    return query.all()
1667
 
5527 anupam.sin 1668
def get_next_invoice_number(orderType):
1669
    if OrderType.B2Cbulk == orderType:
1670
        orderType = OrderType.B2C
1671
    entity_id = InvoiceIDGenerator.query.filter(InvoiceIDGenerator.orderType == orderType).with_lockmode("update").one()
2783 chandransh 1672
    invoice_number = entity_id.id + 1
1673
    entity_id.id = invoice_number
1674
    return invoice_number
1675
 
2536 chandransh 1676
def toggle_doa_flag(order_id):
1677
    order = get_order(order_id)
1678
    if(order.doaFlag):
1679
        order.doaFlag = False
1680
    else:
1681
        order.doaFlag = True
1682
    session.commit()
1683
    return order.doaFlag
1684
 
4454 rajveer 1685
def mark_order_doa_request_received(orderId):
1686
    order = get_order(orderId)
1687
    if order.status != OrderStatus.DELIVERY_SUCCESS:
1688
        return False
1689
    order.status = OrderStatus.DOA_REQUEST_RECEIVED
4459 rajveer 1690
    order.statusDescription = "DOA Request Received"
4454 rajveer 1691
    session.commit()
1692
    return True
1693
 
1694
def mark_order_doa_request_authorized(orderId, isAuthorized):
1695
    order = get_order(orderId)
1696
    if order.status != OrderStatus.DOA_REQUEST_RECEIVED:
1697
        return False
1698
    if isAuthorized:
1699
        order.status = OrderStatus.DOA_REQUEST_AUTHORIZED
4459 rajveer 1700
        order.statusDescription = "DOA Request Authorized"
4454 rajveer 1701
    else:
1702
        order.status = OrderStatus.DELIVERY_SUCCESS
4459 rajveer 1703
        order.statusDescription = "Order delivered"
4454 rajveer 1704
    session.commit()
1705
    return True
1706
 
4488 rajveer 1707
def mark_order_return_request_received(orderId):
1708
    order = get_order(orderId)
1709
    if order.status != OrderStatus.DELIVERY_SUCCESS:
1710
        return False
1711
    order.status = OrderStatus.RET_REQUEST_RECEIVED
1712
    order.statusDescription = "Return Request Received"
1713
    session.commit()
1714
    return True
1715
 
1716
def mark_order_return_request_authorized(orderId, isAuthorized):
1717
    order = get_order(orderId)
1718
    if order.status != OrderStatus.RET_REQUEST_RECEIVED:
1719
        return False
1720
    if isAuthorized:
1721
        order.status = OrderStatus.RET_REQUEST_AUTHORIZED
1722
        order.statusDescription = "Return Request Authorized"
1723
    else:
1724
        order.status = OrderStatus.DELIVERY_SUCCESS
1725
        order.statusDescription = "Order delivered"
1726
    session.commit()
1727
    return True
1728
 
4579 rajveer 1729
def request_pickup_number(order_id, providerId):
2536 chandransh 1730
    order = get_order(order_id)
4495 rajveer 1731
    if order.status != OrderStatus.DELIVERY_SUCCESS and order.status != OrderStatus.DOA_REQUEST_AUTHORIZED and order.status != OrderStatus.RET_REQUEST_AUTHORIZED:
2536 chandransh 1732
        return False
1733
 
4227 rajveer 1734
    subject = "Pickup request from " + order.customer_city
2536 chandransh 1735
    try:
5944 mandeep.dh 1736
        inventory_client = InventoryClient().get_client()
1737
        warehouse = inventory_client.getWarehouse(order.warehouse_id)
2536 chandransh 1738
 
1739
        logistics_client = LogisticsClient().get_client()
4579 rajveer 1740
        provider = logistics_client.getProvider(providerId)
3051 chandransh 1741
        to_addr = provider.details[DeliveryType.PREPAID].email
2536 chandransh 1742
        raw_message = '''
1743
        Dear Sir/Madam,
1744
 
2559 chandransh 1745
        Kindly arrange a pickup today from %(customer_city)s. Pickup and delivery addresses are mentioned below.
2536 chandransh 1746
 
1747
        Pickup CODE: %(provider_code)s
1748
        Total Weight: %(order_weight)s kg
1749
 
2559 chandransh 1750
        Pickup Address:
2536 chandransh 1751
 
1752
        %(customer_address)s
1753
 
1754
        Delivery Address:
1755
 
1756
        %(warehouse_executive)s
1757
        %(warehouse_address)s
1758
        PIN %(warehouse_pin)s
1759
 
1760
        Thanks and Regards
4227 rajveer 1761
        Sandeep Sachdeva
2536 chandransh 1762
        '''
4579 rajveer 1763
 
1764
        if warehouse.id == 7:
1765
            executive = 'Amit Kumar'
1766
        else:
4966 phani.kuma 1767
            executive = 'Dinesh Kumar'
4579 rajveer 1768
 
2559 chandransh 1769
        message = dedent(raw_message) % { 'customer_city' : order.customer_city,
3051 chandransh 1770
                                          'provider_code' : provider.details[DeliveryType.PREPAID].accountNo,
2536 chandransh 1771
                                          'order_weight' : str(order.total_weight),
1772
                                          'customer_address' : __get_order_address(order),
4579 rajveer 1773
                                          'warehouse_executive' : executive,
2536 chandransh 1774
                                          'warehouse_address' : warehouse.location,
4966 phani.kuma 1775
                                          'warehouse_pin' : warehouse.pincode}
5860 rajveer 1776
        to_addresses = [to_addr, 'gaurabh.verma@shop2020.in', 'suraj.sharma@shop2020.in']
4966 phani.kuma 1777
 
5966 rajveer 1778
        mail(mail_user, mail_password, to_addresses, subject, message)
4495 rajveer 1779
        if order.status == OrderStatus.RET_REQUEST_AUTHORIZED:
1780
            order.status = OrderStatus.RET_PICKUP_REQUEST_RAISED
1781
            order.statusDescription = "Pick up requested for RETURN"
1782
        else:
1783
            order.status = OrderStatus.DOA_PICKUP_REQUEST_RAISED
1784
            order.statusDescription = "Pick up requested for DOA"
4815 phani.kuma 1785
        order.doa_logistics_provider_id = providerId
2536 chandransh 1786
        session.commit()
1787
        return True
1788
    except Exception as e:
1789
        print sys.exc_info()[0]
1790
        return False
1791
 
4602 rajveer 1792
def authorize_pickup(order_id, pickup_number, providerId):
2536 chandransh 1793
    order = get_order(order_id)
4495 rajveer 1794
    if order.status not in [OrderStatus.DOA_PICKUP_REQUEST_RAISED, OrderStatus.RET_PICKUP_REQUEST_RAISED]:
2536 chandransh 1795
        return False
4815 phani.kuma 1796
    if order.doa_logistics_provider_id is not None:
1797
        providerId = order.doa_logistics_provider_id    #previously we are not storing this so data is taken from omdashboarduser
2536 chandransh 1798
    subject = 'Pickup details'
1799
    filename = "/tmp/" +  str(order.id) +".pdf" 
1800
    try:
5944 mandeep.dh 1801
        inventory_client = InventoryClient().get_client()
1802
        warehouse = inventory_client.getWarehouse(order.warehouse_id)
2536 chandransh 1803
 
1804
        logistics_client = LogisticsClient().get_client()
4602 rajveer 1805
        provider = logistics_client.getProvider(providerId)
2536 chandransh 1806
        provider_name = provider.name
1807
 
1808
        to_addr = order.customer_email
1809
        today = date.today().strftime("%d-%B-%Y (%A)")
1810
 
1811
        raw_message = '''
1812
        Dear Mr. %(customer_name)s,
1813
 
4815 phani.kuma 1814
        In reference to the return of order ID %(order_id)d, ''' + __generate_mail_body_for_authorize_pickup(order)
1815
        raw_message = raw_message + '''
2536 chandransh 1816
        Take a printout of the attachment in this mail which contains the delivery address and paste it on the packed parcel.
1817
        We have instructed %(provider_name)s to pick up the material from the below mentioned address on %(date)s. %(provider_name)s pickup request number is %(pickup_request_no)s. Kindly keep the parcel ready.
1818
 
1819
        Pickup will happen from the below mentioned address:-
1820
 
1821
        %(customer_address)s
1822
 
1823
        Do let us know in case of any concerns
1824
 
1825
        Thanks & Regards
6029 rajveer 1826
        %(source_name)s Team
2536 chandransh 1827
        '''
1828
        message = dedent(raw_message) % {'customer_name' : order.customer_name,
1829
                                         'order_id' : order_id,
1830
                                         'provider_name' : provider_name,
1831
                                         'date': today,
1832
                                         'pickup_request_no' : pickup_number,
6029 rajveer 1833
                                         'customer_address' : __get_order_address(order),
1834
                                         'source_name' : source_name}
2536 chandransh 1835
        print message
1836
        __generate_return_advice(order, warehouse, provider, filename)
5966 rajveer 1837
        mail(help_user, help_password, [to_addr], subject, message, [get_attachment_part(filename)])
2536 chandransh 1838
        order.pickupRequestNo = pickup_number
4495 rajveer 1839
        if order.status == OrderStatus.RET_PICKUP_REQUEST_RAISED:
1840
            order.status = OrderStatus.RET_PICKUP_CONFIRMED
1841
            order.statusDescription = "RETURN pick up confirmed"
1842
        else:
1843
            order.status = OrderStatus.DOA_PICKUP_CONFIRMED
1844
            order.statusDescription = "DOA pick up confirmed"
2764 chandransh 1845
        order.doa_auth_timestamp = datetime.datetime.now()
2536 chandransh 1846
        session.commit()
1847
        return True
1848
    except Exception as e:
4234 rajveer 1849
        print sys.exc_info()
4506 phani.kuma 1850
        traceback.print_tb(sys.exc_info()[2])
2536 chandransh 1851
        return False
1852
    finally:
1853
        if os.path.exists(filename):
1854
            os.remove(filename)
1855
 
4815 phani.kuma 1856
def __generate_mail_body_for_authorize_pickup(order):
1857
    brand_msgbody_map = {'Nokia' : 'we request you to kindly pack the box properly with bubble wrap, and send it along with the all original accessories, original handset and the invoice.',
1858
                         'Micromax' : 'we request you to kindly pack the box properly with bubble wrap, and send it along with the all original accessories, original handset and the invoice.',
1859
                         'Karbonn' : 'we request you to kindly pack the box properly with bubble wrap, and send it along with the all original accessories, original handset and the invoice.',
1860
                         'Sony Ericsson' : 'we request you to kindly pack the original DOA certificate and the invoice.',
1861
                         'Others Handsets' : 'we request you to kindly pack the box properly with bubble wrap, and send it along with the original "DOA" certificate and the invoice. Please paste the original DOA on the sealed box. Kindly ensure that the product is properly sealed. If we receive open sealed products, we will not be able to accept.',
1862
                         'Handsets Return' : 'we request you to kindly pack the box properly with bubble wrap, and send it along with the all original accessories, original handset and the invoice.',
1863
                         'Accessories' : 'we request you to kindly pack the box properly with bubble wrap, and send it along with the original product and the invoice.'}
1864
    categories = ["Handsets", "Tablets", "Laptops"]
1865
    lineitem = order.lineitems[0]
1866
    brand = lineitem.brand
1867
    productGroup = lineitem.productGroup
1868
    searchkey = None
1869
    if productGroup not in categories:
1870
        searchkey = 'Accessories'
1871
    elif order.status == OrderStatus.RET_PICKUP_REQUEST_RAISED:
1872
        searchkey = 'Handsets Return'
1873
    elif brand == 'Nokia':
1874
        searchkey = 'Nokia'
1875
    elif brand == 'Micromax':
1876
        searchkey = 'Micromax'
1877
    elif brand == 'Karbonn':
1878
        searchkey = 'Karbonn'
1879
    elif brand == 'Sony Ericsson':
1880
        searchkey = 'Sony Ericsson'
1881
    else:
1882
        searchkey = 'Others Handsets'
1883
    return brand_msgbody_map.get(searchkey)
1884
 
2764 chandransh 1885
def mark_doas_as_picked_up(provider_id, pickup_details):
1886
    for pickupNo, doa_pickup_timestamp in pickup_details.iteritems():
4815 phani.kuma 1887
        order = Order.query.filter_by(pickupRequestNo=pickupNo, doa_logistics_provider_id = provider_id).first()
4452 rajveer 1888
        if order == None or order.status != OrderStatus.DOA_PICKUP_CONFIRMED:
2764 chandransh 1889
            #raise TransactionServiceException(102, "No order found for the awb: " + awb)
1890
            continue
1891
        order.status = OrderStatus.DOA_RETURN_IN_TRANSIT
1892
        order.statusDescription = "DOA picked up by Courier Company"
1893
        order.doa_pickup_timestamp = doa_pickup_timestamp
1894
    session.commit()
1895
 
4910 phani.kuma 1896
def get_doas_not_picked_up(provider_id):
2764 chandransh 1897
    current_time = datetime.datetime.now()
1898
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
4815 phani.kuma 1899
    orders_not_picked_up = Order.query.filter_by(doa_logistics_provider_id = provider_id).filter_by(status=OrderStatus.DOA_PICKUP_CONFIRMED).filter(Order.doa_auth_timestamp <= to_datetime).all()
2764 chandransh 1900
    return orders_not_picked_up
1901
 
4741 phani.kuma 1902
def mark_return_orders_as_picked_up(provider_id, pickup_details):
1903
    for pickupNo, return_order_pickup_timestamp in pickup_details.iteritems():
4815 phani.kuma 1904
        order = Order.query.filter_by(pickupRequestNo=pickupNo, doa_logistics_provider_id = provider_id).first()
4741 phani.kuma 1905
        if order == None or order.status != OrderStatus.RET_PICKUP_CONFIRMED:
1906
            #raise TransactionServiceException(102, "No order found for the awb: " + awb)
1907
            continue
1908
        order.status = OrderStatus.RET_RETURN_IN_TRANSIT
1909
        order.statusDescription = "Return order picked up by Courier Company"
1910
        order.doa_pickup_timestamp = return_order_pickup_timestamp
1911
    session.commit()
1912
 
4910 phani.kuma 1913
def get_return_orders_not_picked_up(provider_id):
4741 phani.kuma 1914
    current_time = datetime.datetime.now()
1915
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
4815 phani.kuma 1916
    orders_not_picked_up = Order.query.filter_by(doa_logistics_provider_id = provider_id).filter_by(status=OrderStatus.RET_PICKUP_CONFIRMED).filter(Order.doa_auth_timestamp <= to_datetime).all()
4741 phani.kuma 1917
    return orders_not_picked_up
1918
 
4479 rajveer 1919
def receive_return(order_id, receiveCondition):
2591 chandransh 1920
    order = get_order(order_id)
4365 mandeep.dh 1921
 
4452 rajveer 1922
    if order.status in [OrderStatus.DOA_PICKUP_CONFIRMED, OrderStatus.DOA_RETURN_IN_TRANSIT]:
4479 rajveer 1923
        if receiveCondition == 0:
1924
            order.status = OrderStatus.DOA_RECEIVED_PRESTINE
1925
            order.statusDescription = "DOA package received"
1926
        elif receiveCondition == 1:
1927
            order.status = OrderStatus.DOA_RECEIVED_DAMAGED
1928
            order.statusDescription = "DOA received damaged"
1929
        elif receiveCondition == 2:
1930
            order.status = OrderStatus.DOA_LOST_IN_TRANSIT
1931
            order.statusDescription = "DOA lost in transit"
2764 chandransh 1932
        order.received_return_timestamp = datetime.datetime.now()
4495 rajveer 1933
    elif order.status in [OrderStatus.RET_PICKUP_CONFIRMED, OrderStatus.RET_RETURN_IN_TRANSIT]:
1934
        if receiveCondition == 0:
1935
            order.status = OrderStatus.RET_RECEIVED_PRESTINE
1936
            order.statusDescription = "RETURN package received"
1937
        elif receiveCondition == 1:
1938
            order.status = OrderStatus.RET_RECEIVED_DAMAGED
1939
            order.statusDescription = "RETURN received damaged"
1940
        elif receiveCondition == 2:
1941
            order.status = OrderStatus.RET_LOST_IN_TRANSIT
1942
            order.statusDescription = "RETURN lost in transit"
1943
        order.received_return_timestamp = datetime.datetime.now()
4484 rajveer 1944
    elif order.status == OrderStatus.RTO_IN_TRANSIT :
4486 rajveer 1945
        if receiveCondition == 0:
1946
            order.status = OrderStatus.RTO_RECEIVED_PRESTINE
4515 mandeep.dh 1947
            order.statusDescription = "Returned to origin"
4486 rajveer 1948
        elif receiveCondition == 1:
1949
            order.status = OrderStatus.RTO_RECEIVED_DAMAGED
1950
            order.statusDescription = "RTO received damaged"
1951
        elif receiveCondition == 2:
1952
            order.status = OrderStatus.RTO_LOST_IN_TRANSIT
1953
            order.statusDescription = "RTO lost in transit"
2764 chandransh 1954
        order.received_return_timestamp = datetime.datetime.now()
2616 chandransh 1955
    else:
2591 chandransh 1956
        return False
5110 mandeep.dh 1957
 
1958
    # For OUR warehouses, we need to scan in items for every return
5944 mandeep.dh 1959
    inventoryClient = InventoryClient().get_client()
1960
    warehouse = inventoryClient.getWarehouse(order.warehouse_id)
5110 mandeep.dh 1961
    if warehouse.billingType == BillingType.OURS:
5361 mandeep.dh 1962
        scanMap = {
1963
            OrderStatus.RTO_RECEIVED_PRESTINE : ScanType.SALE_RET,
1964
            OrderStatus.RTO_RECEIVED_DAMAGED  : ScanType.SALE_RET_UNUSABLE,
1965
            OrderStatus.RTO_LOST_IN_TRANSIT   : ScanType.LOST_IN_TRANSIT,
1966
            OrderStatus.DOA_RECEIVED_PRESTINE : ScanType.DOA_IN,
1967
            OrderStatus.DOA_RECEIVED_DAMAGED  : ScanType.DOA_IN,
1968
            OrderStatus.DOA_LOST_IN_TRANSIT   : ScanType.LOST_IN_TRANSIT
1969
        }
5110 mandeep.dh 1970
 
5419 mandeep.dh 1971
        if scanMap.has_key(order.status):
1972
            scanType = scanMap[order.status]
5361 mandeep.dh 1973
            lineitem = order.lineitems[0]
5944 mandeep.dh 1974
            catalogClient = CatalogClient().get_client()
5361 mandeep.dh 1975
            item = catalogClient.getItem(lineitem.item_id)
1976
            warehouseClient = WarehouseClient().get_client()
1977
            if item.type == ItemType.SERIALIZED:
5110 mandeep.dh 1978
                warehouseClient.scanSerializedItemForOrder(lineitem.serial_number, scanType, order.id, order.fulfilmentWarehouseId, lineitem.quantity, order.warehouse_id)
5361 mandeep.dh 1979
            else:
1980
                warehouseClient.scanForOrder(None, scanType, lineitem.quantity, order.id, order.fulfilmentWarehouseId, order.warehouse_id)
5110 mandeep.dh 1981
 
5558 mandeep.dh 1982
    session.commit()
2591 chandransh 1983
    return True
1984
 
1985
def validate_doa(order_id, is_valid):
1986
    order = get_order(order_id)
4452 rajveer 1987
    if order.status != OrderStatus.DOA_RECEIVED_PRESTINE:
2591 chandransh 1988
        return False
1989
    if is_valid:
2609 chandransh 1990
        order.status = OrderStatus.DOA_CERT_VALID
1991
        order.statusDescription = "DOA Certificate Valid"
2591 chandransh 1992
    else:
1993
        order.status = OrderStatus.DOA_CERT_INVALID
1994
        order.statusDescription = "DOA Certificate Invalid"
3578 mandeep.dh 1995
        update_trust_level(order)
2591 chandransh 1996
    session.commit()
1997
    return True
1998
 
4495 rajveer 1999
def validate_return_product(order_id, is_usable):
2000
    order = get_order(order_id)
2001
    if order.status != OrderStatus.RET_RECEIVED_PRESTINE:
2002
        return False
5110 mandeep.dh 2003
 
2004
    scanType = ScanType.SALE_RET_UNUSABLE
4495 rajveer 2005
    if is_usable:
2006
        order.status = OrderStatus.RET_PRODUCT_USABLE
2007
        order.statusDescription = "Return product usable"
5110 mandeep.dh 2008
        scanType = ScanType.SALE_RET
4495 rajveer 2009
    else:
2010
        order.status = OrderStatus.RET_PRODUCT_UNUSABLE
2011
        order.statusDescription = "Return product unusable"
2012
        update_trust_level(order)
2013
    session.commit()
5110 mandeep.dh 2014
 
2015
    # For OUR warehouses, we need to scan in items for every return
5944 mandeep.dh 2016
    inventoryClient = InventoryClient().get_client()
2017
    warehouse = inventoryClient.getWarehouse(order.warehouse_id)
5110 mandeep.dh 2018
    if warehouse.billingType == BillingType.OURS:
2019
        lineitem = order.lineitems[0]
5944 mandeep.dh 2020
        catalogClient = CatalogClient().get_client()
5361 mandeep.dh 2021
        item = catalogClient.getItem(lineitem.item_id)
5110 mandeep.dh 2022
        warehouse_client = WarehouseClient().get_client()
5361 mandeep.dh 2023
        if item.type == ItemType.SERIALIZED:
2024
            warehouse_client.scanSerializedItemForOrder(lineitem.serial_number, scanType, order.id, order.fulfilmentWarehouseId, lineitem.quantity, order.warehouse_id)
2025
        else:
2026
            warehouse_client.scanForOrder(None, scanType, lineitem.quantity, order.id, order.fulfilmentWarehouseId, order.warehouse_id)
4495 rajveer 2027
    return True
2028
 
2628 chandransh 2029
def reship_order(order_id):
2030
    """
4484 rajveer 2031
    If the order is in RTO_RECEIVED_PRESTINE or DOA_CERT_INVALID state, it does the following:
2628 chandransh 2032
        1. Creates a new order for processing in the BILLED state. All billing information is saved.
4484 rajveer 2033
        2. Marks the current order as one of the final states RTO_RESHIPPED and DOA_INVALID_RESHIPPED depending on what state the order started in.
2628 chandransh 2034
 
2035
    If the order is in DOA_CERT_VALID state, it does the following:
2036
        1. Creates a new order for processing in the SUBMITTED_FOR_PROCESSING state.
2037
        2. Creates a return order for the warehouse executive to return the DOA material.
4452 rajveer 2038
        3. Marks the current order as the final DOA_VALID_RESHIPPED state.
2628 chandransh 2039
 
2040
    Returns the id of the newly created order.
2041
 
2042
    Throws an exception if the order with the given id couldn't be found.
2043
 
2044
    Parameters:
2045
     - orderId
2046
    """
2047
    order = get_order(order_id)
4484 rajveer 2048
    if order.status == OrderStatus.RTO_RECEIVED_PRESTINE:
4220 rajveer 2049
        new_order = __clone_order(order, True, order.cod)
4484 rajveer 2050
        order.status = OrderStatus.RTO_RESHIPPED
2628 chandransh 2051
        order.statusDescription = "Order Reshipped"
2052
        session.commit()
3578 mandeep.dh 2053
        update_trust_level(order)
4486 rajveer 2054
    elif order.status == OrderStatus.RTO_RECEIVED_DAMAGED:
2055
        new_order = __clone_order(order, False, order.cod)
2056
        ret_order = __create_return_order(order)
2057
        order.status = OrderStatus.RTO_DAMAGED_RESHIPPED
2058
        order.statusDescription = "Order Reshipped Received Damaged"
2059
        session.commit()
2060
    elif order.status == OrderStatus.RTO_LOST_IN_TRANSIT:
2061
        new_order = __clone_order(order, False, order.cod)
2062
        order.status = OrderStatus.RTO_LOST_IN_TRANSIT_RESHIPPED
2063
        order.statusDescription = "Order Reshipped Lost In Transit"
2064
        session.commit()
4662 rajveer 2065
    elif order.status == OrderStatus.LOST_IN_TRANSIT:
2066
        new_order = __clone_order(order, False, order.cod)
2067
        order.status = OrderStatus.LOST_IN_TRANSIT_RESHIPPED
2068
        order.statusDescription = "Order Reshipped Lost In Transit"
2069
        session.commit()
2628 chandransh 2070
    elif order.status == OrderStatus.DOA_CERT_INVALID:
4220 rajveer 2071
        new_order = __clone_order(order, True, False)
2628 chandransh 2072
        order.status = OrderStatus.DOA_INVALID_RESHIPPED
2073
        order.statusDescription = "Order Reshipped"
2074
        session.commit()
5620 mandeep.dh 2075
        warehouseClient = WarehouseClient().get_client()
2076
        inventoryItem = warehouseClient.getInventoryItemFromOrder(order_id)
2077
        warehouseClient.scanForOrder(inventoryItem, ScanType.SALE, new_order.lineitems[0].quantity, new_order.id, new_order.fulfilmentWarehouseId, new_order.warehouse_id)
2628 chandransh 2078
    elif order.status == OrderStatus.DOA_CERT_VALID:
4220 rajveer 2079
        new_order = __clone_order(order, False, False)
2628 chandransh 2080
        ret_order = __create_return_order(order)
4452 rajveer 2081
        order.status = OrderStatus.DOA_VALID_RESHIPPED
2628 chandransh 2082
        order.statusDescription = "Order Reshipped"
2083
        session.commit()
4479 rajveer 2084
    elif order.status == OrderStatus.DOA_RECEIVED_DAMAGED:
2085
        new_order = __clone_order(order, False, False)
2086
        ret_order = __create_return_order(order)
2087
        order.status = OrderStatus.DOA_RESHIPPED_RCVD_DAMAGED
2088
        order.statusDescription = "Order Reshipped Received Damaged"
2089
        session.commit()
2090
    elif order.status == OrderStatus.DOA_LOST_IN_TRANSIT:
2091
        new_order = __clone_order(order, False, False)
2092
        order.status = OrderStatus.DOA_RESHIPPED_LOST_IN_TRANSIT
2093
        order.statusDescription = "Order Reshipped Lost In Transit"
2094
        session.commit()
4614 rajveer 2095
    elif order.status == OrderStatus.RET_PRODUCT_USABLE:
5620 mandeep.dh 2096
        new_order = __clone_order(order, False, False)
4614 rajveer 2097
        order.status = OrderStatus.RET_PRODUCT_USABLE_RESHIPPED
4495 rajveer 2098
        order.statusDescription = "Order Reshipped"
2099
        session.commit()
4614 rajveer 2100
    elif order.status == OrderStatus.RET_PRODUCT_UNUSABLE:
4495 rajveer 2101
        new_order = __clone_order(order, False, False)
2102
        ret_order = __create_return_order(order)
4614 rajveer 2103
        order.status = OrderStatus.RET_PRODUCT_UNUSABLE_RESHIPPED
4495 rajveer 2104
        order.statusDescription = "Order Reshipped"
2105
        session.commit()
2106
    elif order.status == OrderStatus.RET_RECEIVED_DAMAGED:
2107
        new_order = __clone_order(order, False, False)
2108
        ret_order = __create_return_order(order)
2109
        order.status = OrderStatus.RET_RESHIPPED_RCVD_DAMAGED
2110
        order.statusDescription = "Order Reshipped Received Damaged"
2111
        session.commit()
2112
    elif order.status == OrderStatus.RET_LOST_IN_TRANSIT:
2113
        new_order = __clone_order(order, False, False)
2114
        order.status = OrderStatus.RET_RESHIPPED_LOST_IN_TRANSIT
2115
        order.statusDescription = "Order Reshipped Lost In Transit"
2116
        session.commit()
2628 chandransh 2117
    else:
2118
        raise TransactionServiceException(112, "This order can't be reshipped")
2119
 
2764 chandransh 2120
    order.reship_timestamp = datetime.datetime.now()
2628 chandransh 2121
    order.new_order_id = new_order.id
2122
    session.commit()
2123
 
2124
    return new_order.id
2125
 
3226 chandransh 2126
def refund_order(order_id, refunded_by, reason):
2628 chandransh 2127
    """
4484 rajveer 2128
    If the order is in RTO_RECEIVED_PRESTINE, DOA_CERT_VALID or DOA_CERT_INVALID state, it does the following:
2628 chandransh 2129
        1. Creates a refund request for batch processing.
2130
        2. Creates a return order for the warehouse executive to return the shipped material.
4484 rajveer 2131
        3. Marks the current order as RTO_REFUNDED, DOA_VALID_REFUNDED or DOA_INVALID_REFUNDED final states.
2628 chandransh 2132
 
2133
    If the order is in SUBMITTED_FOR_PROCESSING or INVENTORY_LOW state, it does the following:
2134
        1. Creates a refund request for batch processing.
3226 chandransh 2135
        2. Cancels the reservation of the item in the warehouse.
2136
        3. Marks the current order as the REFUNDED final state.
2628 chandransh 2137
 
3226 chandransh 2138
    For all COD orders, if the order is in INIT, SUBMITTED_FOR_PROCESSING or INVENTORY_LOW state, it does the following:
2139
        1. Cancels the reservation of the item in the warehouse.
2140
        2. Marks the current order as CANCELED.
2141
 
2142
    In all cases, it updates the reason for cancellation or refund and the person who performed the action.
2143
 
2144
    Returns True if it is successful, False otherwise.
2628 chandransh 2145
 
2146
    Throws an exception if the order with the given id couldn't be found.
2147
 
2148
    Parameters:
3226 chandransh 2149
     - order_id
2150
     - refunded_by
2151
     - reason
2628 chandransh 2152
    """
3414 chandransh 2153
    logging.info("Refunding order id: " + str(order_id))
2628 chandransh 2154
    order = get_order(order_id)
3186 chandransh 2155
 
3226 chandransh 2156
    if order.cod:
3414 chandransh 2157
        logging.info("Refunding COD order with status " + str(order.status))
4662 rajveer 2158
        status_transition = { OrderStatus.LOST_IN_TRANSIT : OrderStatus.LOST_IN_TRANSIT_REFUNDED,
2159
                     OrderStatus.RTO_RECEIVED_PRESTINE : OrderStatus.RTO_REFUNDED,
4486 rajveer 2160
                     OrderStatus.RTO_RECEIVED_DAMAGED : OrderStatus.RTO_DAMAGED_REFUNDED,
2161
                     OrderStatus.RTO_LOST_IN_TRANSIT : OrderStatus.RTO_LOST_IN_TRANSIT_REFUNDED,
3366 chandransh 2162
                     OrderStatus.DOA_CERT_INVALID : OrderStatus.DOA_INVALID_REFUNDED,
2163
                     OrderStatus.DOA_CERT_VALID : OrderStatus.DOA_VALID_REFUNDED,
4479 rajveer 2164
                     OrderStatus.DOA_RECEIVED_DAMAGED : OrderStatus.DOA_REFUNDED_RCVD_DAMAGED,
2165
                     OrderStatus.DOA_LOST_IN_TRANSIT : OrderStatus.DOA_REFUNDED_LOST_IN_TRANSIT,
4495 rajveer 2166
                     OrderStatus.RET_PRODUCT_UNUSABLE : OrderStatus.RET_PRODUCT_UNUSABLE_REFUNDED,
2167
                     OrderStatus.RET_PRODUCT_USABLE : OrderStatus.RET_PRODUCT_USABLE_REFUNDED,
2168
                     OrderStatus.RET_RECEIVED_DAMAGED : OrderStatus.RET_REFUNDED_RCVD_DAMAGED,
2169
                     OrderStatus.RET_LOST_IN_TRANSIT : OrderStatus.RET_REFUNDED_LOST_IN_TRANSIT,
4682 rajveer 2170
                     OrderStatus.COD_VERIFICATION_PENDING : OrderStatus.COD_VERIFICATION_FAILED,
2171
                     OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2172
                     OrderStatus.INVENTORY_LOW : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2173
                     OrderStatus.LOW_INV_PO_RAISED : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2174
                     OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2175
                     OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2176
                     OrderStatus.ACCEPTED : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2177
                     OrderStatus.BILLED : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
4247 rajveer 2178
                     OrderStatus.CANCEL_REQUEST_CONFIRMED : OrderStatus.CANCELLED_ON_CUSTOMER_REQUEST
3366 chandransh 2179
                     }
2180
        if order.status not in status_transition.keys():
2181
            raise TransactionServiceException(114, "This order can't be refunded")
2182
 
5199 mandeep.dh 2183
        if order.status in [OrderStatus.COD_VERIFICATION_PENDING, OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT, OrderStatus.ACCEPTED]:
3226 chandransh 2184
            __update_inventory_reservation(order)
2185
            order.statusDescription = "Order Cancelled"
3415 chandransh 2186
        elif order.status == OrderStatus.BILLED:
3413 chandransh 2187
            __create_return_order(order)
2188
            order.statusDescription = "Order Cancelled"
4486 rajveer 2189
        elif order.status in [OrderStatus.RTO_RECEIVED_PRESTINE, OrderStatus.RTO_RECEIVED_DAMAGED, OrderStatus.RTO_LOST_IN_TRANSIT]:
5386 phani.kuma 2190
            if order.status != OrderStatus.RTO_LOST_IN_TRANSIT:
2191
                __create_return_order(order)
4486 rajveer 2192
            order.statusDescription = "RTO Refunded"
4662 rajveer 2193
        elif order.status in [OrderStatus.LOST_IN_TRANSIT]:
5386 phani.kuma 2194
            #__create_return_order(order)
4662 rajveer 2195
            order.statusDescription = "Lost in Transit Refunded"
4479 rajveer 2196
        elif order.status in [OrderStatus.DOA_CERT_INVALID, OrderStatus.DOA_CERT_VALID, OrderStatus.DOA_RECEIVED_DAMAGED, OrderStatus.DOA_LOST_IN_TRANSIT] :
5386 phani.kuma 2197
            if order.status != OrderStatus.DOA_LOST_IN_TRANSIT:
2198
                __create_return_order(order)
3226 chandransh 2199
            __create_refund(order)
4495 rajveer 2200
            order.statusDescription = "DOA Refunded"
2201
        elif order.status in [OrderStatus.RET_PRODUCT_UNUSABLE, OrderStatus.RET_PRODUCT_USABLE, OrderStatus.RET_RECEIVED_DAMAGED, OrderStatus.RET_LOST_IN_TRANSIT] :
5386 phani.kuma 2202
            if order.status != OrderStatus.RET_LOST_IN_TRANSIT:
2203
                __create_return_order(order)
4495 rajveer 2204
            __create_refund(order)
2205
            order.statusDescription = "Return Refunded"
4247 rajveer 2206
        elif order.status == OrderStatus.CANCEL_REQUEST_CONFIRMED:
5199 mandeep.dh 2207
            if order.previousStatus in [OrderStatus.COD_VERIFICATION_PENDING, OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT, OrderStatus.ACCEPTED]:
4247 rajveer 2208
                __update_inventory_reservation(order)
2209
                order.statusDescription = "Order Cancelled on customer request"
2210
            elif order.previousStatus == OrderStatus.BILLED:
2211
                __create_return_order(order)
2212
                order.statusDescription = "Order Cancelled on customer request"
3226 chandransh 2213
    else:
4662 rajveer 2214
        status_transition = {OrderStatus.LOST_IN_TRANSIT : OrderStatus.LOST_IN_TRANSIT_REFUNDED,
2215
                     OrderStatus.RTO_RECEIVED_PRESTINE : OrderStatus.RTO_REFUNDED,
4486 rajveer 2216
                     OrderStatus.RTO_RECEIVED_DAMAGED : OrderStatus.RTO_DAMAGED_REFUNDED,
2217
                     OrderStatus.RTO_LOST_IN_TRANSIT : OrderStatus.RTO_LOST_IN_TRANSIT_REFUNDED,
3366 chandransh 2218
                     OrderStatus.DOA_CERT_INVALID : OrderStatus.DOA_INVALID_REFUNDED,
2219
                     OrderStatus.DOA_CERT_VALID : OrderStatus.DOA_VALID_REFUNDED,
4479 rajveer 2220
                     OrderStatus.DOA_RECEIVED_DAMAGED : OrderStatus.DOA_REFUNDED_RCVD_DAMAGED,
2221
                     OrderStatus.DOA_LOST_IN_TRANSIT : OrderStatus.DOA_REFUNDED_LOST_IN_TRANSIT,
4495 rajveer 2222
                     OrderStatus.RET_PRODUCT_UNUSABLE : OrderStatus.RET_PRODUCT_UNUSABLE_REFUNDED,
2223
                     OrderStatus.RET_PRODUCT_USABLE : OrderStatus.RET_PRODUCT_USABLE_REFUNDED,
2224
                     OrderStatus.RET_RECEIVED_DAMAGED : OrderStatus.RET_REFUNDED_RCVD_DAMAGED,
2225
                     OrderStatus.RET_LOST_IN_TRANSIT : OrderStatus.RET_REFUNDED_LOST_IN_TRANSIT,
4682 rajveer 2226
                     OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2227
                     OrderStatus.INVENTORY_LOW : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2228
                     OrderStatus.LOW_INV_PO_RAISED : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2229
                     OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2230
                     OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2231
                     OrderStatus.ACCEPTED : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
2232
                     OrderStatus.BILLED : OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY,
4247 rajveer 2233
                     OrderStatus.CANCEL_REQUEST_CONFIRMED : OrderStatus.CANCELLED_ON_CUSTOMER_REQUEST,
2234
                     OrderStatus.PAYMENT_FLAGGED : OrderStatus.PAYMENT_FLAGGED_DENIED
3366 chandransh 2235
                     }
2236
        if order.status not in status_transition.keys():
2237
            raise TransactionServiceException(114, "This order can't be refunded")
2238
 
4495 rajveer 2239
        if order.status in [OrderStatus.RTO_RECEIVED_PRESTINE, OrderStatus.RTO_RECEIVED_DAMAGED, OrderStatus.RTO_LOST_IN_TRANSIT] :
5386 phani.kuma 2240
            if order.status != OrderStatus.RTO_LOST_IN_TRANSIT:
2241
                __create_return_order(order)
3226 chandransh 2242
            __create_refund(order)
4495 rajveer 2243
            order.statusDescription = "RTO Refunded"
4662 rajveer 2244
        elif order.status in [OrderStatus.LOST_IN_TRANSIT]:
5386 phani.kuma 2245
            #__create_return_order(order)
4662 rajveer 2246
            __create_refund(order)
2247
            order.statusDescription = "Lost in Transit Refunded"
4495 rajveer 2248
        elif order.status in [OrderStatus.DOA_CERT_INVALID, OrderStatus.DOA_CERT_VALID, OrderStatus.DOA_RECEIVED_DAMAGED, OrderStatus.DOA_LOST_IN_TRANSIT] :
5386 phani.kuma 2249
            if order.status != OrderStatus.DOA_LOST_IN_TRANSIT:
2250
                __create_return_order(order)
4495 rajveer 2251
            __create_refund(order)
2252
            order.statusDescription = "DOA Refunded"
2253
        elif order.status in [OrderStatus.RET_PRODUCT_UNUSABLE, OrderStatus.RET_PRODUCT_USABLE, OrderStatus.RET_RECEIVED_DAMAGED, OrderStatus.RET_LOST_IN_TRANSIT] :
5386 phani.kuma 2254
            if order.status != OrderStatus.RET_LOST_IN_TRANSIT:
2255
                __create_return_order(order)
4495 rajveer 2256
            __create_refund(order)
2257
            order.statusDescription = "Return Refunded"
5199 mandeep.dh 2258
        elif order.status in [OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT, OrderStatus.ACCEPTED]:
3226 chandransh 2259
            __update_inventory_reservation(order)
2260
            order.statusDescription = "Order Refunded"
4247 rajveer 2261
        elif order.status == OrderStatus.CANCEL_REQUEST_CONFIRMED:
5272 mandeep.dh 2262
            if order.previousStatus in [OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT, OrderStatus.PAYMENT_FLAGGED, OrderStatus.ACCEPTED]:
4672 rajveer 2263
                if order.previousStatus == OrderStatus.ACCEPTED:
2264
                    __create_refund(order)
4247 rajveer 2265
                __update_inventory_reservation(order)
2266
                order.statusDescription = "Order Cancelled on customer request"
2267
            elif order.previousStatus == OrderStatus.BILLED:
2268
                __create_return_order(order)
2269
                __create_refund(order)
2270
                order.statusDescription = "Order Cancelled on customer request"
2271
        elif order.status == OrderStatus.PAYMENT_FLAGGED:
2272
            __update_inventory_reservation(order)
2273
            order.statusDescription = "Order Cancelled due to payment flagged"
3195 chandransh 2274
 
4873 mandeep.dh 2275
    # For orders that are cancelled after being billed, we need to scan in the scanned out
2276
    # inventory item and change availability accordingly
5944 mandeep.dh 2277
    inventoryClient = InventoryClient().get_client()
2278
    warehouse = inventoryClient.getWarehouse(order.warehouse_id)
5110 mandeep.dh 2279
    if warehouse.billingType == BillingType.OURS:
4873 mandeep.dh 2280
        if order.status in [OrderStatus.BILLED, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_FROM_WH] or (order.status == OrderStatus.CANCEL_REQUEST_CONFIRMED and order.previousStatus in [OrderStatus.BILLED, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_FROM_WH]):
2281
            lineitem = order.lineitems[0]
5944 mandeep.dh 2282
            catalogClient = CatalogClient().get_client()
5177 mandeep.dh 2283
            item = catalogClient.getItem(lineitem.item_id)
5361 mandeep.dh 2284
            warehouseClient = WarehouseClient().get_client()
5177 mandeep.dh 2285
            if ItemType.SERIALIZED == item.type:
2286
                warehouseClient.scanSerializedItemForOrder(lineitem.serial_number, ScanType.SALE_RET, order.id, order.fulfilmentWarehouseId, lineitem.quantity, order.warehouse_id)
5361 mandeep.dh 2287
            else:
2288
                warehouseClient.scanForOrder(None, ScanType.SALE_RET, lineitem.quantity, order.id, order.fulfilmentWarehouseId, order.warehouse_id)
4873 mandeep.dh 2289
 
3195 chandransh 2290
    order.status = status_transition[order.status]
2291
    order.refund_timestamp = datetime.datetime.now()
3226 chandransh 2292
    order.refunded_by = refunded_by
2293
    order.refund_reason = reason
4873 mandeep.dh 2294
 
2628 chandransh 2295
    session.commit()
2296
    return True
2297
 
3186 chandransh 2298
def __update_inventory_reservation(order):
3191 chandransh 2299
    '''
2300
    Reduce the reservation count for all line items of the given order.
2301
    '''
3186 chandransh 2302
    try:
5944 mandeep.dh 2303
        inventoryClient = InventoryClient().get_client()
3186 chandransh 2304
        for lineitem in order.lineitems:
5966 rajveer 2305
            inventoryClient.reduceReservationCount(lineitem.item_id, order.fulfilmentWarehouseId, sourceId, order.id, lineitem.quantity)
3186 chandransh 2306
    except:
2307
        print "Unable to reduce reservation count"
2308
 
4220 rajveer 2309
def __clone_order(order, should_copy_billing_info, is_cod):
2628 chandransh 2310
    current_time = datetime.datetime.now()
2311
    new_order = Order()
2312
    new_order.customer_id = order.customer_id
2313
    new_order.customer_name = order.customer_name
2314
    new_order.customer_city = order.customer_city
2315
    new_order.customer_state = order.customer_state
2316
    new_order.customer_mobilenumber = order.customer_mobilenumber
2317
    new_order.customer_pincode = order.customer_pincode
2318
    new_order.customer_address1 = order.customer_address1
2319
    new_order.customer_address2 = order.customer_address2
2320
    new_order.customer_email = order.customer_email
2321
    new_order.total_amount = order.total_amount
6318 rajveer 2322
    new_order.gvAmount = order.gvAmount
2628 chandransh 2323
    new_order.total_weight = order.total_weight
2324
    new_order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
2325
    new_order.statusDescription = 'Submitted for Processing'
2326
    new_order.created_timestamp = current_time
5062 varun.gupt 2327
    new_order.originalOrderId = order.originalOrderId if order.originalOrderId is not None else order.id 
2628 chandransh 2328
 
2329
    new_order.transaction = order.transaction
4220 rajveer 2330
    new_order.cod = is_cod
5527 anupam.sin 2331
    new_order.orderType = order.orderType
2628 chandransh 2332
 
2333
    for line_item in order.lineitems:
2334
        litem = LineItem()
2335
        litem.item_id = line_item.item_id
2336
        litem.productGroup = line_item.productGroup
2337
        litem.brand = line_item.brand
2338
        if line_item.model_number:
2339
            litem.model_number = line_item.model_number
2340
        if line_item.model_name:    
2341
            litem.model_name = line_item.model_name
2342
        if line_item.color:
2343
            litem.color = line_item.color
2344
        if line_item.extra_info:
2345
            litem.extra_info = line_item.extra_info
2346
        litem.quantity = line_item.quantity
2347
        litem.unit_price = line_item.unit_price
2348
        litem.unit_weight = line_item.unit_weight
2349
        litem.total_price = line_item.total_price
2350
        litem.total_weight = line_item.total_weight
2351
        litem.transfer_price = line_item.transfer_price
4840 mandeep.dh 2352
        litem.dealText = line_item.dealText
6423 amit.gupta 2353
        litem.vatRate = line_item.vatRate
2628 chandransh 2354
        litem.order = new_order
2355
 
2356
    logistics_client = LogisticsClient().get_client()
2357
    item_id = new_order.lineitems[0].item_id
5785 mandeep.dh 2358
    logistics_info = logistics_client.getLogisticsInfo(new_order.customer_pincode, item_id, is_cod, PickUpType.COURIER) #TODO: We should be able to pass another flag to suggest ignoring the inventory situation.
2628 chandransh 2359
    logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
2360
    new_order.warehouse_id = logistics_info.warehouseId
5110 mandeep.dh 2361
    new_order.fulfilmentWarehouseId = logistics_info.fulfilmentWarehouseId
2628 chandransh 2362
    new_order.logistics_provider_id = logistics_info.providerId
2363
    new_order.airwaybill_no = logistics_info.airway_billno
2364
    new_order.tracking_id = new_order.airwaybill_no
2365
    new_order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
3986 chandransh 2366
    new_order.promised_delivery_time = new_order.expected_delivery_time
4299 rajveer 2367
 
2628 chandransh 2368
    if should_copy_billing_info:
5388 phani.kuma 2369
        new_order.vendorId = order.vendorId
2628 chandransh 2370
        new_order.invoice_number = order.invoice_number
2371
        new_order.billed_by = order.billed_by
2372
        new_order.warehouse_id = order.warehouse_id
5110 mandeep.dh 2373
        new_order.fulfilmentWarehouseId = order.fulfilmentWarehouseId
5620 mandeep.dh 2374
        new_order.jacket_number = order.jacket_number
2375
        new_order.lineitems[0].serial_number = order.lineitems[0].serial_number
2376
        new_order.lineitems[0].item_number   = order.lineitems[0].item_number
2628 chandransh 2377
        new_order.accepted_timestamp = current_time
2378
        new_order.billing_timestamp = current_time    
2379
        new_order.status = OrderStatus.BILLED
2380
        new_order.statusDescription = 'Order Billed'
5966 rajveer 2381
    else:
2382
        session.commit()       
5944 mandeep.dh 2383
        inventoryClient = InventoryClient().get_client()
5966 rajveer 2384
        inventoryClient.reserveItemInWarehouse(item_id, logistics_info.fulfilmentWarehouseId, sourceId, new_order.id, to_java_date(new_order.created_timestamp), to_java_date(new_order.promised_shipping_time), new_order.lineitems[0].quantity)
2628 chandransh 2385
 
2386
    return new_order
2387
 
2388
def __create_return_order(order):
2687 chandransh 2389
    ret_order = ReturnOrder(order)
2628 chandransh 2390
    return ret_order
2391
 
2392
def __create_refund(order):
4078 chandransh 2393
    payment_client = PaymentClient().get_client()
2394
    payment_client.createRefund(order.id, order.transaction.id, order.total_amount)
2747 chandransh 2395
    return
2628 chandransh 2396
 
2536 chandransh 2397
def __get_order_address(order):
2398
    address = order.customer_name + "\n" + order.customer_address1 + "\n"
2399
    if order.customer_address2:
2400
        address = address + order.customer_address2 + "\n"
2401
    address = address + order.customer_city + "\n"
2402
    address = address + order.customer_state + "\n"
2403
    address = address + "PIN " + order.customer_pincode + "\n"
2404
    address = address + "Phone: " + order.customer_mobilenumber
2405
    return address
2406
 
5767 rajveer 2407
def __get_store_address(store):
2408
    address = store.name + "\n" + store.line1 + "\n"
2409
    if store.line2:
2410
        address = address + store.line2 + "\n"
2411
    address = address + store.city + "\n"
2412
    address = address + store.state + "\n"
5802 rajveer 2413
    address = address + "PIN - " + store.pin + "\n"
5793 rajveer 2414
    if store.phone:
5802 rajveer 2415
        address = address + "Phone - " + store.phone + "\n"
5855 rajveer 2416
    address = address +  "Store timings - 11AM to 8PM (7 days)\n"
5767 rajveer 2417
    return address
2418
 
5793 rajveer 2419
def __get_store_address_html(store):
2420
    address = store.name + "<br>" + store.line1 + "<br>"
2421
    if store.line2:
2422
        address = address + store.line2 + "<br>"
2423
    address = address + store.city + "<br>"
2424
    address = address + store.state + "<br>"
5802 rajveer 2425
    address = address + "PIN - " + store.pin + "<br>"
5793 rajveer 2426
    if store.phone:
5802 rajveer 2427
        address = address + "Phone - " + store.phone + "<br>"
5855 rajveer 2428
    address = address +  "Store timings - 7 days: 11AM to 8PM  (7 days)<br>"
5793 rajveer 2429
    return address
2430
 
4815 phani.kuma 2431
def __generate_return_advice(order, warehouse, provider, filename):
2432
    if warehouse.id == 7:
2433
        executive = 'Amit Kumar'
2434
    else:
4966 phani.kuma 2435
        executive = 'Dinesh Kumar'
2536 chandransh 2436
    pdf = Canvas(filename)
2437
    pdf.setFont('Times-Bold',16)
2438
    address_text = pdf.beginText(inch, PAGE_HEIGHT - inch)
2439
    address_text.textLine("To")
4815 phani.kuma 2440
    address_text.textLine(executive)
2536 chandransh 2441
    for line in warehouse.location.split("\n"):
2442
        address_text.textLine(line)
2443
    address_text.textLine("PIN " + warehouse.pincode)
2444
    pdf.drawText(address_text)
2445
 
2446
    pdf.setFont('Times-Roman',12)
2447
    order_text = pdf.beginText(inch, PAGE_HEIGHT - 4 * inch)
3051 chandransh 2448
    order_text.textLine("Pickup CODE: " + provider.details[DeliveryType.PREPAID].accountNo)
2536 chandransh 2449
    lineitem = order.lineitems[0] 
4234 rajveer 2450
    order_text.textLine("Content : " + lineitem.brand + " " + str(lineitem.model_number) + " " + str(lineitem.color))
2536 chandransh 2451
    order_text.textLine("Declared Value: Rs. " + str(order.total_amount))
2452
    order_text.textLine("Order ID : " + str(order.id))
2453
    pdf.drawText(order_text)
2454
    pdf.showPage()
2455
    pdf.save()
2456
    return filename
2457
 
5806 rajveer 2458
def __push_collection_to_hotspot(order, type):
2459
    try:
2460
        logistics_client = LogisticsClient().get_client()
2461
        store = logistics_client.getPickupStore(order.pickupStoreId)
2462
        store_string = "<store>" + store.hotspotId + "</store>"
2463
        order_string = "<orderno>" + str(order.id) + "</orderno>"
2464
        collection_type = "<Type>" + type + "</Type>"
5833 rajveer 2465
        payment_mode = get_order_attribute_value(order.id, "PAYMENT_MODE")
5806 rajveer 2466
        if type == "SALE":
6318 rajveer 2467
            amount = int(order.total_amount-order.gvAmount)
5806 rajveer 2468
            ts = order.delivery_timestamp
2469
        if type == "SALE RETURN":
6318 rajveer 2470
            amount = -int(order.total_amount-order.gvAmount)
5806 rajveer 2471
            ts = order.refund_timestamp
2472
        if payment_mode == "Cash":
2473
            cash_string = "<cash>" + str(amount) + "</cash>"
2474
            card_string = "<hdfc>0</hdfc>"
2475
        if payment_mode == "Card":
2476
            cash_string = "<cash>0</cash>"
2477
            card_string = "<hdfc>" + str(amount) + "</hdfc>"
2478
        date_string = "<date>" + ts.strftime("%Y%m%d") + "</date>"
2479
        time_string = "<time>" + ts.strftime("%H%M%S") + "</time>"
2480
 
2481
        conn = httplib.HTTPConnection("182.71.104.182")
2482
        XML="""
2483
        <soap:Envelope xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:soap="http://schemas.xmlsoap.org/soap/envelope/">
2484
          <soap:Body>
2485
            <SaholicDataTransfer xmlns="http://tempuri.org/">
2486
        """
2487
 
2488
        XML = XML + store_string + order_string + collection_type + cash_string + card_string + date_string + time_string 
2489
 
2490
        footer = """
2491
            </SaholicDataTransfer>
2492
          </soap:Body>
2493
        </soap:Envelope>
2494
        """
2495
        XML = XML + footer
2496
 
2497
        params = urllib.urlencode({'op': 'SaholicDataTransfer'})
2498
        headers = { "Content-type": "text/xml", "Content-Length": "%d" % len(XML)}
2499
        conn.request("POST", "/loaddetect/Service.asmx?"+params, "", headers)
2500
        conn.send(XML)
2501
        response = conn.getresponse()
2502
        print response.status, response.reason
2503
        resp = response.read()
2504
        conn.close()
2505
        print resp
2506
        if "Saved Successfully" in resp:
2507
            return True
2508
        elif "Error in Saving Data" in resp:
2509
            return False
2510
        else: 
2511
            return False
2512
    except Exception as e:
2513
        print e
2514
        return False
2515
 
2697 chandransh 2516
def get_return_orders(warehouse_id, from_date, to_date):
2517
    query = ReturnOrder.query.filter_by(processedStatus = False)
2518
    if warehouse_id != -1:
2519
        query = query.filter_by(warehouseId = warehouse_id)
2520
    if from_date:
4118 chandransh 2521
        query = query.filter(ReturnOrder.createdAt >from_date)
2697 chandransh 2522
    if to_date:
4118 chandransh 2523
        query = query.filter(ReturnOrder.createdAt < to_date)
2697 chandransh 2524
    return [return_order.to_thrift_object() for return_order in query.all()]
2525
 
5481 phani.kuma 2526
def get_all_return_orders(onlyNotProcessed, from_date, to_date):
2527
    query = ReturnOrder.query
2528
    if onlyNotProcessed:
2529
        query = query.filter_by(processedStatus = False)
2530
    if from_date:
2531
        query = query.filter(ReturnOrder.createdAt >= from_date)
2532
    if to_date:
2533
        query = query.filter(ReturnOrder.createdAt <= to_date)
2534
    return [return_order.to_thrift_object() for return_order in query.all()]
2535
 
2700 chandransh 2536
def get_return_order(id):
2537
    ret_order = ReturnOrder.get_by(orderId = id)
2538
    if ret_order is None:
2539
        raise TransactionServiceException(116, "No return order found for the given order id")
2540
    return ret_order.to_thrift_object()
2541
 
2697 chandransh 2542
def process_return(ret_order_id):
2543
    ret_order = ReturnOrder.get_by(orderId = ret_order_id)
2544
    if ret_order is None:
2545
        raise TransactionServiceException(116, "No return order found for the given order id")
2546
    ret_order.processedStatus = True
2701 chandransh 2547
    ret_order.processedAt = datetime.datetime.now()
2697 chandransh 2548
    session.commit()
2549
 
4757 mandeep.dh 2550
def update_orders_as_PORaised(itemIdQuantityMap, purchaseOrderId, warehouseId):
2551
    """
2552
    Updates orders as PO raised. Also updates purchase order id in orders. Pass a map of items mapped to
2553
    the quantities for which the PO is raised.
2554
 
2555
    Parameters:
2556
     - itemIdQuantityMap
2557
     - purchaseOrderId
2558
     - warehouseId
2559
    """
3451 chandransh 2560
    query = Order.query.with_lockmode("update").filter_by(warehouse_id=warehouseId)
4757 mandeep.dh 2561
    query = query.filter(or_(Order.status == OrderStatus.SUBMITTED_FOR_PROCESSING, Order.status == OrderStatus.INVENTORY_LOW))
2819 chandransh 2562
    query = query.filter_by(purchase_order_id=None)
2563
    query = query.order_by(Order.created_timestamp)
2564
    pending_orders = query.all()
4586 mandeep.dh 2565
 
2566
    for order in pending_orders:
4757 mandeep.dh 2567
        lineitem = order.lineitems[0]
2568
        if itemIdQuantityMap.has_key(lineitem.item_id):
2569
            if itemIdQuantityMap[lineitem.item_id] >= lineitem.quantity:
2570
                order.status = OrderStatus.LOW_INV_PO_RAISED
2571
                order.statusDescription = 'In Process'
2572
                order.purchase_order_id = purchaseOrderId
2573
                itemIdQuantityMap[lineitem.item_id] -= lineitem.quantity
4434 mandeep.dh 2574
 
2819 chandransh 2575
    session.commit()
2576
 
3451 chandransh 2577
def update_weight(order_id, weight):
3469 chandransh 2578
    '''
2579
    Update the weight of order
2580
    '''
3451 chandransh 2581
    order = get_order(order_id)
2582
    order.total_weight = weight
3469 chandransh 2583
    lineitem = order.lineitems[0]
2584
    lineitem.total_weight = weight
2585
    lineitem.unit_weight = weight
3451 chandransh 2586
 
3469 chandransh 2587
    session.commit()
2588
    return order
2589
 
2590
def change_product(order_id, item_id):
2591
    '''
2592
    Ship a product of a different color to the customer.
2593
    '''
2594
    order = get_order(order_id)
4662 rajveer 2595
    if order.status not in [OrderStatus.COD_VERIFICATION_PENDING, OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT]:
3469 chandransh 2596
        raise TransactionServiceException(120, "This order has already been processed. Please seek help from engineering.")
2597
 
2598
    lineitem = order.lineitems[0]
3451 chandransh 2599
 
3469 chandransh 2600
    catalog_client = CatalogClient().get_client()
2601
 
2602
    old_item = catalog_client.getItem(lineitem.item_id)
2603
    item = catalog_client.getItem(item_id)
2604
    if old_item.catalogItemId != item.catalogItemId:
2605
        raise TransactionServiceException(121, "You can't ship a different item. You can only ship the same item in a different color.")
2606
 
5944 mandeep.dh 2607
    inventoryClient = InventoryClient().get_client()
5966 rajveer 2608
    inventoryClient.reserveItemInWarehouse(item_id, order.fulfilmentWarehouseId, sourceId, order.id, to_java_date(order.created_timestamp), to_java_date(order.promised_shipping_time), order.lineitems[0].quantity)
5110 mandeep.dh 2609
    __update_inventory_reservation(order)
3469 chandransh 2610
    #TODO: Check that this new item has the same price
2611
 
2612
    lineitem.item_id = item_id
2613
    lineitem.color = item.color
2614
 
3451 chandransh 2615
    session.commit()
3469 chandransh 2616
 
2617
    order = get_order(order_id)
3451 chandransh 2618
    return order
3469 chandransh 2619
 
2620
def change_warehouse(order_id, warehouse_id):
2621
    '''
2622
    Update the warehouse which will be used to fulfil this order.
2623
    '''
2624
    order = get_order(order_id)
6019 rajveer 2625
    if order.status not in [OrderStatus.COD_VERIFICATION_PENDING, OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT, OrderStatus.ACCEPTED]:
3469 chandransh 2626
        raise TransactionServiceException(117, "This order has already been processed. Please seek help from engineering.")
2627
 
2628
    if warehouse_id == order.warehouse_id:
2629
        raise TransactionServiceException(118, "You have selected the current warehouse again. Nothing to do")
2630
 
2631
    lineitem = order.lineitems[0]
2632
    catalog_client = CatalogClient().get_client()
5944 mandeep.dh 2633
    inventoryClient = InventoryClient().get_client()
3469 chandransh 2634
    try:
5944 mandeep.dh 2635
        warehouse = inventoryClient.getWarehouse(warehouse_id)
3469 chandransh 2636
    except:
2637
        raise TransactionServiceException(119, "No warehouse with id" + str(warehouse_id))
5110 mandeep.dh 2638
 
5147 mandeep.dh 2639
    if warehouse.billingWarehouseId is None:
2640
        raise TransactionServiceException(119, "Shifting of warehouse allowed to warehouse with billing support only")
5110 mandeep.dh 2641
 
5620 mandeep.dh 2642
    # If One gives a billing warehouse for warehouse shift, we should assign order to the GOOD warehouse
2643
    # present within it with maximum availability
2644
    if warehouse.id == warehouse.billingWarehouseId:
5944 mandeep.dh 2645
        warehouses = inventoryClient.getWarehouses(None, InventoryType.GOOD, 0, 0, warehouse.id)
5147 mandeep.dh 2646
        if warehouses:
5944 mandeep.dh 2647
            itemInventory = inventoryClient.getItemInventoryByItemId(lineitem.item_id)
5147 mandeep.dh 2648
            warehouse = warehouses[0]
2649
            maxAvailability = 0
2650
            for goodWarehouse in warehouses:
2651
                if itemInventory.availability.has_key(goodWarehouse.id) and maxAvailability < itemInventory.availability[goodWarehouse.id]:
2652
                    warehouse = goodWarehouse
2653
                    maxAvailability = itemInventory.availability[goodWarehouse.id]
2654
 
2655
            # Use preferred vendor's warehouse if not available anywhere
2656
            if not maxAvailability:
2657
                item = catalog_client.getItem(lineitem.item_id)
2658
                if item.preferredVendor:
2659
                    for goodWarehouse in warehouses:
2660
                        if goodWarehouse.vendor.id == item.preferredVendor:
2661
                            warehouse = goodWarehouse
2662
        else:
2663
            raise TransactionServiceException(119, "Shifting of warehouse allowed to GOOD inventory type warehouse with billing support")
2664
 
6007 rajveer 2665
    inventoryClient.reserveItemInWarehouse(lineitem.item_id, warehouse.id, sourceId, order.id, to_java_date(order.created_timestamp), to_java_date(order.promised_shipping_time), lineitem.quantity)
5110 mandeep.dh 2666
    __update_inventory_reservation(order)
2667
 
2668
    order.warehouse_id = warehouse.billingWarehouseId
5147 mandeep.dh 2669
    order.fulfilmentWarehouseId = warehouse.id
3469 chandransh 2670
    session.commit()
2671
    return order
3553 chandransh 2672
 
4647 rajveer 2673
def add_delay_reason(order_id, reason, further_delay, delayReasonText):
3553 chandransh 2674
    '''
3994 chandransh 2675
    Adds a delay reason to the order and updates the expected delivery time.
3553 chandransh 2676
    '''
2677
    order = get_order(order_id)
2678
    try:
3581 chandransh 2679
        order.delay_reason = DelayReason._VALUES_TO_NAMES[reason]
4647 rajveer 2680
        order.delayReasonText = delayReasonText
4009 chandransh 2681
        actual_delivery_delay = adjust_delivery_time(order.expected_delivery_time, further_delay)
2682
        order.expected_delivery_time = order.expected_delivery_time + datetime.timedelta(days=actual_delivery_delay)
4102 chandransh 2683
 
4842 rajveer 2684
        actual_shipping_delay = adjust_delivery_time(order.expected_shipping_time, further_delay)
2685
        order.expected_shipping_time = order.expected_shipping_time + datetime.timedelta(days=actual_shipping_delay)
4102 chandransh 2686
 
3553 chandransh 2687
        session.commit()
4008 mandeep.dh 2688
 
2689
        try:
2690
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
2691
            transaction_requiring_extra_processing.category = 'DELAYED_DELIVERY'
2692
            transaction_requiring_extra_processing.transaction_id = order.transaction_id
2693
            session.commit()
2694
        except Exception as e:
4065 mandeep.dh 2695
            print 'Could not persist transaction id: ' + str(order.transaction_id) + ' for further processing due to: ' + str(e)
4008 mandeep.dh 2696
 
3553 chandransh 2697
        return True
2698
    except:
3994 chandransh 2699
        print sys.exc_info()[0]
3553 chandransh 2700
        return False
3956 chandransh 2701
 
2702
def reconcile_cod_collection(collected_amount_map, xferBy, xferTxnId, xferDate):
2703
    unprocessed_awbs = {}
2704
    for awb, amount in collected_amount_map.iteritems():
2705
        try:
2706
            order = Order.query.filter_by(airwaybill_no=awb).one()
2707
        except NoResultFound:
2708
            unprocessed_awbs[awb] = "No order was found for the given AWB: " + awb
2709
            continue
2710
        except MultipleResultsFound:
2711
            unprocessed_awbs[awb] = "Multiple orders were found for the given AWB:" + awb
2712
            continue
2713
 
2714
        if order.cod_reconciliation_timestamp:
2715
            #This order has been processed already! This may be a re-run. Let's allow other orders to be processed.
2716
            continue
2717
 
2718
        if order.status < OrderStatus.DELIVERY_SUCCESS:
2719
            #Order has not been delivered yet. No way we can receive the payment.
2720
            unprocessed_awbs[awb] = "Payment has been received for order: " + str(order.id) + " before delivery reconciliation"
2721
            continue
2722
 
4252 rajveer 2723
        ##As per ticket #743 if amount difference less than 0.5 we should reconcile
6318 rajveer 2724
        if abs(amount - (order.total_amount - order.gvAmount)) > 0.5:
3956 chandransh 2725
            #Received amount is less than or more than the order amount. Mustn't let the user proceed.
6318 rajveer 2726
            unprocessed_awbs[awb] = "Payment of Rs. " + str(amount) + " has been received against the total value of Rs. " + str(order.total_amount-order.gvAmount) + " for order: " + str(order.id)
3956 chandransh 2727
            continue
2728
 
2729
        try:
2730
            payment_client = PaymentClient().get_client()
2731
            payment_client.partiallyCapturePayment(order.transaction.id, amount, xferBy, xferTxnId, xferDate)
2732
        except Exception:
2733
            unprocessed_awbs[awb] = "We were unable to partially capture payment for order id " + str(order.id) + ", AWB: " + awb
2734
            continue
2735
 
2736
        #Payment has been recorded now. We can update the order peacefully.
2737
        order.cod_reconciliation_timestamp = datetime.datetime.now()
2738
        session.commit()
3469 chandransh 2739
 
3956 chandransh 2740
    return unprocessed_awbs
4008 mandeep.dh 2741
 
2742
def get_transactions_requiring_extra_processing(category):
4065 mandeep.dh 2743
    """
2744
    Returns the list of transaction ids that require some extra processing and
2745
    which belong to a particular category. This is currently used by CRM
2746
    application. If no such transaction ids are present, it returns an empty list.
2747
    """
2748
    query = TransactionRequiringExtraProcessing.query
4008 mandeep.dh 2749
 
4065 mandeep.dh 2750
    if category is not None:
2751
        query = query.filter_by(category = ExtraTransactionProcessingType._VALUES_TO_NAMES.get(category))
4008 mandeep.dh 2752
 
4065 mandeep.dh 2753
    return [a.transaction_id for a in query.all()]
4008 mandeep.dh 2754
 
2755
def mark_transaction_as_processed(transactionId, category):
4065 mandeep.dh 2756
    """
2757
    Marks a particular transaction as processed for a particular cateogory.
2758
    It essentially deletes the transaction id record for a particular
2759
    processing type category (if present) from DB.
2760
    This is currently used by CRM application.
2761
    """
4008 mandeep.dh 2762
    TransactionRequiringExtraProcessing.query.filter_by(transaction_id = transactionId).filter_by(category = ExtraTransactionProcessingType._VALUES_TO_NAMES.get(category)).delete()
2763
    session.commit()
2764
 
4018 chandransh 2765
def get_item_wise_risky_orders_count():
2766
    """
2767
    Returns a map containing the number of risky orders keyed by item id. A risky order
2768
    is defined as one whose shipping date is about to expire.
2769
    """
2770
    item_wise_risky_orders_count = {}
4019 chandransh 2771
    shipping_cutoff_time = (datetime.datetime.now() + datetime.timedelta(days = 1)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4018 chandransh 2772
    orders = Order.query.filter(Order.status.in_(tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.ACCEPTED]))).filter(Order.expected_shipping_time <= shipping_cutoff_time).all()
2773
    for order in orders:
2774
        for lineitem in order.lineitems:
2775
            quantity = lineitem.quantity
2776
            if item_wise_risky_orders_count.has_key(lineitem.item_id):
2777
                quantity = quantity + item_wise_risky_orders_count[lineitem.item_id]
2778
 
2779
            item_wise_risky_orders_count[lineitem.item_id] = quantity
2780
    return item_wise_risky_orders_count
2781
 
4324 mandeep.dh 2782
def update_shipment_address(order_id, address_id):
2783
    """
2784
    Updates shipment address of an order. Delivery and shipping date estimates
2785
    etc. are also updated here.
2786
 
2787
    Throws TransactionServiceException in case address change is not
2788
    possible due to certain reasons such as new pincode in address is
2789
    not serviceable etc.
2790
 
2791
    Parameters:
2792
     - orderId
2793
     - addressId
2794
    """
2795
    user_client = UserClient().get_client()
2796
    address     = user_client.getAddressById(address_id)
2797
    order       = get_order(order_id)
2798
 
4630 mandeep.dh 2799
    type = DeliveryType.PREPAID
2800
    if order.cod:
2801
        type = DeliveryType.COD
2802
 
4324 mandeep.dh 2803
    logistics_client = LogisticsClient().get_client()
5886 mandeep.dh 2804
    logistics_info = logistics_client.getLogisticsEstimation(order.lineitems[0].item_id, address.pin, type)
4324 mandeep.dh 2805
 
2806
    # Check for new estimate. Raise exception if new address is not serviceable
2807
    if logistics_info.deliveryTime == -1:
2808
        raise TransactionServiceException(1, 'Location not serviceable')
2809
 
2810
    # COD services are not available everywhere
2811
    if order.cod:
4868 rajveer 2812
        if not logistics_info.codAllowed:
4324 mandeep.dh 2813
            raise TransactionServiceException(2, 'COD service not available')
2814
 
5593 mandeep.dh 2815
    if logistics_info.providerId != order.logistics_provider_id:
5887 mandeep.dh 2816
        logistics_info = logistics_client.getLogisticsInfo(address.pin, order.lineitems[0].item_id, type, PickUpType.COURIER)
5593 mandeep.dh 2817
        order.tracking_id = logistics_info.airway_billno
2818
        order.airwaybill_no = logistics_info.airway_billno
2819
        order.logistics_provider_id = logistics_info.providerId
2820
 
4324 mandeep.dh 2821
    current_time = datetime.datetime.now()
2822
    logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
2823
    logistics_info.shippingTime = adjust_delivery_time(current_time, logistics_info.shippingTime)
2824
 
2825
    order.customer_name         = address.name
2826
    order.customer_pincode      = address.pin
2827
    order.customer_address1     = address.line1
2828
    order.customer_address2     = address.line2
2829
    order.customer_city         = address.city
2830
    order.customer_state        = address.state
2831
    order.customer_mobilenumber = address.phone
2832
 
2833
    if order.cod:
2834
        order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=COD_SHIPPING_CUTOFF_TIME, minute=0, second=0)
2835
    else:
2836
        order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
2837
 
2838
    order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
2839
    order.promised_shipping_time = order.expected_shipping_time
2840
    order.promised_delivery_time = order.expected_delivery_time 
2841
 
2842
    session.commit()
2843
 
4247 rajveer 2844
def mark_order_cancellation_request_received(orderId):
2845
    """
2846
    Mark order as cancellation request received. If customer sends request of cancellation of
2847
    a particular order, this method will be called. It will just change status of the order
2848
    depending on its current status. It also records the previous status, so that we can move
2849
    back to that status if cancellation request is denied.
2850
 
2851
    Parameters:
2852
     - orderId
2853
    """
4662 rajveer 2854
    status_transition = {OrderStatus.COD_VERIFICATION_PENDING : OrderStatus.CANCEL_REQUEST_RECEIVED,
4247 rajveer 2855
             OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.CANCEL_REQUEST_RECEIVED,
2856
             OrderStatus.INVENTORY_LOW : OrderStatus.CANCEL_REQUEST_RECEIVED,
4306 rajveer 2857
             OrderStatus.LOW_INV_PO_RAISED : OrderStatus.CANCEL_REQUEST_RECEIVED,
2858
             OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.CANCEL_REQUEST_RECEIVED,
2859
             OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.CANCEL_REQUEST_RECEIVED,
4247 rajveer 2860
             OrderStatus.ACCEPTED : OrderStatus.CANCEL_REQUEST_RECEIVED,
2861
             OrderStatus.BILLED : OrderStatus.CANCEL_REQUEST_RECEIVED,
2862
             OrderStatus.PAYMENT_FLAGGED : OrderStatus.CANCEL_REQUEST_RECEIVED
2863
             }
4662 rajveer 2864
 
2865
    order = get_order(orderId)
4247 rajveer 2866
    if order.status not in status_transition.keys():
2867
        raise TransactionServiceException(114, "This order can't be considered for cancellation")
2868
 
4662 rajveer 2869
 
4247 rajveer 2870
    order.previousStatus = order.status
2871
    order.status = OrderStatus.CANCEL_REQUEST_RECEIVED
2872
    order.statusDescription = "Cancellation request received from user"
2873
    session.commit()
2874
 
4662 rajveer 2875
def mark_order_as_lost_in_transit(orderId):
2876
    """
2877
    Parameters:
2878
     - orderId
2879
    """
2880
    order = get_order(orderId)
4910 phani.kuma 2881
    if order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST, OrderStatus.SHIPPED_TO_DESTINATION_CITY, OrderStatus.REACHED_DESTINATION_CITY, OrderStatus.FIRST_DELIVERY_ATTEMPT_MADE]:
4662 rajveer 2882
        raise TransactionServiceException(115, "This order can't be marked as lost in transit")
2883
    order.status = OrderStatus.LOST_IN_TRANSIT
2884
    order.statusDescription = "Lost in Transit"
2885
    session.commit()
2886
 
4258 rajveer 2887
def mark_transaction_as_payment_flag_removed(transactionId):
4247 rajveer 2888
    """
4258 rajveer 2889
    If we and/or payment gateway has decided to accept the payment, this method needs to be called.
2890
    Changed transaction and all orders status to payment accepted.
4247 rajveer 2891
 
2892
    Parameters:
4258 rajveer 2893
     - transactionId
4247 rajveer 2894
    """
4258 rajveer 2895
    transaction = get_transaction(transactionId)
2896
    transaction.status = TransactionStatus.AUTHORIZED
2897
    for order in transaction.orders:
2898
        order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
2899
        order.statusDescription = "Submitted to warehouse"
4247 rajveer 2900
    session.commit()
2901
 
4259 anupam.sin 2902
def refund_transaction(transactionId, refundedBy, reason):
2903
    """
2904
    This method is called when a flagged payment is deemed unserviceable and the corresponding orders
2905
    need to be cancelled
2906
 
2907
    Parameters:
2908
     - transactionId
2909
    """
2910
    transaction = get_transaction(transactionId)
2911
    transaction.status = TransactionStatus.FAILED
2912
    for order in transaction.orders:
2913
        refund_order(order.id, refundedBy, reason)
2914
    session.commit()
2915
 
4247 rajveer 2916
def mark_order_cancellation_request_denied(orderId):
2917
    """
4283 anupam.sin 2918
    If we decide to not to cancel order, we will move the order to previous status.
4247 rajveer 2919
 
2920
    Parameters:
2921
     - orderId
2922
    """
2923
    order = get_order(orderId)
2924
    order.status = order.previousStatus
2925
    order.previousStatus = None
2926
    order.statusDescription = "Cancellation request denied"
2927
    session.commit()
2928
 
2929
def mark_order_cancellation_request_confirmed(orderId):
2930
    """
2931
    If we decide to to cancel order, CRM will call this method to move the status of order to
2932
    cancellation request confirmed. After this OM will be able to cancel the order.
2933
 
2934
    Parameters:
2935
     - orderId
2936
    """
2937
    order = get_order(orderId)
2938
    order.status = OrderStatus.CANCEL_REQUEST_CONFIRMED
2939
    order.statusDescription = "Cancellation request confirmed"
2940
    session.commit()
2941
 
4285 rajveer 2942
def accept_orders_for_item_id(itemId, inventory):
2943
    """
2944
    Marks the orders as ACCEPTED for the given itemId and inventory. It also updates the accepted timestamp. If the
2945
    given order is not a COD order, it also captures the payment if the same has not been captured.
2946
 
2947
    Parameters:
2948
     - itemId
2949
     - inventory
2950
    """
4938 rajveer 2951
    orders = Order.query.filter(Order.warehouse_id == 5).filter(Order.status.in_(tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS]))).order_by(Order.expected_shipping_time).all()
4285 rajveer 2952
    for order in orders:
4301 rajveer 2953
        if inventory > 0:
2954
            lineitem = order.lineitems[0]
2955
            if itemId == lineitem.item_id:
2956
                try:
2957
                    accept_order(order.id)
2958
                    inventory = inventory - lineitem.quantity
2959
                except:
2960
                    logging.info("Unable to accept the order")
2961
        else:
2962
            order_outofstock(order.id)
4285 rajveer 2963
    return
2964
 
4607 rajveer 2965
def get_slipped_sipping_date_orders():
4662 rajveer 2966
    return Order.query.filter(Order.status.in_(tuple([OrderStatus.COD_VERIFICATION_PENDING, OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS, OrderStatus.ACCEPTED, OrderStatus.BILLED]))).filter(Order.promised_shipping_time <= datetime.datetime.now()).order_by(Order.promised_shipping_time).all()
4709 rajveer 2967
 
2968
def get_cancelled_orders(cancelDateFrom, cancelDateTo):
2969
    return Order.query.filter(Order.status == OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY).filter(Order.refund_timestamp >= cancelDateFrom).filter(Order.refund_timestamp <= cancelDateTo).all()
4607 rajveer 2970
 
4369 rajveer 2971
def mark_orders_as_po_raised(vendorId, itemId, quantity, estimate, isReminder):
2972
    if isReminder:
2973
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
2974
    else:
2975
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
2976
 
2977
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 2978
    for order in orders:
2979
        lineitem = order.lineitems[0]
2980
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
2981
            if estimate > 0:
2982
                add_delay_reason(order.id, DelayReason.OTHERS, estimate)
2983
            order.status = OrderStatus.LOW_INV_PO_RAISED
4362 rajveer 2984
            order.statusDescription = "PO Raised"
4303 rajveer 2985
            quantity = quantity - lineitem.quantity
2986
            store_inventory_action(order, itemId, HotspotAction.PO_RAISED, estimate, lineitem.quantity)
2987
            session.commit()
2988
 
4369 rajveer 2989
def mark_orders_as_reversal_initiated(vendorId, itemId, quantity, estimate, isReminder):
2990
    if isReminder:
2991
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
2992
    else:
2993
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
2994
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 2995
    for order in orders:
2996
        lineitem = order.lineitems[0]
2997
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
2998
            if estimate > 0:
2999
                add_delay_reason(order.id, DelayReason.OTHERS, estimate)
3000
            order.status = OrderStatus.LOW_INV_REVERSAL_IN_PROCESS
4362 rajveer 3001
            order.statusDescription = "Reversal Initiated"
4303 rajveer 3002
            quantity = quantity - lineitem.quantity
3003
            store_inventory_action(order, itemId, HotspotAction.REVERSAL_INITIATED, estimate, lineitem.quantity)
3004
            session.commit()
3005
 
4369 rajveer 3006
def mark_orders_as_not_available(vendorId, itemId, quantity, estimate, isReminder):
3007
    if isReminder:
3008
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
3009
    else:
3010
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
3011
 
3012
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 3013
    for order in orders:
3014
        lineitem = order.lineitems[0]
3015
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
3016
            order.status = OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT
4362 rajveer 3017
            order.statusDescription = "Not Available"
4303 rajveer 3018
            quantity = quantity - lineitem.quantity
3019
            store_inventory_action(order, itemId, HotspotAction.NOT_AVAILABLE, estimate, lineitem.quantity)
3020
            session.commit()
3021
 
3022
def store_inventory_action(order, itemId, hotspotAction, estimate, quantity):
4369 rajveer 3023
    orderInv = OrderInventory.get_by(order_id=order.id)
3024
    if not orderInv:
3025
        orderInv = OrderInventory()
3026
        orderInv.order = order
4303 rajveer 3027
    orderInv.hotspotAction = hotspotAction  
3028
    orderInv.estimate = estimate
3029
    orderInv.itemId = itemId
3030
    orderInv.timestamp = datetime.datetime.now()
3031
    orderInv.quantity = quantity
3032
 
4556 mandeep.dh 3033
'''    
4369 rajveer 3034
def mark_orders_as_timeout(vendorId):
3035
    ret_timeouts = {}
3036
    orderInvs = OrderInventory.query.all()
3037
    for orderInv in orderInvs:
3038
        if orderInv.timestamp +  timedelta(days=orderInv.estimate) < datetime.datetime.now():
3039
            if not ret_timeouts.has_key(orderInv.itemId):
3040
                timeout = TimeoutSummary()
3041
                ret_timeouts[orderInv.itemId] = timeout
3042
            else:
3043
                timeout = ret_timeouts.get(orderInv.itemId)
3044
            if orderInv.hotspotAction == HotspotAction.PO_RAISED and orderInv.order.status == OrderStatus.LOW_INV_PO_RAISED:
3045
                if timeout.poRaised:
3046
                    timeout.poRaised += 1
3047
                else:
3048
                    timeout.poRaised = 1
3049
 
3050
                if timeout.poEstimate:
3051
                    timeout.poEstimate += orderInv.estimate
3052
                else:
3053
                    timeout.poEstimate = orderInv.estimate
3054
                orderInv.order.status = OrderStatus.LOW_INV_PO_RAISED_TIMEOUT
3055
                orderInv.order.statusDescription = "PO Raised Timeout"
3056
            elif orderInv.hotspotAction == HotspotAction.REVERSAL_INITIATED and orderInv.order.status == OrderStatus.LOW_INV_REVERSAL_IN_PROCESS:
3057
                if timeout.reversalInitiated:
3058
                    timeout.reversalInitiated += 1
3059
                else:
3060
                    timeout.reversalInitiated = 1
3061
 
3062
                if timeout.reversalEstimate:
3063
                    timeout.reversalEstimate += orderInv.estimate
3064
                else:
3065
                    timeout.reversalEstimate = orderInv.estimate
3066
                orderInv.order.status = OrderStatus.LOW_INV_REVERSAL_TIMEOUT
3067
                orderInv.order.statusDescription = "Reversal Initiated Timeout"
3068
            else:
3069
                orderInv.delete()
3070
    session.commit()
3071
    return ret_timeouts
4556 mandeep.dh 3072
'''
3073
 
3074
 
3075
def mark_orders_as_timeout(vendorId):
3076
    ret_timeouts = {}
3077
    orderInvs = OrderInventory.query.all()
3078
    for orderInv in orderInvs:
3079
        if not ret_timeouts.has_key(orderInv.itemId):
3080
            timeout = TimeoutSummary()
3081
            ret_timeouts[orderInv.itemId] = timeout
3082
        else:
3083
            timeout = ret_timeouts.get(orderInv.itemId)
3084
        if orderInv.hotspotAction == HotspotAction.PO_RAISED and orderInv.order.status == OrderStatus.LOW_INV_PO_RAISED:
3085
            if timeout.poRaised:
3086
                timeout.poRaised += 1
3087
            else:
3088
                timeout.poRaised = 1
3089
 
3090
            if timeout.poEstimate:
3091
                timeout.poEstimate = max(orderInv.estimate, timeout.poEstimate)
3092
            else:
3093
                timeout.poEstimate = orderInv.estimate
3094
        elif orderInv.hotspotAction == HotspotAction.REVERSAL_INITIATED and orderInv.order.status == OrderStatus.LOW_INV_REVERSAL_IN_PROCESS:
3095
            if timeout.reversalInitiated:
3096
                timeout.reversalInitiated += 1
3097
            else:
3098
                timeout.reversalInitiated = 1
3099
 
3100
            if timeout.reversalEstimate:
3101
                timeout.reversalEstimate = max(orderInv.estimate, timeout.reversalEstimate)
3102
            else:
3103
                timeout.reversalEstimate = orderInv.estimate
3104
        orderInv.order.status = OrderStatus.INVENTORY_LOW
3105
        orderInv.order.statusDescription = "Low Inventory"
3106
        orderInv.delete()
3107
    session.commit()
3108
    return ret_timeouts
3109
 
4386 anupam.sin 3110
def get_order_for_awb(awb):
3111
    order = Order.get_by(airwaybill_no=awb)
3112
    if not order:
3113
        raise TransactionServiceException(108, "no such order")
3114
 
3115
    return order
4369 rajveer 3116
 
4910 phani.kuma 3117
def get_orders_for_provider_for_status(provider_id, order_status_list):
3118
    if not provider_id or not order_status_list:
4506 phani.kuma 3119
        raise TransactionServiceException(101, "bad provider id or order status")
4910 phani.kuma 3120
    doa_provider = False
3121
    for order_status in order_status_list:
3122
        if order_status in [OrderStatus.DOA_PICKUP_CONFIRMED, OrderStatus.RET_PICKUP_CONFIRMED]:
3123
            doa_provider = True
3124
    if doa_provider:
3125
        query = Order.query.filter_by(doa_logistics_provider_id = provider_id).filter(Order.status.in_(order_status_list))
4815 phani.kuma 3126
    else:
4910 phani.kuma 3127
        query = Order.query.filter_by(logistics_provider_id = provider_id).filter(Order.status.in_(order_status_list))
4506 phani.kuma 3128
    return query.all()
4386 anupam.sin 3129
 
4600 varun.gupt 3130
def get_billed_orders_for_vendor(vendor_id, billing_date_from, billing_date_to):
5368 mandeep.dh 3131
    query = Order.query.filter(Order.status != OrderStatus.PAYMENT_PENDING)
3132
 
3133
    if vendor_id:
3134
        query = query.filter_by(vendorId = vendor_id)
3135
 
4600 varun.gupt 3136
    if billing_date_from:
3137
        query = query.filter(Order.billing_timestamp >= billing_date_from)
5368 mandeep.dh 3138
 
4600 varun.gupt 3139
    if billing_date_to:
3140
        query = query.filter(Order.billing_timestamp <= billing_date_to)
5368 mandeep.dh 3141
 
4600 varun.gupt 3142
    return query.all()
3143
 
5481 phani.kuma 3144
def get_billed_orders(vendor_id, onlyVendorNotPaid, billing_date_from, billing_date_to):
5386 phani.kuma 3145
    query = Order.query.filter(Order.billing_timestamp != None)
5481 phani.kuma 3146
 
3147
    if onlyVendorNotPaid:
5488 phani.kuma 3148
        query = query.filter_by(vendor_paid = False)
5481 phani.kuma 3149
 
5386 phani.kuma 3150
    if vendor_id != -1:
3151
        query = query.filter_by(vendorId = vendor_id)
3152
 
5481 phani.kuma 3153
    if billing_date_from:
3154
        query = query.filter(Order.billing_timestamp >= billing_date_from)
3155
 
3156
    if billing_date_to:
3157
        query = query.filter(Order.billing_timestamp <= billing_date_to)
3158
 
4885 varun.gupt 3159
    return query.all()
4875 varun.gupt 3160
 
4600 varun.gupt 3161
def save_bluedart_settlements(map_awb_and_amount):
3162
    payment_client = PaymentClient().get_client()
3163
 
3164
    for awb, amount in map_awb_and_amount.iteritems():
3165
        order = Order.get_by(airwaybill_no = awb)
3166
 
3167
        payment = payment_client.getSuccessfulPaymentForTxnId(order.transaction_id)
3168
        settlement = PaymentSettlement()
3169
        settlement.paymentId = payment.paymentId
3170
        settlement.paymentGatewayId = payment.gatewayId
3171
        settlement.netCollection = amount
3172
        session.commit()
3173
 
4905 varun.gupt 3174
def save_payment_settlements(settlementDate, paymentGatewayId, referenceId, serviceTax, otherCharges, netCollection):
5386 phani.kuma 3175
    originalOrderId = None
3176
    if paymentGatewayId == 4:
3177
        cod_order = Order.get_by(id = referenceId)
3178
        if cod_order.originalOrderId:
3179
            originalOrderId = cod_order.originalOrderId
3180
        else:
3181
            originalOrderId = referenceId
4600 varun.gupt 3182
    settlement = PaymentSettlement()
3183
    settlement.settlementDate = settlementDate
3184
    settlement.paymentGatewayId = paymentGatewayId
5386 phani.kuma 3185
    if originalOrderId is not None:
3186
        settlement.originalOrderId = originalOrderId
4905 varun.gupt 3187
    settlement.referenceId = referenceId
4600 varun.gupt 3188
    settlement.serviceTax = serviceTax
3189
    settlement.otherCharges = otherCharges
3190
    settlement.netCollection = netCollection
3191
    session.commit()
3192
 
5386 phani.kuma 3193
def get_settlement_for_Prepaid(reference_id, is_refund = False):
3194
    query = PaymentSettlement.query.filter_by(referenceId = reference_id).filter(PaymentSettlement.paymentGatewayId != 4)
4600 varun.gupt 3195
 
5189 varun.gupt 3196
    if is_refund:
3197
        query = query.filter(PaymentSettlement.netCollection < 0)
3198
    else:
3199
        query = query.filter(PaymentSettlement.netCollection > 0)
3200
 
5386 phani.kuma 3201
    return query.first()
3202
 
3203
def get_settlement_for_Cod(orderId, is_refund = False):
3204
    query = PaymentSettlement.query.filter_by(originalOrderId = orderId)
5382 varun.gupt 3205
 
5386 phani.kuma 3206
    if is_refund:
3207
        query = query.filter(PaymentSettlement.netCollection < 0)
3208
    else:
3209
        query = query.filter(PaymentSettlement.netCollection > 0)
3210
 
5189 varun.gupt 3211
    return query.first()
4600 varun.gupt 3212
 
3213
def save_ebs_settlement_summary(settlement_id, settlement_date, transaction_date_from, transaction_date_to, amount):
5551 phani.kuma 3214
    summary = EBSSettlementSummary.query.filter_by(settlementId = settlement_id).first()
5481 phani.kuma 3215
    if summary is None:
3216
        summary = EBSSettlementSummary()
3217
        summary.settlementId = settlement_id
3218
        summary.settlementDate = settlement_date
3219
        summary.transactionDateFrom = transaction_date_from
3220
        summary.transactionDateTo = transaction_date_to
3221
        summary.amount = amount
3222
        summary.detailsUploaded = False
3223
        session.commit()
4600 varun.gupt 3224
 
3225
def get_ebs_settlement_summaries():
3226
    summaries = EBSSettlementSummary.query.filter_by(detailsUploaded = False).order_by(EBSSettlementSummary.transactionDateFrom).all()
3227
    summary_map = {}
3228
 
3229
    for summary in summaries:
3230
        transactionDateFrom = str(summary.transactionDateFrom.year) + '-' + str(summary.transactionDateFrom.month) + '-' + str(summary.transactionDateFrom.day)
3231
        transactionDateTo = str(summary.transactionDateTo.year) + '-' + str(summary.transactionDateTo.month) + '-' + str(summary.transactionDateTo.day)
3232
 
3233
        summary_map[summary.settlementId] = "%s to %s" %(transactionDateFrom, transactionDateTo)
3234
 
3235
    return summary_map
3236
 
3237
def get_ebs_settlement_date(settlement_id):
3238
    summary = EBSSettlementSummary.get_by(settlementId = settlement_id)
3239
 
3240
    if summary:
3241
        return summary.settlementDate
3242
 
3243
def mark_ebs_settlement_uploaded(settlement_id):
3244
    summary = EBSSettlementSummary.get_by(settlementId = settlement_id)
3245
 
3246
    if summary:
3247
        summary.detailsUploaded = True
3248
        session.commit()
4715 varun.gupt 3249
 
3250
def get_settlements_by_date(settlement_date_from, settlement_date_to, is_refund):
3251
    query = PaymentSettlement.query
4600 varun.gupt 3252
 
4715 varun.gupt 3253
    if settlement_date_from:
3254
        query = query.filter(PaymentSettlement.settlementDate >= settlement_date_from)
3255
 
3256
    if settlement_date_to:
3257
        query = query.filter(PaymentSettlement.settlementDate <= settlement_date_to)
3258
 
4749 varun.gupt 3259
    if is_refund is True:
3260
        query = query.filter(PaymentSettlement.netCollection < 0.0)
3261
    else:
3262
        query = query.filter(PaymentSettlement.netCollection > 0.0)
4715 varun.gupt 3263
 
4749 varun.gupt 3264
    return query.all()
4715 varun.gupt 3265
 
3266
def get_reshipped_order_ids(order_ids):
3267
    reshipped_order_ids = []
3268
    return [order.new_order_id for order in Order.query.filter(Order.new_order_id.in_(order_ids)).all()]
3269
 
4749 varun.gupt 3270
def get_shipping_confirmation_email_body(order):
3271
 
3272
    html = """
3273
<html>
3274
<body>
3275
<div>
3276
    <p>Dear Customer,</p>
3277
    <p>
4885 varun.gupt 3278
        We are pleased to inform that we have shipped the order: $order_id for your product $product_name on $shipping_datetime. 
5139 varun.gupt 3279
        The shipment will be delivered to you via $courier_name Courier Service vide AWB# $awb.
4749 varun.gupt 3280
    </p>
3281
    <p>
3282
        Expected delivery date for your order is $expected_delivery_date. 
3283
        The product can be delivered to you between business hours of 10am to 6pm. 
3284
        You can login to your account on our site and under "Open Orders", please click the AWB and check the current shipment updates for your product.
3285
    </p>
3286
    <p>
5573 amit.gupta 3287
        For scheduling the delivery of your product at some specific timings, we recommend you to contact the helpline for $courier_name with your complete AWB#.
4749 varun.gupt 3288
    </p>
5139 varun.gupt 3289
    <p>$courier_name: $courier_phone</p>
4749 varun.gupt 3290
    <p>
3291
        Warm Regards,<br />
6029 rajveer 3292
        $source_name Team
4749 varun.gupt 3293
    </p>
4750 varun.gupt 3294
</div>
4749 varun.gupt 3295
</body>
3296
</html>
3297
    """
5573 amit.gupta 3298
    logistics_providers = {1: {'name': 'BlueDart', 'phone': '011-66111234'}, 2: {'name': 'Aramex', 'phone': '0124- 39419900'}, 3: {'name': 'Delhivery', 'phone' : '0124- 4212200'}}
5139 varun.gupt 3299
    provider_name = logistics_providers[order.logistics_provider_id]['name']
3300
    provider_phone = logistics_providers[order.logistics_provider_id]['phone']
3301
 
5868 rajveer 3302
    lclient = LogisticsClient().get_client()
3303
    logistics_info = lclient.getLogisticsEstimation(order.lineitems[0].item_id, order.customer_pincode, DeliveryType.PREPAID)
4749 varun.gupt 3304
    dt = datetime.datetime.strptime(str(order.shipping_timestamp), "%Y-%m-%d %H:%M:%S")
4885 varun.gupt 3305
    shipping_date = dt.strftime("%A, %d %B %Y %I:%M%p")
4749 varun.gupt 3306
    product = "%s %s %s" % (order.lineitems[0].brand, order.lineitems[0].model_name, order.lineitems[0].model_number)
5868 rajveer 3307
    new_expected_delivery_timestamp = datetime.datetime.now() + timedelta(adjust_delivery_time(datetime.datetime.now(), (logistics_info.deliveryTime - logistics_info.shippingTime)))
4885 varun.gupt 3308
    new_expected_delivery_date = new_expected_delivery_timestamp.strftime("%A, %d %B %Y")
6029 rajveer 3309
    info = dict(order_id = order.id, product_name = product, shipping_datetime = shipping_date, awb = order.airwaybill_no, expected_delivery_date = new_expected_delivery_date, courier_name = provider_name, courier_phone = provider_phone, source_name = source_name)
4885 varun.gupt 3310
    return Template(html).substitute(info)
4749 varun.gupt 3311
 
5832 rajveer 3312
def enqueue_shipping_confirmation_email(order):
4915 varun.gupt 3313
    # Calling helper service on production
3314
    helperClient = HelperClient(host_key = "helper_service_server_host_prod").get_client()
5832 rajveer 3315
    try:
3316
        emailSubject = 'Shipping Details for Order ID: ' + str(order.id)
3317
        emailBody = get_shipping_confirmation_email_body(order)
5866 rajveer 3318
        helperClient.saveUserEmailForSending([order.customer_email], None, emailSubject, emailBody, str(order.id), 'ShippingConfirmation', [], [])
5832 rajveer 3319
    except Exception as e:
3320
        print e
5139 varun.gupt 3321
 
4915 varun.gupt 3322
def get_order_distribution_by_status(start_date, end_date):
5067 varun.gupt 3323
    query = session.query(Order.status, func.count(Order.id)).group_by(Order.status)
3324
    query = query.filter(Order.created_timestamp >= start_date)
3325
    query = query.filter(Order.created_timestamp <= end_date)
3326
    return query.all()
4915 varun.gupt 3327
 
5067 varun.gupt 3328
def get_order_ids_for_status(status, start_datetime, end_datetime):
3329
    query = Order.query.filter(Order.status == status)
3330
    query = query.filter(Order.created_timestamp >= start_datetime)
3331
    query = query.filter(Order.created_timestamp <= end_datetime)
3332
    return query.all()
3333
 
5099 varun.gupt 3334
def update_orders_as_paid_to_vendor(order_id):
5139 varun.gupt 3335
    paid_order = Order.get_by(id = order_id)
5099 varun.gupt 3336
 
5139 varun.gupt 3337
    if paid_order.originalOrderId is None:
3338
        paid_order.vendor_paid = True
5099 varun.gupt 3339
        session.commit()
5139 varun.gupt 3340
 
3341
        for order in Order.query.filter(Order.originalOrderId == order_id).all():
3342
            order.vendor_paid = True
3343
            session.commit()
3344
    else:
5208 varun.gupt 3345
        original_order_id = paid_order.originalOrderId
5139 varun.gupt 3346
 
5208 varun.gupt 3347
        for order in Order.query.filter(Order.originalOrderId == original_order_id).all():
5139 varun.gupt 3348
            order.vendor_paid = True
3349
            session.commit()
3350
 
3351
        original_order = Order.get_by(id = original_order_id)
3352
        if original_order:  original_order.vendor_paid = True
3353
        session.commit()
5348 anupam.sin 3354
 
3355
def update_COD_agent(agentEmailId, orderId):
3356
    order = Order.get_by(id=orderId)
3357
    if not order:
3358
        raise TransactionServiceException(101, "No order found for the given order id" + str(orderId))
3359
    agent = CodVerificationAgent()
3360
    agent.orderId = orderId
3361
    agent.verificationAgent = agentEmailId
3362
    session.commit()
3363
 
5099 varun.gupt 3364
 
5386 phani.kuma 3365
def update_order_only_as_paid_to_vendor(order_id):
3366
    paid_order = Order.get_by(id = order_id)
3367
    if paid_order is None:
3368
        raise TransactionServiceException(116, "No order found for the given order id")
3369
    paid_order.vendor_paid = True
3370
    session.commit()
3371
 
5208 varun.gupt 3372
def get_refunded_orders_marked_paid():
3373
    orderStatusGroup = OrderStatusGroups()
3374
    return Order.query.filter(Order.vendor_paid == 1).filter(Order.status.in_(orderStatusGroup.refundedOrders)).all()
3375
 
5447 anupam.sin 3376
def get_all_verification_agents(minOrderId, maxOrderId):
3377
    return CodVerificationAgent.query.filter(CodVerificationAgent.orderId >= minOrderId).filter(CodVerificationAgent.orderId <= maxOrderId).all()
3378
 
5527 anupam.sin 3379
def get_all_attributes_for_order_id(orderId):
3380
    return Attribute.query.filter(Attribute.orderId == orderId).all()
3381
 
3382
def set_order_attribute_for_transaction(transactionId, t_attribute):
3383
    for order in Order.query.filter_by(transaction_id = transactionId).all():
3384
        attribute = Attribute()
3385
        attribute.orderId = order.id
3386
        attribute.name = t_attribute.name
3387
        attribute.value = t_attribute.value
3388
        session.commit()
5593 mandeep.dh 3389
 
5558 mandeep.dh 3390
def accept_order_for_item(itemId, quantity, fulfilmentWarehouseId, billingWarehouseId):
5944 mandeep.dh 3391
    inventoryClient = InventoryClient().get_client()
3392
    inventory = inventoryClient.getItemInventoryByItemId(itemId)
5684 mandeep.dh 3393
    orders = Order.query.filter(Order.fulfilmentWarehouseId != fulfilmentWarehouseId).filter(Order.warehouse_id == billingWarehouseId).filter(Order.status.in_(tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.COD_VERIFICATION_PENDING, OrderStatus.ACCEPTED, OrderStatus.INVENTORY_LOW, OrderStatus.LOW_INV_PO_RAISED, OrderStatus.LOW_INV_REVERSAL_IN_PROCESS]))).order_by(Order.expected_shipping_time).all()
5558 mandeep.dh 3394
    for order in orders:
5593 mandeep.dh 3395
        lineitem = order.lineitems[0]
5558 mandeep.dh 3396
        if lineitem.item_id == itemId and lineitem.quantity <= quantity:
5944 mandeep.dh 3397
            warehouse = inventoryClient.getWarehouse(order.fulfilmentWarehouseId)
5684 mandeep.dh 3398
            if not warehouse.billingWarehouseId and warehouse.warehouseType == WarehouseType.THIRD_PARTY:
3399
                if (inventory.availability[fulfilmentWarehouseId] >= inventory.reserved[fulfilmentWarehouseId] + lineitem.quantity) and inventory.availability[order.fulfilmentWarehouseId] < inventory.reserved[order.fulfilmentWarehouseId]:
5966 rajveer 3400
                    inventoryClient.reduceReservationCount(itemId, order.fulfilmentWarehouseId, sourceId, order.id, lineitem.quantity)
3401
                    inventoryClient.reserveItemInWarehouse(itemId, fulfilmentWarehouseId, sourceId, order.id, to_java_date(order.created_timestamp), to_java_date(order.promised_shipping_time), lineitem.quantity)
5684 mandeep.dh 3402
                    order.fulfilmentWarehouseId = fulfilmentWarehouseId
5558 mandeep.dh 3403
                    quantity -= lineitem.quantity
5684 mandeep.dh 3404
                    session.commit()
5593 mandeep.dh 3405
                    if not quantity:
3406
                        break
5558 mandeep.dh 3407
 
6031 rajveer 3408
def get_recharge_order(rechargeOrderId):
6048 rajveer 3409
    return RechargeOrder.get_by(id = rechargeOrderId)
6031 rajveer 3410
 
3411
def get_recharge_orders(userId):
3412
    return RechargeOrder.query.filter_by(userId = userId).all()
3413
 
6279 rajveer 3414
def update_recharge_order_status(rechargeOrderId, rechargeOrderStatus, refundDate=None):
6031 rajveer 3415
    recharge_order = get_recharge_order(rechargeOrderId)
3416
    if rechargeOrderStatus == RechargeOrderStatus.PAYMENT_FAILED:
3417
        recharge_order.status = RechargeOrderStatus.PAYMENT_FAILED
3418
        session.commit()
3419
        return True
6214 rajveer 3420
 
3421
    if rechargeOrderStatus == RechargeOrderStatus.RECHARGE_FAILED:
3422
        recharge_order.status = RechargeOrderStatus.RECHARGE_FAILED
3423
        session.commit()
3424
        update_amount_in_wallet(recharge_order.userId, recharge_order.walletAmount, recharge_order.id)
6441 rajveer 3425
        update_amount_in_wallet(recharge_order.userId, recharge_order.totalAmount - recharge_order.walletAmount - recharge_order.couponAmount, recharge_order.id)
6214 rajveer 3426
        enqueue_recharge_info_email(recharge_order)
3427
        return True
3428
 
3429
    if rechargeOrderStatus == RechargeOrderStatus.RECHARGE_SUCCESSFUL:
3430
        recharge_order.status = RechargeOrderStatus.RECHARGE_SUCCESSFUL
3431
        session.commit()
6285 rajveer 3432
        gvs = allocate_gift_voucher(recharge_order)
3433
        enqueue_recharge_info_email(recharge_order, gvs)
6214 rajveer 3434
        return True
3435
 
6279 rajveer 3436
    if rechargeOrderStatus == RechargeOrderStatus.RECHARGE_FAILED_REFUNDED:
3437
        recharge_order.status = RechargeOrderStatus.RECHARGE_FAILED_REFUNDED
3438
        order.responseTimestamp = refundDate
3439
        session.commit()
3440
        update_amount_in_wallet(recharge_order.userId, recharge_order.walletAmount, recharge_order.id)
6441 rajveer 3441
        update_amount_in_wallet(recharge_order.userId, recharge_order.totalAmount - recharge_order.walletAmount - recharge_order.couponAmount, recharge_order.id)
6279 rajveer 3442
        enqueue_recharge_info_email(recharge_order)
3443
        return True
3444
 
6031 rajveer 3445
    if rechargeOrderStatus == RechargeOrderStatus.PAYMENT_SUCCESSFUL:
6285 rajveer 3446
        gvs = {}
6206 rajveer 3447
        recharge_order.status = RechargeOrderStatus.PAYMENT_SUCCESSFUL
3448
        session.commit()
6031 rajveer 3449
        if recharge_order.walletAmount > 0:
3450
            wallet = get_user_wallet(recharge_order.userId)
3451
            if wallet and wallet.amount >= recharge_order.walletAmount:
3452
                update_amount_in_wallet(recharge_order.userId, -recharge_order.walletAmount, recharge_order.id)
3453
            else:
6441 rajveer 3454
                update_amount_in_wallet(recharge_order.userId, recharge_order.totalAmount - recharge_order.walletAmount - recharge_order.couponAmount, recharge_order.id)
6031 rajveer 3455
                raise TransactionServiceException(103, 'Wallet amount has already been used')
3456
 
6092 rajveer 3457
        try:
3458
            if activate_recharge_txn(recharge_order):
3459
                recharge_order.invoiceNumber = get_next_invoice_number(OrderType.RECHARGE)
3460
                session.commit()
6285 rajveer 3461
                gvs = allocate_gift_voucher(recharge_order)
6092 rajveer 3462
                return True
3463
            else:
3464
                update_amount_in_wallet(recharge_order.userId, recharge_order.walletAmount, recharge_order.id)
6441 rajveer 3465
                update_amount_in_wallet(recharge_order.userId, recharge_order.totalAmount - recharge_order.walletAmount - recharge_order.couponAmount, recharge_order.id)
6092 rajveer 3466
                session.commit()
3467
                return False
3468
        except:
3469
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
3470
            transaction_requiring_extra_processing.category = 'RECHARGE_UNKNOWN'
3471
            transaction_requiring_extra_processing.transaction_id = recharge_order.transaction_id
6031 rajveer 3472
            session.commit()
3473
            return False
6102 rajveer 3474
        finally:
6285 rajveer 3475
            enqueue_recharge_info_email(recharge_order, gvs)
6102 rajveer 3476
 
6031 rajveer 3477
 
6285 rajveer 3478
def allocate_gift_voucher(recharge_order):            
3479
    totalAmount = recharge_order.totalAmount
3480
    gvs = {}
6438 rajveer 3481
    return gvs
6330 rajveer 3482
    if totalAmount >= 100:
3483
        totalAmount=500 if totalAmount >= 500 else totalAmount
6285 rajveer 3484
        try:
3485
            pc = PromotionClient().get_client()
6330 rajveer 3486
            enddate = datetime.datetime.now() + datetime.timedelta(days=45)
6285 rajveer 3487
            while totalAmount > 0:
6330 rajveer 3488
                gvAmount = totalAmount if totalAmount <= 250 else 250
6285 rajveer 3489
                totalAmount = totalAmount - gvAmount
6355 amit.gupta 3490
                gv = pc.createCoupon(26, to_java_date(enddate), recharge_order.userEmailId, gvAmount, False, 1)
6285 rajveer 3491
                gvs[gv] = gvAmount  
3492
        except:
3493
            print "Not able to create gift voucher for user " + recharge_order.userEmailId
3494
    return gvs     
3495
 
6048 rajveer 3496
def activate_recharge_txn(recharge_order):
3497
    if isinstance(recharge_order, MobileRechargeOrder):
3498
        rechargeType = 'MTP'
3499
        deviceNumber = recharge_order.mobileNumber
3500
        strProviderCode = get_provider_code(recharge_order.mobileOperatorId)
6161 rajveer 3501
        plan = recharge_order.rechargePlan
6048 rajveer 3502
    if isinstance(recharge_order, DTHRechargeOrder):
3503
        rechargeType = 'RCH'
3504
        deviceNumber = recharge_order.dthAccountNumber
3505
        strProviderCode = get_provider_code(recharge_order.dthOperatorId)
6161 rajveer 3506
        plan = ''
3507
    retStatus, retCode, spiceTID, description = RechargeService.rechargeDevice(recharge_order.id, rechargeType, strProviderCode, deviceNumber, recharge_order.totalAmount, plan)
6231 rajveer 3508
    recharge_order.spiceTID = spiceTID
3509
    recharge_order.responseTimestamp = datetime.datetime.now()
3510
    recharge_order.description = description
3511
    session.commit()
6092 rajveer 3512
    if retCode == '00' and retStatus == 'S':
3513
        recharge_order.status = RechargeOrderStatus.RECHARGE_SUCCESSFUL
3514
        session.commit()
3515
        return True
6259 rajveer 3516
    elif retCode in ('STO', 'US', 'UP') and retStatus == 'S':
6231 rajveer 3517
        raise TransactionServiceException(111, "No response from operator within specified time")
6092 rajveer 3518
    else:
3519
        recharge_order.status = RechargeOrderStatus.RECHARGE_FAILED
6265 rajveer 3520
        if retCode in ('SVAI', 'SVIA', 'AI', 'IA', 'EAI', 'SVEAI'):
3521
            recharge_order.description = "Invalid Amount"
3522
        elif retCode in ('ECL', 'SVECL', 'SCL'):
3523
            recharge_order.description = "Customer Exceeded Daily Limit"
3524
        elif retCode in ('CI', 'ECI', 'ESP', 'SVCI', 'SVECI', 'SVESP', 'SV_ECI', 'SV ECI'):
6399 rajveer 3525
            recharge_order.description = "Invalid Device Number"
6092 rajveer 3526
        session.commit()
6048 rajveer 3527
        return False
6031 rajveer 3528
 
6092 rajveer 3529
 
6031 rajveer 3530
def get_user_wallet(userId):
3531
    wallet = UserWallet.get_by(userId = userId)
3532
    if not wallet:
3533
        wallet = UserWallet()
3534
        wallet.amount = 0
3535
        wallet.userId = userId
3536
    return wallet
3537
 
3538
def get_user_wallet_history(userId):
3539
    wallet = get_user_wallet(userId)
3540
    if wallet.id:
3541
        return UserWalletHistory.query.filter_by(wallet = wallet).all()
3542
    else:
3543
        return []
3544
 
6125 rajveer 3545
def get_provider(providerId):
3546
    return ServiceProvider.query.filter_by(id = providerId).one()
3547
 
6048 rajveer 3548
def get_provider_code(providerId):
3549
    return ServiceProvider.query.filter_by(id = providerId).one().code
6031 rajveer 3550
 
6310 anupam.sin 3551
def get_provider_by_code(providerCode):
3552
    return ServiceProvider.query.filter_by(code = providerCode).one()
6049 rajveer 3553
 
6206 rajveer 3554
def get_service_providers(recharge_type, only_active):
3555
    query = ServiceProvider.query.filter_by(type = recharge_type)
3556
    if only_active:
3557
        query = query.filter_by(status = only_active)
3558
    providers = query.all()
6048 rajveer 3559
    serviceProviders = {}
3560
    for provider in providers:
3561
        serviceProviders[provider.id] = provider.name
3562
    return serviceProviders
3563
 
6049 rajveer 3564
def get_service_provider_for_device(rechargeType, deviceNumber):
3565
    if rechargeType == RechargeType.MOBILE:
3566
        serviceType = 'MTP'
3567
    if rechargeType == RechargeType.DTH:
3568
        serviceType = 'RCH'
6048 rajveer 3569
 
6313 anupam.sin 3570
    provider = None
6049 rajveer 3571
    try:    
6235 rajveer 3572
        providerCode, operatorCircle = RechargeService.checkServiceProviderApi(serviceType, deviceNumber)
6092 rajveer 3573
        if providerCode == 'SNS' or providerCode == '':
3574
            providerId = 0
3575
        else:
6310 anupam.sin 3576
            provider = get_provider_by_code(providerCode)
3577
            providerId = provider.id
6049 rajveer 3578
    except:
6092 rajveer 3579
        providerId = -1
3580
 
6289 anupam.sin 3581
    deviceNumberInfo = DeviceNumberInfo()
6336 anupam.sin 3582
    deviceNumberInfo.circleCode = operatorCircle
6289 anupam.sin 3583
    deviceNumberInfo.operatorId = providerId
6313 anupam.sin 3584
    if provider:
3585
        deviceNumberInfo.operatorName = provider.name
6289 anupam.sin 3586
    return deviceNumberInfo
6269 rajveer 3587
 
6336 anupam.sin 3588
def validate_recharge(rechargeType, deviceNumber, userSelectedProviderId):
6307 anupam.sin 3589
    deviceNumberInfo = get_service_provider_for_device(rechargeType, deviceNumber)
3590
    if deviceNumberInfo.operatorId == -1:
3591
        return "Oops! There seems to be a problem. Please try after some time."
6336 anupam.sin 3592
    circle = TelecomCircle.query.filter_by(code = deviceNumberInfo.circleCode).first()
6313 anupam.sin 3593
    if circle:
3594
        if not ServiceAvailability.query.filter_by(circleId = circle.id).filter_by(operatorId = userSelectedProviderId).first().isServiceAvailable :
3595
            if deviceNumberInfo.operatorName :
6314 anupam.sin 3596
                if circle.id == 8:
3597
                    return "We currently do not provide this service in J&K. We regret the inconvenience."
3598
                else :
3599
                    return "We are facing technical issues with " + deviceNumberInfo.operatorName + " in " + circle.name + " currently. Please try later." 
6316 anupam.sin 3600
            else :
6314 anupam.sin 3601
                return "We are facing technical issues with this operator in " + circle.name + " currently. Please try later." 
6271 rajveer 3602
    allcounts, successcounts  = get_previous_transactions(rechargeType, deviceNumber)
6269 rajveer 3603
    if allcounts >= 15 or successcounts >= 5:
3604
        return "You have exceeded daily limit for this mobile number."
3605
    return "SUCCESS"
6049 rajveer 3606
 
6271 rajveer 3607
def get_previous_transactions(rechargeType, deviceNumber):
3608
    limittime = datetime.datetime.now() - datetime.timedelta(hours=24)
3609
    if rechargeType == RechargeType.MOBILE:
3610
        allcounts = session.query(func.count(MobileRechargeOrder.mobileNumber)).filter(MobileRechargeOrder.mobileNumber == deviceNumber).filter(MobileRechargeOrder.creationTimestamp  > limittime).scalar()
3611
        successcounts = session.query(func.count(MobileRechargeOrder.mobileNumber)).filter(MobileRechargeOrder.mobileNumber == deviceNumber).filter(MobileRechargeOrder.status == RechargeOrderStatus.RECHARGE_SUCCESSFUL).filter(MobileRechargeOrder.creationTimestamp > limittime).scalar()
3612
    else:
3613
        allcounts = session.query(func.count(DTHRechargeOrder.dthAccountNumber)).filter(DTHRechargeOrder.dthAccountNumber == deviceNumber).filter(MobileRechargeOrder.creationTimestamp  > limittime).scalar()
3614
        successcounts = session.query(func.count(DTHRechargeOrder.dthAccountNumber)).filter(DTHRechargeOrder.dthAccountNumber == deviceNumber).filter(DTHRechargeOrder.status == RechargeOrderStatus.RECHARGE_SUCCESSFUL).filter(MobileRechargeOrder.creationTimestamp > limittime).scalar()
6269 rajveer 3615
    return allcounts, successcounts    
3616
 
6031 rajveer 3617
def update_amount_in_wallet(userId, amount, orderId):
3618
    if amount == 0:
3619
        return
3620
    wallet = get_user_wallet(userId)
3621
    wallet.amount = wallet.amount + amount
3622
 
3623
    history = UserWalletHistory()
3624
    history.amount = amount
3625
    history.orderId = orderId
3626
    history.wallet = wallet
3627
    history.timestamp = datetime.datetime.now()
3628
    session.commit()
6056 amit.gupta 3629
 
6094 rajveer 3630
def get_recharge_orders_for_device(deviceNumber):
3631
    orders = MobileRechargeOrder.query.filter_by(mobileNumber = deviceNumber).all()
3632
    if orders is None:
3633
        orders = DTHRechargeOrder.query.filter_by(dthAccountNumber = deviceNumber).all()
3634
    return orders
3635
 
6154 rajveer 3636
def get_recharge_orders_for_status(status):
3637
    if status == -1:
6161 rajveer 3638
        orders = RechargeOrder.query.order_by(desc(RechargeOrder.id)).all()
6154 rajveer 3639
    else:
6161 rajveer 3640
        orders = RechargeOrder.query.order_by(desc(RechargeOrder.id)).filter_by(status = status).all()
6154 rajveer 3641
    return orders
3642
 
6188 rajveer 3643
def get_recharge_statistics():
3644
    rs = RechargeStatistics()
3645
    statusCounts = {}
3646
    operatorCounts = {}
3647
    currentOrders = {}
3648
    amounts = {}
3649
 
3650
    sorders = session.query(RechargeOrder.status, func.count(RechargeOrder.status)).group_by(RechargeOrder.status).all()
3651
    for sorder in sorders:
6190 rajveer 3652
        statusCounts[sorder[0]] = sorder[1]
6188 rajveer 3653
 
3654
    oorders = session.query(MobileRechargeOrder.mobileOperatorId, func.count(MobileRechargeOrder.mobileOperatorId)).group_by(MobileRechargeOrder.mobileOperatorId).all()
3655
    for oorder in oorders:
6190 rajveer 3656
        operatorCounts[oorder[0]] = oorder[1]
6188 rajveer 3657
 
3658
    oorders = session.query(DTHRechargeOrder.dthOperatorId, func.count(DTHRechargeOrder.dthOperatorId)).group_by(DTHRechargeOrder.dthOperatorId).all()
3659
    for oorder in oorders:
6190 rajveer 3660
        operatorCounts[oorder[0]] = oorder[1]
6188 rajveer 3661
 
6219 rajveer 3662
    forders = RechargeOrder.query.order_by(desc(RechargeOrder.id)).filter_by(status = RechargeOrderStatus.RECHARGE_FAILED).limit(20).all()
6188 rajveer 3663
    currentOrders[RechargeOrderStatus._VALUES_TO_NAMES.get(RechargeOrderStatus.RECHARGE_FAILED)] = [order.to_thrift_object() for order in forders]
3664
    sorders = RechargeOrder.query.order_by(desc(RechargeOrder.id)).filter_by(status = RechargeOrderStatus.RECHARGE_SUCCESSFUL).limit(20).all()
6197 rajveer 3665
    currentOrders[RechargeOrderStatus._VALUES_TO_NAMES.get(RechargeOrderStatus.RECHARGE_SUCCESSFUL)] = [order.to_thrift_object() for order in sorders]
6216 rajveer 3666
    porders = RechargeOrder.query.order_by(desc(RechargeOrder.id)).filter_by(status = RechargeOrderStatus.PAYMENT_SUCCESSFUL).limit(20).all()
3667
    currentOrders[RechargeOrderStatus._VALUES_TO_NAMES.get(RechargeOrderStatus.PAYMENT_SUCCESSFUL)] = [order.to_thrift_object() for order in porders]
6238 rajveer 3668
    rorders = RechargeOrder.query.order_by(desc(RechargeOrder.id)).filter_by(status = RechargeOrderStatus.RECHARGE_FAILED_REFUNDED).limit(20).all()
3669
    currentOrders[RechargeOrderStatus._VALUES_TO_NAMES.get(RechargeOrderStatus.RECHARGE_FAILED_REFUNDED)] = [order.to_thrift_object() for order in rorders]
6188 rajveer 3670
 
6190 rajveer 3671
    twa = session.query(func.sum(UserWallet.amount)).one()
3672
    tra = session.query(func.sum(RechargeOrder.totalAmount)).filter(RechargeOrder.status == RechargeOrderStatus.RECHARGE_SUCCESSFUL).one()
3673
    gwa = session.query(func.sum(UserWalletHistory.amount)).filter(UserWalletHistory.orderId > 100000).one()
6278 rajveer 3674
    #tpa = session.query(func.sum(UserWalletHistory.amount)).filter(UserWalletHistory.orderId < 100000).one()
6482 rajveer 3675
    tpa = session.query(func.sum(RechargeOrder.totalAmount-RechargeOrder.walletAmount-RechargeOrder.couponAmount)).filter(RechargeOrder.status.in_((RechargeOrderStatus.RECHARGE_SUCCESSFUL, RechargeOrderStatus.RECHARGE_FAILED, RechargeOrderStatus.PAYMENT_SUCCESSFUL, RechargeOrderStatus.RECHARGE_FAILED_REFUNDED))).one()
3676
    tda = session.query(func.sum(RechargeOrder.couponAmount)).filter(RechargeOrder.status == RechargeOrderStatus.RECHARGE_SUCCESSFUL).one()
3677
 
6346 rajveer 3678
    balance = RechargeService.getBalance()
6305 rajveer 3679
    amounts['(a) Total Payment Amount'] = tpa[0]
3680
    amounts['(b) Gift Wallet Amount'] = gwa[0]
3681
    amounts['(c) Total Wallet Amount'] = twa[0]
3682
    amounts['(d) Total Recharge Amount'] = tra[0]
3683
    amounts['(e) Account Balance'] = balance
6482 rajveer 3684
    amounts['(f) Discount Balance'] = tda[0]
6278 rajveer 3685
    #amounts['Total Payment Amount'] = tra[0] + twa[0] - gwa[0]
6482 rajveer 3686
    amounts['Saholic Mismatch = (a+b+f-c-d)'] = tpa[0] + gwa[0] + tda[0] - tra[0] - (twa[0] + 53775)
3687
    amounts['SpiceDeck Mismatch = (d-e)'] = tra[0] + balance - 550000 
6278 rajveer 3688
 
6337 rajveer 3689
    '''
6318 rajveer 3690
    if amounts['SpiceDeck Mismatch = (d-e)'] != -739:
3691
        mail(mail_user, mail_password, ['rajneesh.arora@shop2020.in', 'rajveer.singh@shop2020.in', 'anupam.singh@shop2020.in'], "SpiceDeck Account mismatch of amount " + str(amounts['SpiceDeck Mismatch = (d-e)']), "", [], [], [])
3692
    if amounts['Saholic Mismatch = (a+b-c-d)'] != -1919:
3693
        mail(mail_user, mail_password, ['rajneesh.arora@shop2020.in', 'rajveer.singh@shop2020.in', 'anupam.singh@shop2020.in'], "Saholic Recharge Account mismatch of amount " + str(amounts['Saholic Mismatch = (a+b-c-d)']), "", [], [], [])
6337 rajveer 3694
    '''
6318 rajveer 3695
 
6188 rajveer 3696
    rs.statusCounts = statusCounts
3697
    rs.operatorCounts = operatorCounts    
3698
    rs.currentOrders = currentOrders
3699
    rs.amounts = amounts
3700
 
3701
    return rs
3702
 
3703
 
6159 rajveer 3704
def get_plans_for_operator(operator_id):
3705
    return RechargePlan.query.filter_by(operatorId = operator_id).all()
3706
 
6056 amit.gupta 3707
def get_recharge_orders_for_transaction(txId):
3708
    return RechargeOrder.query.filter_by(transaction_id = txId).first()
6102 rajveer 3709
 
6285 rajveer 3710
def enqueue_recharge_info_email(recharge_order, gvs={}):
6102 rajveer 3711
    status = ""
3712
    message = ""
3713
    html_header = """
3714
 
3715
    <html>
3716
    <body>
3717
    <div>
3718
        <p>
3719
            Hello $email,<br /><br />
6127 rajveer 3720
            Thanks for placing recharge order with us. Following are the details of your recharge order:
6102 rajveer 3721
        </p>
3722
        <div>
6136 rajveer 3723
            Order Id : $order_id <br/>
3724
            Order Date : $order_date <br/>
3725
            Order Amount : Rs. $order_amount <br/>
3726
            Order Status : $order_status <br/>
3727
            Device Number : $device_number <br/>
3728
            Service Provider : $service_provider <br/>
6102 rajveer 3729
        </div>
6348 rajveer 3730
    <br />
6102 rajveer 3731
    """
6288 rajveer 3732
 
3733
    FailureReasonMap = {}
3734
    FailureReasonMap["Invalid Amount"] = "the amount you tried seems to be invalid. Please try another amount."
3735
    FailureReasonMap["Customer Exceeded Daily Limit"] = "you have exceeded daily recharge limit for this number. Please try after 12 hours."
6399 rajveer 3736
    FailureReasonMap["Invalid Device Number"] = "it seems that your device number is not being recognized by selected operator. Please make sure the number you tried is correct."
6288 rajveer 3737
 
3738
    failureReason = FailureReasonMap.get(recharge_order.description)
3739
    if failureReason is None:
3740
        failureReason = "of network failure."
6102 rajveer 3741
    if recharge_order.status == RechargeOrderStatus.RECHARGE_SUCCESSFUL:
3742
        status = "Recharge Successful"
3743
        message = ""
6444 rajveer 3744
        if recharge_order.couponAmount > 0:
3745
            message = message + "<b>You have used coupon " + recharge_order.couponCode + " and got discount of Rs. " + str(recharge_order.couponAmount) + ".</b>" 
3746
 
6102 rajveer 3747
    elif recharge_order.status == RechargeOrderStatus.RECHARGE_FAILED:
3748
        status = "Recharge Failed"
6127 rajveer 3749
        message = """
6288 rajveer 3750
                We could not process your recharge because $failureReason We have credited the amount to your recharge wallet. You can use wallet amount to recharge again. <a href="$source_url/static/recharge-faq">Learn More</a>
6127 rajveer 3751
                """
6206 rajveer 3752
    elif recharge_order.status == RechargeOrderStatus.PAYMENT_SUCCESSFUL:
6102 rajveer 3753
        status = "Recharge Under Process"
6127 rajveer 3754
        message = """
6288 rajveer 3755
                Your Payment was successful but due to some internal error with the operator's system we are not sure if the recharge was successful. In case your recharge is not processed, we will credit amount to your recharge wallet. You can use wallet amount to recharge again. <a href="$source_url/static/recharge-faq">Learn More</a>
6127 rajveer 3756
                """
6279 rajveer 3757
    elif recharge_order.status == RechargeOrderStatus.RECHARGE_FAILED_REFUNDED:
3758
        status = "Recharge Failed, Amount Refunded"
3759
        message = """
3760
                It seems operator could not process your recharge request. We are crediting the refund amount to your recharge wallet. You can use wallet amount to recharge again. <a href="$source_url/static/recharge-faq">Learn More</a>
3761
                """
6285 rajveer 3762
    if len(gvs) > 0:
6340 rajveer 3763
        message = message  + '<br />  You have been awarded with following Gift Voucher(s). You can redeem Gift Voucher(s) to purchase any product for Rs. 5000 and above using credit/debit card and net banking. <a href="$source_url/static/giftvoucher-offer">More details</a>'
6330 rajveer 3764
        message = message + " <br />  <b>Voucher Code &#160;&#160;&#160;&#160;&#160;&#160;&#160;&#160;&#160;&#160; Amount</b>"
6285 rajveer 3765
        for gv in gvs.keys():
3766
            gvAmount = gvs.get(gv)
6330 rajveer 3767
            message = message + " <br />  <b>" + gv + "&#160;&#160;&#160;&#160;&#160;&#160;&#160;&#160;&#160;&#160; Rs. "  + str(gvAmount) + "</b>"
6285 rajveer 3768
 
6114 rajveer 3769
    message = message  + '<br />  In case of any problem, please  <a href="$source_url/contact-us">contact us</a>'
6102 rajveer 3770
 
6136 rajveer 3771
    html_header = html_header + message
6102 rajveer 3772
    html_footer = """
3773
    <p>
3774
        Best Wishes,<br />
3775
        $source_name Team
3776
    </p>
3777
    </div>
3778
    </body>
3779
    </html>
3780
    """
3781
 
6125 rajveer 3782
    if isinstance(recharge_order, MobileRechargeOrder):
3783
        deviceNumber = recharge_order.mobileNumber
3784
        providerName = get_provider(recharge_order.mobileOperatorId).name
3785
    if isinstance(recharge_order, DTHRechargeOrder):
3786
        deviceNumber = recharge_order.dthAccountNumber
3787
        providerName = get_provider(recharge_order.dthOperatorId).name
3788
 
6127 rajveer 3789
    display_order_id = recharge_order.idPrefix + str(recharge_order.id)
6102 rajveer 3790
    email = recharge_order.userEmailId
3791
    dt = datetime.datetime.strptime(str(recharge_order.creationTimestamp), "%Y-%m-%d %H:%M:%S")
3792
    formated_order_date = dt.strftime("%A, %d. %B %Y %I:%M%p")
3793
 
6288 rajveer 3794
    email_header = Template(html_header).substitute(dict(order_id = display_order_id, order_date = formated_order_date, email = email, source_url = source_url, order_amount = recharge_order.totalAmount, order_status = status, device_number = deviceNumber, service_provider = providerName, failureReason = failureReason))
6102 rajveer 3795
    email_footer = Template(html_footer).substitute(dict(source_url = source_url, source_name = source_name))
3796
 
3797
    try:
3798
        helper_client = HelperClient().get_client()
6288 rajveer 3799
        helper_client.saveUserEmailForSending([email], "", source_name + " - Recharge Order Details", email_header + email_footer, str(recharge_order.id), "RechargeInfo", [], ['anupam.singh@shop2020.in', 'rajneesh.arora@shop2020.in', 'abhishek.mathur@shop2020.in', 'suraj.sharma@shop2020.in', 'rajveer.singh@shop2020.in'])
6102 rajveer 3800
        return True
3801
    except Exception as e:
3802
        print e
3803
        return False
3804
 
6307 anupam.sin 3805
def get_recharge_denominations(operator_id, circleCode, denomination_type):
3806
    circle_id = TelecomCircle.query.filter_by(code = circleCode).first().id
6289 anupam.sin 3807
    return RechargeDenomination.query.filter_by(operatorId = operator_id).filter_by(circleId = circle_id).filter_by(denominationType = denomination_type).all()
6371 rajveer 3808
 
3809
def update_availability_status(operatorId, circleId, isAvailable):
3810
    serviceAvailability = ServiceAvailability.query.filter_by(circleId = circleId).filter_by(operatorId = operatorId).first()
3811
    serviceAvailability.isServiceAvailable = isAvailable
3812
    session.commit()
3813
 
6389 rajveer 3814
def get_misc_charges(transactionId):
3815
    retCharges = {}
3816
    miscCharges = MiscCharges.get_by(transaction_id = transactionId)
3817
    if miscCharges:
6426 rajveer 3818
        retCharges[1] = miscCharges.chargeAmount 
6389 rajveer 3819
    return retCharges
6396 amit.gupta 3820
 
3821
def get_available_emi_schemes():
3822
    t_emi_schemes = []
3823
    emi_schemes = EmiScheme.query.all()
3824
    if emi_schemes:
3825
        t_emi_schemes = [to_t_emi_scheme(emi_scheme) for emi_scheme in emi_schemes]
3826
    return t_emi_schemes
6389 rajveer 3827
 
6507 anupam.sin 3828
def refund_recharge_order(rechargeOrderId):
3829
    try :
3830
        rechargeOrder = RechargeOrder.query.filter_by(id = rechargeOrderId).first()
3831
        wallet = UserWallet.query.filter_by(userId = rechargeOrder.userId).first()
3832
    except:
3833
        return False
3834
 
3835
    paidAmount = rechargeOrder.totalAmount - rechargeOrder.walletAmount - rechargeOrder.couponAmount
3836
    amountToRefund = 0
3837
    status = None
3838
    if (paidAmount <= wallet.amount) :
3839
        amountToRefund = paidAmount
3840
        status = RechargeOrderStatus.REFUNDED
3841
    else :
3842
        amountToRefund = wallet.amount
3843
        status = RechargeOrderStatus.PARTIALLY_REFUNDED
3844
 
3845
    update_amount_in_wallet(rechargeOrder.userId, -amountToRefund, rechargeOrderId)
3846
 
3847
    try:    
3848
        pc = PaymentClient().get_client()
3849
        if pc.refundPayment(rechargeOrder.transaction_id, amountToRefund, True) :
3850
            rechargeOrder.status = status
6515 anupam.sin 3851
            rechargeOrder.refundTimestamp = datetime.datetime.now()
6507 anupam.sin 3852
            session.commit()
3853
            return True
3854
        else :
3855
            update_amount_in_wallet(rechargeOrder.userId, amountToRefund, rechargeOrderId)
3856
 
3857
    except:
3858
        update_amount_in_wallet(rechargeOrder.userId, amountToRefund, rechargeOrderId)
3859
 
3860
    return False 
3861
 
3862
 
766 rajveer 3863
def close_session():
3864
    if session.is_active:
3865
        print "session is active. closing it."
2783 chandransh 3866
        session.close()
3376 rajveer 3867
 
3868
def is_alive():
3869
    try:
3870
        session.query(Transaction.id).limit(1).one()
3871
        return True
3872
    except:
6000 mandeep.dh 3873
        return False
3874
 
3875
if __name__ == '__main__':
3876
    print 'Inside main'
3877
    DataService.initialize()
3878
    a = MobileRechargeOrder()
3879
    a.mobileNumber = 89564785666
3880
    session.commit()