Subversion Repositories SmartDukaan

#!/usr/bin/python

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.thriftpy.model.v1.catalog.ttypes import status

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

from shop2020.model.v1.catalog.impl.DataService import Item, EntityIDGenerator,\

    ItemChangeLog

from elixir import *

def load_item_data(filename):

    DataService.initialize()

    workbook = xlrd.open_workbook(filename)

    sheet = workbook.sheet_by_index(0)

    num_rows = sheet.nrows

    updatedOn = datetime.datetime.now()

    for rownum in range(1, num_rows):

        print sheet.row_values(rownum)

        hotspot_category, brand, model_number, color, model_name, mrp, mop, dp, xfer_price, weight, start_date, deal_text, deal_value, comments, catalog_item_id, category_id = sheet.row_values(rownum)[0:17]

        if isinstance(model_number, float):

            model_number = str(int(model_number))

        item = Item.get_by(manufacturer_name=brand, model_number=model_number, color=color)

        if item is None:

            print "[ADDING:]{0} {1} {2} to our catalogue.".format(brand, model_number, color)

            item = Item()

            item.manufacturer_name = brand

            item.model_number = model_number

            item.color = color

            item.status = status.IN_PROCESS

            item.addedOn = updatedOn

            session.add(item)

        #Set the catalog_item_id.

        similar_item = Item.query.filter_by(manufacturer_name = brand, model_number=model_number).first()

        print similar_item

        if similar_item is None or similar_item.catalog_item_id is None:

            #Use the entity_id_generator

            entity_id = EntityIDGenerator.query.first()

            item.catalog_item_id = entity_id.id + 1

            entity_id.id = entity_id.id  + 1

        else:

            #If it already exists for a brand and model_number, set it as same.

            item.catalog_item_id = similar_item.catalog_item_id

        #added category and entity id

        '''

        item.catalog_item_id = int(catalog_item_id)

        item.category = int(category_id)

        item.status = status.ACTIVE

        '''

        item.hotspotCategory = hotspot_category

        item.model_name = model_name

        if mrp != "":

            item.mrp = mrp

        if mop != "":

            item.mop = mop

        if dp != "":    

            item.dealerPrice = dp

            item.sellingPrice = dp

        if xfer_price !="":

            item.transfer_price = xfer_price

        if weight != "":    

            item.weight = weight

        if start_date != "":

            item.startDate = datetime.datetime(*xlrd.xldate_as_tuple(start_date, workbook.datemode))

        item.bestDealText = deal_text

        if deal_value != "":

            item.bestDealValue = deal_value

        item.comments = comments

        item.updatedOn = updatedOn

        item_change_log = ItemChangeLog()

        item_change_log.new_status = item.status

        item_change_log.timestamp = updatedOn

        item_change_log.item = item

        session.commit()

    session.execute("UPDATE " + str(Item.table) + " SET status=" + str(status.PHASED_OUT) + ", updatedOn='"+ str(updatedOn) +"' WHERE updatedOn <> '" + str(updatedOn) + "'", mapper=Item)

    session.commit()

    print "Successfully updated the item list information."

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")

    (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)

if __name__ == '__main__':

    main()