Subversion Repositories SmartDukaan

Rev

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