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

 from sqlalchemy.orm.exc import MultipleResultsFound, NoResultFound
 from sqlalchemy.sql import or_
 from sqlalchemy.sql.expression import and_, func, distinct, desc
 from sqlalchemy.sql.functions import count
 import calendar
+from collections import namedtuple
 import datetime
 import json
 import math
 import sys
 import threading
 : [5839] }
 pincodePricingServiceabilityMap = {}
 warehouseMap = {}
+WarehouseInfo = namedtuple('WarehouseInfo', [
+    'id', 'vendor_id', 'inventoryType', 'warehouseType',
+    'billingWarehouseId', 'transferDelayInHours', 'state_id',
+    'logisticsLocation', 'shippingWarehouseId', 'isAvailabilityMonitored'])
+_warehouse_cache = {}
 OOS_CALCULATION_TIME = 23
 # right now these warehouse ids are ignored.
 # physicalWarehouseIdsForFofo = [7678, 7681]
 physicalWarehouseIdsForFofo = []
     #UK
 : 9213
+}
+def _load_warehouse_cache():
+    global _warehouse_cache
+    rows = metadata.bind.execute(
+        "SELECT id, vendor_id, inventoryType, warehouseType, "
+        "billingWarehouseId, transferDelayInHours, state_id, "
+        "logisticsLocation, shippingWarehouseId, isAvailabilityMonitored "
+        "FROM warehouse"
+    )
+    _warehouse_cache = {}
+    for row in rows:
+        wh = WarehouseInfo(*row)
+        _warehouse_cache[wh.id] = wh
+    print "Loaded %d warehouses into cache" % len(_warehouse_cache)
 def initialize(dbname='inventory', db_hostname="localhost"):
     DataService.initialize(dbname, db_hostname)
+    _load_warehouse_cache()
     __populateWarehouseMap()
 @memoized(3600)
 def __populateWarehouseMap():
     ds_warehouse.state = get_state(warehouse.stateId)
     ds_warehouse.warehouseType = WarehouseType._VALUES_TO_NAMES[warehouse.warehouseType]
     if warehouse.vendorString:
         ds_warehouse.vendorString = warehouse.vendorString
     session.commit()
+    _load_warehouse_cache()
     return ds_warehouse.id
 def get_ignored_items(warehouse_id):
     Ignored_inventory_items = IgnoredInventoryUpdateItems.query.filter_by(warehouse_id=warehouse_id).all()
     except:
         return 0
 def get_item_availability_for_our_warehouses(item_ids):
-    our_warehouses = Warehouse.query.filter_by(warehouseType='OURS', inventoryType='GOOD').all()
+    warehouse_ids = [wh.id for wh in _warehouse_cache.values() if wh.warehouseType == 'OURS' and wh.inventoryType == 'GOOD']
-    our_thirdparty_warehouses = Warehouse.query.filter_by(warehouseType='OURS_THIRDPARTY').all()
-    warehouse_ids = []
-    for warehouse in our_warehouses:
-        warehouse_ids.append(warehouse.id)
-    # for warehouse in our_thirdparty_warehouses :
-    #    warehouse_ids.append(warehouse.id)
     availability_map = dict()
     try:
         for item_id in item_ids:
             total_availability = 0
     warehouses = {}
     ourGoodWarehouses = {}
     thirdpartyWarehouses = {}
     preferredThirdpartyWarehouses = {}
-    for warehouse in Warehouse.query.all():
+    for warehouse in _warehouse_cache.values():
         if (warehouse.inventoryType == InventoryType._VALUES_TO_NAMES[InventoryType.BAD] or warehouse.warehouseType == WarehouseType._VALUES_TO_NAMES[WarehouseType.OURS_THIRDPARTY]):
             continue
         warehouses[warehouse.id] = warehouse
         if warehouse.warehouseType == WarehouseType._VALUES_TO_NAMES[WarehouseType.OURS]:
             if warehouse.inventoryType == InventoryType._VALUES_TO_NAMES[InventoryType.GOOD]:
     warehouse = warehouses[warehouse_retid]
     billingWarehouseId = warehouse.billingWarehouseId
     # Fetching billing warehouse of a Good billable warehouse corresponding to the virtual one
     if not warehouse.billingWarehouseId:
+        for w in _warehouse_cache.values():
-        for w in Warehouse.query.filter_by(vendor_id=warehouse.vendor_id, inventoryType=InventoryType._VALUES_TO_NAMES[InventoryType.GOOD], state_id=warehouse.state_id).all():
+            if w.vendor_id == warehouse.vendor_id and w.inventoryType == InventoryType._VALUES_TO_NAMES[InventoryType.GOOD] and w.state_id == warehouse.state_id and w.billingWarehouseId:
-            if w.billingWarehouseId:
                 billingWarehouseId = w.billingWarehouseId
                 break
     expectedDelay = item.expectedDelay
     if expectedDelay is None:
 def get_pending_orders_inventory(vendor_id=1):
     """
     Returns a list of inventory stock for items for which there are pending orders.
     """
-    warehouse_ids = [warehouse.id for warehouse in Warehouse.query.filter_by(vendor_id=vendor_id)]
+    warehouse_ids = [wh.id for wh in _warehouse_cache.values() if wh.vendor_id == vendor_id]
     pending_items_inventory = []
     if warehouse_ids:
         pending_items_inventory = session.query(CurrentInventorySnapshot.item_id, func.sum(CurrentInventorySnapshot.availability), func.sum(CurrentInventorySnapshot.reserved)).filter(CurrentInventorySnapshot.warehouse_id.in_(warehouse_ids)).group_by(CurrentInventorySnapshot.item_id).having(func.sum(CurrentInventorySnapshot.reserved) > 0).all()
     return pending_items_inventory
     """
     Returns a list of inventory Availability and Reserved Count for items which either have real inventory
     or have pending orders.
     """
-    warehouse_ids = [warehouse.id for warehouse in Warehouse.query.filter(Warehouse.isAvailabilityMonitored == 1).filter(or_(Warehouse.inventoryType == 'GOOD', Warehouse.warehouseType == 'OURS'))]
+    warehouse_ids = [wh.id for wh in _warehouse_cache.values() if wh.isAvailabilityMonitored and (wh.inventoryType == 'GOOD' or wh.warehouseType == 'OURS')]
     items_inventory = []
     reserved_items_inventory = []
     available_items_inventory = []
     if warehouse_ids:
         reserved_items_inventory = session.query(CurrentInventorySnapshot.item_id, func.sum(CurrentInventorySnapshot.availability), func.sum(CurrentInventorySnapshot.reserved)).filter(CurrentInventorySnapshot.warehouse_id.in_(warehouse_ids)).group_by(CurrentInventorySnapshot.item_id).having(func.sum(CurrentInventorySnapshot.reserved) > 0).all()
     session.commit()
 def get_our_warehouse_id_for_vendor(vendor_id, billing_warehouse_id):
     try:
+        for wh in _warehouse_cache.values():
-        warehouse = Warehouse.query.filter_by(vendor_id=vendor_id, warehouseType='OURS', inventoryType='GOOD', billingWarehouseId=billing_warehouse_id).first()
+            if wh.vendor_id == vendor_id and wh.warehouseType == 'OURS' and wh.inventoryType == 'GOOD' and wh.billingWarehouseId == billing_warehouse_id:
-        return warehouse.id
+                return wh.id
+        raise InventoryServiceException(101, 'No our warehouse found for vendorId: ' + str(vendor_id))
     except Exception as e:
         print e;
         raise InventoryServiceException(101, 'No our warehouse found for vendorId: ' + str(vendor_id))
         print ex
 def get_shipping_locations():
     shippingLocationIds = {}
-    warehouses = Warehouse.query.all()
-    for warehouse in warehouses:
+    for warehouse in _warehouse_cache.values():
         if warehouse.shippingWarehouseId:
             shippingLocationIds[warehouse.shippingWarehouseId] = 1
     shippingLocations = []
     for shippingLocationId in shippingLocationIds:
         return client.getItem(item_id)
 def get_monitored_warehouses_for_vendors(vendorIds):
     w = []
-    for wh in Warehouse.query.filter_by(isAvailabilityMonitored=1).all():
+    for wh in _warehouse_cache.values():
-        if wh.vendor.id in (vendorIds):
+        if wh.isAvailabilityMonitored and wh.vendor_id in (vendorIds):
-            w.append(to_t_warehouse(wh).id)
+            w.append(wh.id)
     return w
 def get_ignored_warehouseids_and_itemids():
     iw = []

Subversion Repositories SmartDukaan

(root)//trunk/PyProj/src/shop2020/model/v1/inventory/impl/DataAcessors.py – Rev 35718 → 35732