Subversion Repositories SmartDukaan

#!/usr/bin/python
# coding: ascii
'''
@author: Phani Kumar
'''
import optparse
import csv
import xlrd
import datetime

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

from shop2020.thriftpy.model.v1.catalog.ttypes import status
from shop2020.model.v1.catalog.impl import DataService
from shop2020.model.v1.catalog.impl.DataService import Item, Vendor, VendorItemPricing, VendorItemMapping
from elixir import *
from shop2020.utils.EmailAttachmentSender import get_attachment_part, mail

from_user = 'cnc.center@shop2020.in'
from_pwd = '5h0p2o2o'
to = ["khushal.bhatia@shop2020.in", "chaitnaya.vats@shop2020.in", "manish.sharma@shop2020.in", "chandan.kumar@shop2020.in","manoj.kumar@shop2020.in"]

def load_item_data(filename, vendorId):
    DataService.initialize('catalog')
    
    vendor = Vendor.get_by(id=vendorId)
    if vendor is None:
        raise Exception("No vendor found for the id: " + str(vendorId))
    
    workbook = xlrd.open_workbook(filename)
    sheet = workbook.sheet_by_index(0)
    num_rows = sheet.nrows
    new_items = []
    phased_out_items = []
    updated_items = []
    unchanged_items = []
    
    for rownum in range(1, num_rows):
        print sheet.row_values(rownum)
        
        product_group, brand, model_number, model_name, color,\
        dp, mrp, mop, xfer_price = sheet.row_values(rownum)[0:9]
        
        if isinstance(model_number, float):
            model_number = str(int(model_number))
        
        item = None
        vendor_item_pricing = None
        keys = []
        if product_group.strip().lower() == 'New Handset'.strip().lower():
            for our_group in ["handsets", "mobile handset", "smart phone"]:
                new_key = our_group.strip().lower() + '|' + brand.strip().lower() + '|' + model_number.strip().lower() + '|' + color.strip().lower()
                keys.append(new_key)
        else:
            new_key = product_group.strip().lower() + '|' + brand.strip().lower() + '|' + model_number.strip().lower() + '|' + color.strip().lower()
            keys.append(new_key)
        
        i = 0
        for key in keys:
            i = i + 1
            vendor_item_mapping = VendorItemMapping.get_by(vendor=vendor, item_key=key)
            if vendor_item_mapping:
                item = vendor_item_mapping.item
                if item is not None and item.status != status.PHASED_OUT:
                    vendor_item_pricing = VendorItemPricing.get_by(vendor=vendor, item=item)
                    if vendor_item_pricing is None:
                        updated_items.append(sheet.row_values(rownum)[0:9] + [None, item.mrp, None, None, item.sellingPrice])
                    elif item.mrp != mrp or vendor_item_pricing.dealerPrice != dp or vendor_item_pricing.transfer_price != xfer_price or vendor_item_pricing.mop != mop:
                        updated_items.append(sheet.row_values(rownum)[0:9] + [vendor_item_pricing.dealerPrice, item.mrp, vendor_item_pricing.mop, vendor_item_pricing.transfer_price, item.sellingPrice])
                    else:
                        unchanged_items.append(rownum)
                else:
                    phased_out_items.append(rownum)
                break
            elif i == len(keys):
                new_items.append(rownum)

    write_report("/tmp/new_items.csv", new_items, sheet, False)
    write_report("/tmp/phased_out_items.csv", phased_out_items, sheet, False)
    write_report("/tmp/updated_items.csv", updated_items, sheet, True)
    write_report("/tmp/unchanged_items.csv", unchanged_items, sheet, False)
    
    attachment_part = [get_attachment_part("/tmp/new_items.csv"), get_attachment_part("/tmp/phased_out_items.csv"), get_attachment_part("/tmp/updated_items.csv"), get_attachment_part("/tmp/unchanged_items.csv")]
    mail(from_user, from_pwd, to, "Comparison report for " + os.path.basename(filename), "This is a system generated email.Please don't reply to it.", attachment_part)

def write_report(filename, items, sheet, is_item):
    '''
    Iterates through the items list and writes all the values to the specified
    filename in the CSV format. 'is_item' indicates whether the list consists
    of row numbers or complete row data.
    '''
    items_writer = csv.writer(open(filename, "wb"), delimiter=',', quoting=csv.QUOTE_ALL)
    items_writer.writerow(sheet.row_values(0)[0:9] + ['Old DP', 'Old MRP', 'Old FBP', 'Old TP', 'SP'])
    if is_item:
        #The list contains the complete rows.
        for item in items:
            print item
            items_writer.writerow([str(value) for value in item])
    else:
        #The list only has row numbers. We've to fetch the data ourselves.
        for i in items:
            print sheet.row_values(i)
            items_writer.writerow([str(value) for value in sheet.row_values(i)[0:9]])

def main():
    parser = optparse.OptionParser()
    parser.add_option("-f", "--file", dest="filename",
                   default="ItemList.xls", type="string",
                   help="Read the item list from FILE",
                   metavar="FILE")
    parser.add_option("-v", "--vendor", dest="vendor",
                      type="int",
                      help="Update the pricing information for VENDOR",
                      metavar="VENDOR")
    parser.set_defaults(vendor=1)
    (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?")
    filename = options.filename
    load_item_data(filename, options.vendor)

if __name__ == '__main__':
    main()