Subversion Repositories SmartDukaan

'''

Created on 13-Sep-2010

@author: rajveer

'''

from shop2020.logistics.service.impl import DataService

from shop2020.logistics.service.impl.DataService import Awb, AwbUpdate, Provider, \

     DeliveryEstimate, WarehouseAllocation, \

    PublicHolidays, ServiceableLocationDetails, PickupStore

from shop2020.thriftpy.logistics.ttypes import LogisticsServiceException,\

    DeliveryType

from shop2020.utils.Utils import log_entry, to_py_date, to_java_date

from sqlalchemy.sql import or_

from elixir import session

import datetime, time

import sys

import logging

logging.basicConfig(level=logging.DEBUG)

warehouse_allocation_cache = {}

serviceable_location_cache = {}

delivery_estimate_cache = {}

ncr_pincodes = ['110001','110002','110003','110004','110005','110006','110007','110008','110009','110010','110011','110012','110013','110014','110015',\

                '110016','110017','110018','110019','110020','110021','110022','110023','110024','110025','110026','110027','110028','110029','110030',\

                '110031','110032','110033','110034','110035','110037','110038','110041','110042','110044','110045','110046','110047','110048','110049',\

                '110051','110052','110053','110054','110055','110056','110057','110058','110059','110060','110061','110062','110063','110064','110065',\

                '110066','110067','110068','110070','110071','110074','110075','110076','110078','110081','110082','110083','110084','110085','110086',\

                '110087','110088','110089','110091','110092','110093','110094','110095','110096','110101','110103','110104','110105','110106','110107',\

                '110108','110109','110110','110112','110113','110114','110115','110116','110117','110118','110119','110120','110122','110124','110125',\

                '110301','110302','110501','110502','110503','110504','110505','110510','110511','110512','110601','110602','110603','110604','110605',\

                '110606','110607','110608','110609','121001','121002','121003','121004','121005','121006','121007','121008','121009','122001','122002',\

                '122003','122005','122006','122008','122009','122010','122015','122017','201001','201002','201003','201004','201005','201006','201007',\

                '201008','201009','201010','201011','201012','201014','201301','201302','201303','201304','201305','201306','201307','201309','201310',\

                '201311']

'''

This class is for only data transfer. Never used outside this module.

'''

class _DeliveryEstimateObject:

    def __init__(self, delivery_time, reliability, provider_id, codAllowed):

        self.delivery_time = delivery_time

        self.reliability = reliability

        self.provider_id = provider_id

        self.codAllowed = codAllowed

def initialize(dbname="logistics", db_hostname="localhost"):

    log_entry("initialize@DataAccessor", "Initializing data service")

    DataService.initialize(dbname, db_hostname)

    print "Starting cache population at: " + str(datetime.datetime.now())

    __cache_warehouse_allocation_table()

    __cache_serviceable_location_details_table()

    __cache_delivery_estimate_table()

    close_session()

    print "Done cache population at: " + str(datetime.datetime.now())

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.delivery_time, delivery_estimate.reliability

def __cache_serviceable_location_details_table():

    serviceable_locations = ServiceableLocationDetails.query.filter(or_("exp!=0","cod!=0"))

    for serviceable_location in serviceable_locations:

        try:

            provider_pincodes = serviceable_location_cache[serviceable_location.provider_id]

        except:

            provider_pincodes = {}

            serviceable_location_cache[serviceable_location.provider_id] = provider_pincodes 

        provider_pincodes[serviceable_location.dest_pincode] = serviceable_location.dest_code, serviceable_location.exp, serviceable_location.cod, serviceable_location.station_type

def __cache_warehouse_allocation_table():

    warehouse_allocations = WarehouseAllocation.query.all()

    for warehouse_allocation in warehouse_allocations:

        warehouse_allocation_cache[warehouse_allocation.pincode]=warehouse_allocation.primary_warehouse_location

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 

def get_logistics_estimation(destination_pin, item_selling_price, type=DeliveryType.PREPAID):

    logging.info("Getting logistics estimation for pincode:" + destination_pin )

    provider_id, codAllowed = __get_logistics_provider_for_destination_pincode(type, destination_pin, item_selling_price)

    if not provider_id:

        raise LogisticsServiceException(101, "No provider assigned for pincode: " + str(destination_pin))

    try:

        delivery_time = delivery_estimate_cache[destination_pin, provider_id][0]

        reliability = delivery_estimate_cache[destination_pin, provider_id][1]

        delivery_estimate = _DeliveryEstimateObject(delivery_time, reliability, provider_id, codAllowed)

        return delivery_estimate

    except Exception as ex:

        print ex

        raise LogisticsServiceException(103, "No Logistics partner listed for this destination pincode and the primary warehouse")

def __get_logistics_provider_for_destination_pincode(type, destination_pin, item_selling_price):

    if ncr_pincodes.__contains__(destination_pin) and item_selling_price <= 20000:

        return 3, True

    if serviceable_location_cache.get(3).has_key(destination_pin):

        dest_code, exp, iscod, station_type = serviceable_location_cache.get(3).get(destination_pin)

        iscod = iscod and item_selling_price <= 9600

        return 3, iscod

    if serviceable_location_cache.get(1).has_key(destination_pin):

        dest_code, exp, iscod, station_type = serviceable_location_cache.get(1).get(destination_pin)

        iscod = iscod and item_selling_price <= 9600

        return 1, iscod

    else:

        return None, False    

def get_destination_code(providerId, pinCode):

    serviceableLocationDetail = ServiceableLocationDetails.query.filter_by(provider_id = providerId, dest_pincode = pinCode).one()

    return serviceableLocationDetail.dest_code

def add_empty_AWBs(numbers, provider_id, type):

    for number in numbers:

        query = Awb.query.filter_by(awb_number = number, provider_id = provider_id)

        try:

            query.one()

        except:    

            awb = Awb()

            awb.awb_number = number

            awb.provider_id = provider_id

            awb.is_available = True

            awb.type = type

    session.commit()

def set_AWB_as_used(awb_number):

    query = Awb.query.filter_by(awb_number = awb_number)

    awb = query.one() 

    awb.is_available=False

    session.commit()

def get_empty_AWB(provider_id, type = DeliveryType.PREPAID):

    query = Awb.query.with_lockmode("update").filter_by(provider_id = provider_id, is_available = True)  #check the provider thing

    if type == DeliveryType.PREPAID:

        query = query.filter_by(type="Prepaid")

    if type == DeliveryType.COD:

        query = query.filter_by(type="COD")

    try:

        awb = query.first()

        awb.is_available = False

        session.commit()

        return awb.awb_number

    except:

        raise LogisticsServiceException(103, "Unable to get an AWB for the given Provider: " + str(provider_id))

def get_free_awb_count(provider_id, type):

    count = Awb.query.filter_by(provider_id = provider_id, is_available = True, type=type).count()

    if count == None:

        count = 0

    return count

def get_shipment_info(awb_number, provider_id):

    awb = Awb.get_by(awb_number = awb_number, provider_id = provider_id)

    query = AwbUpdate.query.filter_by(awb = awb)

    info = query.all()

    return info

def get_holidays(start_date, end_date):

    query = PublicHolidays.query

    if start_date != -1:

        query = query.filter(PublicHolidays.date >= to_py_date(start_date))

    if end_date != -1:

        query = query.filter(PublicHolidays.date <= to_py_date(end_date))

    holidays = query.all()

    holiday_dates = [to_java_date(datetime.datetime(*time.strptime(str(holiday.date), '%Y-%m-%d')[:3])) for holiday in holidays] 

    return holiday_dates

def get_provider_for_pickup_type(pickUp):

    return Provider.query.filter(Provider.pickup == pickUp).first().id

def close_session():

    if session.is_active:

        print "session is active. closing it."

        session.close()

def is_alive():

    try:

        session.query(Awb.id).limit(1).one()

        return True

    except:

        return False

def get_all_pickup_stores():

    pickupStores = PickupStore.query.all()

    return pickupStores

def get_pickup_store(storeId):

    return PickupStore.query.filter_by(id = storeId).one()

def get_pickup_store_by_hotspot_id(hotspotId):

    return PickupStore.query.filter_by(hotspotId = hotspotId).one()

def add_pincode(provider, pincode, destCode, exp, cod, stationType):

    provider = Provider().query.filter_by(id = provider).one()

    serviceableLocationDetails = ServiceableLocationDetails()

    serviceableLocationDetails.provider = provider

    serviceableLocationDetails.dest_pincode = pincode

    serviceableLocationDetails.dest_code = destCode

    serviceableLocationDetails.exp = exp

    serviceableLocationDetails.cod = cod

    serviceableLocationDetails.station_type = stationType

    session.commit()

def update_pincode(providerId, pincode, exp, cod):

    serviceableLocationDetails = ServiceableLocationDetails().get_by(provider_id = providerId, dest_pincode = pincode).one()

    serviceableLocationDetails.exp = exp

    serviceableLocationDetails.cod = cod

    session.commit()