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

from shop2020.thriftpy.logistics.ttypes import LogisticsServiceException

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

from elixir import session

import datetime

def initialize():

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

    DataService.initialize()

def get_provider(provider_id):

        return Provider.get_by(id=provider_id)

def get_providers():

    return Provider.query.all()

def get_logistics_estimation(destination_pin):

    try:

        warehouse_location = WarehouseAllocation.get_by(pincode = destination_pin).primary_warehouse_location

    except:

        raise LogisticsServiceException(101, "No Warehouse assigned to this pincode.")

    query = DeliveryEstimate.query.filter_by(warehouse_location = warehouse_location)

    query = query.filter_by(destination_pin = destination_pin)

    query = query.order_by(DeliveryEstimate.delivery_time)

    try:

        return query.first()

    except:

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

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):

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

    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_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)

    return query.all()