Subversion Repositories SmartDukaan

#!/usr/bin/python

'''

One-time script to update the data of RTO orders since the time

BlueDart changes their report format.

Created on 16-Nov-2011

@author: Chandranshu

'''

import optparse

import xlrd

import time

import datetime

if __name__ == '__main__' and __package__ is None:

    import os

    import sys

    sys.path.insert(0, os.getcwd())

from shop2020.clients.TransactionClient import TransactionClient

from shop2020.thriftpy.model.v1.order.ttypes import TransactionServiceException

REASON = 'reason'

DATE = 'date'

RECEIVER = 'receiver'

def load_rto_data(filename):

    print "Reading RTO details from:" + filename

    workbook = xlrd.open_workbook(filename)

    sheet = workbook.sheet_by_index(0)

    num_rows = sheet.nrows

    orders = {}

    returned_orders = {}

    orderIdString = ''

    for rownum in range(1, num_rows):

        orderId, awb, reason, unused_return_awb, date, unused_receiver = sheet.row_values(rownum)[0:7]

        orderIdString = orderIdString + str(int(orderId)) + ','

        orders[int(orderId)] = reason

        delivery_date = str(get_py_datetime(date, None))

        returned_orders[str(int(awb))] = delivery_date + "|" + reason

    print orderIdString

    return orders, returned_orders

def mark_orders_as_sales_returned(returned_orders):

    txnClient = TransactionClient().get_client()

    try:

        txnClient.markAsRTOrders(1, returned_orders)

    except TransactionServiceException as tex:

        print tex.message        

def receive_returns(orders):

    txnClient = TransactionClient().get_client()

    for orderId, unused_reason in orders.iteritems():

        try:

            txnClient.receiveReturn(orderId)

        except TransactionServiceException as tex:

            print tex.message

def refund_orders(orders):

    txnClient = TransactionClient().get_client()

    for orderId, reason in orders.iteritems():

        try:

            txnClient.refundOrder(orderId, 'ys-hadmin', reason);

        except TransactionServiceException as tex:

            print tex.message

def get_py_datetime(date, timeval):

    # This should be a command line argument.

    # Refer http://docs.python.org/library/time.html#time.strftime to

    # get a complete list of format specifiers available for date time.

    time_format = "%d-%b-%y %H%M"

    if timeval is None or timeval == '--':

        timeval='0000'

    time_string = date + " " + timeval

    mytime = time.strptime(time_string, time_format)

    return datetime.datetime(*mytime[:6])

def main():

    parser = optparse.OptionParser()

    parser.set_usage("%prog [options] filename")

    (options, args) = parser.parse_args()

    if len(args) != 1:

        parser.error("You've not supplied required arguments. Please see the usage.")

    filename = args[0]

    orders, returned_orders = load_rto_data(filename)

    mark_orders_as_sales_returned(returned_orders)

    receive_returns(orders)

    refund_orders(orders)

if __name__ == '__main__':

    main()