WebSVN – SmartDukaan – Diff – /trunk/PyProj/src/shop2020/purchase/main/handler/PurchaseServiceHandler.py

 @author: Chandranshu
 '''
 from elixir import metadata, setup_all, session
 from shop2020.clients.CatalogClient import CatalogClient
+from shop2020.clients.InventoryClient import InventoryClient
 from shop2020.clients.TransactionClient import TransactionClient
 from shop2020.purchase.main.model.Invoice import Invoice
 from shop2020.purchase.main.model.LineItem import LineItem
 from shop2020.purchase.main.model.Purchase import Purchase
 from shop2020.purchase.main.model.PurchaseOrder import PurchaseOrder
 from shop2020.purchase.main.model.RevisionedPurchaseOrder import \
     RevisionedPurchaseOrder
 from shop2020.purchase.main.model.Supplier import Supplier
 from shop2020.thriftpy.generic.ttypes import ExceptionType
-from shop2020.thriftpy.model.v1.catalog.ttypes import WarehouseType, \
+from shop2020.thriftpy.model.v1.inventory.ttypes import WarehouseType, \
     InventoryType
 from shop2020.thriftpy.model.v1.order.ttypes import OrderStatus
 from shop2020.thriftpy.purchase.ttypes import PurchaseServiceException, POStatus, \
     PurchaseOrder as TPurchaseOrder, LineItem as TLineItem
 from shop2020.utils.Utils import to_py_date
             pending_orders = transactionClient.getOrdersInBatch([OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.ACCEPTED], 0, 0, warehouseId)
             if not pending_orders:
                 return purchaseOrders
-            catalog_client = CatalogClient().get_client()
+            inventory_client = InventoryClient().get_client()
             availability = {}
-            warehouses = catalog_client.getWarehouses(WarehouseType.OURS, InventoryType.GOOD, 0, None, warehouseId)
+            ourGoodWarehouseIds = [w.id for w in inventory_client.getWarehouses(WarehouseType.OURS, InventoryType.GOOD, 0, None, warehouseId)]
+            itemInventorySnapshot = inventory_client.getInventorySnapshot(0)
-            for warehouse in warehouses:
+            for itemId, itemInventory in itemInventorySnapshot.iteritems():
-                goodWarehouseId = warehouse.id
+                item = self.__get_item_from_master(itemId)
-                for item in catalog_client.getAllItemsForWarehouse(goodWarehouseId):
+                for warehouseId, quantity in itemInventory.availability.iteritems():
-                    itemInventory = catalog_client.getItemInventoryByItemId(item.id)
+                    if warehouseId in ourGoodWarehouseIds:
-                    if availability.has_key(item.id):
+                        if availability.has_key(item.id):
-                        availability[item.id] = [availability[item.id][0] + itemInventory.availability[goodWarehouseId], item]
+                            availability[item.id] = [availability[item.id][0] + quantity, item]
-                    else:
+                        else:
-                        availability[item.id] = [itemInventory.availability[goodWarehouseId], item]
+                            availability[item.id] = [quantity, item]
             unfulfilledPurchaseOrders = PurchaseOrder.query.filter(or_(PurchaseOrder.status == POStatus.PARTIALLY_FULFILLED, PurchaseOrder.status == POStatus.READY)).all()
             for purchaseOrder in unfulfilledPurchaseOrders:
                 for lineitem in purchaseOrder.lineitems:
                     if availability.has_key(lineitem.itemId):
                         availability[lineitem.itemId] = [availability[lineitem.itemId][0] + lineitem.unfulfilledQuantity, availability[lineitem.itemId][1]]
                     else:
-                        item = catalog_client.getItem(lineitem.itemId)
+                        item = self.__get_item_from_master(lineitem.itemId)
                         availability[item.id] = [lineitem.unfulfilledQuantity, item]
             codRequirements = {}
             requirements = {}
             for order in pending_orders:
                         if (netRequirements.has_key(item.preferredVendor)):
                             netRequirements[item.preferredVendor].append([item, requirementsCount - availabilityCount])
                         else:
                             netRequirements[item.preferredVendor] = [[item, requirementsCount - availabilityCount]];
                 else:
-                    item = catalog_client.getItem(itemId);
+                    item = self.__get_item_from_master(itemId)
                     if item.preferredVendor is None:
                         raise PurchaseServiceException(101, 'Preferred Vendor missing for ' + " ".join([str(item.brand), str(item.modelName), str(item.modelNumber), str(item.color)]))
                     if (netRequirements.has_key(item.preferredVendor)):
                         netRequirements[item.preferredVendor].append([item, requirementsCount])
                     else:
                     t_po_lineitem.itemId = item.id
                     t_po_lineitem.quantity = quantity
                     if codRequirements.has_key(item.id):
                         t_po_lineitem.codCount = min(codRequirements[item.id], quantity)
                     try:
-                        item_pricing = catalog_client.getItemPricing(item.id, vendorId)
+                        item_pricing = inventory_client.getItemPricing(item.id, vendorId)
                     except Exception as e:
                         vendor = self.getSupplier(vendorId)
                         print 'Could not find transfer price for Item id: ' + str(item.id) + ' and vendor id: ' + str(vendorId)
                         print e
                         raise PurchaseServiceException(101, 'Transfer price missing for ' + vendor.name + ' and ' + " ".join([item.brand, item.modelName, item.modelNumber, item.color]))
             return True
         except:
             return False
         finally:
             self.close_session()
+    def __get_item_from_master(self, item_id):
+        client = CatalogClient("catalog_service_server_host_master", "catalog_service_server_port").get_client()
+        return client.getItem(item_id)

Subversion Repositories SmartDukaan

(root)/trunk/PyProj/src/shop2020/purchase/main/handler/PurchaseServiceHandler.py – Rev 5768 → 5944