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, LineItem,  TransactionStatus, Order

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 getTransaction(self, transaction_id):

        return self.client.getTransaction(transaction_id)

    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.statusDescription = "INIT phase"

    transaction.transactionStatus = TransactionStatus.INIT

    #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 = Order()

    order.status = 5

    order.warehouse_id = 1

    order.total_weight = 10

    order.total_amount = 8

    order.statusDescription = "new order"

    transaction.orders =[]

    transaction.orders.append(order)

    order.lineitems = []

    line_item = LineItem()

    line_item.addedOn = to_java_date(datetime.datetime.now())

    line_item.unit_weight = 10

    line_item.total_weight = 10

    line_item.unit_price = 8

    line_item.total_price = 8

    order.lineitems.append(line_item)

    #transaction_client.createTransaction(transaction)

    tran = transaction_client.getTransaction(1);

    print tran.id

    print tran.shoppingCartid

    print tran

    #transaction_client.createAlert(2, 1, "Not yet shipped")