Subversion Repositories SmartDukaan

'''

Created on 23-Mar-2010

@author: ashish

'''

from elixir import *

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

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

    Warehouse, ItemInventoryHistory, CurrentInventorySnapshot, ItemInfo,\

    ItemChangeLog, Price

from shop2020.thriftpy.model.v1.catalog.ttypes import \

    InventoryServiceException, status

from shop2020.thriftpy.model.v1.catalog.ttypes import Item as TItem

from shop2020.thriftpy.model.v1.catalog.ttypes import Warehouse as TWarehouse

import datetime

from shop2020.utils.Utils import log_entry, to_py_date, to_java_date

def initialize():

    DataService.initialize()

def get_all_items(is_active):

    if is_active:

        return Item.query.filter_by(status= status.ACTIVE).all()

    else:

        return Item.query.all()

def get_all_items_by_status(status):

    if not status:

        #return all items

        return get_all_items(False)

    else:

        return Item.query.filter_by(status=status).all()

def get_Item(item_id):

    return Item.get_by(id=item_id)

def get_item_by_catalog_id(catalog_id):

    query = Item.get_by(catalog_item_id=catalog_id)

    try:

        return query.one()

    except:

        raise InventoryServiceException(109, "Item not found")

def get_Warehouse(warehouse_id):

    return Warehouse.get_by(id=warehouse_id)

def get_all_warehouses_by_status(status):

    if not status:

        return Warehouse.query.all()

    else:

        return Warehouse.query.filter_by(status=status)

def get_all_warehouses_for_item(item_id):

    item = get_Item(item_id)

    if not item:

        raise InventoryServiceException(108, "Some unforeseen error while obtaining item")

    return item.get_all_warehouses

def get_all_items_for_warehouse(warehouse_id):

    warehouse = get_Warehouse(warehouse_id)

    if not warehouse:

        raise InventoryServiceException(108, "bad warehouse")

    return warehouse.all_items

def add_warehouse(warehouse):

    if not warehouse:

        raise InventoryServiceException(108, "Bad warehouse")

    if get_Warehouse(warehouse.id):

        #warehouse is already present.

        raise InventoryServiceException(101, "Warehouse already present")

    ds_warehouse = Warehouse()

    ds_warehouse.id = warehouse.id

    ds_warehouse.location = warehouse.location

    ds_warehouse.status = status.ACTIVE

    ds_warehouse.addedOn = datetime.datetime.now()

    session.commit()

    return ds_warehouse.id

def update_item(item):

    if not item:

        raise InventoryServiceException(108, "Bad item in request")

    if not item.id:

        raise InventoryServiceException(101, "non existent item")

    ds_item = get_Item(item.id)

    if not ds_item:

        raise InventoryServiceException(101, "Item already exists")

    if item.catalogItemId:

        ds_item.catalog_item_id = item.catalogItemId

    if item.vendorItemId:

        ds_item.vendor_item_id = item.vendorItemId

    else:

        raise InventoryServiceException(101, "vendor item id cannot be missing")

    if item.weight:

        ds_item.weight = item.weight

    if item.featureId:

        ds_item.feature_id = item.featureId

    if item.featureDescription:

        ds_item.feature_description = item.featureDescription

    session.commit();

    return ds_item.id

def add_item(item):

    if not item:

        raise InventoryServiceException(108, "Bad item in request")

    if get_Item(item.id):

        raise InventoryServiceException(101, "Item already exists")

    ds_item = Item()

    if item.id:

        ds_item.id = item.id

    if item.catalogItemId:

        ds_item.catalog_item_id = item.catalogItemId

    if item.vendorItemId:

        ds_item.vendor_item_id = item.vendorItemId

    else:

        raise InventoryServiceException(101, "vendor item id cannot be missing")

    if item.itemStatus:

        ds_item.status = item.itemStatus

    else: #this is for update calls

        ds_item.status = status.IN_PROCESS

    ds_item.addedOn = datetime.datetime.now()

    if item.price:

        update_price(item, ds_item)

    if item.weight:

        ds_item.weight = item.weight

    if item.featureId:

        ds_item.feature_id = item.featureId

    if item.featureDescription:

        ds_item.feature_description = item.featureDescription

    if item.otherInfo:

        for k,v in item.otherInfo.iteritems():

            info = ItemInfo()

            info.key = k

            info.value = v

            ds_item.iteminfo.append(info)

    #check if categories present. If yes, add them to system

    session.commit();

    return ds_item.id

def update_price(item, ds_item):

    log_entry("update_price", "updating price for item")

    for k,v in item.price.iteritems():

        price = Price()

        price.warehouse_id = k

        price.price = v

        price.item = ds_item

        session.commit() 

def update_price_for_warehouse(item_id, warehouse_id, price):

    log_entry("update_price", "updating price for item and warehouse")

    price = Price.query.filter_by(warehouse_id=warehouse_id, item_id=item_id).one()

    if not price:

        price = Price()

        item = get_Item(item_id)

        price.warehouse_id = warehouse_id

        price.item = item

    price.price = price    

    session.commit()

def update_inventory(item_id, warehouse_id, quantity, timestamp):

    item = get_Item(item_id)

    warehouse = get_Warehouse(warehouse_id) 

    time = to_py_date(timestamp)

    if not item:

        raise InventoryServiceException(106, "Item not found, how did you get this?")

    if not warehouse:

        raise InventoryServiceException(107, "Warehouse? Where?")

    try:

        item_inventory_history = ItemInventoryHistory()

        item_inventory_history.warehouse = warehouse

        item_inventory_history.item = item

        item_inventory_history.timestamp = time

        item_inventory_history.availibility = quantity

        current_inventory_snapshot = CurrentInventorySnapshot.get_by(item=item, warehouse=warehouse)

        if not current_inventory_snapshot:

            current_inventory_snapshot = CurrentInventorySnapshot()

            current_inventory_snapshot.item = item

            current_inventory_snapshot.warehouse = warehouse

        current_inventory_snapshot.availibility = quantity

        current_inventory_snapshot.checkedOn = time

        session.commit()

    except:

        raise InventoryServiceException(108, "Some unforeseen error while updating inventory")

def get_item_inventoy(item_id):

    inventory = Item.get_by(id=item_id).currentInventory

    if not inventory:

        raise InventoryServiceException(108, "Some unforeseen error while updating inventory")

    return inventory

def get_item_inventory_by_catalog_id(item_id):

    inventory = Item.get_by(catalog_item_id=item_id).currentInventory

    if not inventory:

        raise InventoryServiceException(108, "Some unforeseen error while updating inventory")

    return inventory

def retire_warehouse(warehouse_id):

    if not warehouse_id:

        raise InventoryServiceException(101, "Bad warehouse id")

    warehouse = get_Warehouse(warehouse_id)

    if not warehouse:

        raise InventoryServiceException(108, "warehouse id not present")

    warehouse.status = status.DELETED;

    session.commit()

def retire_item(item_id):

    if not item_id:

        raise InventoryServiceException(101, "bad item id")

    item = get_Item(item_id)

    if not item:

        raise InventoryServiceException(108, "item id not present")

    item.status = status.PHASED_OUT

    item.retireDate = datetime.datetime.now()

    session.commit()

#need to implement threads based solution here

def start_item_on(item_id, timestamp):

    if not item_id:

        raise InventoryServiceException(101, "bad item id")

    item = get_Item(item_id)

    if not item:

        raise InventoryServiceException(108, "item id not present")

    item.status = status.ACTIVE

    item.startDate = datetime.datetime.fromtimestamp(to_py_date(timestamp))

    add_status_change_log(item, status.ACTIVE)

    session.commit()

#need to implement threads here

def retire_item_on(item_id, timestamp):

    if not item_id:

        raise InventoryServiceException(101, "bad item id")

    item = get_Item(item_id)

    if not item:

        raise InventoryServiceException(108, "item id not present")

    item.status = status.PHASED_OUT

    item.retireDate = datetime.datetime.fromtimestamp(to_py_date(timestamp))

    add_status_change_log(item, status.PHASED_OUT)

    session.commit()

def add_status_change_log(item, new_status):

    item_change_log = ItemChangeLog()

    item_change_log.new_status = new_status

    item_change_log.old_status = item.status

    item_change_log.timestamp = datetime.datetime.now()

    item_change_log.item = item

    session.commit()

def change_item_status(item_id, new_status):

    if not item_id:

        raise InventoryServiceException(101, "bad item id")

    item = get_Item(item_id)

    if not item:

        raise InventoryServiceException(108, "item id not present")

    item.status = new_status

    add_status_change_log(item, new_status)

    session.commit()

def get_item_availability_for_warehouse(warehouse_id, item_id):

    if not warehouse_id:

        raise InventoryServiceException(101, "bad warehouse_id")

    if not item_id:

        raise InventoryServiceException(101, "bad item_id")

    warehouse = get_Warehouse(warehouse_id)

    if not warehouse:

        raise InventoryServiceException(108, "warehouse does not exist")

    item = get_Item(item_id)

    if not item:

            raise InventoryServiceException(108, "item does not exist")

    current_inventory_snapshot = CurrentInventorySnapshot.query.filter(CurrentInventorySnapshot.warehouse.id == warehouse_id, CurrentInventorySnapshot.item.id == item_id).one()

    if not current_inventory_snapshot:

        return 0

    else:

        return current_inventory_snapshot.availibility

def get_warehouses_for_item(item_id):

    if not item_id:

        raise InventoryServiceException(101, "bad item_id")

    item = get_Item(item_id)

    if not item:

        raise InventoryServiceException(101, "bad item")

    warehouses = item.currentInventory.warehouse