Subversion Repositories SmartDukaan

#!/usr/bin/python

import optparse

import sys

from thrift.Thrift import TException

import traceback

if __name__ == '__main__' and __package__ is None:

    import os

    sys.path.insert(0, os.getcwd())

from shop2020.model.v1.order.impl import DataAccessors, DataService

from shop2020.thriftpy.model.v1.order.ttypes import TransactionStatus

from shop2020.thriftpy.model.v1.user.ttypes import ShoppingCartException,\

    PromotionException

from shop2020.clients.UserClient import UserClient

from shop2020.clients.PromotionClient import PromotionClient

def reset_cart(transaction, userServiceClient):

    items = {}

    for order in transaction.orders:

        for line_item in order.lineitems:

            item_id = line_item.item_id

            if items.has_key(item_id):

                quantity = items[item_id] + line_item.quantity

            else:

                quantity = line_item.quantity

            items[item_id] = quantity

    print items

    try:

        userServiceClient.get_client().resetCart(transaction.shopping_cart_id, items);

    except ShoppingCartException, TException:

        print "Error while resetting the cart in the cart database"

        traceback.print_exc()

    except:

        print "Unexpected exception"

        traceback.print_exc()

def track_coupon_usage(transaction, userServiceClient):

    try:

        cart = userServiceClient.get_client().getCart(transaction.shopping_cart_id)

        coupon_code = cart.couponCode

        if coupon_code is not None and coupon_code != '':

            PromotionClient().get_client().trackCouponUsage(coupon_code, transaction.id, transaction.customer_id);

    except ShoppingCartException:

        print "Error occurred in reading CardId for transaction"

        traceback.print_exc()

    except PromotionException:

        print "Promotion Exception: "

        traceback.print_exc()

    except TException:

        print "Transport from Promotion Service failed:"

        traceback.print_exc()

    except Exception:

        print "Unexpected exception:"

        traceback.print_exc()

def main():

    parser = optparse.OptionParser()

    parser.add_option("-t", "--txn-id", dest="txn_id",

                   type="int",

                   help="mark the transaction TXN_ID and all its orders as successful",

                   metavar="TXN_ID")

    (options, args) = parser.parse_args()

    if len(args) != 0:

        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")

    if options.txn_id == None:

        parser.error("No Transaction id supplied")

    DataService.initialize(echoOn=True)

    transaction = DataAccessors.get_transaction(options.txn_id)

    DataAccessors.change_transaction_status(options.txn_id, TransactionStatus.IN_PROCESS, "Payment received for the order")

    DataAccessors.enqueue_transaction_info_email(options.txn_id)

    user_service_client = UserClient()

    reset_cart(transaction, user_service_client)

    track_coupon_usage(transaction, user_service_client)

if __name__ == '__main__':

    main()