WebSVN – SmartDukaan – Diff – //trunk/PyProj/src/shop2020/logistics/service/impl/DataAccessor.py

 from shop2020.clients.CatalogClient import CatalogClient
 from shop2020.clients.InventoryClient import InventoryClient
 from shop2020.logistics.service.impl import DataService
 from shop2020.logistics.service.impl.DataService import Awb, AwbUpdate, Provider, \
     DeliveryEstimate, WarehouseAllocation, PublicHolidays, \
-    ServiceableLocationDetails, PickupStore
+    ServiceableLocationDetails, PickupStore, Locations
 from shop2020.thriftpy.logistics.ttypes import LogisticsServiceException, \
-    DeliveryType, LogisticsLocationInfo
+    DeliveryType, LogisticsLocationInfo, LocationInfo, ProviderInfo
-from shop2020.thriftpy.model.v1.inventory.ttypes import Warehouse, BillingType,\
+from shop2020.thriftpy.model.v1.inventory.ttypes import Warehouse, BillingType, \
     WarehouseType
 from shop2020.utils.Utils import log_entry, to_py_date, to_java_date
 from sqlalchemy.sql import or_
 import datetime
 import logging
 import os
 import sys
 import time
+from collections import namedtuple
 logging.basicConfig(level=logging.DEBUG)
+PincodeProvider = namedtuple("PincodeProvider", ["dest_pin", "provider"])
 warehouse_allocation_cache = {}
 serviceable_location_cache = {}
 delivery_estimate_cache = {}
 warehouse_location_cache = {}
+location_state={}
+state_locations = {}
+#pincode location map is ab
+#{'110001':{1:{"sameState":False, "providerInfo":{1:(1,1), 2:(1,1)}}},}
+pincode_locations = {}
+#{(dest_pin, provider):(otgAvailable,providerCodLimit, websiteCodLimit, storeCodLimit, providerPrepaidLimit)}
+serviceable_map = {}
+#Need to identify a better way
+statepinmap={'11':0,
+             '40':1,'41':1,'42':1,'43':1,'44':1,
+             '56':2, '57':2, '58':2, '59':2,
+             '12':3, '13':3,
+             '30':4, '31':4, '32':4, '33':4, '34':4,
+             '36':6,'37':6,'38':6,'39':6
+             }
 '''
 This class is for only data transfer. Never used outside this module.
 '''
 class _DeliveryEstimateObject:
     print "Starting cache population at: " + str(datetime.datetime.now())
     #__cache_warehouse_allocation_table()
     __cache_warehouse_locations()
     __cache_serviceable_location_details_table()
     __cache_delivery_estimate_table()
+    __cache_locations()
+    __cache_pincode_estimates()
     close_session()
     print "Done cache population at: " + str(datetime.datetime.now())
+def __cache_locations():
+    locations = Locations.query.all()
+    for location in locations:
+        location_state[location.id] = location.state_id
+        if not state_locations.has_key(location.state_id):
+            state_locations[location.state_id] = []
+        state_locations[location.state_id] = state_locations[location.state_id].append(location.state_id)
+def __cache_pincode_estimates():
+    delivery_estimates = DeliveryEstimate.query.all()
+    for delivery_estimate in delivery_estimates:
+        if not pincode_locations.has_key(delivery_estimate.destination_pin):
+            pincode_locations[delivery_estimate.destination_pin] = {}
+        destination_pinMap = pincode_locations[delivery_estimate.destination_pin]
+        if not destination_pinMap.has_key(delivery_estimate.warehouse_location):
+            destination_pinMap[delivery_estimate.warehouse_location] = {}
+            state_id = location_state.get(delivery_estimate.warehouse_location)
+            sameState = __getStateFromPin(delivery_estimate.destination_pin) == state_id
+            destination_pinMap[delivery_estimate.warehouse_location]['sameState']=sameState
+            destination_pinMap[delivery_estimate.warehouse_location]['providerInfo'] = {}
+        destination_pinMap[delivery_estimate.warehouse_location]['providerInfo'][delivery_estimate.provider_id] = delivery_estimate.delivery_time + delivery_estimate.delivery_delay
+def __cache_pincode_provider_serviceability():
+    pincode_providers = ServiceableLocationDetails.query.filter(ServiceableLocationDetails.exp==1)
+    for pp in pincode_providers:
+        serviceable_map[PincodeProvider(pp.dest_pincode, pp.provider_id)] = (pp.cod, pp.otgAvailable,pp.providerCodLimit, pp.websiteCodLimit, pp.storeCodLimit, pp.providerPrepaidLimit)
 def __cache_delivery_estimate_table():
     delivery_estimates = DeliveryEstimate.query.all()
     for delivery_estimate in delivery_estimates:
         delivery_estimate_cache[delivery_estimate.destination_pin, delivery_estimate.provider_id, delivery_estimate.warehouse_location]\
         =delivery_estimate.delivery_time, delivery_estimate.delivery_delay
     for warehouse_allocation in warehouse_allocations:
         warehouse_allocation_cache[warehouse_allocation.pincode]=warehouse_allocation.primary_warehouse_location
 def __cache_warehouse_locations():
     client = InventoryClient().get_client()
-    client.getAllWarehouses(True)
+    #client.getAllWarehouses(True)
-    #for WarehouseAllocation.query.all()
-    warehouse = Warehouse()
     for warehouse in client.getAllWarehouses(True):
         warehouse_location_cache[warehouse.billingWarehouseId]=warehouse.logisticsLocation
 def get_provider(provider_id):
     provider =  Provider.get_by(id=provider_id)
     return provider
 def get_providers():
     providers = Provider.query.filter_by(isActive = 1).all()
     return providers
+#it would return only states where our warehouses are present else return -1
+def __getStateFromPin(pin):
+    if pin[:3] in ['744', '682']:
+        return -1
+    elif statepinmap.has_key(pin[:2]):
+        return statepinmap[pin[:2]]
+    else:
+        return -1
+def get_logistics_locations(destination_pin, selling_price_list):
+    #pincode_locations = {1:{"sameState":False, "providerInfo":{<provider_id>:<delay_days>}},}
+    #pp.otgAvailable,pp.providerCodLimit, pp.websiteCodLimit, pp.storeCodLimit, pp.providerPrepaidLimit
+    returnMap = {}
+    if pincode_locations.has_key(destination_pin):
+        locations = pincode_locations[destination_pin]
+        for item_selling_price in selling_price_list:
+            returnMap[item_selling_price] = {}
+            for location_id, value in locations.iteritems():
+                returnMap[item_selling_price][location_id] = LocationInfo(locationId = location_id, sameState=value['sameState'])
+                #put weight logic here
+                otgAvailable = False
+                codAvailable = False
+                minDelay = -1
+                maxDelay = -1
+                serviceable = False
+                for provider_id, delay_days in value['providerInfo'].iteritems():
+                    iscod, isotg, providerCodLimit, websiteCodLimit, storeCodLimit, providerPrepaidLimit = serviceable_map[PincodeProvider(destination_pin, provider_id)]
+                    if item_selling_price <= providerPrepaidLimit:
+                        if not serviceable:
+                            serviceable = True
+                            minDelay = delay_days
+                            maxDelay = delay_days
+                        else:
+                            minDelay = min(delay_days,minDelay)
+                            maxDelay = max(delay_days,maxDelay)
+                        iscod = iscod and item_selling_price <= min(providerCodLimit, websiteCodLimit)
+                        isotg = (isotg and item_selling_price >= 2000)
+                        otgAvailable = otgAvailable or isotg
+                        codAvailable = codAvailable or iscod
+                if serviceable:
+                    returnMap[item_selling_price][location_id].otgAvailable = otgAvailable
+                    returnMap[item_selling_price][location_id].codAvailable = codAvailable
+                    returnMap[item_selling_price][location_id].minDelay = minDelay
+                    returnMap[item_selling_price][location_id].maxDelay = maxDelay
+    return returnMap
 def get_logistics_estimation(destination_pin, item_selling_price, weight, type, billingWarehouseId):
     logging.info("Getting logistics estimation for pincode:" + destination_pin )
     provider_id, codAllowed, otgAvailable = __get_logistics_provider_for_destination_pincode(destination_pin, item_selling_price, weight, type, billingWarehouseId)
             awb.awb_number = number
             awb.provider_id = provider_id
             awb.is_available = True
             awb.type = type
     session.commit()
+def get_pincode_item_serviceability():
+    pass
 def set_AWB_as_used(awb_number):
     query = Awb.query.filter_by(awb_number = awb_number)
     awb = query.one()
     awb.is_available=False

Subversion Repositories SmartDukaan

(root)//trunk/PyProj/src/shop2020/logistics/service/impl/DataAccessor.py – Rev 19230 → 19413