Subversion Repositories SmartDukaan

Rev

Rev 4454 | Rev 4479 | 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
2536 chandransh 8
from elixir import *
3499 mandeep.dh 9
from reportlab.lib.units import inch
2536 chandransh 10
from reportlab.pdfgen.canvas import Canvas
11
from reportlab.rl_config import defaultPageSize
3499 mandeep.dh 12
from shop2020.clients.CatalogClient import CatalogClient
13
from shop2020.clients.HelperClient import HelperClient
14
from shop2020.clients.LogisticsClient import LogisticsClient
2747 chandransh 15
from shop2020.clients.PaymentClient import PaymentClient
3499 mandeep.dh 16
from shop2020.clients.UserClient import UserClient
17
from shop2020.clients.WarehouseClient import WarehouseClient
4421 mandeep.dh 18
from shop2020.config.client.ConfigClient import ConfigClient
3499 mandeep.dh 19
from shop2020.model.v1 import order
20
from shop2020.model.v1.order.impl.DataService import Transaction, LineItem, \
4435 mandeep.dh 21
    Order, BatchNoGenerator, InvoiceIDGenerator, TransactionRequiringExtraProcessing, \
22
    OrderInventory, Alert
2687 chandransh 23
from shop2020.model.v1.order.impl.model.ReturnOrder import ReturnOrder
3499 mandeep.dh 24
from shop2020.thriftpy.logistics.ttypes import DeliveryType
4435 mandeep.dh 25
from shop2020.thriftpy.model.v1.catalog.ttypes import BillingType, \
26
    InventoryServiceException
27
from shop2020.thriftpy.model.v1.order.ttypes import TransactionServiceException, \
28
    TransactionStatus, OrderStatus, DelayReason, ExtraTransactionProcessingType, \
29
    HotspotAction, TimeoutSummary
4421 mandeep.dh 30
from shop2020.thriftpy.payments.ttypes import PaymentException
3499 mandeep.dh 31
from shop2020.thriftpy.warehouse.ttypes import PurchaseOrder, \
4365 mandeep.dh 32
    LineItem as POLineItem, ScanType
3499 mandeep.dh 33
from shop2020.utils.EmailAttachmentSender import mail, get_attachment_part
855 rajveer 34
from shop2020.utils.Utils import to_py_date, to_java_date
4008 mandeep.dh 35
from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound
3499 mandeep.dh 36
from sqlalchemy.sql import func
37
from sqlalchemy.sql.expression import and_, or_, desc, not_, distinct
38
from string import Template
39
from textwrap import dedent
40
import datetime
41
import logging
42
import os
4421 mandeep.dh 43
import sys
4435 mandeep.dh 44
import time
4421 mandeep.dh 45
import traceback
3499 mandeep.dh 46
logging.basicConfig(level=logging.DEBUG)
104 ashish 47
 
3499 mandeep.dh 48
 
4019 chandransh 49
try:
50
    config_client = ConfigClient()
51
    PREPAID_SHIPPING_CUTOFF_TIME = int(config_client.get_property('delivery_cutoff_time'))
52
except Exception as ex:
53
    print "[ERROR] Unexpected config error:", sys.exc_info()[0]
54
    PREPAID_SHIPPING_CUTOFF_TIME = 15
3499 mandeep.dh 55
 
4019 chandransh 56
COD_SHIPPING_CUTOFF_TIME = 12
3499 mandeep.dh 57
 
2819 chandransh 58
PAGE_HEIGHT=defaultPageSize[1]
59
PAGE_WIDTH=defaultPageSize[0]
60
 
3499 mandeep.dh 61
ORDER_STATUS_TO_USER_TRUST_LEVEL_DELTA_DICT = {
62
        OrderStatus.SALES_RET_RESHIPPED     :  3,
63
        OrderStatus.SALES_RETURN_IN_TRANSIT : -5,
64
        OrderStatus.DELIVERY_SUCCESS        :  1,
3578 mandeep.dh 65
        OrderStatus.DOA_CERT_INVALID        : -5
3499 mandeep.dh 66
}
67
 
104 ashish 68
def get_new_transaction():
69
    transaction = Transaction()
70
    transaction.createdOn = datetime.datetime.now()
738 chandransh 71
    transaction.status = TransactionStatus.INIT
104 ashish 72
    transaction.status_message = "New transaction"
483 rajveer 73
    return transaction
74
 
75
def create_order(t_order):
76
    order = Order()
77
    if t_order.warehouse_id:
78
        order.warehouse_id = t_order.warehouse_id
509 rajveer 79
    if t_order.logistics_provider_id:    
483 rajveer 80
        order.logistics_provider_id = t_order.logistics_provider_id
81
        order.airwaybill_no = t_order.airwaybill_no
82
        order.tracking_id = t_order.tracking_id
4009 chandransh 83
    if t_order.expected_shipping_time:
4019 chandransh 84
        order.expected_shipping_time = to_py_date(t_order.expected_shipping_time).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4102 chandransh 85
        order.promised_shipping_time = order.expected_shipping_time
4009 chandransh 86
    if t_order.expected_delivery_time:
483 rajveer 87
        order.expected_delivery_time = to_py_date(t_order.expected_delivery_time)
4102 chandransh 88
        order.promised_delivery_time = order.expected_delivery_time
483 rajveer 89
    if t_order.customer_id:    
90
        order.customer_id = t_order.customer_id
91
        order.customer_name = t_order.customer_name
860 rajveer 92
        order.customer_city = t_order.customer_city
93
        order.customer_state = t_order.customer_state
483 rajveer 94
        order.customer_mobilenumber = t_order.customer_mobilenumber
95
        order.customer_pincode = t_order.customer_pincode
738 chandransh 96
        order.customer_address1 = t_order.customer_address1
97
        order.customer_address2 = t_order.customer_address2
483 rajveer 98
        order.customer_email = t_order.customer_email
99
    #new status and status description to be added    
716 rajveer 100
    order.status = t_order.status  
483 rajveer 101
    order.statusDescription = t_order.statusDescription
102
    order.total_amount = t_order.total_amount
103
    order.total_weight = t_order.total_weight
104
    if t_order.created_timestamp:
105
        order.created_timestamp = to_py_date(t_order.created_timestamp)
106
    else:
685 chandransh 107
        order.created_timestamp = datetime.datetime.now()
483 rajveer 108
    return order
109
 
110
def create_transaction(t_transaction):
111
    t_orders = t_transaction.orders
112
    if not t_orders:
113
        raise TransactionServiceException(101, "Orders missing from the transaction")
114
    transaction = get_new_transaction()
115
    transaction.customer_id = t_transaction.customer_id 
116
    transaction.shopping_cart_id = t_transaction.shoppingCartid
2219 varun.gupt 117
    transaction.coupon_code = t_transaction.coupon_code
2815 vikas 118
    transaction.session_source = t_transaction.sessionSource
119
    transaction.session_start_time = to_py_date(t_transaction.sessionStartTime)
3858 vikas 120
    transaction.first_source = t_transaction.firstSource
121
    transaction.first_source_start_time = to_py_date(t_transaction.firstSourceTime)
483 rajveer 122
 
123
    for t_order in t_orders:
124
        order = create_order(t_order)
125
        order.transaction = transaction 
126
        for line_item in t_order.lineitems:
127
            litem = LineItem()
716 rajveer 128
            litem.item_id = line_item.item_id
963 chandransh 129
            litem.productGroup = line_item.productGroup
483 rajveer 130
            litem.brand = line_item.brand
131
            if line_item.model_number:
132
                litem.model_number = line_item.model_number
133
            if line_item.model_name:    
134
                litem.model_name = line_item.model_name
908 chandransh 135
            if line_item.color:
136
                litem.color = line_item.color
483 rajveer 137
            if line_item.extra_info:
138
                litem.extra_info = line_item.extra_info
139
            litem.quantity = line_item.quantity
140
            litem.unit_price = line_item.unit_price
141
            litem.unit_weight = line_item.unit_weight
142
            litem.total_price = line_item.total_price
143
            litem.total_weight = line_item.total_weight
4283 anupam.sin 144
            '''litem.transfer_price = line_item.transfer_price'''
4172 rajveer 145
            litem.dealText = line_item.dealText
4295 varun.gupt 146
            litem.warranty_expiry_timestamp = to_py_date(line_item.warrantry_expiry_timestamp)
1976 varun.gupt 147
 
483 rajveer 148
            """
149
            if line_item.addedOn:
150
                litem.added_on = to_py_date(line_item.addedOn)
151
            else:
152
                litem.added_on = datetime.datetime.now()
153
            """
154
            litem.order = order
155
    session.commit()
156
    return transaction.id
157
 
158
def get_transaction(transaction_id):
159
    transaction = Transaction.get_by(id=transaction_id)
160
    if not transaction:
161
        raise TransactionServiceException(108, "no such transaction")
162
 
163
    return transaction
164
 
165
def get_transactions_for_customer(customer_id, from_date, to_date, status):
166
    if not customer_id:
167
        raise TransactionServiceException(101, "bad customer id")
168
    query = Transaction.query.filter(Transaction.customer_id == customer_id)
1493 vikas 169
    if status is not None:
483 rajveer 170
        query = query.filter(Transaction.status == status)
171
    if from_date:
172
        query = query.filter(Transaction.createdOn >from_date)
173
    if to_date:
174
        query = query.filter(Transaction.createdOn < to_date)
175
    return query.all()
176
 
177
def get_transactions_for_shopping_cart_id(shopping_cart_id):
178
    transactions = Transaction.query.filter_by(shopping_cart_id=shopping_cart_id).all()
179
    return transactions
180
 
181
def get_transaction_status(transaction_id):
182
    if not transaction_id:
183
        raise TransactionServiceException(101, "bad transaction id")
184
 
185
    transaction = get_transaction(transaction_id)
186
 
187
    return transaction.status
188
 
189
def change_transaction_status(transaction_id, new_status, description):
190
    transaction = get_transaction(transaction_id)
191
    transaction.status = new_status
192
    transaction.status_message = description
690 chandransh 193
    if new_status == TransactionStatus.FAILED:
194
        for order in transaction.orders:
195
            order.status = OrderStatus.PAYMENT_FAILED
2339 chandransh 196
            order.statusDescription = "Payment Failed"
4247 rajveer 197
    elif new_status == TransactionStatus.AUTHORIZED or new_status == TransactionStatus.FLAGGED:
690 chandransh 198
        for order in transaction.orders:
4247 rajveer 199
            if new_status == TransactionStatus.AUTHORIZED:
200
                order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
201
                order.statusDescription = "Submitted to warehouse"
202
            elif new_status == TransactionStatus.FLAGGED:
203
                order.status = OrderStatus.PAYMENT_FLAGGED
204
                order.statusDescription = "Payment flagged by gateway"
3092 chandransh 205
            order.cod = False
743 rajveer 206
            #After we got payment success, we will set logistics info also 
207
            logistics_client = LogisticsClient().get_client()
208
            #FIXME line item is only one now. If multiple will come, need to fix.
871 chandransh 209
            item_id = order.lineitems[0].item_id
3044 chandransh 210
            logistics_info = logistics_client.getLogisticsInfo(order.customer_pincode, item_id, DeliveryType.PREPAID)
743 rajveer 211
 
1730 ankur.sing 212
            current_time = datetime.datetime.now()
213
            logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
4009 chandransh 214
            logistics_info.shippingTime = adjust_delivery_time(current_time, logistics_info.shippingTime)
1730 ankur.sing 215
 
743 rajveer 216
            order.warehouse_id = logistics_info.warehouseId
217
            order.logistics_provider_id = logistics_info.providerId
218
            order.airwaybill_no = logistics_info.airway_billno
219
            order.tracking_id = order.airwaybill_no
4019 chandransh 220
            order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4102 chandransh 221
            order.promised_shipping_time = order.expected_shipping_time
1730 ankur.sing 222
            order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
3986 chandransh 223
            order.promised_delivery_time = order.expected_delivery_time 
871 chandransh 224
 
3133 rajveer 225
            catalog_client = CatalogClient().get_client()
871 chandransh 226
            catalog_client.reserveItemInWarehouse(item_id, logistics_info.warehouseId, order.lineitems[0].quantity)
1351 varun.gupt 227
 
4316 anupam.sin 228
            try:
229
                item_pricing = catalog_client.getItemPricing(item_id, -1)
230
                order.lineitems[0].transfer_price = item_pricing.transferPrice
231
            except:
232
                print "Not able to get transfer price. Skipping"
3064 chandransh 233
    elif new_status == TransactionStatus.COD_IN_PROCESS:
234
        for order in transaction.orders:
235
            order.status = OrderStatus.INIT
236
            order.statusDescription = "Verification Pending"
237
            order.cod = True
238
            #After we got payment success, we will set logistics info also 
239
            logistics_client = LogisticsClient().get_client()
240
            #FIXME line item is only one now. If multiple will come, need to fix.
241
            item_id = order.lineitems[0].item_id
242
            logistics_info = logistics_client.getLogisticsInfo(order.customer_pincode, item_id, DeliveryType.COD)
243
 
244
            current_time = datetime.datetime.now()
245
            logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
4009 chandransh 246
            logistics_info.shippingTime = adjust_delivery_time(current_time, logistics_info.shippingTime)
3064 chandransh 247
 
248
            order.warehouse_id = logistics_info.warehouseId
249
            order.logistics_provider_id = logistics_info.providerId
250
            order.airwaybill_no = logistics_info.airway_billno
251
            order.tracking_id = order.airwaybill_no
4019 chandransh 252
            order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=COD_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4102 chandransh 253
            order.promised_shipping_time = order.expected_shipping_time
3064 chandransh 254
            order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
3986 chandransh 255
            order.promised_delivery_time = order.expected_delivery_time
3064 chandransh 256
 
3133 rajveer 257
            catalog_client = CatalogClient().get_client()
3064 chandransh 258
            catalog_client.reserveItemInWarehouse(item_id, logistics_info.warehouseId, order.lineitems[0].quantity)
259
 
4316 anupam.sin 260
            try:
261
                item_pricing = catalog_client.getItemPricing(item_id, -1)
262
                order.lineitems[0].transfer_price = item_pricing.transferPrice        
263
            except:
264
                print "Not able to get transfer price. Skipping"
483 rajveer 265
    session.commit()
3499 mandeep.dh 266
 
3530 mandeep.dh 267
    try:
268
        if new_status == TransactionStatus.COD_IN_PROCESS:
4008 mandeep.dh 269
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
270
            transaction_requiring_extra_processing.category = 'COD_VERIFICATION'
271
            transaction_requiring_extra_processing.transaction_id = transaction_id
272
            session.commit()
4247 rajveer 273
        elif new_status == TransactionStatus.FLAGGED:
274
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
275
            transaction_requiring_extra_processing.category = 'PAYMENT_FLAGGED'
276
            transaction_requiring_extra_processing.transaction_id = transaction_id
277
            session.commit()
3530 mandeep.dh 278
    except Exception as e:
4065 mandeep.dh 279
        print "Error inserting transaction Id: " + str(transaction_id) + " due to " + str(e)
3499 mandeep.dh 280
 
483 rajveer 281
    return True
282
 
3994 chandransh 283
def __get_holidays(start_time=0L, to_time=-1L):
284
    '''
285
    Get the list of all public holidays since start time including the
286
    day of start time if it's a holiday
287
    '''
288
    midnightTime = start_time.replace(hour=0, minute=0, second=0, microsecond=0)
1730 ankur.sing 289
    fromDate = to_java_date(midnightTime)
290
    logistics_client = LogisticsClient().get_client()
3994 chandransh 291
    #TODO: This list should be cached after the first call.
292
    #Calling logistics server is also unnecessary.
293
    holidays = logistics_client.getHolidays(fromDate, to_time)
294
    holidays = [to_py_date(holiday).date() for holiday in holidays]
295
    return holidays
296
 
297
 
298
def adjust_delivery_time(start_time, delivery_days):
299
    '''
300
    Returns the actual no. of days which will pass while 'delivery_days'
301
    no. of business days will pass since 'order_time'. 
302
    '''
303
    start_date = start_time.date()
304
    end_date = start_date + datetime.timedelta(days = delivery_days)
305
    holidays = __get_holidays(start_time)
1730 ankur.sing 306
 
3994 chandransh 307
    while start_date <= end_date:
308
        if start_date.weekday() == 6 or start_date in holidays:
1730 ankur.sing 309
            delivery_days = delivery_days + 1
3994 chandransh 310
            end_date = end_date + datetime.timedelta(days = 1)
4088 chandransh 311
        start_date = start_date + datetime.timedelta(days = 1)
3994 chandransh 312
 
1730 ankur.sing 313
    return delivery_days
314
 
1398 varun.gupt 315
def enqueue_transaction_info_email(transaction_id):
1262 varun.gupt 316
    transaction = get_transaction(transaction_id)
317
 
318
    html_header = """
319
<html>
320
<body>
321
<div>
322
    <p>
323
        Hello,<br /><br />
1742 rajveer 324
        Thanks for placing order with us. Following are the details of your order:
1262 varun.gupt 325
    </p>
326
    <div>
327
        Order Date: $order_date
1569 varun.gupt 328
    </div>
1262 varun.gupt 329
    <div>
330
        <table>
1569 varun.gupt 331
        <tr><td colspan="5"><hr /></td></tr>
332
        <tr><td colspan="5" align="left"><b>Order Details</b></td></tr>
333
        <tr><td colspan="5"><hr /></td></tr>
1262 varun.gupt 334
        <tr>
2050 varun.gupt 335
            <th width="100">Order No.</th>
336
            <th>Product</th>
337
            <th width="100">Quantity</th>
338
            <th width="100">Unit Price</th>
339
            <th width="100">Amount</th>
1262 varun.gupt 340
        </tr>
341
    """
1398 varun.gupt 342
    user_email = ""
1262 varun.gupt 343
    total_amount = 0.0
344
    html_table = ""
1569 varun.gupt 345
    order_date = datetime.datetime.now()
1262 varun.gupt 346
 
347
    for order in transaction.orders:
348
        order_date = order.created_timestamp
1398 varun.gupt 349
        user_email = order.customer_email
1569 varun.gupt 350
        html_tr = "<tr><td align='center'>" + str(order.id) + "</td>"
1262 varun.gupt 351
 
352
        for lineitem in order.lineitems:
2050 varun.gupt 353
            lineitem_total_price = round(lineitem.total_price, 2)
354
 
1276 chandransh 355
            html_tr += "<td>" + str(lineitem) + "</td>"
2050 varun.gupt 356
            html_tr += "<td align='center'>" + ("%.0f" % lineitem.quantity) + "</td>"
357
            html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem.unit_price) + "</td>"
358
            html_tr += "<td align='center'> Rs. " + ("%.2f" % lineitem_total_price) + "</td></tr>"
359
            total_amount += lineitem_total_price
1262 varun.gupt 360
            html_table += html_tr
2050 varun.gupt 361
 
1262 varun.gupt 362
    html_footer = """
363
        <tr><td colspan=5>&nbsp;</td></tr>
1565 varun.gupt 364
        <tr><td colspan=5><hr /></td></tr>
1262 varun.gupt 365
        <tr>
366
            <td colspan=4>Total Amount</td>
2050 varun.gupt 367
            <td> Rs. $total_amount</td>
1262 varun.gupt 368
        </tr>
1569 varun.gupt 369
        <tr><td colspan=5><hr /></td></tr>
4232 varun.gupt 370
        <tr>
371
            <td colspan=5>
372
                Note:- With the onset of dense fog in North India, it severely hits the operations of our courier partners. 
373
                We would like to mention that our deliveries might be slightly delayed. 
374
                We look forward to your support and understanding with regards to anticipated delays.
375
            </td>
376
        </tr>
1262 varun.gupt 377
        </table>
1569 varun.gupt 378
    </div>
1262 varun.gupt 379
    <p>
380
        Best Wishes,<br />
381
        Saholic Team
382
    </p>
383
</div>
384
</body>
385
</html>
386
    """
1398 varun.gupt 387
 
1569 varun.gupt 388
    dt = datetime.datetime.strptime(str(order_date), "%Y-%m-%d %H:%M:%S")
1565 varun.gupt 389
    formated_order_date = dt.strftime("%A, %d. %B %Y %I:%M%p")
390
 
1262 varun.gupt 391
    email_header = Template(html_header).substitute(dict(order_date = formated_order_date))
2050 varun.gupt 392
    email_footer = Template(html_footer).substitute(dict(total_amount = "%.2f" % total_amount))
1262 varun.gupt 393
 
1351 varun.gupt 394
    try:
395
        helper_client = HelperClient().get_client()
1398 varun.gupt 396
        helper_client.saveUserEmailForSending(user_email, "", "Saholic - Order Details", email_header + html_table + email_footer, str(transaction_id), "TransactionInfo")
1351 varun.gupt 397
        return True
398
    except Exception as e:
399
        print e
400
        return False
1262 varun.gupt 401
 
483 rajveer 402
def get_order(order_id):
403
    order = Order.get_by(id=order_id)
404
    if not order:
405
        raise TransactionServiceException(108, "no such order")
406
 
407
    return order
408
 
3014 chandransh 409
def get_orders_for_customer(customer_id, from_date, to_date, statuses):
483 rajveer 410
    if not customer_id:
411
        raise TransactionServiceException(101, "bad customer id")
412
    query = Order.query.filter(Order.customer_id == customer_id)
3014 chandransh 413
    if statuses:
414
        query = query.filter(Order.status.in_(tuple(statuses)))
483 rajveer 415
    if from_date:
416
        query = query.filter(Order.created_timestamp >from_date)
417
    if to_date:
418
        query = query.filter(Order.created_timestamp < to_date)
419
    return query.all()
420
 
1528 ankur.sing 421
def get_order_for_customer(order_id, customer_id):
422
    if not customer_id:
423
        raise TransactionServiceException(101, "bad customer id")
424
    order = get_order(order_id)
425
    if order.customer_id != customer_id:
426
        raise TransactionServiceException(108, "order: " + str(order_id) + " does not belong to customer: " + str(customer_id))
427
    return order
428
 
483 rajveer 429
def get_all_orders(status, from_date, to_date, warehouse_id):
430
    if not warehouse_id:
431
        raise TransactionServiceException(101, "bad warehouse id")
432
    query = Order.query.filter(Order.warehouse_id == warehouse_id)
433
    if status:
434
        query = query.filter(Order.status == status)
435
    if from_date:
436
        query = query.filter(Order.created_timestamp >from_date)
437
    if to_date:
438
        query = query.filter(Order.created_timestamp < to_date)
439
    return query.all()
440
 
4133 chandransh 441
def get_orders_in_batch(statuses=[], offset=0, limit=0, warehouse_id=None):
4296 rajveer 442
    query = Order.query
443
    if warehouse_id:
444
        query = query.filter(Order.warehouse_id == warehouse_id)
4133 chandransh 445
    if statuses:
446
        query = query.filter(Order.status.in_(statuses))
447
    query = query.offset(offset)
448
    if limit:
449
        query = query.limit(limit)
450
    return query.all()
451
 
452
def get_order_count(statuses=[], warehouse_id=None):
453
    if not warehouse_id:
454
        raise TransactionServiceException(101, "bad warehouse id")
455
    query = session.query(func.count(Order.id)).filter(Order.warehouse_id == warehouse_id)
456
    if statuses:
457
        query = query.filter(Order.status.in_(statuses))
458
    return query.scalar()
459
 
1382 varun.gupt 460
def get_returnable_orders_for_customer(customer_id, limit = None):
461
    if not customer_id:
462
        raise TransactionServiceException(101, "bad customer id")
463
 
464
    query = Order.query.filter(Order.customer_id == customer_id)
465
    query = query.filter(Order.status != OrderStatus.PAYMENT_FAILED)
466
    query = query.filter(Order.status != OrderStatus.REJECTED)
467
    query = query.filter(Order.status > OrderStatus.SHIPPED_FROM_WH)
468
    query = query.filter(or_(Order.status < OrderStatus.DELIVERY_SUCCESS, Order.delivery_timestamp < datetime.datetime.now() + datetime.timedelta(hours = 48)))
469
    orders = query.all()
470
    order_ids = [order.id for order in orders]
471
    return order_ids
472
 
473
def get_cancellable_orders_for_customer(customer_id, limit = None):
474
    if not customer_id:
475
        raise TransactionServiceException(101, "bad customer id")
476
 
477
    query = Order.query.filter(Order.customer_id == customer_id)
4343 varun.gupt 478
    query = query.filter(Order.status != OrderStatus.PAYMENT_FAILED)
1382 varun.gupt 479
    query = query.filter(Order.status != OrderStatus.REJECTED)
4288 anupam.sin 480
    query = query.filter(Order.status <= OrderStatus.BILLED)
1382 varun.gupt 481
    orders = query.all()
482
    return [order.id for order in orders]
483
 
995 varun.gupt 484
def get_orders_by_billing_date(status, start_billing_date, end_billing_date, warehouse_id):
485
    if not warehouse_id:
486
        raise TransactionServiceException(101, "bad warehouse id")
487
 
488
    query = Order.query.filter(Order.warehouse_id == warehouse_id)
489
 
490
    if status:
491
        query = query.filter(Order.status == status)
492
 
493
    if start_billing_date:
494
        query = query.filter(Order.billing_timestamp >= start_billing_date)
495
 
496
    if end_billing_date:
497
        query = query.filter(Order.billing_timestamp <= end_billing_date)
498
 
499
    return query.all()
500
 
3451 chandransh 501
def get_orders_by_shipping_date(from_shipping_date, to_shipping_date, provider_id, warehouse_id, cod):
502
    query = Order.query.filter(Order.cod == cod)
3427 chandransh 503
 
504
    if warehouse_id and warehouse_id != -1:
505
        query = query.filter(Order.warehouse_id == warehouse_id)
506
 
507
    if provider_id and provider_id != -1:
508
        query = query.filter(Order.logistics_provider_id == provider_id)
509
 
510
    if from_shipping_date:
511
        query = query.filter(Order.shipping_timestamp >= from_shipping_date)
512
 
513
    if to_shipping_date:
514
        query = query.filter(Order.shipping_timestamp <= to_shipping_date)
515
 
516
    query = query.order_by(Order.airwaybill_no)
517
 
518
    return query.all()
519
 
1528 ankur.sing 520
def get_orders_for_transaction(transaction_id, customer_id):
521
    orders = Order.query.filter_by(transaction_id=transaction_id, customer_id=customer_id).all()
522
 
1208 chandransh 523
    if not orders:
524
        raise TransactionServiceException(101, "No order for the transaction")
525
    return orders
526
 
1405 ankur.sing 527
def get_undelivered_orders(provider_id, warehouse_id):
528
    query = Order.query
529
    if provider_id != -1:
530
        query = query.filter_by(Order.logistics_provider_id == provider_id)
531
    if warehouse_id != -1:
532
        query = query.filter(Order.warehouse_id == warehouse_id)
533
    query = query.filter(and_(Order.status != OrderStatus.DELIVERY_SUCCESS, Order.status != OrderStatus.FAILED, \
534
                              Order.status > OrderStatus.SHIPPED_FROM_WH))
1479 ankur.sing 535
    #Get last midnight time instead of current time as courier company would have sent the 
536
    #delivery records updated at least till midnight and not by the current time.
537
    time_format = "%Y-%m-%d %H:%M:%S.%f"
538
    t = time.strptime(str(datetime.datetime.now()), time_format)
539
    midnightTime = datetime.datetime(*t[:3])
540
    query = query.filter(Order.expected_delivery_time < midnightTime)
1405 ankur.sing 541
    orders = query.all()
542
    return orders
543
 
1208 chandransh 544
def get_line_items_for_order(order_id):
545
    query = LineItem.query.filter(LineItem.order_id == order_id)
546
    return query.all()
547
 
483 rajveer 548
def change_order_status(self, orderId, status, description):
549
    order = get_order(orderId)
550
    order.status = status
551
    order.statusDescription = description
552
    session.commit()
3578 mandeep.dh 553
    update_trust_level(order)
554
    return True
3499 mandeep.dh 555
 
3578 mandeep.dh 556
def update_trust_level(order):
3530 mandeep.dh 557
    try:
3578 mandeep.dh 558
        trust_level_delta = 0
559
        if order.cod == True:
560
            trust_level_delta = ORDER_STATUS_TO_USER_TRUST_LEVEL_DELTA_DICT.get(order.status, 0)
561
 
3530 mandeep.dh 562
        if (trust_level_delta != 0):
563
            user_client = UserClient().get_client()
564
            user_client.increaseTrustLevel(order.customer_id, trust_level_delta)
565
    except Exception as e:
566
        print e
3499 mandeep.dh 567
 
483 rajveer 568
    return True
3064 chandransh 569
 
570
def verify_order(self, orderId):
571
    logging.info("Verifying order no: " + str(orderId))
572
    order = get_order(orderId)
573
    if order.status == OrderStatus.INIT:
574
        order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
575
        order.statusDescription = "Submitted for processing"
576
        order.verification_timestamp = datetime.datetime.now()
577
        session.commit()
578
        logging.info("Successfully verified order no: " + str(orderId))
579
        return True
580
    else:
581
        logging.warning("Verify called for order no." + str(orderId) +" which is not in verification pending state");
582
        return False
583
 
4285 rajveer 584
def accept_order(orderId):
3064 chandransh 585
    logging.info("Accepting order no: " + str(orderId))
921 rajveer 586
    order = get_order(orderId)
4306 rajveer 587
    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]:
3064 chandransh 588
        if not order.cod:
589
            if order.transaction.status == TransactionStatus.AUTHORIZED:
4421 mandeep.dh 590
                __capture_txn(order)
3010 chandransh 591
        order.status = OrderStatus.ACCEPTED
592
        order.statusDescription = "Order Accepted"
593
        order.accepted_timestamp = datetime.datetime.now()
594
        session.commit()
3064 chandransh 595
        logging.info("Successfully accepted the order no.:" + str(orderId))
3010 chandransh 596
        return True
597
    else:
3064 chandransh 598
        logging.warning("Accept called for the unacceptable order: " + str(orderId))
3010 chandransh 599
        return False
921 rajveer 600
 
4421 mandeep.dh 601
def __capture_txn(order):
602
    txnId = order.transaction_id
3010 chandransh 603
    logging.info("Capturing payment for merchant txn:" + str(txnId))
4421 mandeep.dh 604
    try:
605
        payment_client = PaymentClient().get_client()
606
        capture_result = payment_client.capturePayment(txnId)
607
        if capture_result:
608
            logging.info("Successfully captured payment for merchant txn:" + str(txnId))
609
            change_transaction_status(txnId, TransactionStatus.IN_PROCESS, "Payment received")
610
        else:
611
            raise TransactionServiceException(115, "Payment capture failed.")
612
    except PaymentException as e:
613
        if e.error_code == 106:
614
            order.status = OrderStatus.CAPTURE_IN_PROCESS
615
            session.commit()
616
            raise TransactionServiceException(122, "Unable to capture payment due to connection issue.")
617
        else:
618
            raise TransactionServiceException(115, "Payment capture failed.")
3010 chandransh 619
 
4285 rajveer 620
def add_billing_details(orderId, invoice_number, imeiNumber, itemNumber, billedBy, jacketNumber, billingType, vendorId):
621
    if jacketNumber is None or jacketNumber <= 0:
2783 chandransh 622
        raise TransactionServiceException(110, "Invalid jacket number")
623
 
4285 rajveer 624
    if billedBy is None or billedBy.strip() == "":
2783 chandransh 625
        raise TransactionServiceException(110, "Invalid Biller")
626
 
1149 chandransh 627
    order = Order.get_by(id=orderId)
628
    if not order:
2783 chandransh 629
        raise TransactionServiceException(101, "No order found for the given order id" + str(orderId))
630
 
4283 anupam.sin 631
    if order.status == OrderStatus.ACCEPTED:
4285 rajveer 632
        order.jacket_number = jacketNumber
633
        if billingType == BillingType.OURS:
634
            if itemNumber is None or itemNumber.strip() == "":
4283 anupam.sin 635
                raise TransactionServiceException(110, "Invalid item number")
636
 
637
            order.invoice_number = get_next_invoice_number()
638
            lineitem = order.lineitems[0]
4285 rajveer 639
            if lineitem.productGroup == "Handsets" and imeiNumber <= 0:
4283 anupam.sin 640
                raise TransactionServiceException(110, "No IMEI number supplied for a handset.")
4285 rajveer 641
            lineitem.item_number = itemNumber
642
            if imeiNumber > 0:
643
                lineitem.imei_number = str(imeiNumber)
4283 anupam.sin 644
        else:
645
            order.invoice_number = invoice_number
646
 
2842 chandransh 647
        order.status = OrderStatus.BILLED
648
        order.statusDescription = "Order Billed"
649
        order.billing_timestamp = datetime.datetime.now()
4285 rajveer 650
        order.billed_by = billedBy
2842 chandransh 651
        lineitem = order.lineitems[0]
4283 anupam.sin 652
        item_id = lineitem.item_id
4435 mandeep.dh 653
 
654
        # Not letting the billing process fail in cases where we are unable to 
655
        # fill in transfer price
656
        item_pricing = None
657
        try:
658
            catalog_client = CatalogClient().get_client()
659
            item_pricing = catalog_client.getItemPricing(item_id, vendorId)
660
        except InventoryServiceException as e:
661
            print sys.exc_info()[0]
662
            print e.message
663
 
4285 rajveer 664
        order.vendorId = vendorId
4283 anupam.sin 665
        lineitems = order.lineitems
666
        for lineitem in lineitems:
667
            catalog_client.reduceReservationCount(item_id, order.warehouse_id, lineitem.quantity)
4435 mandeep.dh 668
 
669
        # Putting NULL as transfer price for cases where its missing
670
        if item_pricing is not None:        
671
            order.lineitems[0].transfer_price = item_pricing.transferPrice
4283 anupam.sin 672
        session.commit()
4365 mandeep.dh 673
 
674
        # For Mahipal warehouse, we need to scan out items for every billed order
675
        if order.warehouse_id == 7:
676
            try:
677
                warehouse_client = WarehouseClient().get_client()
678
                warehouse_client.scanOut(itemNumber, str(imeiNumber), ScanType.SALE)
679
            except:
680
                print sys.exc_info()[0]
681
                print 'Could not scan out orders'
682
 
4283 anupam.sin 683
        return True
684
    else:
685
        return False
1149 chandransh 686
 
1220 chandransh 687
def batch_orders(warehouseId):
688
    if not warehouseId:
689
        raise TransactionServiceException(101, "bad warehouse id")
1225 chandransh 690
 
691
    batchno_generator = BatchNoGenerator()
692
    session.commit()
1220 chandransh 693
    query = Order.query.filter_by(warehouse_id=warehouseId)
4381 mandeep.dh 694
    query = query.filter(or_(Order.status == OrderStatus.SUBMITTED_FOR_PROCESSING, Order.status == OrderStatus.INVENTORY_LOW))
1220 chandransh 695
    query = query.order_by(Order.created_timestamp)
696
    pending_orders = query.all()
1225 chandransh 697
 
1220 chandransh 698
    serial_no = 1
699
    for order in pending_orders:
1276 chandransh 700
        order.batchNo = batchno_generator.id
1220 chandransh 701
        order.serialNo = serial_no
702
        serial_no += 1
703
    session.commit() 
704
    pending_orders = query.all()
705
    return pending_orders
706
 
707
def order_outofstock(orderId):
1911 chandransh 708
    '''
709
    Mark order as out of stock
710
    '''
1220 chandransh 711
    order = get_order(orderId)
712
    order.status = OrderStatus.INVENTORY_LOW
713
    order.statusDescription = "Low Inventory"
714
    order.outofstock_timestamp = datetime.datetime.now()
715
    session.commit()
716
    return True
717
 
3064 chandransh 718
def mark_orders_as_manifested(warehouse_id, provider_id, cod):
759 chandransh 719
    try:
3064 chandransh 720
        orders = Order.query.filter_by(warehouse_id = warehouse_id, logistics_provider_id = provider_id, status = OrderStatus.BILLED, cod = cod).all()
759 chandransh 721
        for order in orders:
4410 rajveer 722
            order.status = OrderStatus.MANIFESTED
723
            order.statusDescription = "Order manifested"
724
        session.commit()
725
        return True
726
    except:
727
        return False
728
 
729
def mark_orders_as_shipped_from_warehouse(warehouse_id, provider_id, cod):
730
    try:
731
        current_timestamp = datetime.datetime.now()
732
        orders = Order.query.filter_by(warehouse_id = warehouse_id, logistics_provider_id = provider_id, status = OrderStatus.MANIFESTED, cod = cod).all()
733
        for order in orders:
759 chandransh 734
            order.status = OrderStatus.SHIPPED_FROM_WH
987 chandransh 735
            order.statusDescription = "Order shipped from warehouse"
986 chandransh 736
            order.shipping_timestamp = current_timestamp
759 chandransh 737
        session.commit()
738
        return True
739
    except:
740
        return False
4410 rajveer 741
 
1113 chandransh 742
def mark_orders_as_picked_up(provider_id, pickup_details):
743
    for awb, pickup_timestamp in pickup_details.iteritems():
1135 chandransh 744
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
2684 chandransh 745
        if order == None or order.status != OrderStatus.SHIPPED_FROM_WH:
2220 chandransh 746
            #raise TransactionServiceException(102, "No order found for the awb: " + awb)
747
            continue
1113 chandransh 748
        order.status = OrderStatus.SHIPPED_TO_LOGST
749
        order.statusDescription = "Order picked up by Courier Company"
1246 chandransh 750
        order.pickup_timestamp = pickup_timestamp
1113 chandransh 751
    session.commit()
1217 chandransh 752
 
1132 chandransh 753
    current_time = datetime.datetime.now()
754
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
1246 chandransh 755
    from_datetime = to_datetime - datetime.timedelta(2)
1132 chandransh 756
    orders_not_picked_up = Order.query.filter_by(logistics_provider_id = provider_id).filter_by(status=OrderStatus.SHIPPED_FROM_WH).filter(Order.shipping_timestamp >= from_datetime).filter(Order.shipping_timestamp <= to_datetime).all()
1113 chandransh 757
    return orders_not_picked_up
758
 
1132 chandransh 759
def mark_orders_as_delivered(provider_id, delivered_orders):
760
    for awb, detail in delivered_orders.iteritems():
1135 chandransh 761
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
2684 chandransh 762
        if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST]:
1911 chandransh 763
            continue
764
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
1132 chandransh 765
        order.status = OrderStatus.DELIVERY_SUCCESS
766
        order.statusDescription = "Order delivered"
767
        order.delivery_timestamp, order.receiver = detail.split('|')
3578 mandeep.dh 768
        update_trust_level(order)
1132 chandransh 769
    session.commit()
770
 
1135 chandransh 771
def mark_orders_as_failed(provider_id, returned_orders):
772
    for awb, detail in returned_orders.iteritems():
773
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
2684 chandransh 774
        if order == None or order.status not in [OrderStatus.SHIPPED_FROM_WH, OrderStatus.SHIPPED_TO_LOGST]:
775
            continue
776
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
777
        order.status = OrderStatus.SALES_RETURN_IN_TRANSIT
1135 chandransh 778
        order.delivery_timestamp, reason = detail.split('|')
779
        order.statusDescription = "Order Returned to Origin:" + reason
3578 mandeep.dh 780
        update_trust_level(order)
1911 chandransh 781
    session.commit()
1135 chandransh 782
 
1246 chandransh 783
def update_non_delivery_reason(provider_id, undelivered_orders):
784
    for awb, reason in undelivered_orders.iteritems():
785
        order = Order.query.filter_by(airwaybill_no=awb, logistics_provider_id = provider_id).first()
786
        if order == None:
2684 chandransh 787
            continue
788
            #raise TransactionServiceException(103, "No order found for the awb:" + awb)
1246 chandransh 789
        order.statusDescription = reason
790
    session.commit()
791
 
4444 rajveer 792
def get_alerts(type, warehouseId, status, timestamp):
793
    query = Alert.query.filter(Alert.warehouseId == warehouseId)
794
    if type != -1 :
795
        query = query.filter(Alert.type == type)
796
    if status != -1 :
797
        query = query.filter(Alert.status == status)
4394 rajveer 798
    if timestamp:
799
        query = query.filter(Alert.timestamp >= to_py_date(timestamp))
4412 rajveer 800
    query = query.order_by(desc(Alert.timestamp))
483 rajveer 801
    return query.all()
802
 
4447 rajveer 803
def add_alert(type, warehouseId, description):
4394 rajveer 804
    alert = Alert()
805
    alert.type = type
806
    alert.status = 1
807
    alert.description = description
808
    alert.timestamp = datetime.datetime.now()
4447 rajveer 809
    alert.warehouseId = warehouseId
4444 rajveer 810
    session.commit()
766 rajveer 811
 
4444 rajveer 812
def mark_alerts_as_seen(warehouseId):
813
    query = Alert.query.filter(Alert.warehouseId == warehouseId).filter(Alert.status == 1)
814
    alerts = query.all()
815
    for alert in alerts:
816
        alert.status = 0    
817
    session.commit()
818
 
1596 ankur.sing 819
def get_valid_order_count():
2851 chandransh 820
    '''
821
    Returns the number of orders which we processed. The reshipped orders are not
822
    counted since there'll always be a new compensating order present.
823
    '''
4452 rajveer 824
    return Order.query.filter(Order.status >= OrderStatus.SUBMITTED_FOR_PROCESSING).filter(not_(Order.status.in_((OrderStatus.SALES_RET_RESHIPPED, OrderStatus.DOA_INVALID_RESHIPPED, OrderStatus.DOA_VALID_RESHIPPED, OrderStatus.CANCELED)))).count()
1596 ankur.sing 825
 
1627 ankur.sing 826
def get_cust_count_with_successful_txn():
2851 chandransh 827
    '''
828
    Returns the number of distinct customers who have done successful transactions.
829
    It uses the sqlalchemy func module to construct an efficient query.
830
    '''
831
    return session.query(func.count(distinct(Order.customer_id))).filter(Order.status >= OrderStatus.SUBMITTED_FOR_PROCESSING).scalar()
1627 ankur.sing 832
 
1730 ankur.sing 833
def get_valid_orders_amount_range():
2851 chandransh 834
    '''
835
    Returns a list containing the minimum and maximum amount across all
836
    orders placed with us. No need to consider reshippped orders for min/max
837
    amount of orders.
838
    '''
839
    return session.query(func.min(Order.total_amount), func.max(Order.total_amount)).filter(Order.status >= OrderStatus.SUBMITTED_FOR_PROCESSING).one()
1627 ankur.sing 840
 
1886 ankur.sing 841
def get_valid_orders(limit):
2851 chandransh 842
    '''
843
    Returns the last limit number of valid orders placed with us. Ignored the
844
    limit if it's passed as 0. Raises an excpetion if the supplied limit is 
845
    less than 0. 
846
    '''
847
    if limit < 0:
848
        raise TransactionServiceException(101, "Invalid argument limit " + limit)
3072 chandransh 849
    query = Order.query.filter(Order.status >= OrderStatus.INIT)
1886 ankur.sing 850
    query = query.order_by(desc(Order.created_timestamp))
2851 chandransh 851
    if limit != 0:
1886 ankur.sing 852
        query = query.limit(limit)
853
    return query.all()
854
 
2783 chandransh 855
def get_next_invoice_number():
856
    entity_id = InvoiceIDGenerator.query.with_lockmode("update").one()
857
    invoice_number = entity_id.id + 1
858
    entity_id.id = invoice_number
859
    return invoice_number
860
 
2536 chandransh 861
def toggle_doa_flag(order_id):
862
    order = get_order(order_id)
863
    if(order.doaFlag):
864
        order.doaFlag = False
865
    else:
866
        order.doaFlag = True
867
    session.commit()
868
    return order.doaFlag
869
 
4454 rajveer 870
def mark_order_doa_request_received(orderId):
871
    order = get_order(orderId)
872
    if order.status != OrderStatus.DELIVERY_SUCCESS:
873
        return False
874
    order.status = OrderStatus.DOA_REQUEST_RECEIVED
4459 rajveer 875
    order.statusDescription = "DOA Request Received"
4454 rajveer 876
    session.commit()
877
    return True
878
 
879
def mark_order_doa_request_authorized(orderId, isAuthorized):
880
    order = get_order(orderId)
881
    if order.status != OrderStatus.DOA_REQUEST_RECEIVED:
882
        return False
883
    if isAuthorized:
884
        order.status = OrderStatus.DOA_REQUEST_AUTHORIZED
4459 rajveer 885
        order.statusDescription = "DOA Request Authorized"
4454 rajveer 886
    else:
887
        order.status = OrderStatus.DELIVERY_SUCCESS
4459 rajveer 888
        order.statusDescription = "Order delivered"
4454 rajveer 889
    session.commit()
890
    return True
891
 
2536 chandransh 892
def request_pickup_number(order_id):
893
    order = get_order(order_id)
4452 rajveer 894
    if order.status != OrderStatus.DELIVERY_SUCCESS and order.status != OrderStatus.DOA_PICKUP_REQUEST_RAISED:
2536 chandransh 895
        return False
896
    from_user = 'cnc.center@shop2020.in'
897
    from_pwd = '5h0p2o2o'
898
 
4227 rajveer 899
    subject = "Pickup request from " + order.customer_city
2536 chandransh 900
    try:
3133 rajveer 901
        catalog_client = CatalogClient().get_client()
2536 chandransh 902
        warehouse = catalog_client.getWarehouse(order.warehouse_id)
903
 
904
        logistics_client = LogisticsClient().get_client()
905
        provider = logistics_client.getProvider(order.logistics_provider_id)
3051 chandransh 906
        to_addr = provider.details[DeliveryType.PREPAID].email
2536 chandransh 907
        raw_message = '''
908
        Dear Sir/Madam,
909
 
2559 chandransh 910
        Kindly arrange a pickup today from %(customer_city)s. Pickup and delivery addresses are mentioned below.
2536 chandransh 911
 
912
        Pickup CODE: %(provider_code)s
913
        Total Weight: %(order_weight)s kg
914
 
2559 chandransh 915
        Pickup Address:
2536 chandransh 916
 
917
        %(customer_address)s
918
 
919
        Delivery Address:
920
 
921
        %(warehouse_executive)s
922
        %(warehouse_address)s
923
        PIN %(warehouse_pin)s
924
        Phone: %(warehouse_phone)s
925
 
926
        Thanks and Regards
4227 rajveer 927
        Sandeep Sachdeva
2536 chandransh 928
        '''
2559 chandransh 929
        message = dedent(raw_message) % { 'customer_city' : order.customer_city,
3051 chandransh 930
                                          'provider_code' : provider.details[DeliveryType.PREPAID].accountNo,
2536 chandransh 931
                                          'order_weight' : str(order.total_weight),
932
                                          'customer_address' : __get_order_address(order),
933
                                          'warehouse_executive' : 'Parmod Kumar',
934
                                          'warehouse_address' : warehouse.location,
935
                                          'warehouse_pin' : warehouse.pincode,
936
                                          'warehouse_phone': '9971573026'}
4227 rajveer 937
        #FIXME Email is hardcoded for bluedart
938
        mail(from_user, from_pwd, [to_addr, 'sonikap@bluedart.com'], subject, message)
4452 rajveer 939
        order.status = OrderStatus.DOA_PICKUP_REQUEST_RAISED
2536 chandransh 940
        order.statusDescription = "Pick up requested for DOA"
941
        session.commit()
942
        return True
943
    except Exception as e:
944
        print sys.exc_info()[0]
945
        return False
946
 
947
def authorize_pickup(order_id, pickup_number):
948
    order = get_order(order_id)
4452 rajveer 949
    if order.status != OrderStatus.DOA_PICKUP_REQUEST_RAISED:
2536 chandransh 950
        return False
951
    from_user = 'help@shop2020.in'
952
    from_pwd = '5h0p2o2o'
953
    subject = 'Pickup details'
954
    filename = "/tmp/" +  str(order.id) +".pdf" 
955
    try:
3133 rajveer 956
        catalog_client = CatalogClient().get_client()
2536 chandransh 957
        warehouse = catalog_client.getWarehouse(order.warehouse_id)
958
 
959
        logistics_client = LogisticsClient().get_client()
960
        provider = logistics_client.getProvider(order.logistics_provider_id)
961
        provider_name = provider.name
962
 
963
        to_addr = order.customer_email
964
        today = date.today().strftime("%d-%B-%Y (%A)")
965
 
966
        raw_message = '''
967
        Dear Mr. %(customer_name)s,
968
 
969
        In reference to the return of order ID %(order_id)d. We request you to kindly pack the box properly with bubble wrap, and send it along with the original "DOA" certificate and the invoice.
970
        Take a printout of the attachment in this mail which contains the delivery address and paste it on the packed parcel.
971
        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.
972
 
973
        Pickup will happen from the below mentioned address:-
974
 
975
        %(customer_address)s
976
 
977
        Do let us know in case of any concerns
978
 
979
        Thanks & Regards
980
        Saholic Team
981
        '''
982
        message = dedent(raw_message) % {'customer_name' : order.customer_name,
983
                                         'order_id' : order_id,
984
                                         'provider_name' : provider_name,
985
                                         'date': today,
986
                                         'pickup_request_no' : pickup_number,
987
                                         'customer_address' : __get_order_address(order)}
988
        print message
989
        __generate_return_advice(order, warehouse, provider, filename)
3527 chandransh 990
        mail(from_user, from_pwd, [to_addr], subject, message, [get_attachment_part(filename)])
2536 chandransh 991
        order.pickupRequestNo = pickup_number
4452 rajveer 992
        order.status = OrderStatus.DOA_PICKUP_CONFIRMED
993
        order.statusDescription = "DOA pick up confirmed"
2764 chandransh 994
        order.doa_auth_timestamp = datetime.datetime.now()
2536 chandransh 995
        session.commit()
996
        return True
997
    except Exception as e:
4234 rajveer 998
        print sys.exc_info()
999
	traceback.print_tb(sys.exc_info()[2])
2536 chandransh 1000
        return False
1001
    finally:
1002
        if os.path.exists(filename):
1003
            os.remove(filename)
1004
 
2764 chandransh 1005
def mark_doas_as_picked_up(provider_id, pickup_details):
1006
    for pickupNo, doa_pickup_timestamp in pickup_details.iteritems():
1007
        order = Order.query.filter_by(pickupRequestNo=pickupNo, logistics_provider_id = provider_id).first()
4452 rajveer 1008
        if order == None or order.status != OrderStatus.DOA_PICKUP_CONFIRMED:
2764 chandransh 1009
            #raise TransactionServiceException(102, "No order found for the awb: " + awb)
1010
            continue
1011
        order.status = OrderStatus.DOA_RETURN_IN_TRANSIT
1012
        order.statusDescription = "DOA picked up by Courier Company"
1013
        order.doa_pickup_timestamp = doa_pickup_timestamp
1014
    session.commit()
1015
 
1016
    current_time = datetime.datetime.now()
1017
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
1018
    from_datetime = to_datetime - datetime.timedelta(2)
4452 rajveer 1019
    orders_not_picked_up = Order.query.filter_by(logistics_provider_id = provider_id).filter_by(status=OrderStatus.DOA_PICKUP_CONFIRMED).filter(Order.doa_auth_timestamp >= from_datetime).filter(Order.doa_auth_timestamp <= to_datetime).all()
2764 chandransh 1020
    return orders_not_picked_up
1021
 
2616 chandransh 1022
def receive_return(order_id):
2591 chandransh 1023
    order = get_order(order_id)
4365 mandeep.dh 1024
 
1025
    # FIXME: For Mahipal warehouse, we need to scan in items for every sales return
1026
    if order.warehouse_id == 7:
1027
        try:
1028
            warehouse_client = WarehouseClient().get_client()
1029
            lineitem = order.lineitems[0]
1030
 
1031
            # Inventory should not be updated for DOA returns 
1032
            if order.status == OrderStatus.SALES_RETURN_IN_TRANSIT:
1033
                warehouse_client.scanIn(order.id, lineitem.item_number, lineitem.imei_number, ScanType.SALE_RET)
1034
                catalog_client = CatalogClient().get_client()
1035
                catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity)
4452 rajveer 1036
            elif order.status in [OrderStatus.DOA_PICKUP_CONFIRMED, OrderStatus.DOA_RETURN_IN_TRANSIT]:
4365 mandeep.dh 1037
                warehouse_client.scanIn(order.id, lineitem.item_number, lineitem.imei_number, ScanType.DOA_IN)
1038
        except:
1039
            print sys.exc_info()[0]
1040
            print 'Could not scan in sales return orders'
1041
 
4452 rajveer 1042
    if order.status in [OrderStatus.DOA_PICKUP_CONFIRMED, OrderStatus.DOA_RETURN_IN_TRANSIT]:
1043
        order.status = OrderStatus.DOA_RECEIVED_PRESTINE
2616 chandransh 1044
        order.statusDescription = "DOA package received"
2764 chandransh 1045
        order.received_return_timestamp = datetime.datetime.now()
2616 chandransh 1046
        session.commit()
1047
    elif order.status == OrderStatus.SALES_RETURN_IN_TRANSIT :
1048
        order.status = OrderStatus.SALES_RET_RECEIVED
1049
        order.statusDescription = "Sales return received"
2764 chandransh 1050
        order.received_return_timestamp = datetime.datetime.now()
2616 chandransh 1051
        session.commit()
1052
    else:
2591 chandransh 1053
        return False
1054
    return True
1055
 
1056
def validate_doa(order_id, is_valid):
1057
    order = get_order(order_id)
4452 rajveer 1058
    if order.status != OrderStatus.DOA_RECEIVED_PRESTINE:
2591 chandransh 1059
        return False
1060
    if is_valid:
2609 chandransh 1061
        order.status = OrderStatus.DOA_CERT_VALID
1062
        order.statusDescription = "DOA Certificate Valid"
2591 chandransh 1063
    else:
1064
        order.status = OrderStatus.DOA_CERT_INVALID
1065
        order.statusDescription = "DOA Certificate Invalid"
3578 mandeep.dh 1066
        update_trust_level(order)
2591 chandransh 1067
    session.commit()
1068
    return True
1069
 
2628 chandransh 1070
def reship_order(order_id):
1071
    """
1072
    If the order is in SALES_RET_RECEIVED or DOA_CERT_INVALID state, it does the following:
1073
        1. Creates a new order for processing in the BILLED state. All billing information is saved.
1074
        2. Marks the current order as one of the final states SALES_RET_RESHIPPED and DOA_INVALID_RESHIPPED depending on what state the order started in.
1075
 
1076
    If the order is in DOA_CERT_VALID state, it does the following:
1077
        1. Creates a new order for processing in the SUBMITTED_FOR_PROCESSING state.
1078
        2. Creates a return order for the warehouse executive to return the DOA material.
4452 rajveer 1079
        3. Marks the current order as the final DOA_VALID_RESHIPPED state.
2628 chandransh 1080
 
1081
    Returns the id of the newly created order.
1082
 
1083
    Throws an exception if the order with the given id couldn't be found.
1084
 
1085
    Parameters:
1086
     - orderId
1087
    """
1088
    order = get_order(order_id)
1089
    if order.status == OrderStatus.SALES_RET_RECEIVED:
4220 rajveer 1090
        new_order = __clone_order(order, True, order.cod)
2628 chandransh 1091
        order.status = OrderStatus.SALES_RET_RESHIPPED
1092
        order.statusDescription = "Order Reshipped"
1093
        session.commit()
3578 mandeep.dh 1094
        update_trust_level(order)
2628 chandransh 1095
    elif order.status == OrderStatus.DOA_CERT_INVALID:
4220 rajveer 1096
        new_order = __clone_order(order, True, False)
2628 chandransh 1097
        order.status = OrderStatus.DOA_INVALID_RESHIPPED
1098
        order.statusDescription = "Order Reshipped"
1099
        session.commit()
1100
    elif order.status == OrderStatus.DOA_CERT_VALID:
4220 rajveer 1101
        new_order = __clone_order(order, False, False)
2628 chandransh 1102
        ret_order = __create_return_order(order)
4452 rajveer 1103
        order.status = OrderStatus.DOA_VALID_RESHIPPED
2628 chandransh 1104
        order.statusDescription = "Order Reshipped"
1105
        session.commit()
1106
    else:
1107
        raise TransactionServiceException(112, "This order can't be reshipped")
1108
 
2764 chandransh 1109
    order.reship_timestamp = datetime.datetime.now()
2628 chandransh 1110
    order.new_order_id = new_order.id
1111
    session.commit()
1112
 
1113
    return new_order.id
1114
 
3226 chandransh 1115
def refund_order(order_id, refunded_by, reason):
2628 chandransh 1116
    """
1117
    If the order is in SALES_RET_RECEIVED, DOA_CERT_VALID or DOA_CERT_INVALID state, it does the following:
1118
        1. Creates a refund request for batch processing.
1119
        2. Creates a return order for the warehouse executive to return the shipped material.
1120
        3. Marks the current order as SALES_RET_REFUNDED, DOA_VALID_REFUNDED or DOA_INVALID_REFUNDED final states.
1121
 
1122
    If the order is in SUBMITTED_FOR_PROCESSING or INVENTORY_LOW state, it does the following:
1123
        1. Creates a refund request for batch processing.
3226 chandransh 1124
        2. Cancels the reservation of the item in the warehouse.
1125
        3. Marks the current order as the REFUNDED final state.
2628 chandransh 1126
 
3226 chandransh 1127
    For all COD orders, if the order is in INIT, SUBMITTED_FOR_PROCESSING or INVENTORY_LOW state, it does the following:
1128
        1. Cancels the reservation of the item in the warehouse.
1129
        2. Marks the current order as CANCELED.
1130
 
1131
    In all cases, it updates the reason for cancellation or refund and the person who performed the action.
1132
 
1133
    Returns True if it is successful, False otherwise.
2628 chandransh 1134
 
1135
    Throws an exception if the order with the given id couldn't be found.
1136
 
1137
    Parameters:
3226 chandransh 1138
     - order_id
1139
     - refunded_by
1140
     - reason
2628 chandransh 1141
    """
3414 chandransh 1142
    logging.info("Refunding order id: " + str(order_id))
2628 chandransh 1143
    order = get_order(order_id)
3186 chandransh 1144
 
1145
 
3366 chandransh 1146
 
3226 chandransh 1147
    if order.cod:
3414 chandransh 1148
        logging.info("Refunding COD order with status " + str(order.status))
3366 chandransh 1149
        status_transition = {OrderStatus.SALES_RET_RECEIVED : OrderStatus.SALES_RET_REFUNDED,
1150
                     OrderStatus.DOA_CERT_INVALID : OrderStatus.DOA_INVALID_REFUNDED,
1151
                     OrderStatus.DOA_CERT_VALID : OrderStatus.DOA_VALID_REFUNDED,
1152
                     OrderStatus.INIT : OrderStatus.CANCELED,
1153
                     OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.CANCELED,
1154
                     OrderStatus.INVENTORY_LOW : OrderStatus.CANCELED,
4306 rajveer 1155
                     OrderStatus.LOW_INV_PO_RAISED : OrderStatus.CANCELED,
1156
                     OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.CANCELED,
1157
                     OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.CANCELED,
3366 chandransh 1158
                     OrderStatus.ACCEPTED : OrderStatus.CANCELED,
4247 rajveer 1159
                     OrderStatus.BILLED : OrderStatus.CANCELED,
1160
                     OrderStatus.CANCEL_REQUEST_CONFIRMED : OrderStatus.CANCELLED_ON_CUSTOMER_REQUEST
3366 chandransh 1161
                     }
1162
        if order.status not in status_transition.keys():
1163
            raise TransactionServiceException(114, "This order can't be refunded")
1164
 
3413 chandransh 1165
        if order.status < OrderStatus.BILLED:
3226 chandransh 1166
            __update_inventory_reservation(order)
1167
            order.statusDescription = "Order Cancelled"
3415 chandransh 1168
        elif order.status == OrderStatus.BILLED:
3413 chandransh 1169
            __create_return_order(order)
1170
            order.statusDescription = "Order Cancelled"
4078 chandransh 1171
        elif order.status == OrderStatus.SALES_RET_RECEIVED:
3226 chandransh 1172
            __create_return_order(order)
4078 chandransh 1173
            order.statusDescription = "Order Returned to origin"
1174
        elif order.status in [OrderStatus.DOA_CERT_INVALID, OrderStatus.DOA_CERT_VALID] :
1175
            __create_return_order(order)
3226 chandransh 1176
            __create_refund(order)
1177
            order.statusDescription = "Order Refunded"
4247 rajveer 1178
        elif order.status == OrderStatus.CANCEL_REQUEST_CONFIRMED:
1179
            if order.previousStatus  < OrderStatus.BILLED:
1180
                __update_inventory_reservation(order)
1181
                order.statusDescription = "Order Cancelled on customer request"
1182
            elif order.previousStatus == OrderStatus.BILLED:
1183
                __create_return_order(order)
1184
                order.statusDescription = "Order Cancelled on customer request"
3226 chandransh 1185
    else:
3366 chandransh 1186
        status_transition = {OrderStatus.SALES_RET_RECEIVED : OrderStatus.SALES_RET_REFUNDED,
1187
                     OrderStatus.DOA_CERT_INVALID : OrderStatus.DOA_INVALID_REFUNDED,
1188
                     OrderStatus.DOA_CERT_VALID : OrderStatus.DOA_VALID_REFUNDED,
1189
                     OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.REFUNDED,
3731 chandransh 1190
                     OrderStatus.INVENTORY_LOW : OrderStatus.REFUNDED,
4306 rajveer 1191
                     OrderStatus.LOW_INV_PO_RAISED : OrderStatus.REFUNDED,
1192
                     OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.REFUNDED,
1193
                     OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.REFUNDED,
3731 chandransh 1194
                     OrderStatus.ACCEPTED : OrderStatus.REFUNDED,
4247 rajveer 1195
                     OrderStatus.BILLED : OrderStatus.REFUNDED,
1196
                     OrderStatus.CANCEL_REQUEST_CONFIRMED : OrderStatus.CANCELLED_ON_CUSTOMER_REQUEST,
1197
                     OrderStatus.PAYMENT_FLAGGED : OrderStatus.PAYMENT_FLAGGED_DENIED
3366 chandransh 1198
                     }
1199
        if order.status not in status_transition.keys():
1200
            raise TransactionServiceException(114, "This order can't be refunded")
1201
 
3226 chandransh 1202
        if order.status in [OrderStatus.SALES_RET_RECEIVED, OrderStatus.DOA_CERT_INVALID, OrderStatus.DOA_CERT_VALID] :
1203
            __create_return_order(order)
1204
            __create_refund(order)
1205
            order.statusDescription = "Order Refunded"
4306 rajveer 1206
        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]:
3191 chandransh 1207
            __create_cancellation(order)
3226 chandransh 1208
            __update_inventory_reservation(order)
1209
            order.statusDescription = "Order Refunded"
4247 rajveer 1210
        elif order.status == OrderStatus.CANCEL_REQUEST_CONFIRMED:
1211
            if order.previousStatus  < OrderStatus.BILLED:
1212
                __create_refund(order)
1213
                __update_inventory_reservation(order)
1214
                order.statusDescription = "Order Cancelled on customer request"
1215
            elif order.previousStatus == OrderStatus.BILLED:
1216
                __create_return_order(order)
1217
                __create_refund(order)
1218
                order.statusDescription = "Order Cancelled on customer request"
1219
        elif order.status == OrderStatus.PAYMENT_FLAGGED:
1220
            __update_inventory_reservation(order)
1221
            order.statusDescription = "Order Cancelled due to payment flagged"
3195 chandransh 1222
 
1223
    order.status = status_transition[order.status]
1224
    order.refund_timestamp = datetime.datetime.now()
3226 chandransh 1225
    order.refunded_by = refunded_by
1226
    order.refund_reason = reason
2628 chandransh 1227
 
1228
    session.commit()
1229
    return True
1230
 
3186 chandransh 1231
def __update_inventory_reservation(order):
3191 chandransh 1232
    '''
1233
    Reduce the reservation count for all line items of the given order.
1234
    '''
3186 chandransh 1235
    try:
1236
        catalog_client = CatalogClient().get_client()
1237
        for lineitem in order.lineitems:
1238
            catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity);
1239
    except:
1240
        print "Unable to reduce reservation count"
1241
 
4220 rajveer 1242
def __clone_order(order, should_copy_billing_info, is_cod):
2628 chandransh 1243
    current_time = datetime.datetime.now()
1244
    new_order = Order()
1245
    new_order.customer_id = order.customer_id
1246
    new_order.customer_name = order.customer_name
1247
    new_order.customer_city = order.customer_city
1248
    new_order.customer_state = order.customer_state
1249
    new_order.customer_mobilenumber = order.customer_mobilenumber
1250
    new_order.customer_pincode = order.customer_pincode
1251
    new_order.customer_address1 = order.customer_address1
1252
    new_order.customer_address2 = order.customer_address2
1253
    new_order.customer_email = order.customer_email
1254
    new_order.total_amount = order.total_amount
1255
    new_order.total_weight = order.total_weight
1256
    new_order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
1257
    new_order.statusDescription = 'Submitted for Processing'
1258
    new_order.created_timestamp = current_time
1259
 
1260
    new_order.transaction = order.transaction
4220 rajveer 1261
    new_order.cod = is_cod
2628 chandransh 1262
 
1263
    for line_item in order.lineitems:
1264
        litem = LineItem()
1265
        litem.item_id = line_item.item_id
1266
        litem.productGroup = line_item.productGroup
1267
        litem.brand = line_item.brand
1268
        if line_item.model_number:
1269
            litem.model_number = line_item.model_number
1270
        if line_item.model_name:    
1271
            litem.model_name = line_item.model_name
1272
        if line_item.color:
1273
            litem.color = line_item.color
1274
        if line_item.extra_info:
1275
            litem.extra_info = line_item.extra_info
1276
        litem.quantity = line_item.quantity
1277
        litem.unit_price = line_item.unit_price
1278
        litem.unit_weight = line_item.unit_weight
1279
        litem.total_price = line_item.total_price
1280
        litem.total_weight = line_item.total_weight
1281
        litem.transfer_price = line_item.transfer_price
1282
        litem.order = new_order
1283
 
1284
    logistics_client = LogisticsClient().get_client()
1285
    item_id = new_order.lineitems[0].item_id
3239 chandransh 1286
    logistics_info = logistics_client.getLogisticsInfo(new_order.customer_pincode, item_id, DeliveryType.PREPAID) #TODO: We should be able to pass another flag to suggest ignoring the inventory situation.
2628 chandransh 1287
    logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
1288
 
1289
    new_order.warehouse_id = logistics_info.warehouseId
1290
    new_order.logistics_provider_id = logistics_info.providerId
1291
    new_order.airwaybill_no = logistics_info.airway_billno
1292
    new_order.tracking_id = new_order.airwaybill_no
1293
    new_order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
3986 chandransh 1294
    new_order.promised_delivery_time = new_order.expected_delivery_time
4299 rajveer 1295
 
2628 chandransh 1296
    if should_copy_billing_info:
1297
        new_order.invoice_number = order.invoice_number
1298
        new_order.billed_by = order.billed_by
1299
        new_order.warehouse_id = order.warehouse_id
1300
        new_order.accepted_timestamp = current_time
1301
        new_order.billing_timestamp = current_time    
1302
        new_order.status = OrderStatus.BILLED
1303
        new_order.statusDescription = 'Order Billed'
1304
    else:       
3133 rajveer 1305
        catalog_client = CatalogClient().get_client()
2628 chandransh 1306
        catalog_client.reserveItemInWarehouse(item_id, logistics_info.warehouseId, new_order.lineitems[0].quantity)
1307
 
1308
    return new_order
1309
 
1310
def __create_return_order(order):
2687 chandransh 1311
    ret_order = ReturnOrder(order)
2628 chandransh 1312
    return ret_order
1313
 
1314
def __create_refund(order):
4078 chandransh 1315
    payment_client = PaymentClient().get_client()
1316
    payment_client.createRefund(order.id, order.transaction.id, order.total_amount)
2747 chandransh 1317
    return
2628 chandransh 1318
 
3035 chandransh 1319
def __create_cancellation(order):
1320
    return
1321
 
2536 chandransh 1322
def __get_order_address(order):
1323
    address = order.customer_name + "\n" + order.customer_address1 + "\n"
1324
    if order.customer_address2:
1325
        address = address + order.customer_address2 + "\n"
1326
    address = address + order.customer_city + "\n"
1327
    address = address + order.customer_state + "\n"
1328
    address = address + "PIN " + order.customer_pincode + "\n"
1329
    address = address + "Phone: " + order.customer_mobilenumber
1330
    return address
1331
 
1332
def __generate_return_advice(order, warehouse, provider, filename):    
1333
    pdf = Canvas(filename)
1334
    pdf.setFont('Times-Bold',16)
1335
    address_text = pdf.beginText(inch, PAGE_HEIGHT - inch)
1336
    address_text.textLine("To")
1337
    address_text.textLine("Parmod Kumar")
1338
    for line in warehouse.location.split("\n"):
1339
        address_text.textLine(line)
1340
    address_text.textLine("PIN " + warehouse.pincode)
1341
    address_text.textLine("")
1342
    address_text.textLine("Phone: 9971573026")
1343
    pdf.drawText(address_text)
1344
 
1345
    pdf.setFont('Times-Roman',12)
1346
    order_text = pdf.beginText(inch, PAGE_HEIGHT - 4 * inch)
3051 chandransh 1347
    order_text.textLine("Pickup CODE: " + provider.details[DeliveryType.PREPAID].accountNo)
2536 chandransh 1348
    lineitem = order.lineitems[0] 
4234 rajveer 1349
    order_text.textLine("Content : " + lineitem.brand + " " + str(lineitem.model_number) + " " + str(lineitem.color))
2536 chandransh 1350
    order_text.textLine("Declared Value: Rs. " + str(order.total_amount))
1351
    order_text.textLine("Order ID : " + str(order.id))
1352
    pdf.drawText(order_text)
1353
    pdf.showPage()
1354
    pdf.save()
1355
    return filename
1356
 
2697 chandransh 1357
def get_return_orders(warehouse_id, from_date, to_date):
1358
    query = ReturnOrder.query.filter_by(processedStatus = False)
1359
    if warehouse_id != -1:
1360
        query = query.filter_by(warehouseId = warehouse_id)
1361
    if from_date:
4118 chandransh 1362
        query = query.filter(ReturnOrder.createdAt >from_date)
2697 chandransh 1363
    if to_date:
4118 chandransh 1364
        query = query.filter(ReturnOrder.createdAt < to_date)
2697 chandransh 1365
    return [return_order.to_thrift_object() for return_order in query.all()]
1366
 
2700 chandransh 1367
def get_return_order(id):
1368
    ret_order = ReturnOrder.get_by(orderId = id)
1369
    if ret_order is None:
1370
        raise TransactionServiceException(116, "No return order found for the given order id")
1371
    return ret_order.to_thrift_object()
1372
 
2697 chandransh 1373
def process_return(ret_order_id):
1374
    ret_order = ReturnOrder.get_by(orderId = ret_order_id)
1375
    if ret_order is None:
1376
        raise TransactionServiceException(116, "No return order found for the given order id")
1377
    ret_order.processedStatus = True
2701 chandransh 1378
    ret_order.processedAt = datetime.datetime.now()
2697 chandransh 1379
    session.commit()
1380
 
2819 chandransh 1381
def create_purchase_order(warehouseId):
1382
    if not warehouseId:
1383
        raise TransactionServiceException(101, "bad warehouse id")
1384
 
3451 chandransh 1385
    query = Order.query.with_lockmode("update").filter_by(warehouse_id=warehouseId)
4380 mandeep.dh 1386
    query = query.filter(or_(Order.status == OrderStatus.SUBMITTED_FOR_PROCESSING, Order.status == OrderStatus.INVENTORY_LOW))
2819 chandransh 1387
    query = query.filter_by(purchase_order_id=None)
1388
    query = query.order_by(Order.created_timestamp)
1389
    pending_orders = query.all()
1390
 
1391
    if not pending_orders:
1392
        return -1
4434 mandeep.dh 1393
 
1394
    catalog_client = CatalogClient().get_client()
1395
    availability = {}
1396
    for item in catalog_client.getAllItemsForWarehouse(warehouseId):
1397
        availability[item.id] = item.itemInventory.availability[warehouseId]
1398
 
1399
    query = Order.query.with_lockmode("update").filter_by(warehouse_id=warehouseId)
1400
    query = query.filter(Order.status == OrderStatus.ACCEPTED)
1401
    accepted_orders = query.all()
1402
 
2819 chandransh 1403
    t_purchase_order = PurchaseOrder()
4365 mandeep.dh 1404
    t_purchase_order.supplierId = 3 # Putting supplier id as 3 for sarvottam for time being
2819 chandransh 1405
    t_purchase_order.warehouseId = warehouseId
1406
    t_purchase_order.lineitems = []
1407
 
4434 mandeep.dh 1408
    t_po_lineitems = []
2819 chandransh 1409
    lineitems = {}
4434 mandeep.dh 1410
 
1411
    # Considering all pending orders and accepted as well
1412
    accepted_orders.extend(pending_orders)
1413
    for order in accepted_orders:
2819 chandransh 1414
        for lineitem in order.lineitems:
1415
            item_id = lineitem.item_id
1416
            if lineitems.has_key(item_id):
4434 mandeep.dh 1417
                lineitems[item_id].quantity += lineitem.quantity
2819 chandransh 1418
            else:
1419
                t_po_lineitem = POLineItem()
1420
                t_po_lineitem.productGroup = lineitem.productGroup
1421
                t_po_lineitem.brand = lineitem.brand
1422
                t_po_lineitem.modelNumber = lineitem.model_number
1423
                t_po_lineitem.modelName = lineitem.model_name
1424
                t_po_lineitem.color = lineitem.color
1425
                t_po_lineitem.itemId = lineitem.item_id
4434 mandeep.dh 1426
                t_po_lineitem.quantity = lineitem.quantity - availability.get(item_id, 0)
2819 chandransh 1427
                lineitems[item_id] = t_po_lineitem
4434 mandeep.dh 1428
                t_po_lineitems.append(t_po_lineitem)
1429
                #FIXME vendor is hardcoded for sarvottam
4292 rajveer 1430
                item_pricing = catalog_client.getItemPricing(item_id, 3)
1431
                t_po_lineitem.unitPrice = item_pricing.transferPrice
4434 mandeep.dh 1432
 
1433
    for lineitem in t_po_lineitems:
1434
        if lineitem.quantity > 0:
1435
            t_purchase_order.lineitems.append(lineitem)
1436
 
1437
    if not t_purchase_order.lineitems:
1438
        return -1
1439
 
2819 chandransh 1440
    warehouse_client = WarehouseClient().get_client()
1441
    po_id = warehouse_client.createPurchaseOrder(t_purchase_order)
1442
 
1443
    for order in pending_orders:
1444
        order.purchase_order_id = po_id
4434 mandeep.dh 1445
        order.status = OrderStatus.LOW_INV_PO_RAISED
1446
        order.statusDescription = 'In Process'
2819 chandransh 1447
    session.commit()
1448
    return po_id
1449
 
3451 chandransh 1450
def update_weight(order_id, weight):
3469 chandransh 1451
    '''
1452
    Update the weight of order
1453
    '''
3451 chandransh 1454
    order = get_order(order_id)
1455
    order.total_weight = weight
3469 chandransh 1456
    lineitem = order.lineitems[0]
1457
    lineitem.total_weight = weight
1458
    lineitem.unit_weight = weight
3451 chandransh 1459
 
3469 chandransh 1460
    session.commit()
1461
    return order
1462
 
1463
def change_product(order_id, item_id):
1464
    '''
1465
    Ship a product of a different color to the customer.
1466
    '''
1467
    order = get_order(order_id)
4306 rajveer 1468
    if order.status not in [OrderStatus.INIT, 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 1469
        raise TransactionServiceException(120, "This order has already been processed. Please seek help from engineering.")
1470
 
1471
    lineitem = order.lineitems[0]
3451 chandransh 1472
 
3469 chandransh 1473
    catalog_client = CatalogClient().get_client()
1474
 
1475
    old_item = catalog_client.getItem(lineitem.item_id)
1476
    item = catalog_client.getItem(item_id)
1477
    if old_item.catalogItemId != item.catalogItemId:
1478
        raise TransactionServiceException(121, "You can't ship a different item. You can only ship the same item in a different color.")
1479
 
1480
    catalog_client.reserveItemInWarehouse(item_id, order.warehouse_id, lineitem.quantity)
1481
    catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity)
1482
    #TODO: Check that this new item has the same price
1483
 
1484
    lineitem.item_id = item_id
1485
    lineitem.color = item.color
1486
 
3451 chandransh 1487
    session.commit()
3469 chandransh 1488
 
1489
    order = get_order(order_id)
3451 chandransh 1490
    return order
3469 chandransh 1491
 
1492
def change_warehouse(order_id, warehouse_id):
1493
    '''
1494
    Update the warehouse which will be used to fulfil this order.
1495
    '''
1496
    order = get_order(order_id)
4306 rajveer 1497
    if order.status not in [OrderStatus.INIT, 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 1498
        raise TransactionServiceException(117, "This order has already been processed. Please seek help from engineering.")
1499
 
1500
    if warehouse_id == order.warehouse_id:
1501
        raise TransactionServiceException(118, "You have selected the current warehouse again. Nothing to do")
1502
 
1503
    lineitem = order.lineitems[0]
1504
    catalog_client = CatalogClient().get_client()
1505
    try:
1506
        warehouse = catalog_client.getWarehouse(warehouse_id)
1507
    except:
1508
        raise TransactionServiceException(119, "No warehouse with id" + str(warehouse_id))
1509
 
1510
    catalog_client.reserveItemInWarehouse(lineitem.item_id, warehouse_id, lineitem.quantity)
1511
    catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity)
3451 chandransh 1512
 
3469 chandransh 1513
    order.warehouse_id = warehouse_id
1514
    session.commit()
1515
    return order
3553 chandransh 1516
 
3986 chandransh 1517
def add_delay_reason(order_id, reason, further_delay=0):
3553 chandransh 1518
    '''
3994 chandransh 1519
    Adds a delay reason to the order and updates the expected delivery time.
3553 chandransh 1520
    '''
1521
    order = get_order(order_id)
1522
    try:
3581 chandransh 1523
        order.delay_reason = DelayReason._VALUES_TO_NAMES[reason]
4009 chandransh 1524
        actual_delivery_delay = adjust_delivery_time(order.expected_delivery_time, further_delay)
1525
        order.expected_delivery_time = order.expected_delivery_time + datetime.timedelta(days=actual_delivery_delay)
4102 chandransh 1526
 
1527
        if order.status < OrderStatus.SHIPPED_FROM_WH:
1528
            actual_shipping_delay = adjust_delivery_time(order.expected_shipping_time, further_delay)
1529
            order.expected_shipping_time = order.expected_shipping_time + datetime.timedelta(days=actual_shipping_delay)
1530
 
3553 chandransh 1531
        session.commit()
4008 mandeep.dh 1532
 
1533
        try:
1534
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
1535
            transaction_requiring_extra_processing.category = 'DELAYED_DELIVERY'
1536
            transaction_requiring_extra_processing.transaction_id = order.transaction_id
1537
            session.commit()
1538
        except Exception as e:
4065 mandeep.dh 1539
            print 'Could not persist transaction id: ' + str(order.transaction_id) + ' for further processing due to: ' + str(e)
4008 mandeep.dh 1540
 
3553 chandransh 1541
        return True
1542
    except:
3994 chandransh 1543
        print sys.exc_info()[0]
3553 chandransh 1544
        return False
3956 chandransh 1545
 
1546
def reconcile_cod_collection(collected_amount_map, xferBy, xferTxnId, xferDate):
1547
    unprocessed_awbs = {}
1548
    for awb, amount in collected_amount_map.iteritems():
1549
        try:
1550
            order = Order.query.filter_by(airwaybill_no=awb).one()
1551
        except NoResultFound:
1552
            unprocessed_awbs[awb] = "No order was found for the given AWB: " + awb
1553
            continue
1554
        except MultipleResultsFound:
1555
            unprocessed_awbs[awb] = "Multiple orders were found for the given AWB:" + awb
1556
            continue
1557
 
1558
        if order.cod_reconciliation_timestamp:
1559
            #This order has been processed already! This may be a re-run. Let's allow other orders to be processed.
1560
            continue
1561
 
1562
        if order.status < OrderStatus.DELIVERY_SUCCESS:
1563
            #Order has not been delivered yet. No way we can receive the payment.
1564
            unprocessed_awbs[awb] = "Payment has been received for order: " + str(order.id) + " before delivery reconciliation"
1565
            continue
1566
 
4252 rajveer 1567
        ##As per ticket #743 if amount difference less than 0.5 we should reconcile
1568
        if abs(amount - order.total_amount) > 0.5:
3956 chandransh 1569
            #Received amount is less than or more than the order amount. Mustn't let the user proceed.
1570
            unprocessed_awbs[awb] = "Payment of Rs. " + str(amount) + " has been received against the total value of Rs. " + str(order.total_amount) + " for order: " + str(order.id)
1571
            continue
1572
 
1573
        try:
1574
            payment_client = PaymentClient().get_client()
1575
            payment_client.partiallyCapturePayment(order.transaction.id, amount, xferBy, xferTxnId, xferDate)
1576
        except Exception:
1577
            unprocessed_awbs[awb] = "We were unable to partially capture payment for order id " + str(order.id) + ", AWB: " + awb
1578
            continue
1579
 
1580
        #Payment has been recorded now. We can update the order peacefully.
1581
        order.cod_reconciliation_timestamp = datetime.datetime.now()
1582
        session.commit()
3469 chandransh 1583
 
3956 chandransh 1584
    return unprocessed_awbs
4008 mandeep.dh 1585
 
1586
def get_transactions_requiring_extra_processing(category):
4065 mandeep.dh 1587
    """
1588
    Returns the list of transaction ids that require some extra processing and
1589
    which belong to a particular category. This is currently used by CRM
1590
    application. If no such transaction ids are present, it returns an empty list.
1591
    """
1592
    query = TransactionRequiringExtraProcessing.query
4008 mandeep.dh 1593
 
4065 mandeep.dh 1594
    if category is not None:
1595
        query = query.filter_by(category = ExtraTransactionProcessingType._VALUES_TO_NAMES.get(category))
4008 mandeep.dh 1596
 
4065 mandeep.dh 1597
    return [a.transaction_id for a in query.all()]
4008 mandeep.dh 1598
 
1599
def mark_transaction_as_processed(transactionId, category):
4065 mandeep.dh 1600
    """
1601
    Marks a particular transaction as processed for a particular cateogory.
1602
    It essentially deletes the transaction id record for a particular
1603
    processing type category (if present) from DB.
1604
    This is currently used by CRM application.
1605
    """
4008 mandeep.dh 1606
    TransactionRequiringExtraProcessing.query.filter_by(transaction_id = transactionId).filter_by(category = ExtraTransactionProcessingType._VALUES_TO_NAMES.get(category)).delete()
1607
    session.commit()
1608
 
4018 chandransh 1609
def get_item_wise_risky_orders_count():
1610
    """
1611
    Returns a map containing the number of risky orders keyed by item id. A risky order
1612
    is defined as one whose shipping date is about to expire.
1613
    """
1614
    item_wise_risky_orders_count = {}
4019 chandransh 1615
    shipping_cutoff_time = (datetime.datetime.now() + datetime.timedelta(days = 1)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4018 chandransh 1616
    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()
1617
    for order in orders:
1618
        for lineitem in order.lineitems:
1619
            quantity = lineitem.quantity
1620
            if item_wise_risky_orders_count.has_key(lineitem.item_id):
1621
                quantity = quantity + item_wise_risky_orders_count[lineitem.item_id]
1622
 
1623
            item_wise_risky_orders_count[lineitem.item_id] = quantity
1624
    return item_wise_risky_orders_count
1625
 
4324 mandeep.dh 1626
def update_shipment_address(order_id, address_id):
1627
    """
1628
    Updates shipment address of an order. Delivery and shipping date estimates
1629
    etc. are also updated here.
1630
 
1631
    Throws TransactionServiceException in case address change is not
1632
    possible due to certain reasons such as new pincode in address is
1633
    not serviceable etc.
1634
 
1635
    Parameters:
1636
     - orderId
1637
     - addressId
1638
    """
1639
    user_client = UserClient().get_client()
1640
    address     = user_client.getAddressById(address_id)
1641
    order       = get_order(order_id)
1642
 
1643
    logistics_client = LogisticsClient().get_client()
1644
    logistics_info = logistics_client.getLogisticsEstimation(order.lineitems[0].item_id, address.pin)
1645
 
1646
    # Check for new estimate. Raise exception if new address is not serviceable
1647
    if logistics_info.deliveryTime == -1:
1648
        raise TransactionServiceException(1, 'Location not serviceable')
1649
 
1650
    # COD services are not available everywhere
1651
    if order.cod:
1652
        if not logistics_client.isCodAllowed(address.pin):
1653
            raise TransactionServiceException(2, 'COD service not available')
1654
 
1655
    current_time = datetime.datetime.now()
1656
    logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
1657
    logistics_info.shippingTime = adjust_delivery_time(current_time, logistics_info.shippingTime)
1658
 
1659
    order.customer_name         = address.name
1660
    order.customer_pincode      = address.pin
1661
    order.customer_address1     = address.line1
1662
    order.customer_address2     = address.line2
1663
    order.customer_city         = address.city
1664
    order.customer_state        = address.state
1665
    order.customer_mobilenumber = address.phone
1666
 
1667
    if order.cod:
1668
        order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=COD_SHIPPING_CUTOFF_TIME, minute=0, second=0)
1669
    else:
1670
        order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
1671
 
1672
    order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
1673
    order.promised_shipping_time = order.expected_shipping_time
1674
    order.promised_delivery_time = order.expected_delivery_time 
1675
 
1676
    session.commit()
1677
 
4247 rajveer 1678
def mark_order_cancellation_request_received(orderId):
1679
    """
1680
    Mark order as cancellation request received. If customer sends request of cancellation of
1681
    a particular order, this method will be called. It will just change status of the order
1682
    depending on its current status. It also records the previous status, so that we can move
1683
    back to that status if cancellation request is denied.
1684
 
1685
    Parameters:
1686
     - orderId
1687
    """
1688
    status_transition = {OrderStatus.INIT : OrderStatus.CANCEL_REQUEST_RECEIVED,
1689
             OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.CANCEL_REQUEST_RECEIVED,
1690
             OrderStatus.INVENTORY_LOW : OrderStatus.CANCEL_REQUEST_RECEIVED,
4306 rajveer 1691
             OrderStatus.LOW_INV_PO_RAISED : OrderStatus.CANCEL_REQUEST_RECEIVED,
1692
             OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.CANCEL_REQUEST_RECEIVED,
1693
             OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.CANCEL_REQUEST_RECEIVED,
4247 rajveer 1694
             OrderStatus.ACCEPTED : OrderStatus.CANCEL_REQUEST_RECEIVED,
1695
             OrderStatus.BILLED : OrderStatus.CANCEL_REQUEST_RECEIVED,
1696
             OrderStatus.PAYMENT_FLAGGED : OrderStatus.CANCEL_REQUEST_RECEIVED
1697
             }
1698
    if order.status not in status_transition.keys():
1699
        raise TransactionServiceException(114, "This order can't be considered for cancellation")
1700
 
1701
    order = get_order(orderId)
1702
    order.previousStatus = order.status
1703
    order.status = OrderStatus.CANCEL_REQUEST_RECEIVED
1704
    order.statusDescription = "Cancellation request received from user"
1705
    session.commit()
1706
 
4258 rajveer 1707
def mark_transaction_as_payment_flag_removed(transactionId):
4247 rajveer 1708
    """
4258 rajveer 1709
    If we and/or payment gateway has decided to accept the payment, this method needs to be called.
1710
    Changed transaction and all orders status to payment accepted.
4247 rajveer 1711
 
1712
    Parameters:
4258 rajveer 1713
     - transactionId
4247 rajveer 1714
    """
4258 rajveer 1715
    transaction = get_transaction(transactionId)
1716
    transaction.status = TransactionStatus.AUTHORIZED
1717
    for order in transaction.orders:
1718
        order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
1719
        order.statusDescription = "Submitted to warehouse"
4247 rajveer 1720
    session.commit()
1721
 
4259 anupam.sin 1722
def refund_transaction(transactionId, refundedBy, reason):
1723
    """
1724
    This method is called when a flagged payment is deemed unserviceable and the corresponding orders
1725
    need to be cancelled
1726
 
1727
    Parameters:
1728
     - transactionId
1729
    """
1730
    transaction = get_transaction(transactionId)
1731
    transaction.status = TransactionStatus.FAILED
1732
    for order in transaction.orders:
1733
        refund_order(order.id, refundedBy, reason)
1734
    session.commit()
1735
 
4247 rajveer 1736
def mark_order_cancellation_request_denied(orderId):
1737
    """
4283 anupam.sin 1738
    If we decide to not to cancel order, we will move the order to previous status.
4247 rajveer 1739
 
1740
    Parameters:
1741
     - orderId
1742
    """
1743
    order = get_order(orderId)
1744
    order.status = order.previousStatus
1745
    order.previousStatus = None
1746
    order.statusDescription = "Cancellation request denied"
1747
    session.commit()
1748
 
1749
def mark_order_cancellation_request_confirmed(orderId):
1750
    """
1751
    If we decide to to cancel order, CRM will call this method to move the status of order to
1752
    cancellation request confirmed. After this OM will be able to cancel the order.
1753
 
1754
    Parameters:
1755
     - orderId
1756
    """
1757
    order = get_order(orderId)
1758
    order.status = OrderStatus.CANCEL_REQUEST_CONFIRMED
1759
    order.statusDescription = "Cancellation request confirmed"
1760
    session.commit()
1761
 
4285 rajveer 1762
def accept_orders_for_item_id(itemId, inventory):
1763
    """
1764
    Marks the orders as ACCEPTED for the given itemId and inventory. It also updates the accepted timestamp. If the
1765
    given order is not a COD order, it also captures the payment if the same has not been captured.
1766
 
1767
    Parameters:
1768
     - itemId
1769
     - inventory
1770
    """
4351 rajveer 1771
    orders = Order.query.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 1772
    for order in orders:
4301 rajveer 1773
        if inventory > 0:
1774
            lineitem = order.lineitems[0]
1775
            if itemId == lineitem.item_id:
1776
                try:
1777
                    accept_order(order.id)
1778
                    inventory = inventory - lineitem.quantity
1779
                except:
1780
                    logging.info("Unable to accept the order")
1781
        else:
1782
            order_outofstock(order.id)
4285 rajveer 1783
    return
1784
 
4369 rajveer 1785
def mark_orders_as_po_raised(vendorId, itemId, quantity, estimate, isReminder):
1786
    if isReminder:
1787
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
1788
    else:
1789
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
1790
 
1791
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 1792
    for order in orders:
1793
        lineitem = order.lineitems[0]
1794
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
1795
            if estimate > 0:
1796
                add_delay_reason(order.id, DelayReason.OTHERS, estimate)
1797
            order.status = OrderStatus.LOW_INV_PO_RAISED
4362 rajveer 1798
            order.statusDescription = "PO Raised"
4303 rajveer 1799
            quantity = quantity - lineitem.quantity
1800
            store_inventory_action(order, itemId, HotspotAction.PO_RAISED, estimate, lineitem.quantity)
1801
            session.commit()
1802
 
4369 rajveer 1803
def mark_orders_as_reversal_initiated(vendorId, itemId, quantity, estimate, isReminder):
1804
    if isReminder:
1805
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
1806
    else:
1807
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
1808
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 1809
    for order in orders:
1810
        lineitem = order.lineitems[0]
1811
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
1812
            if estimate > 0:
1813
                add_delay_reason(order.id, DelayReason.OTHERS, estimate)
1814
            order.status = OrderStatus.LOW_INV_REVERSAL_IN_PROCESS
4362 rajveer 1815
            order.statusDescription = "Reversal Initiated"
4303 rajveer 1816
            quantity = quantity - lineitem.quantity
1817
            store_inventory_action(order, itemId, HotspotAction.REVERSAL_INITIATED, estimate, lineitem.quantity)
1818
            session.commit()
1819
 
4369 rajveer 1820
def mark_orders_as_not_available(vendorId, itemId, quantity, estimate, isReminder):
1821
    if isReminder:
1822
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
1823
    else:
1824
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
1825
 
1826
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 1827
    for order in orders:
1828
        lineitem = order.lineitems[0]
1829
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
1830
            order.status = OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT
4362 rajveer 1831
            order.statusDescription = "Not Available"
4303 rajveer 1832
            quantity = quantity - lineitem.quantity
1833
            store_inventory_action(order, itemId, HotspotAction.NOT_AVAILABLE, estimate, lineitem.quantity)
1834
            session.commit()
1835
 
1836
def store_inventory_action(order, itemId, hotspotAction, estimate, quantity):
4369 rajveer 1837
    orderInv = OrderInventory.get_by(order_id=order.id)
1838
    if not orderInv:
1839
        orderInv = OrderInventory()
1840
        orderInv.order = order
4303 rajveer 1841
    orderInv.hotspotAction = hotspotAction  
1842
    orderInv.estimate = estimate
1843
    orderInv.itemId = itemId
1844
    orderInv.timestamp = datetime.datetime.now()
1845
    orderInv.quantity = quantity
1846
 
4369 rajveer 1847
 
1848
def mark_orders_as_timeout(vendorId):
1849
    ret_timeouts = {}
1850
    orderInvs = OrderInventory.query.all()
1851
    for orderInv in orderInvs:
1852
        if orderInv.timestamp +  timedelta(days=orderInv.estimate) < datetime.datetime.now():
1853
            if not ret_timeouts.has_key(orderInv.itemId):
1854
                timeout = TimeoutSummary()
1855
                ret_timeouts[orderInv.itemId] = timeout
1856
            else:
1857
                timeout = ret_timeouts.get(orderInv.itemId)
1858
            if orderInv.hotspotAction == HotspotAction.PO_RAISED and orderInv.order.status == OrderStatus.LOW_INV_PO_RAISED:
1859
                if timeout.poRaised:
1860
                    timeout.poRaised += 1
1861
                else:
1862
                    timeout.poRaised = 1
1863
 
1864
                if timeout.poEstimate:
1865
                    timeout.poEstimate += orderInv.estimate
1866
                else:
1867
                    timeout.poEstimate = orderInv.estimate
1868
                orderInv.order.status = OrderStatus.LOW_INV_PO_RAISED_TIMEOUT
1869
                orderInv.order.statusDescription = "PO Raised Timeout"
1870
            elif orderInv.hotspotAction == HotspotAction.REVERSAL_INITIATED and orderInv.order.status == OrderStatus.LOW_INV_REVERSAL_IN_PROCESS:
1871
                if timeout.reversalInitiated:
1872
                    timeout.reversalInitiated += 1
1873
                else:
1874
                    timeout.reversalInitiated = 1
1875
 
1876
                if timeout.reversalEstimate:
1877
                    timeout.reversalEstimate += orderInv.estimate
1878
                else:
1879
                    timeout.reversalEstimate = orderInv.estimate
1880
                orderInv.order.status = OrderStatus.LOW_INV_REVERSAL_TIMEOUT
1881
                orderInv.order.statusDescription = "Reversal Initiated Timeout"
1882
            else:
1883
                orderInv.delete()
1884
    session.commit()
1885
    return ret_timeouts
1886
 
4386 anupam.sin 1887
def get_order_for_awb(awb):
1888
    order = Order.get_by(airwaybill_no=awb)
1889
    if not order:
1890
        raise TransactionServiceException(108, "no such order")
1891
 
1892
    return order
4369 rajveer 1893
 
4386 anupam.sin 1894
 
766 rajveer 1895
def close_session():
1896
    if session.is_active:
1897
        print "session is active. closing it."
2783 chandransh 1898
        session.close()
3376 rajveer 1899
 
1900
def is_alive():
1901
    try:
1902
        session.query(Transaction.id).limit(1).one()
1903
        return True
1904
    except:
4343 varun.gupt 1905
        return False