Subversion Repositories SmartDukaan

#!/usr/bin/python

'''

Load the whole inventory for a warehouse.

With the amount of data that we have now and the amount

of traffic we get, it's now impracitcal to do this for 

already existing warehouses. It should only be used for

a newly provisioned warehouse.

@author: Chandranshu

'''

import optparse

import xlrd

import datetime

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

    import sys

    import os

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

from shop2020.model.v1.catalog.impl import DataService

from shop2020.model.v1.catalog.impl.DataAcessors import update_inventory

from elixir import *

def read_inventory_data(filename, product_group):

    workbook = xlrd.open_workbook(filename)

    sheet = workbook.sheet_by_index(0)

    num_rows = sheet.nrows

    availability = {}

    for rownum in range(1, num_rows):

        if product_group:

            brand, model_number, color, quantity = sheet.row_values(rownum)[0:4]

        else:

            product_group, brand, model_number, color, quantity = sheet.row_values(rownum)[0:5]

        if isinstance(model_number, float):

            model_number = str(int(model_number))

        print sheet.row_values(rownum)

        itemid = product_group + ";" + brand + ";" + model_number + ";" + color

        if availability.has_key(itemid):

            quantity = quantity + availability.get(itemid)

        availability[itemid] = quantity

    return availability

def load_inventory_data(filename, warehouse_id, product_group):

    DataService.initialize('catalog')    

    availability = read_inventory_data(filename, product_group)

    update_inventory(warehouse_id, datetime.datetime.now(), availability)

    session.commit()

    print "Successfully updated the item inventory information."

def main():

    parser = optparse.OptionParser()

    parser.add_option("-f", "--file", dest="filename",

                   default="ItemInventory.xls", type="string",

                   help="Read the item inventory from FILE",

                   metavar="FILE")

    parser.add_option("-w", "--warehouse", dest="warehouse_id",

                   default="1", type="string",

                   help="Load the item inventory for the warehouse with WarehouseId as id",

                   metavar="WarehouseId")

    parser.add_option("-p", "--product-group", dest="product_group",

                      default=None, type="string",

                      help="Assume PRODUCT_GROUP to be the product group of all items",

                      metavar="PRODUCT_GROUP")

    (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

    warehouse_id = options.warehouse_id

    load_inventory_data(filename, warehouse_id, options.product_group)

if __name__ == '__main__':

    main()