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.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