Subversion Repositories SmartDukaan

'''
Created on 25-Mar-2010

@author: ashish
'''
import urlparse
from shop2020.thriftpy.config.ttypes import ConfigException
from thrift.transport import TSocket
from thrift.transport.TTransport import TFramedTransport
from thrift.protocol.TBinaryProtocol import TBinaryProtocol
from shop2020.thriftpy.config import Configuration
from shop2020.thriftpy.model.v1.catalog import InventoryService
from shop2020.thriftpy.model.v1.catalog.ttypes import Item, status, Warehouse
from shop2020.config.client.ConfigClient import ConfigClient
import datetime
from shop2020.utils.Utils import to_java_date
from shop2020.thriftpy.model.v1.order import TransactionService
from shop2020.thriftpy.model.v1.order.ttypes import Transaction, PaymentInfo,\
    Payment, PaymentStatus, OrderInfo, LineItem, Item, TransactionStatus,\
    ShipmentInfo, Shipment, ShipmentStatus


class TransactionClient:
    file_path = '/tmp/config.properties'
    
    def __init__(self):
        config_client = ConfigClient()
        self.host = config_client.get_property("transaction_service_server_host")
        self.port = config_client.get_property("transaction_service_server_port")
        
        self.start_client()
    
    
    def start_client(self):
        try:
            self.transport = TSocket.TSocket(self.host, self.port)
            self.transport = TFramedTransport(self.transport)
            self.protocol = TBinaryProtocol(self.transport)
            self.client = TransactionService.Client(self.protocol)
            self.transport.open()
        except:
            raise ConfigException(101, 'unable to load the client')
    
    def createTransaction(self, transaction):
        self.client.createTransaction(transaction)
        
    def createAlert(self, transaction_id, type, comment):
        self.client.setAlert(transaction_id, False, type, comment)
    
    def get_client(self):
        return self.client
    
if __name__ == '__main__':
    transaction_client = TransactionClient()
    
    transaction = Transaction()
    transaction.shoppingCartid = 2
    transaction.customer_id = 4
    transaction.createdOn = to_java_date(datetime.datetime.now())
    transaction.expectedDeliveryTime = to_java_date(datetime.datetime.now())
    payment_info = PaymentInfo()
    payment_info.payments
    payment_info.payments = {}
    payment = Payment()
    payment.merchant_tx_id = "123"
    payment.bank_tx_id = "ICIC0001"
    payment.mode = "CARD"
    payment.amount = 23.5
    payment.status = PaymentStatus.SUCCESS
    payment.submissionTimestamp = to_java_date(datetime.datetime.now())
    payment.completionTimestamp = to_java_date(datetime.datetime.now())
    payment.description = "Sample payment"
    payment_info.payments[to_java_date(datetime.datetime.now())] = payment
    transaction.paymentInfo = payment_info
    order_info = OrderInfo()
    order_info.lineitems = []
    line_item = LineItem()
    line_item.addedOn = to_java_date(datetime.datetime.now())
    line_item.item = Item()
    line_item.item.id = 23
    line_item.item.price = 2.40
    line_item.item.weight = 2.42
    order_info.lineitems.append(line_item)
    transaction.orderInfo = order_info
    transaction.statusDescription = "INIT phase"
    transaction.transactionStatus = TransactionStatus.INIT
    shipping_info = ShipmentInfo()
    shipping_info.shipments = []
    shipment = Shipment()
    shipment.weight = 2.42
    shipment.lineitems = order_info.lineitems;
    shipment.insurance = 2;
    shipment.value = 2.40
    shipment.status = ShipmentStatus.YET_TO_SHIP
    shipment.address = "Ashish\nD-15-15,Ardee City,\nSector-52\nGurgaon\nHaryana\n122003"
    shipping_info.shipments.append(shipment)
    transaction.shipmentInfo = shipping_info
    
    transaction_client.createTransaction(transaction)
    
    transaction_client.createAlert(2, 1, "Not yet shipped")
    
    transaction_client.get_client().setExtraInfo(2, 3, "E 71", "Black", "Nokia")