Subversion Repositories SmartDukaan

#!/usr/bin/python 

import datetime
import optparse
import sys
import csv
import xlrd

if __name__ == '__main__' and __package__ is None:
    import os
    sys.path.insert(0, os.getcwd())

from shop2020.clients.LogisticsClient import LogisticsClient
from shop2020.clients.TransactionClient import TransactionClient
from shop2020.thriftpy.model.v1.order.ttypes import TransactionServiceException
from shop2020.utils.EmailAttachmentDownloader import download_attachment
from shop2020.utils.Utils import to_py_date

def process_pickup_records(provider):
    filename = fetch_report('PICKUP REPORT')
    pickup_details = read_pickup_report(filename)
    orders_not_picked_up = update_picked_orders(provider.id, pickup_details)
    if orders_not_picked_up:
        print "Some of our orders were not picked up. Printing report to: PickupMismatch.csv"
        print_discrepancy_report("PickupMismatch.csv", orders_not_picked_up)

def process_delivery_report(provider):
    filename = fetch_report('DELIVERED AND RTO REPORT')
    #filename = 'delivery_report.xls'
    delivered_orders, returned_orders = read_delivery_report(filename)
    update_delivered_orders(provider.id, delivered_orders)
    update_returned_orders(provider.id, returned_orders)
    #TODO: Send returned orders in a mail

def process_non_delivery_report(provider):
    filename = fetch_report('UNDELIVERED REPORT')
                
def get_provider_by_name(provider_name):
    logistics_client = LogisticsClient().get_client()
    #FIXME: Write a thrift call to get a provider by name
    provider = None
    providers = logistics_client.getAllProviders()
    for p in providers:
        if p.name == provider_name:
            provider=p
            break
    if provider == None:
        sys.exit("Can't continue execution: No such provider")
    return provider

def fetch_report(type):
    filename = download_attachment(type, todays_date_string())
    if filename is None:
        sys.exit("The " + type + " report is not yet available.")
    return filename

def read_pickup_report(filename):
    print "Reading pickup report from:" + filename
    workbook = xlrd.open_workbook(filename)
    sheet = workbook.sheet_by_index(0)
    num_rows = sheet.nrows
    picked_up_orders = {}
    for rownum in range(1, num_rows):
        unused_customer_code, awb, date, time = sheet.row_values(rownum)[0:4]
        picked_up_orders[awb] = 0 #TODO: Use the date and time value in the report
    
    print "Picked up Orders:"
    print picked_up_orders
    return picked_up_orders

def read_delivery_report(filename):
    print "Reading delivery details from:" + filename
    workbook = xlrd.open_workbook(filename)
    sheet = workbook.sheet_by_index(0)
    num_rows = sheet.nrows
    delivered_orders = {}
    returned_orders = {}
    for rownum in range(1, num_rows):
        unused_customer_code, awb, date, time, status, receiver, reason_for_return = sheet.row_values(rownum)[0:7]
        if receiver: #TODO: Use status for this check
            delivered_orders[awb] = str(datetime.datetime.now()) + "|" +  receiver #TODO: Use the date and time value in the report
        else:
            returned_orders[awb] = str(datetime.datetime.now()) + "|" + reason_for_return
    
    print "Delivered Orders:"
    print delivered_orders
    
    print "Returned Orders:"
    print returned_orders
    return delivered_orders, returned_orders

def update_picked_orders(provider_id, pickup_details):
    txnClient = TransactionClient().get_client()
    try:
        orders_not_picked_up = txnClient.markOrdersAsPickedUp(provider_id, pickup_details)
        return orders_not_picked_up
    except TransactionServiceException as tex:
        print tex.message

def update_delivered_orders(provider_id, delivered_orders):
    txnClient = TransactionClient().get_client()
    try:
        txnClient.markOrdersAsDelivered(provider_id, delivered_orders)
    except TransactionServiceException as tex:
        print tex.message

def update_returned_orders(provider_id, returned_orders):
    txnClient = TransactionClient().get_client()
    try:
        txnClient.markOrdersAsFailed(provider_id, returned_orders)
    except TransactionServiceException as tex:
        print tex.message

def print_discrepancy_report(filename, orders):
    writer = csv.writer(open(filename, "wb"), delimiter=',', quoting=csv.QUOTE_NONE)
    writer.writerow(['Order Id', 'AWB No', 'Shipping timestamp'])
    for order in orders:
        writer.writerow([order.id, order.airwaybill_no, to_py_date(order.shipping_timestamp)])

def todays_date_string():
    return '"24-Mar-2011"'

def main():
    parser = optparse.OptionParser()
    parser.add_option("-p", "--pickup", dest="pickup_report",
                   action="store_true",
                   help="Run the pickup reconciliation")
    parser.add_option("-d", "--delivery", dest="delivery_report",
                   action="store_true",
                   help="Run the delivery reconciliation")
    parser.add_option("-n", "--non-delivery", dest="non_delivery_report",
                   action="store_true",
                   help="Run the non delivery reconciliation")
    parser.add_option("-a", "--all", dest="all_reports",
                   action="store_true",
                   help="Run all reconciliations")
    parser.add_option("-P", "--provider", dest="provider",
                   default="BlueDart", type="string",
                   help="The PROVIDER this report is for",
                   metavar="PROVIDER")
    parser.set_defaults(pickup_report=False, delivery_report=False, non_delivery_report=False, all_reports=False)
    (options, args) = parser.parse_args()
    if len(args) != 0:
        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")
    
    if options.all_reports:
        options.pickup_report = True
        options.delivery_report = True
        options.non_delivery_report = True
    
    provider = get_provider_by_name(options.provider)
    
    if options.pickup_report:
        process_pickup_records(provider)
    if options.delivery_report:
        process_delivery_report(provider)
    if options.non_delivery_report:
        process_non_delivery_report(provider)

if __name__ == '__main__':
    main()