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:
        coupon_code = transaction.coupon_code
        
        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")
    parser.add_option("-c", "--cod", dest="cod",
                      action="store_true", default=False,
                      help="process the transaction as a COD transaction")
    (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)
    if(options.cod):
        DataAccessors.change_transaction_status(options.txn_id, TransactionStatus.COD_IN_PROCESS, "COD payment awaited")
    else:
        DataAccessors.change_transaction_status(options.txn_id, TransactionStatus.AUTHORIZED, "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()