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 Category, Item, Vendor,\
    Warehouse, ItemInventoryHistory, CurrentInventorySnapshot, ItemInfo,\
    ItemChangeLog
from shop2020.thriftpy.model.v1.catalog.ttypes import Category as TCategory,\
    CatalogServiceException
from shop2020.thriftpy.model.v1.catalog.ttypes import Item as TItem
from shop2020.thriftpy.model.v1.catalog.ttypes import Vendor as TVendor
from shop2020.thriftpy.model.v1.catalog.ttypes import Warehouse as TWarehouse
import datetime
from shop2020.utils.Utils import status_active, status_added, status_retired


def initialize():
    DataService.initialize()
    
def is_category_present(id_key):
    category = Category.get_by(id=id_key)
    if category:
        return True    
    return False

def get_category(id_key):
    category = Category.get_by(id=id_key)
    return category

def get_all_categories():
    return Category.query.all()

def get_all_items_for_category(category_id):
    category = Category.get_by(id=category_id)
    if not category:
        raise CatalogServiceException(108, "Some unforeseen error while obtaining")
    return category.items

def get_all_items(is_active):
    if is_active:
        return Item.get_by(status=status_active)
    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.get_by(status=status).all()

def get_Item(item_id):
    return Item.get_by(id=item_id)

def get_all_vendors():
    return Vendor.query.all()

def get_all_active_vendors():
    return Vendor.get_by(status=status_active).all()

def get_Vendor(vendor_id):
    return Vendor.get_by(id=vendor_id)

def get_Warehouse(warehouse_id):
    return Warehouse.get_by(id=warehouse_id)

def get_all_warehouses_for_item(item_id):
    item = get_Item(item_id)
    if not item:
        raise CatalogServiceException(108, "Some unforeseen error while obtaining item")
    return item.get_all_warehouses()

def add_warehouse(warehouse):
    if not warehouse:
        raise CatalogServiceException(108, "Bad warehouse")
    if get_Warehouse(warehouse.id):
        #warehouse is already present.
        raise CatalogServiceException(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 add_vendor(vendor):
    if get_Vendor(vendor.id):
        raise CatalogServiceException(101, "Vendor already present")
    ds_vendor = Vendor()
    if not vendor:
        raise CatalogServiceException(108, "Bad vendor in request")
    ds_vendor.id = vendor.id
    ds_vendor.name = vendor.name
    session.commit()
    return ds_vendor.id

def add_item(item):
    if not item:
        raise CatalogServiceException(108, "Bad item in request")
    if get_Item(item.id):
        raise CatalogServiceException(101, "Item already exists")
    
    ds_item = Item()
    ds_item.id = item.id
    ds_item.status = status_added
    ds_item.addedOn = datetime.datetime.now()
    ds_item.price = item.price
    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
    if item.categories:
        for category_id in item.categories:
            category = get_category(category_id)
            if category:
                ds_item.category.append(category)
            else:
                raise CatalogServiceException(108, "Category not found ".category_id)
    session.commit();
    return ds_item.id
    
def add_category(category):
    try: 
        ds_category = Category()
        
        ds_category.id = category.id
        ds_category.name = category.name
        ds_category.addedOn = datetime.datetime.now()
        ds_category.isRoot = category.isTopLevel
        if category.parent_id:
            ds_category.parent= Category.get_by(id=category.parent_id)
        
        if category.childCategories:
            for child in category.childCategories:
                ds_category.children.append(get_category(child))
        ds_category.status = status_active
        #save the category now
        session.commit()
        return ds_category.id
    except:
        return 0

def add_item_to_category(id, item):
    category = get_category(id)
    if not category:
        raise CatalogServiceException(101, "no such category exists")
    if not item:
        raise CatalogServiceException(104, "bad item")
    try:
        item_id = add_item(item)
        added_item = get_Item(item_id)
        added_item.categories.append(category)
        session.commit()
        return item_id
    except:
        raise CatalogServiceException(101, "intenral error")
    
def add_item_to_category_by_id(category_id, item_id):
    category = get_category(category_id)
    item = get_Item(item_id)
    if not category:
        raise CatalogServiceException(101, "no such category exists")
    if not item:
        raise CatalogServiceException(104, "bad item")
    item.categories.append(category)
    session.commit()

def update_inventory(item_id, warehouse_id, quantity, timestamp):
    item = get_Item(item_id)
    warehouse = get_Warehouse(warehouse_id) 
    time = datetime.datetime.fromtimestamp(timestamp)
    
    if not item:
        raise CatalogServiceException(106, "Item not found, how did you get this?")
    if not warehouse:
        raise CatalogServiceException(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.availability = quantity
        
        current_inventory_snapshot = CurrentInventorySnapshot.get_by(item=item, warehouse=warehouse)
        current_inventory_snapshot.availibility = quantity
        current_inventory_snapshot.checkedOn = time
        session.commit()
    except:
        raise CatalogServiceException(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 CatalogServiceException(108, "Some unforeseen error while updating inventory")
    return inventory

def retire_category(category_id):
    
    if not category_id:
        raise CatalogServiceException(101, "Bad category id")
    category = get_category(category_id)
    
    if not category:
        raise CatalogServiceException(108, "Category not present")
    
    category.status = status_retired
    
    session.commit()
    
def retire_warehouse(warehouse_id):
    if not warehouse_id:
        raise CatalogServiceException(101, "Bad warehouse id")
    warehouse = get_Warehouse(warehouse_id)
    if not warehouse:
        raise CatalogServiceException(108, "warehouse id not present")
    warehouse.status = status_retired;
    session.commit()
    
def retire_vendor(vendor_id):
    if not vendor_id:
        raise CatalogServiceException(101, "bad vendor id")
    vendor = get_Vendor(vendor_id)
    
    if not vendor:
        raise CatalogServiceException(108, "vendor id not present")
    
    vendor.status = status_retired
    session.commit()
    
def retire_item(item_id):
    if not item_id:
        raise CatalogServiceException(101, "bad item id")
    item = get_Item(item_id)
    if not item:
        raise CatalogServiceException(108, "item id not present")
    item.status = status_retired
    session.commit()

def start_item_on(item_id, timestamp):
    if not item_id:
        raise CatalogServiceException(101, "bad item id")
    item = get_Item(item_id)
    if not item:
        raise CatalogServiceException(108, "item id not present")
    
    item.status = status_active
    item.startDate = datetime.datetime.fromtimestamp(timestamp)
    add_status_change_log(item, status_active)
    session.commit()
    
def retire_item_on(item_id, timestamp):
    if not item_id:
        raise CatalogServiceException(101, "bad item id")
    item = get_Item(item_id)
    if not item:
        raise CatalogServiceException(108, "item id not present")
    
    item.status = status_retired
    item.retireDate = datetime.datetime.fromtimestamp(timestamp)
    add_status_change_log(item, status_retired)
    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 CatalogServiceException(101, "bad item id")
    item = get_Item(item_id)
    if not item:
        raise CatalogServiceException(108, "item id not present")
    
    item.status = status_retired
    add_status_change_log(item, status_retired)
    session.commit()
    
def get_item_availability_for_warehouse(warehouse_id, item_id):
    if not warehouse_id:
        raise CatalogServiceException(101, "bad warehouse_id")
    if not item_id:
        raise CatalogServiceException(101, "bad item_id")
    
    warehouse = get_Warehouse(warehouse_id)
    if not warehouse:
        raise CatalogServiceException(108, "warehouse does not exist")
    item = get_Item(item_id)
    
    if not item:
            raise CatalogServiceException(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