Subversion Repositories SmartDukaan

'''

Created on 06-Jul-2011

@author: Varun Gupta

'''

from shop2020.clients.CatalogClient import CatalogClient

from shop2020.clients.UserClient import UserClient

from shop2020.model.v1.user.impl.PromotionRuleDataUtilities import \

    get_coupon_usage_count, get_coupon_usage_count_by_user

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

import datetime

from math import floor

def execute(cart, coupon_code, args):

    appliedOnce = False

    if 'handset_display_name' not in args:

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

    if 'end_on' in args and datetime.datetime.strptime(args['end_on'], '%Y-%m-%d') <= datetime.datetime.now():

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

    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

    discountJ = None

    if 'discount' in args:

        discountJ  = args ['discount']

    min_amount = None

    if 'min_amount' in args:

        min_amount  = args ['min_amount']

    max_discount = 10000

    if 'max_discount' in args:

        max_discount  = args ['max_discount']

    discount_left = max_discount

    isPercentageDiscount = False

    if 'is_percentage_discount' in args:

        isPercentageDiscount = args['is_percentage_discount']

    total_discounts = 0

    cart_has_specified_item = False

    discounts = []

    for line in cart.lines:

        if discount_left > 0 :

            discount_value = 0

            if min_amount is not None and line.actualPrice < min_amount:

                continue

            if 'items' in args:

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

                    if type(args['items'])==dict: 

                        discount_value = args['items'][line.itemId]

                        cart_has_specified_item = True

                    elif discountJ is not None:

                        discount_value = discountJ

                        cart_has_specified_item = True

                else:

                    continue

            elif 'categories' in args:

                catalog_client = CatalogClient().get_client()

                category = catalog_client.getItem(line.itemId).category

                if category in args['categories']:

                    if type(args['categories'])==dict: 

                        discount_value = args['categories'][category]

                        cart_has_specified_item = True

                    elif discountJ is not None:

                        discount_value = discountJ 

                        cart_has_specified_item = True

                else:

                    continue

            elif 'brands' in args:

                catalog_client = CatalogClient().get_client()

                brand = catalog_client.getItem(line.itemId).brand

                if brand in args['brands']:

                    if type(args['brands'])==dict: 

                        discount_value = args['brands'][brand]

                        cart_has_specified_item = True

                    elif discountJ is not None:

                        discount_value = discountJ 

                        cart_has_specified_item = True

                else:

                    continue

            elif discountJ is not None:

                discount_value = discountJ 

                cart_has_specified_item = True

            else :

                continue

            if isPercentageDiscount:

                discount_value = round((line.actualPrice*discount_value)/100, 4)

            if 'appliedOnce' in args and args['appliedOnce']:

                quantity = 1

                appliedOnce = True

            else:

                quantity = min(floor(discount_left/discount_value), line.quantity)

            if quantity > 0:

                discount = Discount()

                discount.cart_id = cart.id

                discount.item_id = line.itemId

                discount.quantity = quantity

                discount.discount = discount_value

                discounts.append(discount)

                discount_left = discount_left - quantity*discount_value

                total_discounts += round(discount_value * quantity, 4)

            if appliedOnce :

                break

            if discount_left > 0 and line.quantity - quantity > 0:

                discountOne = Discount()

                discountOne.cart_id = cart.id

                discountOne.item_id = line.itemId

                discountOne.quantity = 1

                discountOne.discount = discount_left

                discounts.append(discountOne)

                discount_left = 0

                total_discounts = max_discount

        else:

            break

    if cart_has_specified_item is False:

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

    cart.discountedPrice = cart.totalPrice - round(total_discounts, 0)

    return cart, discounts

def getDiscountOnItem(item):

    return None