WebSVN – SmartDukaan – Diff – //trunk/PyProj/src/shop2020/model/v1/catalog/impl/DataAcessors.py

 @author: ashish
 '''
 from elixir import *
 from shop2020.model.v1.catalog.impl import DataService
-from shop2020.model.v1.catalog.impl.DataService import Category, Item, Vendor,\
+from shop2020.model.v1.catalog.impl.DataService import Item, \
     Warehouse, ItemInventoryHistory, CurrentInventorySnapshot, ItemInfo,\
-    ItemChangeLog
+    ItemChangeLog, Price
-from shop2020.thriftpy.model.v1.catalog.ttypes import Category as TCategory,\
+from shop2020.thriftpy.model.v1.catalog.ttypes import \
-    CatalogServiceException
+    InventoryServiceException, status
 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
+from shop2020.utils.Utils import log_entry, to_py_date, to_java_date
 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)
+        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.get_by(status=status).all()
+        return Item.query.filter_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_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 CatalogServiceException(108, "Some unforeseen error while obtaining item")
+        raise InventoryServiceException(108, "Some unforeseen error while obtaining item")
-    return item.get_all_warehouses()
+    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 CatalogServiceException(108, "Bad warehouse")
+        raise InventoryServiceException(108, "Bad warehouse")
     if get_Warehouse(warehouse.id):
         #warehouse is already present.
-        raise CatalogServiceException(101, "Warehouse 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.status = status.ACTIVE
     ds_warehouse.addedOn = datetime.datetime.now()
     session.commit()
     return ds_warehouse.id
-def add_vendor(vendor):
+def update_item(item):
+    if not item:
+        raise InventoryServiceException(108, "Bad item in request")
-    if get_Vendor(vendor.id):
+    if not item.id:
-        raise CatalogServiceException(101, "Vendor already present")
+        raise InventoryServiceException(101, "non existent item")
-    ds_vendor = Vendor()
+    ds_item = get_Item(item.id)
-    if not vendor:
+    if not ds_item:
-        raise CatalogServiceException(108, "Bad vendor in request")
+        raise InventoryServiceException(101, "Item already exists")
+    if item.catalogItemId:
+        ds_item.catalog_item_id = item.catalogItemId
-    ds_vendor.id = vendor.id
+    if item.vendorItemId:
-    ds_vendor.name = vendor.name
+        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()
+    session.commit();
-    return ds_vendor.id
+    return ds_item.id
 def add_item(item):
     if not item:
-        raise CatalogServiceException(108, "Bad item in request")
+        raise InventoryServiceException(108, "Bad item in request")
     if get_Item(item.id):
-        raise CatalogServiceException(101, "Item already exists")
+        raise InventoryServiceException(101, "Item already exists")
     ds_item = Item()
+    if item.id:
-    ds_item.id = 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_added
+        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.price = item.price
+        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
-    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):
+def update_price(item, ds_item):
+    log_entry("update_price", "updating price for item")
+    for k,v in item.price.iteritems():
-    try:
+        price = Price()
-        ds_category = Category()
+        price.warehouse_id = k
+        price.price = v
-        ds_category.id = category.id
+        price.item = ds_item
+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:
-        ds_category.name = category.name
+        price = Price()
-        ds_category.addedOn = datetime.datetime.now()
+        item = get_Item(item_id)
-        ds_category.isRoot = category.isTopLevel
+        price.warehouse_id = warehouse_id
-        if category.parent_id:
+        price.item = item
-            ds_category.parent= Category.get_by(id=category.parent_id)
+    price.price = price
+    session.commit()
-        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)
+    time = to_py_date(timestamp)
     if not item:
-        raise CatalogServiceException(106, "Item not found, how did you get this?")
+        raise InventoryServiceException(106, "Item not found, how did you get this?")
     if not warehouse:
-        raise CatalogServiceException(107, "Warehouse? Where?")
+        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.availability = quantity
+        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 CatalogServiceException(108, "Some unforeseen error while updating inventory")
+        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 CatalogServiceException(108, "Some unforeseen error while updating inventory")
+        raise InventoryServiceException(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")
+        raise InventoryServiceException(101, "Bad warehouse id")
     warehouse = get_Warehouse(warehouse_id)
     if not warehouse:
-        raise CatalogServiceException(108, "warehouse id not present")
+        raise InventoryServiceException(108, "warehouse id not present")
-    warehouse.status = status_retired;
+    warehouse.status = status.DELETED;
-    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")
+        raise InventoryServiceException(101, "bad item id")
     item = get_Item(item_id)
     if not item:
-        raise CatalogServiceException(108, "item id not present")
+        raise InventoryServiceException(108, "item id not present")
-    item.status = status_retired
+    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 CatalogServiceException(101, "bad item id")
+        raise InventoryServiceException(101, "bad item id")
     item = get_Item(item_id)
     if not item:
-        raise CatalogServiceException(108, "item id not present")
+        raise InventoryServiceException(108, "item id not present")
-    item.status = status_active
+    item.status = status.ACTIVE
-    item.startDate = datetime.datetime.fromtimestamp(timestamp)
+    item.startDate = datetime.datetime.fromtimestamp(to_py_date(timestamp))
-    add_status_change_log(item, status_active)
+    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 CatalogServiceException(101, "bad item id")
+        raise InventoryServiceException(101, "bad item id")
     item = get_Item(item_id)
     if not item:
-        raise CatalogServiceException(108, "item id not present")
+        raise InventoryServiceException(108, "item id not present")
-    item.status = status_retired
+    item.status = status.PHASED_OUT
-    item.retireDate = datetime.datetime.fromtimestamp(timestamp)
+    item.retireDate = datetime.datetime.fromtimestamp(to_py_date(timestamp))
-    add_status_change_log(item, status_retired)
+    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.item = item
     session.commit()
 def change_item_status(item_id, new_status):
     if not item_id:
-        raise CatalogServiceException(101, "bad item id")
+        raise InventoryServiceException(101, "bad item id")
     item = get_Item(item_id)
     if not item:
-        raise CatalogServiceException(108, "item id not present")
+        raise InventoryServiceException(108, "item id not present")
-    item.status = status_retired
+    item.status = new_status
-    add_status_change_log(item, status_retired)
+    add_status_change_log(item, new_status)
     session.commit()
 def get_item_availability_for_warehouse(warehouse_id, item_id):
     if not warehouse_id:
-        raise CatalogServiceException(101, "bad warehouse_id")
+        raise InventoryServiceException(101, "bad warehouse_id")
     if not item_id:
-        raise CatalogServiceException(101, "bad item_id")
+        raise InventoryServiceException(101, "bad item_id")
     warehouse = get_Warehouse(warehouse_id)
     if not warehouse:
-        raise CatalogServiceException(108, "warehouse does not exist")
+        raise InventoryServiceException(108, "warehouse does not exist")
     item = get_Item(item_id)
     if not item:
-            raise CatalogServiceException(108, "item does not exist")
+            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

Subversion Repositories SmartDukaan

(root)//trunk/PyProj/src/shop2020/model/v1/catalog/impl/DataAcessors.py – Rev 103 → 122