Subversion Repositories SmartDukaan

Rev

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