Subversion Repositories SmartDukaan

Rev

Rev 1730 | Rev 2341 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
412 ashish 1 
'''
2 
Created on 05-Aug-2010
3 
 
4 
@author: ashish
5 
'''
644 chandransh 6 
from shop2020.thriftpy.logistics.ttypes import LogisticsInfo,\
7 
    LogisticsServiceException
8 
from shop2020.logistics.service.impl.DataAccessor import get_empty_AWB,\
675 chandransh 9 
    get_shipment_info, initialize, get_logistics_estimation, get_provider,\
1730 ankur.sing 10 
    get_providers, close_session, get_free_awb_count, get_holidays
669 chandransh 11 
from shop2020.logistics.service.impl.Converters import to_t_awbupdate,\
12 
    to_t_provider
494 rajveer 13 
from shop2020.clients.InventoryClient import InventoryClient
14 
from shop2020.config.client.ConfigClient import ConfigClient
15 
import datetime
644 chandransh 16 
import math
1687 vikas 17 
import sys
1730 ankur.sing 18 
from shop2020.logistics.service.impl.DataService import ServiceableLocationDetails,\
19 
    PublicHolidays
472 rajveer 20 
 
412 ashish 21 
class LogisticsServiceHandler:
22 
 
1248 chandransh 23 
    def __init__(self, dbname='logistics'):
24 
        initialize(dbname)
746 rajveer 25 
        try:
26 
            config_client = ConfigClient()
27 
            self.cutoff_time = int(config_client.get_property('delivery_cutoff_time'))
28 
            self.stock_threshold = int(config_client.get_property('inventory_stock_threshold'))
29 
            self.time_to_producre = int(config_client.get_property('inventory_time_to_procure'))
776 rajveer 30 
            self.default_pincode = int(config_client.get_property('default_pincode'))
1687 vikas 31 
        except Exception as ex:
32 
            print "[ERROR] Unexpected config error:", sys.exc_info()[0]
746 rajveer 33 
            self.cutoff_time = 15
34 
            self.stock_threshold = 2
776 rajveer 35 
            self.time_to_producre = 48
36 
            self.default_pincode = "110001"
746 rajveer 37 
 
669 chandransh 38 
    def getProvider(self, providerId):
675 chandransh 39 
        """
40 
        Returns a provider for a given provider ID. Throws an exception if none found.
41 
 
42 
        Parameters:
43 
         - providerId
44 
        """
796 rajveer 45 
        try:
1137 chandransh 46 
            provider = get_provider(providerId)
47 
            if provider:
48 
                return to_t_provider(provider)
49 
            else:
50 
                raise LogisticsServiceException(101, "No Provider found for the given id")
796 rajveer 51 
        finally:
52 
            close_session()
53 
 
675 chandransh 54 
    def getAllProviders(self, ):
55 
        """
56 
        Returns a list containing all the providers.
57 
        """
796 rajveer 58 
        try:
59 
            return [to_t_provider(provider) for provider in get_providers()]
60 
        finally:
61 
            close_session()
62 
 
644 chandransh 63 
    def getLogisticsInfo(self, destination_pincode, itemId):
483 rajveer 64 
        """
65 
        Parameters:
66 
         - destination_pincode
716 rajveer 67 
         - item_id
483 rajveer 68 
        """
796 rajveer 69 
        try:
70 
            logistics_info = self.getLogisticsEstimation(itemId, destination_pincode)
71 
            logistics_info.airway_billno = get_empty_AWB(logistics_info.providerId)
72 
            return logistics_info
73 
        finally:
74 
            close_session()
75 
 
412 ashish 76 
    def getEmptyAWB(self, provider_id):
77 
        """
78 
        Parameters:
79 
         - provider_id
80 
        """
796 rajveer 81 
        try:
82 
            return get_empty_AWB(provider_id)
83 
        finally:
84 
            close_session()
85 
 
644 chandransh 86 
    def getShipmentInfo(self, awb, providerId):
412 ashish 87 
        """
88 
        Parameters:
89 
         - awb
766 rajveer 90 
         - providerId
412 ashish 91 
        """
796 rajveer 92 
        try:
93 
            awb_updates = get_shipment_info(awb, providerId)
94 
            t_updates = []
95 
            for update in awb_updates:
96 
                t_updates.append(to_t_awbupdate(update))
97 
            return t_updates
98 
        finally:
99 
            close_session()
100 
 
644 chandransh 101 
    def getLogisticsEstimation(self, itemId, destination_pin):
472 rajveer 102 
        """
103 
        Parameters:
104 
         - itemId
105 
         - destination_pin
106 
        """
644 chandransh 107 
        try:
796 rajveer 108 
            if not destination_pin:
109 
                destination_pin = self.default_pincode
1504 ankur.sing 110 
            try:
111 
                client = InventoryClient().get_client()
112 
                item = client.getItem(itemId)
113 
            except Exception as ex:
114 
                raise LogisticsServiceException(103, "Unable to fetch inventory information about this item.")
115 
            delivery_estimate = get_logistics_estimation(destination_pin, item.sellingPrice)
796 rajveer 116 
            if delivery_estimate is None:
117 
                raise LogisticsServiceException(103, "Unable to fetch delivery estimate for this pincode.")
118 
            delivery_time = 24*delivery_estimate.delivery_time
119 
            if self.cutoff_time <= datetime.datetime.now().hour:
120 
                delivery_time = delivery_time + 24
121 
            warehouse_loc = delivery_estimate.warehouse_location
122 
            try:
123 
                warehouse_id, items_in_inventory = client.getItemAvailabilityAtLocation(warehouse_loc, itemId)
1504 ankur.sing 124 
            except Exception as ex:
796 rajveer 125 
                raise LogisticsServiceException(103, "Unable to fetch inventory information about this item.")
126 
            if items_in_inventory < self.stock_threshold :
127 
                delivery_time = delivery_time + self.time_to_producre
2231 chandransh 128 
 
129 
            if itemId == 1502:
130 
                delivery_time = delivery_time + 48
131 
 
796 rajveer 132 
            delivery_time = int(math.ceil(delivery_time/24.0))
133 
 
134 
            logistics_info = LogisticsInfo()
135 
            logistics_info.deliveryTime = delivery_time
136 
            logistics_info.providerId = delivery_estimate.provider_id
137 
            logistics_info.warehouseId = warehouse_id
138 
 
139 
            return logistics_info
140 
        finally:
141 
            close_session()
142 
 
731 chandransh 143 
    def getDestinationCode(self, providerId, pinCode):
144 
        """
145 
        Returns the short three letter code of a pincode for the given provider.
146 
        Raises an exception if the pin code is not serviced by the given provider.
147 
 
148 
        Parameters:
149 
         - providerId
150 
         - pinCode
151 
        """
796 rajveer 152 
        try:
153 
            sld = ServiceableLocationDetails.get_by(provider_id = providerId, dest_pincode = pinCode)
154 
            if sld != None:
155 
                return sld.dest_code
156 
            else:
157 
                raise LogisticsServiceException(101, "The pincode " + pinCode + " is not serviced by this provider: " + str(providerId))
158 
        finally:
159 
            close_session()
1137 chandransh 160 
 
161 
    def getFreeAwbCount(self, providerId):
162 
        """
163 
        Returns the number of unused AWB numbers for the given provider
796 rajveer 164 
 
1137 chandransh 165 
        Parameters:
166 
         - providerId
167 
        """
168 
        try:
169 
            return get_free_awb_count(providerId)
170 
        finally:
171 
            close_session()
1730 ankur.sing 172 
 
173 
    def getHolidays(self, fromDate, toDate):
174 
        """
175 
        Returns list of Holiday dates between fromDate and toDate (both inclusive)
176 
        fromDate should be passed as milliseconds corresponding to the start of the day.
177 
        If fromDate is passed as -1, fromDate is not considered for filtering
178 
        If toDate is passed as -1, toDate is not considered for filtering
1137 chandransh 179 
 
1730 ankur.sing 180 
        Parameters:
181 
         - fromDate
182 
         - toDate
183 
        """
184 
        try:
185 
            return get_holidays(fromDate, toDate)
186 
        finally:
187 
            close_session()
188 
 
766 rajveer 189 
    def closeSession(self, ):
190 
        close_session()