Subversion Repositories SmartDukaan

Rev

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