Subversion Repositories SmartDukaan

#!/usr/bin/python
'''
It is used to process transactions for which the payment 
was received but the order was not submitted for processing.
A few known cases:
 1. Missing transfer price for an item
 2. Lack of AWB numbers
 3. Amount mismatch
 4. Payment was successful but the user was not redirected to
 our website.

@author: Chandranshu
'''
import optparse
import sys

from thrift.Thrift import TException
import traceback
from shop2020.thriftpy.logistics.ttypes import PickUpType
from shop2020.thriftpy.model.v1.order.ttypes import OrderType, OrderSource

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")
    parser.add_option("-H", "--host", dest="hostname",
                      default="localhost",
                      type="string", help="The HOST where the DB server is running",
                      metavar="HOST")
    parser.add_option("-O", "--type", dest="orderType",
                      type="string", help="The type of order viz. B2B, B2C, B2Cbulk")
    (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")
    if options.orderType == None:
        parser.error("Please pass an orderType")
        
    orderTypeInteger = OrderType._NAMES_TO_VALUES.get(options.orderType)
    
    if orderTypeInteger == None:
        parser.error("Invalid OrderType. Allowed values are B2B, B2C and B2Cbulk")
        
    DataService.initialize(db_hostname=options.hostname, 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", PickUpType.COURIER, orderTypeInteger, OrderSource.WEBSITE)
    else:
        DataAccessors.change_transaction_status(options.txn_id, TransactionStatus.AUTHORIZED, "Payment received for the order", PickUpType.COURIER, orderTypeInteger, OrderSource.WEBSITE)
    
    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()