Subversion Repositories SmartDukaan

#!/usr/bin/python

'''
Creates a CSV report of all the shipped orders and send mail to Sandeep.

Created on 20-Feb-2012

@author: Rajveer
'''
import xlwt
import datetime
from optparse import OptionParser



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

from shop2020.utils.Utils import to_py_date, to_java_date
from shop2020.clients.TransactionClient import TransactionClient
from shop2020.clients.LogisticsClient import LogisticsClient
from shop2020.clients.CatalogClient import CatalogClient
from shop2020.thriftpy.model.v1.order.ttypes import  Order, DelayReason
from shop2020.utils.EmailAttachmentSender import mail, get_attachment_part
to = ["sandeep.sachdeva@shop2020.in","ashutosh.saxena@shop2020.in","suraj.sharma@shop2020.in","pankaj.jain@spiceglobal.com", "pankaj.kankar@shop2020.in", "chaitnaya.vats@shop2020.in", "asghar.bilgrami@shop2020.in"]

def generate_delayed_orders_file():
    txn_client = TransactionClient().get_client()
    slipped_orders = txn_client.getSlippedSippingDateOrders()
    
    wbk = xlwt.Workbook()
    sheet = wbk.add_sheet('main')

    heading_xf = xlwt.easyxf('font: bold on; align: wrap on, vert centre, horiz center')
    sheet.set_panes_frozen(True)
    sheet.set_horz_split_pos(1)
    sheet.set_remove_splits(True)
    
    excel_integer_format = '0'
    integer_style = xlwt.XFStyle()
    integer_style.num_format_str = excel_integer_format

    date_style = xlwt.XFStyle()
    date_style.num_format_str = "M/D/YY"
    
    sheet.write(0, 0, "Order ID", heading_xf)
    sheet.write(0, 1, "Mode.", heading_xf)
    sheet.write(0, 2, "Brand", heading_xf)
    sheet.write(0, 3, "Model Name", heading_xf)
    sheet.write(0, 4, "Model Number", heading_xf)
    sheet.write(0, 5, "Color", heading_xf)
    sheet.write(0, 6, "Creation Date", heading_xf)
    sheet.write(0, 7, "Promised Shipping Date", heading_xf)
    sheet.write(0, 8, "Expected Shipping Date", heading_xf)
    sheet.write(0, 9, "Delay Reason", heading_xf)
    sheet.write(0, 10, "Detailed Reason Text", heading_xf)    

    order = Order()
    mode = {}
    mode[0] = 'Prepaid'
    mode[1] = 'COD'
    
    i = 1
    for order in slipped_orders:
        sheet.write(i, 0, order.id) 
        sheet.write(i, 1, mode[order.cod])
        
        lineitem = order.lineitems[0]
        sheet.write(i, 2, lineitem.brand)
        sheet.write(i, 3, lineitem.model_name)
        sheet.write(i, 4, lineitem.model_number)
        sheet.write(i, 5, lineitem.color)
        
        sheet.write(i, 6, to_py_date(order.created_timestamp), date_style)
        sheet.write(i, 7, to_py_date(order.promised_shipping_time), date_style)
        sheet.write(i, 8, to_py_date(order.expected_shipping_time), date_style)
        if order.delayReason:
            sheet.write(i, 9, DelayReason._VALUES_TO_NAMES[order.delayReason])
        sheet.write(i, 10, order.delayReasonText)
        
        i= i+1
    
    today = datetime.date.today()
    datestr = str(today.year) + "-" + str(today.month) + "-" + str(today.day)
    filename = "/tmp/SlippedOrders-" + datestr + ".xls" 
    wbk.save(filename)
    
    try:
        part = get_attachment_part(filename)
        mail("cnc.center@shop2020.in", "5h0p2o2o", to, "Slipped Orders as on: " + datestr, "PFA attached xls files for the orders which missed the shipping date.", [part])
    finally:
        os.remove(filename)

def generate_cancelled_orders_file():
    txn_client = TransactionClient().get_client()
    #cancelled_orders = txn_client.getAllOrders(OrderStatus.CANCELLED_DUE_TO_LOW_INVENTORY, to_java_date(datetime.datetime.now() - datetime.timedelta(hours = 24)), to_java_date(datetime.datetime.now()), -1)
    cancelled_orders = txn_client.getCancelledOrders(to_java_date(datetime.datetime.now() - datetime.timedelta(hours = 24)), to_java_date(datetime.datetime.now()))
    
    wbk = xlwt.Workbook()
    sheet = wbk.add_sheet('main')

    heading_xf = xlwt.easyxf('font: bold on; align: wrap on, vert centre, horiz center')
    sheet.set_panes_frozen(True)
    sheet.set_horz_split_pos(1)
    sheet.set_remove_splits(True)
    
    excel_integer_format = '0'
    integer_style = xlwt.XFStyle()
    integer_style.num_format_str = excel_integer_format

    date_style = xlwt.XFStyle()
    date_style.num_format_str = "M/D/YY"
    
    sheet.write(0, 0, "Order ID", heading_xf)
    sheet.write(0, 1, "Mode.", heading_xf)
    sheet.write(0, 2, "Brand", heading_xf)
    sheet.write(0, 3, "Model Name", heading_xf)
    sheet.write(0, 4, "Model Number", heading_xf)
    sheet.write(0, 5, "Color", heading_xf)
    sheet.write(0, 6, "Creation Date", heading_xf)
    sheet.write(0, 7, "Promised Shipping Date", heading_xf)
    sheet.write(0, 8, "Expected Shipping Date", heading_xf)
    sheet.write(0, 9, "Cancellation Reason", heading_xf)
    sheet.write(0, 10, "Cancellation Date", heading_xf)    

    order = Order()
    mode = {}
    mode[0] = 'Prepaid'
    mode[1] = 'COD'
    
    i = 1
    for order in cancelled_orders:
        sheet.write(i, 0, order.id) 
        sheet.write(i, 1, mode[order.cod])
        
        lineitem = order.lineitems[0]
        sheet.write(i, 2, lineitem.brand)
        sheet.write(i, 3, lineitem.model_name)
        sheet.write(i, 4, lineitem.model_number)
        sheet.write(i, 5, lineitem.color)
        
        sheet.write(i, 6, to_py_date(order.created_timestamp), date_style)
        sheet.write(i, 7, to_py_date(order.promised_shipping_time), date_style)
        sheet.write(i, 8, to_py_date(order.expected_shipping_time), date_style)
        sheet.write(i, 9, order.refundReason)
        sheet.write(i, 10, to_py_date(order.refund_timestamp), date_style)
        
        i= i+1
    
    today = datetime.date.today()
    datestr = str(today.year) + "-" + str(today.month) + "-" + str(today.day)
    filename = "/tmp/CancelledOrders-" + datestr + ".xls" 
    wbk.save(filename)
    
    try:
        part = get_attachment_part(filename)
        mail("cnc.center@shop2020.in", "5h0p2o2o", to, "Low Inventory Cancelled Orders as on: " + datestr, "PFA attached xls files for the orders which has been cancelled.", [part])
    finally:
        os.remove(filename)

def generate_expected_delivery_missed_orders_file():
    
    txn_client = TransactionClient().get_client()
    orders_not_delivered = txn_client.getUndeliveredOrdersExpectedDeliveryDateNotMet()
    
    print "orders_not_delivered"
    print orders_not_delivered
    
    logistics_client = LogisticsClient().get_client()
    providers = logistics_client.getAllProviders()
    providers_map = {}
    for provider in providers:
        providers_map[provider.id] = provider.name
    
    catalog_client = CatalogClient().get_client()
    warehouses = catalog_client.getAllWarehouses(True)
    warehouses_map = {}
    for warehouse in warehouses:
        warehouses_map[warehouse.id] = warehouse.displayName
    
    wbk = xlwt.Workbook()
    sheet = wbk.add_sheet('main')

    heading_xf = xlwt.easyxf('font: bold on; align: wrap on, vert centre, horiz center')
    sheet.set_panes_frozen(True)
    sheet.set_horz_split_pos(1)
    sheet.set_remove_splits(True)
    
    excel_integer_format = '0'
    integer_style = xlwt.XFStyle()
    integer_style.num_format_str = excel_integer_format

    date_style = xlwt.XFStyle()
    date_style.num_format_str = "M/D/YY"
    
    sheet.write(0, 0, "Order ID", heading_xf)
    sheet.write(0, 1, "Mode.", heading_xf)
    sheet.write(0, 2, "Airway BillNo", heading_xf)
    sheet.write(0, 3, "Current Status", heading_xf)
    sheet.write(0, 4, "Shipping Date", heading_xf)
    sheet.write(0, 5, "Pickup Date", heading_xf)
    sheet.write(0, 6, "Expected Delivery Date", heading_xf)
    sheet.write(0, 7, "Warehouse", heading_xf)
    sheet.write(0, 8, "Warehouse ID", heading_xf)
    sheet.write(0, 9, "Logistics Provider", heading_xf)
    sheet.write(0, 10, "Logistics Provider ID", heading_xf)
    sheet.write(0, 11, "Customer City", heading_xf)
    sheet.write(0, 12, "Customer State", heading_xf)
    sheet.write(0, 13, "Customer Pincode", heading_xf)

    order = Order()
    mode = {}
    mode[0] = 'Prepaid'
    mode[1] = 'COD'
    
    i = 1
    for order in orders_not_delivered:
        sheet.write(i, 0, order.id) 
        sheet.write(i, 1, mode[order.cod])
        sheet.write(i, 2, order.airwaybill_no)
        sheet.write(i, 3, order.statusDescription)
        sheet.write(i, 4, to_py_date(order.shipping_timestamp), date_style)
        sheet.write(i, 5, to_py_date(order.pickup_timestamp), date_style)
        sheet.write(i, 6, to_py_date(order.expected_delivery_time), date_style)
        sheet.write(i, 7, warehouses_map.get(order.warehouse_id))
        sheet.write(i, 8, order.warehouse_id)
        sheet.write(i, 9, providers_map.get(order.logistics_provider_id))
        sheet.write(i, 10, order.logistics_provider_id)
        sheet.write(i, 11, order.customer_city)
        sheet.write(i, 12, order.customer_state)
        sheet.write(i, 13, order.customer_pincode)
        
        i= i+1
    
    today = datetime.date.today()
    datestr = str(today.year) + "-" + str(today.month) + "-" + str(today.day)
    filename = "/tmp/UndeliveredOrders-" + datestr + ".xls" 
    wbk.save(filename)
    
    try:
        part = get_attachment_part(filename)
        mail("cnc.center@shop2020.in", "5h0p2o2o", to, "Orders which missed Expected Delivery Date as on: " + datestr, "PFA attached xls files for the orders which missed the Expected Delivery Date.", [part])
    finally:
        os.remove(filename)

def main():
    parser = OptionParser()
    parser.add_option("-d", "--delayed", dest="delayed",
                      action="store_true",
                      help="")
    parser.add_option("-c", "--cancelled", dest="cancelled",
                      action="store_true",
                      help="")
    parser.add_option("-u", "--undelivered", dest="undelivered",
                      action="store_true",
                      help="")

    (options, args) = parser.parse_args()
    if options.delayed:
        generate_delayed_orders_file()
    if options.cancelled:
        generate_cancelled_orders_file()
    if options.undelivered:
        generate_expected_delivery_missed_orders_file()

if __name__ == '__main__':
    main()