Subversion Repositories SmartDukaan

Rev

Rev 4486 | Rev 4495 | 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 = {
4484 rajveer 62
        OrderStatus.RTO_RESHIPPED     :  3,
63
        OrderStatus.RTO_IN_TRANSIT : -5,
3499 mandeep.dh 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)
4484 rajveer 777
        order.status = OrderStatus.RTO_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
    '''
4484 rajveer 824
    return Order.query.filter(Order.status >= OrderStatus.SUBMITTED_FOR_PROCESSING).filter(not_(Order.status.in_((OrderStatus.RTO_RESHIPPED, OrderStatus.DOA_INVALID_RESHIPPED, OrderStatus.DOA_VALID_RESHIPPED, OrderStatus.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
 
4488 rajveer 892
def mark_order_return_request_received(orderId):
893
    order = get_order(orderId)
894
    if order.status != OrderStatus.DELIVERY_SUCCESS:
895
        return False
896
    order.status = OrderStatus.RET_REQUEST_RECEIVED
897
    order.statusDescription = "Return Request Received"
898
    session.commit()
899
    return True
900
 
901
def mark_order_return_request_authorized(orderId, isAuthorized):
902
    order = get_order(orderId)
903
    if order.status != OrderStatus.RET_REQUEST_RECEIVED:
904
        return False
905
    if isAuthorized:
906
        order.status = OrderStatus.RET_REQUEST_AUTHORIZED
907
        order.statusDescription = "Return Request Authorized"
908
    else:
909
        order.status = OrderStatus.DELIVERY_SUCCESS
910
        order.statusDescription = "Order delivered"
911
    session.commit()
912
    return True
913
 
2536 chandransh 914
def request_pickup_number(order_id):
915
    order = get_order(order_id)
4452 rajveer 916
    if order.status != OrderStatus.DELIVERY_SUCCESS and order.status != OrderStatus.DOA_PICKUP_REQUEST_RAISED:
2536 chandransh 917
        return False
918
    from_user = 'cnc.center@shop2020.in'
919
    from_pwd = '5h0p2o2o'
920
 
4227 rajveer 921
    subject = "Pickup request from " + order.customer_city
2536 chandransh 922
    try:
3133 rajveer 923
        catalog_client = CatalogClient().get_client()
2536 chandransh 924
        warehouse = catalog_client.getWarehouse(order.warehouse_id)
925
 
926
        logistics_client = LogisticsClient().get_client()
927
        provider = logistics_client.getProvider(order.logistics_provider_id)
3051 chandransh 928
        to_addr = provider.details[DeliveryType.PREPAID].email
2536 chandransh 929
        raw_message = '''
930
        Dear Sir/Madam,
931
 
2559 chandransh 932
        Kindly arrange a pickup today from %(customer_city)s. Pickup and delivery addresses are mentioned below.
2536 chandransh 933
 
934
        Pickup CODE: %(provider_code)s
935
        Total Weight: %(order_weight)s kg
936
 
2559 chandransh 937
        Pickup Address:
2536 chandransh 938
 
939
        %(customer_address)s
940
 
941
        Delivery Address:
942
 
943
        %(warehouse_executive)s
944
        %(warehouse_address)s
945
        PIN %(warehouse_pin)s
946
        Phone: %(warehouse_phone)s
947
 
948
        Thanks and Regards
4227 rajveer 949
        Sandeep Sachdeva
2536 chandransh 950
        '''
2559 chandransh 951
        message = dedent(raw_message) % { 'customer_city' : order.customer_city,
3051 chandransh 952
                                          'provider_code' : provider.details[DeliveryType.PREPAID].accountNo,
2536 chandransh 953
                                          'order_weight' : str(order.total_weight),
954
                                          'customer_address' : __get_order_address(order),
955
                                          'warehouse_executive' : 'Parmod Kumar',
956
                                          'warehouse_address' : warehouse.location,
957
                                          'warehouse_pin' : warehouse.pincode,
958
                                          'warehouse_phone': '9971573026'}
4227 rajveer 959
        #FIXME Email is hardcoded for bluedart
960
        mail(from_user, from_pwd, [to_addr, 'sonikap@bluedart.com'], subject, message)
4452 rajveer 961
        order.status = OrderStatus.DOA_PICKUP_REQUEST_RAISED
2536 chandransh 962
        order.statusDescription = "Pick up requested for DOA"
963
        session.commit()
964
        return True
965
    except Exception as e:
966
        print sys.exc_info()[0]
967
        return False
968
 
969
def authorize_pickup(order_id, pickup_number):
970
    order = get_order(order_id)
4452 rajveer 971
    if order.status != OrderStatus.DOA_PICKUP_REQUEST_RAISED:
2536 chandransh 972
        return False
973
    from_user = 'help@shop2020.in'
974
    from_pwd = '5h0p2o2o'
975
    subject = 'Pickup details'
976
    filename = "/tmp/" +  str(order.id) +".pdf" 
977
    try:
3133 rajveer 978
        catalog_client = CatalogClient().get_client()
2536 chandransh 979
        warehouse = catalog_client.getWarehouse(order.warehouse_id)
980
 
981
        logistics_client = LogisticsClient().get_client()
982
        provider = logistics_client.getProvider(order.logistics_provider_id)
983
        provider_name = provider.name
984
 
985
        to_addr = order.customer_email
986
        today = date.today().strftime("%d-%B-%Y (%A)")
987
 
988
        raw_message = '''
989
        Dear Mr. %(customer_name)s,
990
 
991
        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.
992
        Take a printout of the attachment in this mail which contains the delivery address and paste it on the packed parcel.
993
        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.
994
 
995
        Pickup will happen from the below mentioned address:-
996
 
997
        %(customer_address)s
998
 
999
        Do let us know in case of any concerns
1000
 
1001
        Thanks & Regards
1002
        Saholic Team
1003
        '''
1004
        message = dedent(raw_message) % {'customer_name' : order.customer_name,
1005
                                         'order_id' : order_id,
1006
                                         'provider_name' : provider_name,
1007
                                         'date': today,
1008
                                         'pickup_request_no' : pickup_number,
1009
                                         'customer_address' : __get_order_address(order)}
1010
        print message
1011
        __generate_return_advice(order, warehouse, provider, filename)
3527 chandransh 1012
        mail(from_user, from_pwd, [to_addr], subject, message, [get_attachment_part(filename)])
2536 chandransh 1013
        order.pickupRequestNo = pickup_number
4452 rajveer 1014
        order.status = OrderStatus.DOA_PICKUP_CONFIRMED
1015
        order.statusDescription = "DOA pick up confirmed"
2764 chandransh 1016
        order.doa_auth_timestamp = datetime.datetime.now()
2536 chandransh 1017
        session.commit()
1018
        return True
1019
    except Exception as e:
4234 rajveer 1020
        print sys.exc_info()
1021
	traceback.print_tb(sys.exc_info()[2])
2536 chandransh 1022
        return False
1023
    finally:
1024
        if os.path.exists(filename):
1025
            os.remove(filename)
1026
 
2764 chandransh 1027
def mark_doas_as_picked_up(provider_id, pickup_details):
1028
    for pickupNo, doa_pickup_timestamp in pickup_details.iteritems():
1029
        order = Order.query.filter_by(pickupRequestNo=pickupNo, logistics_provider_id = provider_id).first()
4452 rajveer 1030
        if order == None or order.status != OrderStatus.DOA_PICKUP_CONFIRMED:
2764 chandransh 1031
            #raise TransactionServiceException(102, "No order found for the awb: " + awb)
1032
            continue
1033
        order.status = OrderStatus.DOA_RETURN_IN_TRANSIT
1034
        order.statusDescription = "DOA picked up by Courier Company"
1035
        order.doa_pickup_timestamp = doa_pickup_timestamp
1036
    session.commit()
1037
 
1038
    current_time = datetime.datetime.now()
1039
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
1040
    from_datetime = to_datetime - datetime.timedelta(2)
4452 rajveer 1041
    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 1042
    return orders_not_picked_up
1043
 
4479 rajveer 1044
def receive_return(order_id, receiveCondition):
2591 chandransh 1045
    order = get_order(order_id)
4365 mandeep.dh 1046
 
1047
    # FIXME: For Mahipal warehouse, we need to scan in items for every sales return
1048
    if order.warehouse_id == 7:
1049
        try:
1050
            warehouse_client = WarehouseClient().get_client()
1051
            lineitem = order.lineitems[0]
1052
 
1053
            # Inventory should not be updated for DOA returns 
4484 rajveer 1054
            if order.status == OrderStatus.RTO_IN_TRANSIT:
4365 mandeep.dh 1055
                warehouse_client.scanIn(order.id, lineitem.item_number, lineitem.imei_number, ScanType.SALE_RET)
1056
                catalog_client = CatalogClient().get_client()
1057
                catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity)
4452 rajveer 1058
            elif order.status in [OrderStatus.DOA_PICKUP_CONFIRMED, OrderStatus.DOA_RETURN_IN_TRANSIT]:
4365 mandeep.dh 1059
                warehouse_client.scanIn(order.id, lineitem.item_number, lineitem.imei_number, ScanType.DOA_IN)
1060
        except:
1061
            print sys.exc_info()[0]
1062
            print 'Could not scan in sales return orders'
1063
 
4452 rajveer 1064
    if order.status in [OrderStatus.DOA_PICKUP_CONFIRMED, OrderStatus.DOA_RETURN_IN_TRANSIT]:
4479 rajveer 1065
        if receiveCondition == 0:
1066
            order.status = OrderStatus.DOA_RECEIVED_PRESTINE
1067
            order.statusDescription = "DOA package received"
1068
        elif receiveCondition == 1:
1069
            order.status = OrderStatus.DOA_RECEIVED_DAMAGED
1070
            order.statusDescription = "DOA received damaged"
1071
        elif receiveCondition == 2:
1072
            order.status = OrderStatus.DOA_LOST_IN_TRANSIT
1073
            order.statusDescription = "DOA lost in transit"
2764 chandransh 1074
        order.received_return_timestamp = datetime.datetime.now()
2616 chandransh 1075
        session.commit()
4484 rajveer 1076
    elif order.status == OrderStatus.RTO_IN_TRANSIT :
4486 rajveer 1077
        if receiveCondition == 0:
1078
            order.status = OrderStatus.RTO_RECEIVED_PRESTINE
1079
            order.statusDescription = "RTO package received"
1080
        elif receiveCondition == 1:
1081
            order.status = OrderStatus.RTO_RECEIVED_DAMAGED
1082
            order.statusDescription = "RTO received damaged"
1083
        elif receiveCondition == 2:
1084
            order.status = OrderStatus.RTO_LOST_IN_TRANSIT
1085
            order.statusDescription = "RTO lost in transit"
2764 chandransh 1086
        order.received_return_timestamp = datetime.datetime.now()
2616 chandransh 1087
        session.commit()
1088
    else:
2591 chandransh 1089
        return False
1090
    return True
1091
 
1092
def validate_doa(order_id, is_valid):
1093
    order = get_order(order_id)
4452 rajveer 1094
    if order.status != OrderStatus.DOA_RECEIVED_PRESTINE:
2591 chandransh 1095
        return False
1096
    if is_valid:
2609 chandransh 1097
        order.status = OrderStatus.DOA_CERT_VALID
1098
        order.statusDescription = "DOA Certificate Valid"
2591 chandransh 1099
    else:
1100
        order.status = OrderStatus.DOA_CERT_INVALID
1101
        order.statusDescription = "DOA Certificate Invalid"
3578 mandeep.dh 1102
        update_trust_level(order)
2591 chandransh 1103
    session.commit()
1104
    return True
1105
 
2628 chandransh 1106
def reship_order(order_id):
1107
    """
4484 rajveer 1108
    If the order is in RTO_RECEIVED_PRESTINE or DOA_CERT_INVALID state, it does the following:
2628 chandransh 1109
        1. Creates a new order for processing in the BILLED state. All billing information is saved.
4484 rajveer 1110
        2. Marks the current order as one of the final states RTO_RESHIPPED and DOA_INVALID_RESHIPPED depending on what state the order started in.
2628 chandransh 1111
 
1112
    If the order is in DOA_CERT_VALID state, it does the following:
1113
        1. Creates a new order for processing in the SUBMITTED_FOR_PROCESSING state.
1114
        2. Creates a return order for the warehouse executive to return the DOA material.
4452 rajveer 1115
        3. Marks the current order as the final DOA_VALID_RESHIPPED state.
2628 chandransh 1116
 
1117
    Returns the id of the newly created order.
1118
 
1119
    Throws an exception if the order with the given id couldn't be found.
1120
 
1121
    Parameters:
1122
     - orderId
1123
    """
1124
    order = get_order(order_id)
4484 rajveer 1125
    if order.status == OrderStatus.RTO_RECEIVED_PRESTINE:
4220 rajveer 1126
        new_order = __clone_order(order, True, order.cod)
4484 rajveer 1127
        order.status = OrderStatus.RTO_RESHIPPED
2628 chandransh 1128
        order.statusDescription = "Order Reshipped"
1129
        session.commit()
3578 mandeep.dh 1130
        update_trust_level(order)
4486 rajveer 1131
    elif order.status == OrderStatus.RTO_RECEIVED_DAMAGED:
1132
        new_order = __clone_order(order, False, order.cod)
1133
        ret_order = __create_return_order(order)
1134
        order.status = OrderStatus.RTO_DAMAGED_RESHIPPED
1135
        order.statusDescription = "Order Reshipped Received Damaged"
1136
        session.commit()
1137
    elif order.status == OrderStatus.RTO_LOST_IN_TRANSIT:
1138
        new_order = __clone_order(order, False, order.cod)
1139
        order.status = OrderStatus.RTO_LOST_IN_TRANSIT_RESHIPPED
1140
        order.statusDescription = "Order Reshipped Lost In Transit"
1141
        session.commit()
2628 chandransh 1142
    elif order.status == OrderStatus.DOA_CERT_INVALID:
4220 rajveer 1143
        new_order = __clone_order(order, True, False)
2628 chandransh 1144
        order.status = OrderStatus.DOA_INVALID_RESHIPPED
1145
        order.statusDescription = "Order Reshipped"
1146
        session.commit()
1147
    elif order.status == OrderStatus.DOA_CERT_VALID:
4220 rajveer 1148
        new_order = __clone_order(order, False, False)
2628 chandransh 1149
        ret_order = __create_return_order(order)
4452 rajveer 1150
        order.status = OrderStatus.DOA_VALID_RESHIPPED
2628 chandransh 1151
        order.statusDescription = "Order Reshipped"
1152
        session.commit()
4479 rajveer 1153
    elif order.status == OrderStatus.DOA_RECEIVED_DAMAGED:
1154
        new_order = __clone_order(order, False, False)
1155
        ret_order = __create_return_order(order)
1156
        order.status = OrderStatus.DOA_RESHIPPED_RCVD_DAMAGED
1157
        order.statusDescription = "Order Reshipped Received Damaged"
1158
        session.commit()
1159
    elif order.status == OrderStatus.DOA_LOST_IN_TRANSIT:
1160
        new_order = __clone_order(order, False, False)
1161
        order.status = OrderStatus.DOA_RESHIPPED_LOST_IN_TRANSIT
1162
        order.statusDescription = "Order Reshipped Lost In Transit"
1163
        session.commit()
2628 chandransh 1164
    else:
1165
        raise TransactionServiceException(112, "This order can't be reshipped")
1166
 
2764 chandransh 1167
    order.reship_timestamp = datetime.datetime.now()
2628 chandransh 1168
    order.new_order_id = new_order.id
1169
    session.commit()
1170
 
1171
    return new_order.id
1172
 
3226 chandransh 1173
def refund_order(order_id, refunded_by, reason):
2628 chandransh 1174
    """
4484 rajveer 1175
    If the order is in RTO_RECEIVED_PRESTINE, DOA_CERT_VALID or DOA_CERT_INVALID state, it does the following:
2628 chandransh 1176
        1. Creates a refund request for batch processing.
1177
        2. Creates a return order for the warehouse executive to return the shipped material.
4484 rajveer 1178
        3. Marks the current order as RTO_REFUNDED, DOA_VALID_REFUNDED or DOA_INVALID_REFUNDED final states.
2628 chandransh 1179
 
1180
    If the order is in SUBMITTED_FOR_PROCESSING or INVENTORY_LOW state, it does the following:
1181
        1. Creates a refund request for batch processing.
3226 chandransh 1182
        2. Cancels the reservation of the item in the warehouse.
1183
        3. Marks the current order as the REFUNDED final state.
2628 chandransh 1184
 
3226 chandransh 1185
    For all COD orders, if the order is in INIT, SUBMITTED_FOR_PROCESSING or INVENTORY_LOW state, it does the following:
1186
        1. Cancels the reservation of the item in the warehouse.
1187
        2. Marks the current order as CANCELED.
1188
 
1189
    In all cases, it updates the reason for cancellation or refund and the person who performed the action.
1190
 
1191
    Returns True if it is successful, False otherwise.
2628 chandransh 1192
 
1193
    Throws an exception if the order with the given id couldn't be found.
1194
 
1195
    Parameters:
3226 chandransh 1196
     - order_id
1197
     - refunded_by
1198
     - reason
2628 chandransh 1199
    """
3414 chandransh 1200
    logging.info("Refunding order id: " + str(order_id))
2628 chandransh 1201
    order = get_order(order_id)
3186 chandransh 1202
 
1203
 
3366 chandransh 1204
 
3226 chandransh 1205
    if order.cod:
3414 chandransh 1206
        logging.info("Refunding COD order with status " + str(order.status))
4484 rajveer 1207
        status_transition = {OrderStatus.RTO_RECEIVED_PRESTINE : OrderStatus.RTO_REFUNDED,
4486 rajveer 1208
                     OrderStatus.RTO_RECEIVED_DAMAGED : OrderStatus.RTO_DAMAGED_REFUNDED,
1209
                     OrderStatus.RTO_LOST_IN_TRANSIT : OrderStatus.RTO_LOST_IN_TRANSIT_REFUNDED,
3366 chandransh 1210
                     OrderStatus.DOA_CERT_INVALID : OrderStatus.DOA_INVALID_REFUNDED,
1211
                     OrderStatus.DOA_CERT_VALID : OrderStatus.DOA_VALID_REFUNDED,
4479 rajveer 1212
                     OrderStatus.DOA_RECEIVED_DAMAGED : OrderStatus.DOA_REFUNDED_RCVD_DAMAGED,
1213
                     OrderStatus.DOA_LOST_IN_TRANSIT : OrderStatus.DOA_REFUNDED_LOST_IN_TRANSIT,
3366 chandransh 1214
                     OrderStatus.INIT : OrderStatus.CANCELED,
1215
                     OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.CANCELED,
1216
                     OrderStatus.INVENTORY_LOW : OrderStatus.CANCELED,
4306 rajveer 1217
                     OrderStatus.LOW_INV_PO_RAISED : OrderStatus.CANCELED,
1218
                     OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.CANCELED,
1219
                     OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.CANCELED,
3366 chandransh 1220
                     OrderStatus.ACCEPTED : OrderStatus.CANCELED,
4247 rajveer 1221
                     OrderStatus.BILLED : OrderStatus.CANCELED,
1222
                     OrderStatus.CANCEL_REQUEST_CONFIRMED : OrderStatus.CANCELLED_ON_CUSTOMER_REQUEST
3366 chandransh 1223
                     }
1224
        if order.status not in status_transition.keys():
1225
            raise TransactionServiceException(114, "This order can't be refunded")
1226
 
3413 chandransh 1227
        if order.status < OrderStatus.BILLED:
3226 chandransh 1228
            __update_inventory_reservation(order)
1229
            order.statusDescription = "Order Cancelled"
3415 chandransh 1230
        elif order.status == OrderStatus.BILLED:
3413 chandransh 1231
            __create_return_order(order)
1232
            order.statusDescription = "Order Cancelled"
4486 rajveer 1233
        elif order.status in [OrderStatus.RTO_RECEIVED_PRESTINE, OrderStatus.RTO_RECEIVED_DAMAGED, OrderStatus.RTO_LOST_IN_TRANSIT]:
3226 chandransh 1234
            __create_return_order(order)
4486 rajveer 1235
            order.statusDescription = "RTO Refunded"
4479 rajveer 1236
        elif order.status in [OrderStatus.DOA_CERT_INVALID, OrderStatus.DOA_CERT_VALID, OrderStatus.DOA_RECEIVED_DAMAGED, OrderStatus.DOA_LOST_IN_TRANSIT] :
4078 chandransh 1237
            __create_return_order(order)
3226 chandransh 1238
            __create_refund(order)
1239
            order.statusDescription = "Order Refunded"
4247 rajveer 1240
        elif order.status == OrderStatus.CANCEL_REQUEST_CONFIRMED:
1241
            if order.previousStatus  < OrderStatus.BILLED:
1242
                __update_inventory_reservation(order)
1243
                order.statusDescription = "Order Cancelled on customer request"
1244
            elif order.previousStatus == OrderStatus.BILLED:
1245
                __create_return_order(order)
1246
                order.statusDescription = "Order Cancelled on customer request"
3226 chandransh 1247
    else:
4484 rajveer 1248
        status_transition = {OrderStatus.RTO_RECEIVED_PRESTINE : OrderStatus.RTO_REFUNDED,
4486 rajveer 1249
                     OrderStatus.RTO_RECEIVED_DAMAGED : OrderStatus.RTO_DAMAGED_REFUNDED,
1250
                     OrderStatus.RTO_LOST_IN_TRANSIT : OrderStatus.RTO_LOST_IN_TRANSIT_REFUNDED,
3366 chandransh 1251
                     OrderStatus.DOA_CERT_INVALID : OrderStatus.DOA_INVALID_REFUNDED,
1252
                     OrderStatus.DOA_CERT_VALID : OrderStatus.DOA_VALID_REFUNDED,
4479 rajveer 1253
                     OrderStatus.DOA_RECEIVED_DAMAGED : OrderStatus.DOA_REFUNDED_RCVD_DAMAGED,
1254
                     OrderStatus.DOA_LOST_IN_TRANSIT : OrderStatus.DOA_REFUNDED_LOST_IN_TRANSIT,
3366 chandransh 1255
                     OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.REFUNDED,
3731 chandransh 1256
                     OrderStatus.INVENTORY_LOW : OrderStatus.REFUNDED,
4306 rajveer 1257
                     OrderStatus.LOW_INV_PO_RAISED : OrderStatus.REFUNDED,
1258
                     OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.REFUNDED,
1259
                     OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.REFUNDED,
3731 chandransh 1260
                     OrderStatus.ACCEPTED : OrderStatus.REFUNDED,
4247 rajveer 1261
                     OrderStatus.BILLED : OrderStatus.REFUNDED,
1262
                     OrderStatus.CANCEL_REQUEST_CONFIRMED : OrderStatus.CANCELLED_ON_CUSTOMER_REQUEST,
1263
                     OrderStatus.PAYMENT_FLAGGED : OrderStatus.PAYMENT_FLAGGED_DENIED
3366 chandransh 1264
                     }
1265
        if order.status not in status_transition.keys():
1266
            raise TransactionServiceException(114, "This order can't be refunded")
1267
 
4486 rajveer 1268
        if order.status in [OrderStatus.RTO_RECEIVED_PRESTINE, OrderStatus.RTO_RECEIVED_DAMAGED, OrderStatus.RTO_LOST_IN_TRANSIT, OrderStatus.DOA_CERT_INVALID, OrderStatus.DOA_CERT_VALID, OrderStatus.DOA_RECEIVED_DAMAGED, OrderStatus.DOA_LOST_IN_TRANSIT] :
3226 chandransh 1269
            __create_return_order(order)
1270
            __create_refund(order)
1271
            order.statusDescription = "Order Refunded"
4306 rajveer 1272
        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 1273
            __create_cancellation(order)
3226 chandransh 1274
            __update_inventory_reservation(order)
1275
            order.statusDescription = "Order Refunded"
4247 rajveer 1276
        elif order.status == OrderStatus.CANCEL_REQUEST_CONFIRMED:
1277
            if order.previousStatus  < OrderStatus.BILLED:
1278
                __create_refund(order)
1279
                __update_inventory_reservation(order)
1280
                order.statusDescription = "Order Cancelled on customer request"
1281
            elif order.previousStatus == OrderStatus.BILLED:
1282
                __create_return_order(order)
1283
                __create_refund(order)
1284
                order.statusDescription = "Order Cancelled on customer request"
1285
        elif order.status == OrderStatus.PAYMENT_FLAGGED:
1286
            __update_inventory_reservation(order)
1287
            order.statusDescription = "Order Cancelled due to payment flagged"
3195 chandransh 1288
 
1289
    order.status = status_transition[order.status]
1290
    order.refund_timestamp = datetime.datetime.now()
3226 chandransh 1291
    order.refunded_by = refunded_by
1292
    order.refund_reason = reason
2628 chandransh 1293
 
1294
    session.commit()
1295
    return True
1296
 
3186 chandransh 1297
def __update_inventory_reservation(order):
3191 chandransh 1298
    '''
1299
    Reduce the reservation count for all line items of the given order.
1300
    '''
3186 chandransh 1301
    try:
1302
        catalog_client = CatalogClient().get_client()
1303
        for lineitem in order.lineitems:
1304
            catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity);
1305
    except:
1306
        print "Unable to reduce reservation count"
1307
 
4220 rajveer 1308
def __clone_order(order, should_copy_billing_info, is_cod):
2628 chandransh 1309
    current_time = datetime.datetime.now()
1310
    new_order = Order()
1311
    new_order.customer_id = order.customer_id
1312
    new_order.customer_name = order.customer_name
1313
    new_order.customer_city = order.customer_city
1314
    new_order.customer_state = order.customer_state
1315
    new_order.customer_mobilenumber = order.customer_mobilenumber
1316
    new_order.customer_pincode = order.customer_pincode
1317
    new_order.customer_address1 = order.customer_address1
1318
    new_order.customer_address2 = order.customer_address2
1319
    new_order.customer_email = order.customer_email
1320
    new_order.total_amount = order.total_amount
1321
    new_order.total_weight = order.total_weight
1322
    new_order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
1323
    new_order.statusDescription = 'Submitted for Processing'
1324
    new_order.created_timestamp = current_time
1325
 
1326
    new_order.transaction = order.transaction
4220 rajveer 1327
    new_order.cod = is_cod
2628 chandransh 1328
 
1329
    for line_item in order.lineitems:
1330
        litem = LineItem()
1331
        litem.item_id = line_item.item_id
1332
        litem.productGroup = line_item.productGroup
1333
        litem.brand = line_item.brand
1334
        if line_item.model_number:
1335
            litem.model_number = line_item.model_number
1336
        if line_item.model_name:    
1337
            litem.model_name = line_item.model_name
1338
        if line_item.color:
1339
            litem.color = line_item.color
1340
        if line_item.extra_info:
1341
            litem.extra_info = line_item.extra_info
1342
        litem.quantity = line_item.quantity
1343
        litem.unit_price = line_item.unit_price
1344
        litem.unit_weight = line_item.unit_weight
1345
        litem.total_price = line_item.total_price
1346
        litem.total_weight = line_item.total_weight
1347
        litem.transfer_price = line_item.transfer_price
1348
        litem.order = new_order
1349
 
1350
    logistics_client = LogisticsClient().get_client()
1351
    item_id = new_order.lineitems[0].item_id
4479 rajveer 1352
    logistics_info = logistics_client.getLogisticsInfo(new_order.customer_pincode, item_id, is_cod) #TODO: We should be able to pass another flag to suggest ignoring the inventory situation.
2628 chandransh 1353
    logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
1354
 
1355
    new_order.warehouse_id = logistics_info.warehouseId
1356
    new_order.logistics_provider_id = logistics_info.providerId
1357
    new_order.airwaybill_no = logistics_info.airway_billno
1358
    new_order.tracking_id = new_order.airwaybill_no
1359
    new_order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
3986 chandransh 1360
    new_order.promised_delivery_time = new_order.expected_delivery_time
4299 rajveer 1361
 
2628 chandransh 1362
    if should_copy_billing_info:
1363
        new_order.invoice_number = order.invoice_number
1364
        new_order.billed_by = order.billed_by
1365
        new_order.warehouse_id = order.warehouse_id
1366
        new_order.accepted_timestamp = current_time
1367
        new_order.billing_timestamp = current_time    
1368
        new_order.status = OrderStatus.BILLED
1369
        new_order.statusDescription = 'Order Billed'
1370
    else:       
3133 rajveer 1371
        catalog_client = CatalogClient().get_client()
2628 chandransh 1372
        catalog_client.reserveItemInWarehouse(item_id, logistics_info.warehouseId, new_order.lineitems[0].quantity)
1373
 
1374
    return new_order
1375
 
1376
def __create_return_order(order):
2687 chandransh 1377
    ret_order = ReturnOrder(order)
2628 chandransh 1378
    return ret_order
1379
 
1380
def __create_refund(order):
4078 chandransh 1381
    payment_client = PaymentClient().get_client()
1382
    payment_client.createRefund(order.id, order.transaction.id, order.total_amount)
2747 chandransh 1383
    return
2628 chandransh 1384
 
3035 chandransh 1385
def __create_cancellation(order):
1386
    return
1387
 
2536 chandransh 1388
def __get_order_address(order):
1389
    address = order.customer_name + "\n" + order.customer_address1 + "\n"
1390
    if order.customer_address2:
1391
        address = address + order.customer_address2 + "\n"
1392
    address = address + order.customer_city + "\n"
1393
    address = address + order.customer_state + "\n"
1394
    address = address + "PIN " + order.customer_pincode + "\n"
1395
    address = address + "Phone: " + order.customer_mobilenumber
1396
    return address
1397
 
1398
def __generate_return_advice(order, warehouse, provider, filename):    
1399
    pdf = Canvas(filename)
1400
    pdf.setFont('Times-Bold',16)
1401
    address_text = pdf.beginText(inch, PAGE_HEIGHT - inch)
1402
    address_text.textLine("To")
1403
    address_text.textLine("Parmod Kumar")
1404
    for line in warehouse.location.split("\n"):
1405
        address_text.textLine(line)
1406
    address_text.textLine("PIN " + warehouse.pincode)
1407
    address_text.textLine("")
1408
    address_text.textLine("Phone: 9971573026")
1409
    pdf.drawText(address_text)
1410
 
1411
    pdf.setFont('Times-Roman',12)
1412
    order_text = pdf.beginText(inch, PAGE_HEIGHT - 4 * inch)
3051 chandransh 1413
    order_text.textLine("Pickup CODE: " + provider.details[DeliveryType.PREPAID].accountNo)
2536 chandransh 1414
    lineitem = order.lineitems[0] 
4234 rajveer 1415
    order_text.textLine("Content : " + lineitem.brand + " " + str(lineitem.model_number) + " " + str(lineitem.color))
2536 chandransh 1416
    order_text.textLine("Declared Value: Rs. " + str(order.total_amount))
1417
    order_text.textLine("Order ID : " + str(order.id))
1418
    pdf.drawText(order_text)
1419
    pdf.showPage()
1420
    pdf.save()
1421
    return filename
1422
 
2697 chandransh 1423
def get_return_orders(warehouse_id, from_date, to_date):
1424
    query = ReturnOrder.query.filter_by(processedStatus = False)
1425
    if warehouse_id != -1:
1426
        query = query.filter_by(warehouseId = warehouse_id)
1427
    if from_date:
4118 chandransh 1428
        query = query.filter(ReturnOrder.createdAt >from_date)
2697 chandransh 1429
    if to_date:
4118 chandransh 1430
        query = query.filter(ReturnOrder.createdAt < to_date)
2697 chandransh 1431
    return [return_order.to_thrift_object() for return_order in query.all()]
1432
 
2700 chandransh 1433
def get_return_order(id):
1434
    ret_order = ReturnOrder.get_by(orderId = id)
1435
    if ret_order is None:
1436
        raise TransactionServiceException(116, "No return order found for the given order id")
1437
    return ret_order.to_thrift_object()
1438
 
2697 chandransh 1439
def process_return(ret_order_id):
1440
    ret_order = ReturnOrder.get_by(orderId = ret_order_id)
1441
    if ret_order is None:
1442
        raise TransactionServiceException(116, "No return order found for the given order id")
1443
    ret_order.processedStatus = True
2701 chandransh 1444
    ret_order.processedAt = datetime.datetime.now()
2697 chandransh 1445
    session.commit()
1446
 
2819 chandransh 1447
def create_purchase_order(warehouseId):
1448
    if not warehouseId:
1449
        raise TransactionServiceException(101, "bad warehouse id")
1450
 
3451 chandransh 1451
    query = Order.query.with_lockmode("update").filter_by(warehouse_id=warehouseId)
4380 mandeep.dh 1452
    query = query.filter(or_(Order.status == OrderStatus.SUBMITTED_FOR_PROCESSING, Order.status == OrderStatus.INVENTORY_LOW))
2819 chandransh 1453
    query = query.filter_by(purchase_order_id=None)
1454
    query = query.order_by(Order.created_timestamp)
1455
    pending_orders = query.all()
1456
 
1457
    if not pending_orders:
1458
        return -1
4434 mandeep.dh 1459
 
1460
    catalog_client = CatalogClient().get_client()
1461
    availability = {}
1462
    for item in catalog_client.getAllItemsForWarehouse(warehouseId):
1463
        availability[item.id] = item.itemInventory.availability[warehouseId]
1464
 
1465
    query = Order.query.with_lockmode("update").filter_by(warehouse_id=warehouseId)
1466
    query = query.filter(Order.status == OrderStatus.ACCEPTED)
1467
    accepted_orders = query.all()
1468
 
2819 chandransh 1469
    t_purchase_order = PurchaseOrder()
4365 mandeep.dh 1470
    t_purchase_order.supplierId = 3 # Putting supplier id as 3 for sarvottam for time being
2819 chandransh 1471
    t_purchase_order.warehouseId = warehouseId
1472
    t_purchase_order.lineitems = []
1473
 
4434 mandeep.dh 1474
    t_po_lineitems = []
2819 chandransh 1475
    lineitems = {}
4434 mandeep.dh 1476
 
1477
    # Considering all pending orders and accepted as well
1478
    accepted_orders.extend(pending_orders)
1479
    for order in accepted_orders:
2819 chandransh 1480
        for lineitem in order.lineitems:
1481
            item_id = lineitem.item_id
1482
            if lineitems.has_key(item_id):
4434 mandeep.dh 1483
                lineitems[item_id].quantity += lineitem.quantity
2819 chandransh 1484
            else:
1485
                t_po_lineitem = POLineItem()
1486
                t_po_lineitem.productGroup = lineitem.productGroup
1487
                t_po_lineitem.brand = lineitem.brand
1488
                t_po_lineitem.modelNumber = lineitem.model_number
1489
                t_po_lineitem.modelName = lineitem.model_name
1490
                t_po_lineitem.color = lineitem.color
1491
                t_po_lineitem.itemId = lineitem.item_id
4434 mandeep.dh 1492
                t_po_lineitem.quantity = lineitem.quantity - availability.get(item_id, 0)
2819 chandransh 1493
                lineitems[item_id] = t_po_lineitem
4434 mandeep.dh 1494
                t_po_lineitems.append(t_po_lineitem)
1495
                #FIXME vendor is hardcoded for sarvottam
4292 rajveer 1496
                item_pricing = catalog_client.getItemPricing(item_id, 3)
1497
                t_po_lineitem.unitPrice = item_pricing.transferPrice
4434 mandeep.dh 1498
 
1499
    for lineitem in t_po_lineitems:
1500
        if lineitem.quantity > 0:
1501
            t_purchase_order.lineitems.append(lineitem)
1502
 
1503
    if not t_purchase_order.lineitems:
1504
        return -1
1505
 
2819 chandransh 1506
    warehouse_client = WarehouseClient().get_client()
1507
    po_id = warehouse_client.createPurchaseOrder(t_purchase_order)
1508
 
1509
    for order in pending_orders:
1510
        order.purchase_order_id = po_id
4434 mandeep.dh 1511
        order.status = OrderStatus.LOW_INV_PO_RAISED
1512
        order.statusDescription = 'In Process'
2819 chandransh 1513
    session.commit()
1514
    return po_id
1515
 
3451 chandransh 1516
def update_weight(order_id, weight):
3469 chandransh 1517
    '''
1518
    Update the weight of order
1519
    '''
3451 chandransh 1520
    order = get_order(order_id)
1521
    order.total_weight = weight
3469 chandransh 1522
    lineitem = order.lineitems[0]
1523
    lineitem.total_weight = weight
1524
    lineitem.unit_weight = weight
3451 chandransh 1525
 
3469 chandransh 1526
    session.commit()
1527
    return order
1528
 
1529
def change_product(order_id, item_id):
1530
    '''
1531
    Ship a product of a different color to the customer.
1532
    '''
1533
    order = get_order(order_id)
4306 rajveer 1534
    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 1535
        raise TransactionServiceException(120, "This order has already been processed. Please seek help from engineering.")
1536
 
1537
    lineitem = order.lineitems[0]
3451 chandransh 1538
 
3469 chandransh 1539
    catalog_client = CatalogClient().get_client()
1540
 
1541
    old_item = catalog_client.getItem(lineitem.item_id)
1542
    item = catalog_client.getItem(item_id)
1543
    if old_item.catalogItemId != item.catalogItemId:
1544
        raise TransactionServiceException(121, "You can't ship a different item. You can only ship the same item in a different color.")
1545
 
1546
    catalog_client.reserveItemInWarehouse(item_id, order.warehouse_id, lineitem.quantity)
1547
    catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity)
1548
    #TODO: Check that this new item has the same price
1549
 
1550
    lineitem.item_id = item_id
1551
    lineitem.color = item.color
1552
 
3451 chandransh 1553
    session.commit()
3469 chandransh 1554
 
1555
    order = get_order(order_id)
3451 chandransh 1556
    return order
3469 chandransh 1557
 
1558
def change_warehouse(order_id, warehouse_id):
1559
    '''
1560
    Update the warehouse which will be used to fulfil this order.
1561
    '''
1562
    order = get_order(order_id)
4306 rajveer 1563
    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 1564
        raise TransactionServiceException(117, "This order has already been processed. Please seek help from engineering.")
1565
 
1566
    if warehouse_id == order.warehouse_id:
1567
        raise TransactionServiceException(118, "You have selected the current warehouse again. Nothing to do")
1568
 
1569
    lineitem = order.lineitems[0]
1570
    catalog_client = CatalogClient().get_client()
1571
    try:
1572
        warehouse = catalog_client.getWarehouse(warehouse_id)
1573
    except:
1574
        raise TransactionServiceException(119, "No warehouse with id" + str(warehouse_id))
1575
 
1576
    catalog_client.reserveItemInWarehouse(lineitem.item_id, warehouse_id, lineitem.quantity)
1577
    catalog_client.reduceReservationCount(lineitem.item_id, order.warehouse_id, lineitem.quantity)
3451 chandransh 1578
 
3469 chandransh 1579
    order.warehouse_id = warehouse_id
1580
    session.commit()
1581
    return order
3553 chandransh 1582
 
3986 chandransh 1583
def add_delay_reason(order_id, reason, further_delay=0):
3553 chandransh 1584
    '''
3994 chandransh 1585
    Adds a delay reason to the order and updates the expected delivery time.
3553 chandransh 1586
    '''
1587
    order = get_order(order_id)
1588
    try:
3581 chandransh 1589
        order.delay_reason = DelayReason._VALUES_TO_NAMES[reason]
4009 chandransh 1590
        actual_delivery_delay = adjust_delivery_time(order.expected_delivery_time, further_delay)
1591
        order.expected_delivery_time = order.expected_delivery_time + datetime.timedelta(days=actual_delivery_delay)
4102 chandransh 1592
 
1593
        if order.status < OrderStatus.SHIPPED_FROM_WH:
1594
            actual_shipping_delay = adjust_delivery_time(order.expected_shipping_time, further_delay)
1595
            order.expected_shipping_time = order.expected_shipping_time + datetime.timedelta(days=actual_shipping_delay)
1596
 
3553 chandransh 1597
        session.commit()
4008 mandeep.dh 1598
 
1599
        try:
1600
            transaction_requiring_extra_processing = TransactionRequiringExtraProcessing()
1601
            transaction_requiring_extra_processing.category = 'DELAYED_DELIVERY'
1602
            transaction_requiring_extra_processing.transaction_id = order.transaction_id
1603
            session.commit()
1604
        except Exception as e:
4065 mandeep.dh 1605
            print 'Could not persist transaction id: ' + str(order.transaction_id) + ' for further processing due to: ' + str(e)
4008 mandeep.dh 1606
 
3553 chandransh 1607
        return True
1608
    except:
3994 chandransh 1609
        print sys.exc_info()[0]
3553 chandransh 1610
        return False
3956 chandransh 1611
 
1612
def reconcile_cod_collection(collected_amount_map, xferBy, xferTxnId, xferDate):
1613
    unprocessed_awbs = {}
1614
    for awb, amount in collected_amount_map.iteritems():
1615
        try:
1616
            order = Order.query.filter_by(airwaybill_no=awb).one()
1617
        except NoResultFound:
1618
            unprocessed_awbs[awb] = "No order was found for the given AWB: " + awb
1619
            continue
1620
        except MultipleResultsFound:
1621
            unprocessed_awbs[awb] = "Multiple orders were found for the given AWB:" + awb
1622
            continue
1623
 
1624
        if order.cod_reconciliation_timestamp:
1625
            #This order has been processed already! This may be a re-run. Let's allow other orders to be processed.
1626
            continue
1627
 
1628
        if order.status < OrderStatus.DELIVERY_SUCCESS:
1629
            #Order has not been delivered yet. No way we can receive the payment.
1630
            unprocessed_awbs[awb] = "Payment has been received for order: " + str(order.id) + " before delivery reconciliation"
1631
            continue
1632
 
4252 rajveer 1633
        ##As per ticket #743 if amount difference less than 0.5 we should reconcile
1634
        if abs(amount - order.total_amount) > 0.5:
3956 chandransh 1635
            #Received amount is less than or more than the order amount. Mustn't let the user proceed.
1636
            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)
1637
            continue
1638
 
1639
        try:
1640
            payment_client = PaymentClient().get_client()
1641
            payment_client.partiallyCapturePayment(order.transaction.id, amount, xferBy, xferTxnId, xferDate)
1642
        except Exception:
1643
            unprocessed_awbs[awb] = "We were unable to partially capture payment for order id " + str(order.id) + ", AWB: " + awb
1644
            continue
1645
 
1646
        #Payment has been recorded now. We can update the order peacefully.
1647
        order.cod_reconciliation_timestamp = datetime.datetime.now()
1648
        session.commit()
3469 chandransh 1649
 
3956 chandransh 1650
    return unprocessed_awbs
4008 mandeep.dh 1651
 
1652
def get_transactions_requiring_extra_processing(category):
4065 mandeep.dh 1653
    """
1654
    Returns the list of transaction ids that require some extra processing and
1655
    which belong to a particular category. This is currently used by CRM
1656
    application. If no such transaction ids are present, it returns an empty list.
1657
    """
1658
    query = TransactionRequiringExtraProcessing.query
4008 mandeep.dh 1659
 
4065 mandeep.dh 1660
    if category is not None:
1661
        query = query.filter_by(category = ExtraTransactionProcessingType._VALUES_TO_NAMES.get(category))
4008 mandeep.dh 1662
 
4065 mandeep.dh 1663
    return [a.transaction_id for a in query.all()]
4008 mandeep.dh 1664
 
1665
def mark_transaction_as_processed(transactionId, category):
4065 mandeep.dh 1666
    """
1667
    Marks a particular transaction as processed for a particular cateogory.
1668
    It essentially deletes the transaction id record for a particular
1669
    processing type category (if present) from DB.
1670
    This is currently used by CRM application.
1671
    """
4008 mandeep.dh 1672
    TransactionRequiringExtraProcessing.query.filter_by(transaction_id = transactionId).filter_by(category = ExtraTransactionProcessingType._VALUES_TO_NAMES.get(category)).delete()
1673
    session.commit()
1674
 
4018 chandransh 1675
def get_item_wise_risky_orders_count():
1676
    """
1677
    Returns a map containing the number of risky orders keyed by item id. A risky order
1678
    is defined as one whose shipping date is about to expire.
1679
    """
1680
    item_wise_risky_orders_count = {}
4019 chandransh 1681
    shipping_cutoff_time = (datetime.datetime.now() + datetime.timedelta(days = 1)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
4018 chandransh 1682
    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()
1683
    for order in orders:
1684
        for lineitem in order.lineitems:
1685
            quantity = lineitem.quantity
1686
            if item_wise_risky_orders_count.has_key(lineitem.item_id):
1687
                quantity = quantity + item_wise_risky_orders_count[lineitem.item_id]
1688
 
1689
            item_wise_risky_orders_count[lineitem.item_id] = quantity
1690
    return item_wise_risky_orders_count
1691
 
4324 mandeep.dh 1692
def update_shipment_address(order_id, address_id):
1693
    """
1694
    Updates shipment address of an order. Delivery and shipping date estimates
1695
    etc. are also updated here.
1696
 
1697
    Throws TransactionServiceException in case address change is not
1698
    possible due to certain reasons such as new pincode in address is
1699
    not serviceable etc.
1700
 
1701
    Parameters:
1702
     - orderId
1703
     - addressId
1704
    """
1705
    user_client = UserClient().get_client()
1706
    address     = user_client.getAddressById(address_id)
1707
    order       = get_order(order_id)
1708
 
1709
    logistics_client = LogisticsClient().get_client()
1710
    logistics_info = logistics_client.getLogisticsEstimation(order.lineitems[0].item_id, address.pin)
1711
 
1712
    # Check for new estimate. Raise exception if new address is not serviceable
1713
    if logistics_info.deliveryTime == -1:
1714
        raise TransactionServiceException(1, 'Location not serviceable')
1715
 
1716
    # COD services are not available everywhere
1717
    if order.cod:
1718
        if not logistics_client.isCodAllowed(address.pin):
1719
            raise TransactionServiceException(2, 'COD service not available')
1720
 
1721
    current_time = datetime.datetime.now()
1722
    logistics_info.deliveryTime = adjust_delivery_time(current_time, logistics_info.deliveryTime)
1723
    logistics_info.shippingTime = adjust_delivery_time(current_time, logistics_info.shippingTime)
1724
 
1725
    order.customer_name         = address.name
1726
    order.customer_pincode      = address.pin
1727
    order.customer_address1     = address.line1
1728
    order.customer_address2     = address.line2
1729
    order.customer_city         = address.city
1730
    order.customer_state        = address.state
1731
    order.customer_mobilenumber = address.phone
1732
 
1733
    if order.cod:
1734
        order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=COD_SHIPPING_CUTOFF_TIME, minute=0, second=0)
1735
    else:
1736
        order.expected_shipping_time = (current_time + datetime.timedelta(days=logistics_info.shippingTime)).replace(hour=PREPAID_SHIPPING_CUTOFF_TIME, minute=0, second=0)
1737
 
1738
    order.expected_delivery_time = current_time + datetime.timedelta(days=logistics_info.deliveryTime)
1739
    order.promised_shipping_time = order.expected_shipping_time
1740
    order.promised_delivery_time = order.expected_delivery_time 
1741
 
1742
    session.commit()
1743
 
4247 rajveer 1744
def mark_order_cancellation_request_received(orderId):
1745
    """
1746
    Mark order as cancellation request received. If customer sends request of cancellation of
1747
    a particular order, this method will be called. It will just change status of the order
1748
    depending on its current status. It also records the previous status, so that we can move
1749
    back to that status if cancellation request is denied.
1750
 
1751
    Parameters:
1752
     - orderId
1753
    """
1754
    status_transition = {OrderStatus.INIT : OrderStatus.CANCEL_REQUEST_RECEIVED,
1755
             OrderStatus.SUBMITTED_FOR_PROCESSING : OrderStatus.CANCEL_REQUEST_RECEIVED,
1756
             OrderStatus.INVENTORY_LOW : OrderStatus.CANCEL_REQUEST_RECEIVED,
4306 rajveer 1757
             OrderStatus.LOW_INV_PO_RAISED : OrderStatus.CANCEL_REQUEST_RECEIVED,
1758
             OrderStatus.LOW_INV_REVERSAL_IN_PROCESS : OrderStatus.CANCEL_REQUEST_RECEIVED,
1759
             OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT : OrderStatus.CANCEL_REQUEST_RECEIVED,
4247 rajveer 1760
             OrderStatus.ACCEPTED : OrderStatus.CANCEL_REQUEST_RECEIVED,
1761
             OrderStatus.BILLED : OrderStatus.CANCEL_REQUEST_RECEIVED,
1762
             OrderStatus.PAYMENT_FLAGGED : OrderStatus.CANCEL_REQUEST_RECEIVED
1763
             }
1764
    if order.status not in status_transition.keys():
1765
        raise TransactionServiceException(114, "This order can't be considered for cancellation")
1766
 
1767
    order = get_order(orderId)
1768
    order.previousStatus = order.status
1769
    order.status = OrderStatus.CANCEL_REQUEST_RECEIVED
1770
    order.statusDescription = "Cancellation request received from user"
1771
    session.commit()
1772
 
4258 rajveer 1773
def mark_transaction_as_payment_flag_removed(transactionId):
4247 rajveer 1774
    """
4258 rajveer 1775
    If we and/or payment gateway has decided to accept the payment, this method needs to be called.
1776
    Changed transaction and all orders status to payment accepted.
4247 rajveer 1777
 
1778
    Parameters:
4258 rajveer 1779
     - transactionId
4247 rajveer 1780
    """
4258 rajveer 1781
    transaction = get_transaction(transactionId)
1782
    transaction.status = TransactionStatus.AUTHORIZED
1783
    for order in transaction.orders:
1784
        order.status = OrderStatus.SUBMITTED_FOR_PROCESSING
1785
        order.statusDescription = "Submitted to warehouse"
4247 rajveer 1786
    session.commit()
1787
 
4259 anupam.sin 1788
def refund_transaction(transactionId, refundedBy, reason):
1789
    """
1790
    This method is called when a flagged payment is deemed unserviceable and the corresponding orders
1791
    need to be cancelled
1792
 
1793
    Parameters:
1794
     - transactionId
1795
    """
1796
    transaction = get_transaction(transactionId)
1797
    transaction.status = TransactionStatus.FAILED
1798
    for order in transaction.orders:
1799
        refund_order(order.id, refundedBy, reason)
1800
    session.commit()
1801
 
4247 rajveer 1802
def mark_order_cancellation_request_denied(orderId):
1803
    """
4283 anupam.sin 1804
    If we decide to not to cancel order, we will move the order to previous status.
4247 rajveer 1805
 
1806
    Parameters:
1807
     - orderId
1808
    """
1809
    order = get_order(orderId)
1810
    order.status = order.previousStatus
1811
    order.previousStatus = None
1812
    order.statusDescription = "Cancellation request denied"
1813
    session.commit()
1814
 
1815
def mark_order_cancellation_request_confirmed(orderId):
1816
    """
1817
    If we decide to to cancel order, CRM will call this method to move the status of order to
1818
    cancellation request confirmed. After this OM will be able to cancel the order.
1819
 
1820
    Parameters:
1821
     - orderId
1822
    """
1823
    order = get_order(orderId)
1824
    order.status = OrderStatus.CANCEL_REQUEST_CONFIRMED
1825
    order.statusDescription = "Cancellation request confirmed"
1826
    session.commit()
1827
 
4285 rajveer 1828
def accept_orders_for_item_id(itemId, inventory):
1829
    """
1830
    Marks the orders as ACCEPTED for the given itemId and inventory. It also updates the accepted timestamp. If the
1831
    given order is not a COD order, it also captures the payment if the same has not been captured.
1832
 
1833
    Parameters:
1834
     - itemId
1835
     - inventory
1836
    """
4351 rajveer 1837
    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 1838
    for order in orders:
4301 rajveer 1839
        if inventory > 0:
1840
            lineitem = order.lineitems[0]
1841
            if itemId == lineitem.item_id:
1842
                try:
1843
                    accept_order(order.id)
1844
                    inventory = inventory - lineitem.quantity
1845
                except:
1846
                    logging.info("Unable to accept the order")
1847
        else:
1848
            order_outofstock(order.id)
4285 rajveer 1849
    return
1850
 
4369 rajveer 1851
def mark_orders_as_po_raised(vendorId, itemId, quantity, estimate, isReminder):
1852
    if isReminder:
1853
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
1854
    else:
1855
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
1856
 
1857
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 1858
    for order in orders:
1859
        lineitem = order.lineitems[0]
1860
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
1861
            if estimate > 0:
1862
                add_delay_reason(order.id, DelayReason.OTHERS, estimate)
1863
            order.status = OrderStatus.LOW_INV_PO_RAISED
4362 rajveer 1864
            order.statusDescription = "PO Raised"
4303 rajveer 1865
            quantity = quantity - lineitem.quantity
1866
            store_inventory_action(order, itemId, HotspotAction.PO_RAISED, estimate, lineitem.quantity)
1867
            session.commit()
1868
 
4369 rajveer 1869
def mark_orders_as_reversal_initiated(vendorId, itemId, quantity, estimate, isReminder):
1870
    if isReminder:
1871
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
1872
    else:
1873
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
1874
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 1875
    for order in orders:
1876
        lineitem = order.lineitems[0]
1877
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
1878
            if estimate > 0:
1879
                add_delay_reason(order.id, DelayReason.OTHERS, estimate)
1880
            order.status = OrderStatus.LOW_INV_REVERSAL_IN_PROCESS
4362 rajveer 1881
            order.statusDescription = "Reversal Initiated"
4303 rajveer 1882
            quantity = quantity - lineitem.quantity
1883
            store_inventory_action(order, itemId, HotspotAction.REVERSAL_INITIATED, estimate, lineitem.quantity)
1884
            session.commit()
1885
 
4369 rajveer 1886
def mark_orders_as_not_available(vendorId, itemId, quantity, estimate, isReminder):
1887
    if isReminder:
1888
        statuses = tuple([OrderStatus.LOW_INV_PO_RAISED_TIMEOUT, OrderStatus.LOW_INV_REVERSAL_TIMEOUT])
1889
    else:
1890
        statuses = tuple([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW])
1891
 
1892
    orders = Order.query.filter(Order.status.in_(statuses)).order_by(Order.expected_shipping_time).all()
4303 rajveer 1893
    for order in orders:
1894
        lineitem = order.lineitems[0]
1895
        if quantity >= lineitem.quantity and itemId == lineitem.item_id:
1896
            order.status = OrderStatus.LOW_INV_NOT_AVAILABLE_AT_HOTSPOT
4362 rajveer 1897
            order.statusDescription = "Not Available"
4303 rajveer 1898
            quantity = quantity - lineitem.quantity
1899
            store_inventory_action(order, itemId, HotspotAction.NOT_AVAILABLE, estimate, lineitem.quantity)
1900
            session.commit()
1901
 
1902
def store_inventory_action(order, itemId, hotspotAction, estimate, quantity):
4369 rajveer 1903
    orderInv = OrderInventory.get_by(order_id=order.id)
1904
    if not orderInv:
1905
        orderInv = OrderInventory()
1906
        orderInv.order = order
4303 rajveer 1907
    orderInv.hotspotAction = hotspotAction  
1908
    orderInv.estimate = estimate
1909
    orderInv.itemId = itemId
1910
    orderInv.timestamp = datetime.datetime.now()
1911
    orderInv.quantity = quantity
1912
 
4369 rajveer 1913
 
1914
def mark_orders_as_timeout(vendorId):
1915
    ret_timeouts = {}
1916
    orderInvs = OrderInventory.query.all()
1917
    for orderInv in orderInvs:
1918
        if orderInv.timestamp +  timedelta(days=orderInv.estimate) < datetime.datetime.now():
1919
            if not ret_timeouts.has_key(orderInv.itemId):
1920
                timeout = TimeoutSummary()
1921
                ret_timeouts[orderInv.itemId] = timeout
1922
            else:
1923
                timeout = ret_timeouts.get(orderInv.itemId)
1924
            if orderInv.hotspotAction == HotspotAction.PO_RAISED and orderInv.order.status == OrderStatus.LOW_INV_PO_RAISED:
1925
                if timeout.poRaised:
1926
                    timeout.poRaised += 1
1927
                else:
1928
                    timeout.poRaised = 1
1929
 
1930
                if timeout.poEstimate:
1931
                    timeout.poEstimate += orderInv.estimate
1932
                else:
1933
                    timeout.poEstimate = orderInv.estimate
1934
                orderInv.order.status = OrderStatus.LOW_INV_PO_RAISED_TIMEOUT
1935
                orderInv.order.statusDescription = "PO Raised Timeout"
1936
            elif orderInv.hotspotAction == HotspotAction.REVERSAL_INITIATED and orderInv.order.status == OrderStatus.LOW_INV_REVERSAL_IN_PROCESS:
1937
                if timeout.reversalInitiated:
1938
                    timeout.reversalInitiated += 1
1939
                else:
1940
                    timeout.reversalInitiated = 1
1941
 
1942
                if timeout.reversalEstimate:
1943
                    timeout.reversalEstimate += orderInv.estimate
1944
                else:
1945
                    timeout.reversalEstimate = orderInv.estimate
1946
                orderInv.order.status = OrderStatus.LOW_INV_REVERSAL_TIMEOUT
1947
                orderInv.order.statusDescription = "Reversal Initiated Timeout"
1948
            else:
1949
                orderInv.delete()
1950
    session.commit()
1951
    return ret_timeouts
1952
 
4386 anupam.sin 1953
def get_order_for_awb(awb):
1954
    order = Order.get_by(airwaybill_no=awb)
1955
    if not order:
1956
        raise TransactionServiceException(108, "no such order")
1957
 
1958
    return order
4369 rajveer 1959
 
4386 anupam.sin 1960
 
766 rajveer 1961
def close_session():
1962
    if session.is_active:
1963
        print "session is active. closing it."
2783 chandransh 1964
        session.close()
3376 rajveer 1965
 
1966
def is_alive():
1967
    try:
1968
        session.query(Transaction.id).limit(1).one()
1969
        return True
1970
    except:
4343 varun.gupt 1971
        return False