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, Provider, Shipment,\

    ShipmentUpdate

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_by_id(provider_id):

    query = Provider.query.filter_by(id=provider_id)

    try:

        return query.one()

    except:

        provider = Provider()

        provider.name = "Default"

        session.commit()

        return provider

def add_empty_AWBs(numbers, provider_id):

    provider = get_provider_by_id(provider_id)

    for number in numbers:

        query = AWB.query.filter(AWB.awb_number == number)

        query = query.filter(AWB.provider_id == provider_id)

        try:

            query.one()

        except:    

            awb = AWB()

            awb.awb_number = number

            awb.provider = provider

            awb.is_available = True

    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)  #check the provider thing

    query = query.filter_by(is_available=True)

    try:

        awbs = query.all()

        awb = awbs[0]

        set_AWB_as_used(awb.awb_number)

        return awb.awb_number

    except:

        return "129102" #generate new exception

def get_providers():

    return Provider.query.all()

def get_provider(provider_id):

        return Provider.get_by(id=provider_id)

def get_shipment_info(awb_number):

    shipment = Shipment.get_by(awb = awb_number)

    query = ShipmentUpdate.query.filter_by(shipment = shipment)

    return query.all()

def create_shipment(shipment): #add date for the shipment

    new_shipment = Shipment()

    new_shipment.destination = shipment.destination 

    new_shipment.origin = shipment.origin

    new_shipment.recepient_address = shipment.recepient_address

    new_shipment.recepient_name = shipment.recepient_name

    new_shipment.recepient_phone = shipment.recepient_phone

    new_shipment.recepient_pincode = shipment.recepient_pincode

    new_shipment.shipment_contents = shipment.shipment_contents

    new_shipment.shipment_weight = shipment.shipment_weight

    new_shipment.warehouse_name = shipment.warehouse_name

    #new_shipment.timestamp = shipment.timestamp 

    if shipment.timestamp is None: 

        new_shipment.timestamp = datetime.datetime.today()

    else:

        new_shipment.timestamp = shipment.timestamp

    new_shipment.awb = shipment.awb

    session.commit() 

    return

def update_shipment_status(awb, shipment_update):

    shipment = Shipment.get_by(awb = awb)

    new_shipment_update = ShipmentUpdate()

    new_shipment_update.shipment = shipment

    new_shipment_update.description = shipment_update.description

    new_shipment_update.pin = shipment_update.pin

    new_shipment_update.shipment 

    if shipment_update.timestamp is None:

        new_shipment_update.timestamp = datetime.datetime.today()

    else:

        new_shipment_update.timestamp = shipment_update.timestamp    

    session.commit()

def get_shipments(warehouse_id, from_date, to_date, provider_id):

    awbs = AWB.query.filter_by(provider_id = provider_id).all()

    shipments = []

    for awb in awbs:

        awb_number = awb.awb_number

        query =  Shipment.query.filter(Shipment.awb == awb_number)

        query = query.filter(Shipment.timestamp > from_date)

        query = query.filter(Shipment.timestamp < to_date)

        try:

            shipment = query.one()

            shipments.append(shipment)

        except:

            continue    

    return shipments

def get_todays_shipments(warehouse_id, provider_id):

    from_date = datetime.date.today()

    from_datetime = datetime.datetime(from_date.year, from_date.month, from_date.day)

    to_date = from_date + datetime.timedelta(days=1)

    to_datetime = datetime.datetime(to_date.year, to_date.month, to_date.day)

    return get_shipments(warehouse_id, from_datetime, to_datetime, provider_id)