Subversion Repositories SmartDukaan

Rev

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