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

 '''
 Created on 27-Apr-2010
 @author: ashish
 '''
-from shop2020.model.v1.catalog.impl import DataService
-from shop2020.utils.Utils import log_entry, to_py_date
-from shop2020.model.v1.catalog.impl.DataAcessors import add_item, add_warehouse,\
-    update_inventory, retire_warehouse, retire_item, change_item_status,\
-    get_item, get_all_items, get_all_items_by_status,\
-    get_item_availability_for_warehouse, get_item_availability_for_location, get_all_warehouses_by_status,\
-    get_Warehouse, get_all_warehouses_for_item, get_all_items_for_warehouse,\
-    update_item, start_item_on, close_session, \
-    add_vendor, add_warehouse_vendor_mapping, get_warehouses_for_vendor, get_vendors_for_warehouse, \
-    retire_item_on, get_item_inventory_by_item_id, get_items_by_catalog_id,\
-    get_best_deals, get_best_deals_catalog_ids, get_best_deals_count, get_best_sellers, get_latest_arrivals, get_latest_arrivals_catalog_ids, get_latest_arrivals_count,\
-    is_active, get_best_sellers_catalog_ids, get_best_sellers_count, put_category_object,\
-    get_category_object, mark_item_as_content_complete,\
-    reserve_item_in_warehouse, reduce_reservation_count, generate_new_entity_id,\
-    get_all_item_pricing, add_vendor_pricing, \
-    get_item_pricing, get_category, get_all_categories, add_category,\
-    get_item_status_description, get_all_vendors,\
-    get_item_mappings, add_vendor_item_mapping, check_similar_item,\
-    change_risky_flag, get_items_for_mastersheet, get_risky_items,\
-    get_similar_items_catalog_ids, add_product_notification,\
-    send_product_notifications, update_inventory_history,\
-    get_all_brands_by_category, is_alive, get_item_pricing_by_source,\
-    add_source_item_pricing, get_all_sources, get_item_for_source,\
-    get_all_source_pricing, get_item_count_by_status, search_items, get_search_result_count,\
-    get_pending_orders_inventory, get_product_notifications, get_product_notification_request_count,\
-    add_inventory, get_all_similar_items_catalog_ids, add_similar_item_catalog_id, delete_similar_item_catalog_id, add_authorization_log_for_item,\
-    get_thrift_item_list, get_all_brands, mark_missed_inventory_updates_as_processed, \
-    reset_availability, get_item_keys_to_be_processed
-from shop2020.model.v1.catalog.impl.Convertors import to_t_item,\
-    to_t_item_inventory, to_t_warehouse, to_t_vendor_item_pricing, to_t_category, to_t_vendor,\
-    to_t_vendor_item_mapping, to_t_source_item_pricing, to_t_product_notification_request,\
-    to_t_product_notification_request_count
-from shop2020.thriftpy.model.v1.catalog.ttypes import status,\
-    InventoryServiceException, AvailableAndReservedStock
-from shop2020.config.client.ConfigClient import ConfigClient
 from shop2020.clients.CatalogClient import CatalogClient
 from shop2020.clients.PurchaseClient import PurchaseClient
+from shop2020.config.client.ConfigClient import ConfigClient
+from shop2020.model.v1.catalog.impl import DataService
+from shop2020.model.v1.catalog.impl.Convertors import to_t_item, \
+    to_t_item_inventory, to_t_warehouse, to_t_vendor_item_pricing, to_t_category, \
+    to_t_vendor, to_t_vendor_item_mapping, to_t_source_item_pricing, \
+    to_t_product_notification_request, to_t_product_notification_request_count
+from shop2020.model.v1.catalog.impl.DataAcessors import add_item, add_warehouse, \
+    update_inventory, retire_warehouse, retire_item, change_item_status, get_item, \
+    get_all_items, get_all_items_by_status, get_item_availability_for_warehouse, \
+    get_item_availability_for_location, get_all_warehouses_by_status, get_Warehouse, \
+    get_all_warehouses_for_item, get_all_items_for_warehouse, update_item, \
+    start_item_on, close_session, add_vendor, retire_item_on, get_item_inventory_by_item_id, \
+    get_items_by_catalog_id, get_best_deals, get_best_deals_catalog_ids, \
+    get_best_deals_count, get_best_sellers, get_latest_arrivals, \
+    get_latest_arrivals_catalog_ids, get_latest_arrivals_count, is_active, \
+    get_best_sellers_catalog_ids, get_best_sellers_count, put_category_object, \
+    get_category_object, mark_item_as_content_complete, reserve_item_in_warehouse, \
+    reduce_reservation_count, generate_new_entity_id, get_all_item_pricing, \
+    add_vendor_pricing, get_item_pricing, get_category, get_all_categories, \
+    add_category, get_item_status_description, get_all_vendors, get_item_mappings, \
+    add_vendor_item_mapping, check_similar_item, change_risky_flag, \
+    get_items_for_mastersheet, get_risky_items, get_similar_items_catalog_ids, \
+    add_product_notification, send_product_notifications, update_inventory_history, \
+    get_all_brands_by_category, is_alive, get_item_pricing_by_source, \
+    add_source_item_pricing, get_all_sources, get_item_for_source, \
+    get_all_source_pricing, get_item_count_by_status, search_items, \
+    get_search_result_count, get_pending_orders_inventory, get_product_notifications, \
+    get_product_notification_request_count, add_inventory, \
+    get_all_similar_items_catalog_ids, add_similar_item_catalog_id, \
+    delete_similar_item_catalog_id, add_authorization_log_for_item, \
+    get_thrift_item_list, get_all_brands, add_bad_inventory, \
+    mark_missed_inventory_updates_as_processed, \
+    get_item_keys_to_be_processed, reset_availability
+from shop2020.model.v1.catalog.impl.DataService import Warehouse, \
+    MissedInventoryUpdate
+from shop2020.thriftpy.model.v1.catalog.ttypes import status, \
+    InventoryServiceException, AvailableAndReservedStock, WarehouseType, InventoryType
 from shop2020.thriftpy.purchase.ttypes import PurchaseOrder, LineItem
-import xlrd
+from shop2020.utils.Utils import log_entry, to_py_date, to_java_date
 import datetime
+import xlrd
 class InventoryServiceHandler:
     '''
     classdocs
         log_entry(self, "add Inventory called")
         try:
             return add_inventory(itemId, warehouseId, quantity)
         finally:
             close_session()
     def isActive(self, item_id):
         """
         Parameters:
          - item_id
         try:
             warehouse = get_Warehouse(warehouse_id)
             if warehouse:
                 return to_t_warehouse(warehouse)
             else:
-                raise InventoryServiceException(101, "no such warehouse")
+                raise InventoryServiceException(101, "no such warehouse: " + str(warehouse_id))
         finally:
             close_session()
     def getAllWarehousesForItem(self, item_id):
         """
         """
         try:
             return add_vendor(vendor)
         finally:
             close_session()
-    def addWarehouseVendorMapping(self, warehouse_id, vendorId):
-        """
-        map given warehouse id to given vendor id
-        """
-        try:
-            return add_warehouse_vendor_mapping(warehouse_id, vendorId)
-        finally:
-            close_session()
-    def getVendorsForWarehouse(self, warehouse_id):
-        """
-        get all vendors for the given warehouse_id
-        """
-        try:
-            vendors = get_vendors_for_warehouse(warehouse_id)
-            ret_vendors = []
-            for vendor in vendors:
-                if vendor:
-                    ret_vendors.append(to_t_vendor(vendor))
-            return ret_vendors
-        finally:
-            close_session()
-    def getWarehousesForVendor(self, vendorId):
-        '''
-        get all warehouses for the given vendorId
-        '''
-        try:
-            warehouses = get_warehouses_for_vendor(vendorId)
-            ret_warehouses = []
-            for warehouse in warehouses:
-                if warehouse:
-                    ret_warehouses.append(to_t_warehouse(warehouse))
-            return ret_warehouses
-        finally:
-            close_session()
-    def processPurchaseOrder(self, filename, vendorId):
-        '''
-        create purchase order for given filename
-        '''
-        workbook = xlrd.open_workbook(filename)
-        sheet = workbook.sheet_by_index(0)
-        num_rows = sheet.nrows
-        if(sheet.ncols != 2):
-            raise InventoryServiceException(101, "Sheet format Invalid")
-        catalogClient = CatalogClient().get_client()
-        lineitems = []
-        purchaseOrder = PurchaseOrder()
-        purchaseOrder.supplierId = vendorId
-        purchaseOrder.warehouseId = 7
-        totalCost = 0
-        for rownum in range(1, num_rows):
-            itemId = sheet.row_values(rownum)[0]
-            quantity = sheet.row_values(rownum)[1]
-            lineitem = LineItem()
-            item = catalogClient.getItem(itemId)
-            lineitem.itemId = item.id
-            lineitem.productGroup = item.productGroup
-            lineitem.brand = item.brand
-            lineitem.modelNumber = item.modelNumber
-            lineitem.modelName = item.modelName
-            lineitem.color = item.color
-            lineitem.quantity = quantity
-            lineitem.unfulfilledQuantity = quantity
-            try:
-                item_pricing = catalogClient.getItemPricing(itemId, vendorId)
-                lineitem.unitPrice = item_pricing.transferPrice
-            except:
-                raise InventoryServiceException(101, "Unable to get pricing for " + item.brand + " " + item.modelName \
-                                + " " + item.modelNumber + "(Item id : " + str(int(itemId)) + \
-                                ")" + " for VendorId : " + str(vendorId))
-            lineitem.fulfilled = 0
-            totalCost += quantity*lineitem.unitPrice
-            lineitems.append(lineitem)
-        purchaseOrder.totalCost = totalCost
-        purchaseOrder.lineitems = lineitems
-        purchaseClient = PurchaseClient().get_client()
-        po_id = purchaseClient.createPurchaseOrder(purchaseOrder)
-        return "POId:" + str(po_id)
     def addAuthorizationLog(self, itemId, username, reason):
         """
         add a log to authorize table.
         try:
             return add_authorization_log_for_item(itemId, username, reason)
         finally:
             close_session()
+    def getWarehouses(self, warehouseType, inventoryType, vendorId, billingWarehouseId, shippingWarehouseId):
+        """
+        This method returns all warehouses for a given warehosueType, inventoryType, vendor, billingWarehouse and shippingWarehouse.
+        getWarehouses(WarehouseType.OURS, InventoryType.GOOD, 3, 7, 7) would return ours warehouse with GOOD type inventory for vendor 1 for billing warehouse 7 and shipping warehouse 7
+        getWarehouses(WarehouseType.OURS, InventoryType.GOOD, 3, 7, 7) would return ours warehouse with GOOD type inventory for vendor 3 for billing warehouse 7 and shipping warehouse 7
+        getWarehouses(null, null, 3, 7, 7) would return all type warehouses with all type inventory for vendor 3 for billing warehouse 7 and shipping warehouse 7
+        getWarehouses(null, null, 0, 0, 7) would return all type warehouses with all type inventory for all vendors for all billing warehouses at shipping warehouse 7
+        Parameters:
+         - warehouseType
+         - inventoryType
+         - vendorId
+         - billingWarehouseId
+         - shippingWarehouseId
+        """
+        try:
+            query = Warehouse.query
+            if warehouseType is not None:
+                query = query.filter_by(warehouseType = WarehouseType._VALUES_TO_NAMES[warehouseType])
+            if inventoryType is not None:
+                query = query.filter_by(inventoryType = InventoryType._VALUES_TO_NAMES[inventoryType])
+            if vendorId:
+                query = query.filter_by(vendor_id = vendorId)
+            if billingWarehouseId:
+                query = query.filter_by(billingWarehouseId = billingWarehouseId)
+            if shippingWarehouseId:
+                query = query.filter_by(shippingWarehouseId = shippingWarehouseId)
+            return [to_t_warehouse(w) for w in query.all()]
+        finally:
+            close_session()
     def getItemKeysToBeProcessed(self, warehouseId):
         """
         Returns the list of item keys which need to be processed for a given warehouse.
         This is currently used by Support application to send item keys whose inventory needs
         to be updated from PLB
         try:
             reset_availability(itemKey, vendorId, quantity, warehouseId)
         finally:
             close_session()
+    def getIgnoredItemKeys(self, ):
+        """
+        Returns all the item key mappings that have been ignored until date. Value of map has the warehouse id
+        and the timestamp from where alert was raised.
+        """
+        try:
+            itemKeysMap = {}
+            for key in MissedInventoryUpdate.query.filter_by(isIgnored = 1).all():
+                itemKeysMap[key.itemKey] = { key.warehouseId : to_java_date(key.timestamp) }
+            return itemKeysMap
+        finally:
+            close_session()
+    def addBadInventory(self, itemId, warehouseId, quantity):
+        """
+        Add the BAD type inventory to existing stock.
+        Parameters:
+         - itemId
+         - warehouseId
+         - quantity
+        """
+        try:
+            add_bad_inventory(itemId, warehouseId, quantity)
+        finally:
+            close_session()
 def is_valid(item):
     if item.status in [status.ACTIVE, status.PAUSED, status.PAUSED_BY_RISK]:
         if item.startDate:
             return not(datetime.datetime.now() < item.startDate or item.sellingPrice == 0)
         else:

Subversion Repositories SmartDukaan

(root)/trunk/PyProj/src/shop2020/model/v1/catalog/impl/InventoryServiceHandler.py @ 5167 – Rev 4988 → 5110