Subversion Repositories SmartDukaan

'''

Created on 29-Mar-2010

@author: ashish

'''

from shop2020.thriftpy.model.v1.order.ttypes import Transaction as T_Transaction,\

    Item as T_Item, LineItem as T_LineItem, PaymentInfo, Payment, BillingInfo,\

    ShipmentInfo, Item, OrderInfo as T_OrderInfo

from shop2020.thriftpy.model.v1.order.ttypes import Billing as T_Billing

from shop2020.thriftpy.model.v1.order.ttypes import Shipment as T_Shipment

from shop2020.thriftpy.model.v1.order.ttypes import Alert as T_Alert

from shop2020.thriftpy.model.v1.order.ttypes import ExtraOrderInfo as T_ExtraOrderInfo

from shop2020.utils.Utils import to_java_date

from shop2020.utils import OrderStatus

from shop2020.model.v1.order.impl.DataService import Billing, Alerts,\

    ExtraItemInfo

from shop2020.utils.OrderStatus import get_t_status

def to_t_transaction(transaction):

    t_transaction = T_Transaction()

    t_transaction.id = transaction.id

    t_transaction.warehouse_id = transaction.warehouse_id

    t_transaction.createdOn = to_java_date(transaction.createdOn)

    t_transaction.transactionStatus = OrderStatus.get_t_status(transaction.status)

    t_transaction.statusDescription = transaction.status_message

    t_transaction.expectedDeliveryTime = to_java_date(transaction.expected_delivery_time)

    t_transaction.customer_id = transaction.customer_id

    t_transaction.shoppingCartid = transaction.shopping_cart_id

    t_transaction.orderInfo = to_t_order_info(transaction.lineitems)

    t_transaction.shipmentInfo = to_t_shipment_info(transaction.shipments)

    t_transaction.billingInfo = to_t_billing_info(transaction.lineitems)

    t_transaction.paymentInfo = to_t_payment_info(transaction.payments)

    #populate order info

    return t_transaction

def to_t_payment_info(payments):

    t_payment_info = PaymentInfo()

    if payments:

        payment_map = {}

        for payment in payments:

            t_payment_info.id = payment.id

            t_payment = Payment()

            t_payment.merchant_tx_id = payment.merchant_tx_id

            t_payment.bank_tx_id = payment.bank_tx_id

            t_payment.mode = payment.mode

            t_payment.amount = payment.amount

            t_payment.status = payment.status

            t_payment.description = payment.status_message

            t_payment.submissionTimestamp = to_java_date(payment.start_time)

            t_payment.completionTimestamp = to_java_date(payment.finish_time)

            payment_map[t_payment.submissionTimestamp] = t_payment

        t_payment_info.payments = payment_map

    return t_payment_info

def to_t_billing_info(lineitems):

    t_billing_info = BillingInfo()

    if lineitems:

        billed_items = []

        billings = []

        for item in lineitems:

            #each item has a billing

            t_billing_info.id = item.transaction.id 

            billing = item.billing

            if billing:

                if billed_items.count(billing.id) == 0:

                    billed_items.append(billing.id)

                    billings.append(to_t_billing(billing))

        t_billing_info.billings = billings

    return t_billing_info

def to_t_order_info(lineitems):

    t_order_info = T_OrderInfo()

    if lineitems:

        line_item_list = []

        for lineitem in lineitems:

            t_order_info.id = lineitem.transaction.id

            line_item_list.append(to_t_lineitem(lineitem))

        t_order_info.lineitems = line_item_list

    return t_order_info

def to_t_shipment_info(shipments):

    shipment_info = ShipmentInfo()

    if shipments:

        shipment_list = []

        for shipment in shipments:

            shipment_info.id = shipment.transaction.id

            shipment_list.append(to_t_shipment(shipment))

        shipment_info.shipments = shipment_list

    return shipment_info

def to_t_billing(billing):

    t_billing = T_Billing()

    t_billing.id = billing.id

    t_billing.billNumber = billing.bill_number

    t_billing.generatedBy = billing.generated_by

    try:

        t_billing.generatedTimestamp = to_java_date(billing.created_on)

    except:

        print "date not found"

    t_billing.lineItem = []

    if billing.line_item:

        for item in billing.line_item:

            t_billing.lineItem.append(to_t_lineitem(item))

    return t_billing

def to_t_lineitem(lineitem):

    t_line_item = T_LineItem()

    try:

        t_line_item.id = lineitem.id

    except:

        t_line_item.id = 0

    if lineitem.item_info:

        t_line_item.item = to_t_item_info(lineitem.item_info)

    t_line_item.addedOn = to_java_date(lineitem.added_on)

    t_line_item.additionalInfo = {}

    if lineitem.additional_info:

        for k,v in lineitem.additional_info.iteritems():

            t_line_item.additionalInfo[k] = v

    return t_line_item

def to_t_item_info(item_info):

    t_item = Item()

    t_item.id = item_info.line_item.item_id

    t_item.price = item_info.price

    t_item.weight = item_info.weight

    return t_item

def to_t_shipment(shipment):

    t_shipment = T_Shipment()

    t_shipment.id = shipment.id

    t_shipment.address = shipment.address

    t_shipment.airwayBillNo = shipment.airway_bill_no

    try:

        t_shipment.deliveryTimestamp = to_java_date(shipment.delivery_time)

    except:

        print "no delivery timestamp"

    t_shipment.insurance = shipment.insurance_value

    t_shipment.provider = shipment.provider

    t_shipment.value = shipment.declared_value

    try:

        t_shipment.shipmentTimestamp = to_java_date(shipment.shipment_time)

    except:

        print "no shipment timestamp"

    t_shipment.status = get_t_status(shipment.status)

    t_shipment.weight = shipment.weight

    t_shipment.trackingId = shipment.tracking_id

    t_shipment.lineItems = []

    if shipment.lineitems:

        for line_item in shipment.lineitems:

            t_shipment.lineItems.append(to_t_lineitem(line_item))

    return t_shipment

def to_t_alert(alert):

    t_alert = T_Alert()

    t_alert.id = alert.id

    t_alert.comment = alert.comment

    try:

        t_alert.time_set = to_java_date(alert.time_set)

    except:

        pass

    try:

        t_alert.time_unset = to_java_date(alert.time_unset)

    except:

        pass

    t_alert.transaction_id = alert.transaction_id

    t_alert.type = alert.type

    return t_alert

def to_t_extra_item_info(extra_item_info, transaction_id):

    extra_order_info = T_ExtraOrderInfo()

    if not extra_item_info:

        extra_order_info.id = 2

        extra_order_info.model_name = "N 72"

        extra_order_info.sku_id = 34

        extra_order_info.transaction_id = transaction_id

        extra_order_info.vendor_info = "Nokia"

        extra_order_info.colour = "Black"

        return extra_order_info

    extra_order_info.id = extra_item_info.id

    extra_order_info.model_name = extra_item_info.model

    extra_order_info.vendor_info = extra_item_info.vendor

    extra_order_info.transaction_id = extra_item_info.transaction_id

    extra_order_info.sku_id = extra_item_info.sku_id

    return extra_order_info