Subversion Repositories SmartDukaan

'''

Created on 06-Jul-2011

@author: Varun Gupta

'''

from shop2020.thriftpy.model.v1.user.ttypes import PromotionException, Discount

from shop2020.model.v1.user.impl.PromotionRuleDataUtilities import get_coupon_usage_count, get_coupon_usage_count_by_user

from shop2020.clients.UserClient import UserClient

def execute(cart, coupon_code, args):

    if 'items' not in args or 'discount' not in args or 'handset_display_name' not in args:

        raise PromotionException(109, 'Error in rule arguments.')

    if 'usage_limit' in args and args['usage_limit'] is not None:

        if args['usage_limit'] <= get_coupon_usage_count(coupon_code):

            raise PromotionException(112, 'This promotion is over.')

    user_client = UserClient().get_client()

    user = user_client.getUserByCartId(cart.id)

    if 'usage_limit_for_user' in args and args['usage_limit_for_user'] is not None:

        if args['usage_limit_for_user'] <= get_coupon_usage_count_by_user(coupon_code, user.userId):

            raise PromotionException(111, 'You have already used this coupon maximum possible times.')

    if not cart.lines:

        return cart

    total_selling_price = 0

    total_discounted_price = 0

    cart_has_specified_item = False

    discounts = []

    for line in cart.lines:

        total_selling_price += line.actualPrice * line.quantity

        discount_value = 0

        if line.itemId in args['items']:

            cart_has_specified_item = True

            line.discountedPrice = round(line.actualPrice - args['discount'], 4)

            total_discounted_price += round(line.discountedPrice * line.quantity, 4)

            discount_value = args['discount']

        else:

            total_discounted_price += round(line.actualPrice * line.quantity, 4)

        if discount_value > 0:

            discount = Discount()

            discount.discount = discount_value

            discount.quantity = line.quantity

            discount.cart_id = cart.id

            discount.item_id = line.itemId

            discounts.append(discount)

    if cart_has_specified_item is False:

        raise PromotionException(115, 'This coupon is applicable only for ' + args['handset_display_name'])

    cart.totalPrice = round(total_selling_price, 4)

    cart.discountedPrice = total_discounted_price

    return cart, discounts

def getDiscountOnItem(item):

    return 100