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

def execute(cart, coupon_code, args):
    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 = None
    if 'max_discount' in args:
        max_discount  = args ['max_discount']
        
    
    isPercentageDiscount = False
    if 'is_percentage_discount' in args:
        isPercentageDiscount = args['is_percentage_discount']
    

    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 min_amount is not None and line.actualPrice < min_amount:
            total_discounted_price += round(line.actualPrice * line.quantity, 4)
            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:
                total_discounted_price += round(line.actualPrice * line.quantity, 4)
                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:
                total_discounted_price += round(line.actualPrice * line.quantity, 4)
                continue
        elif discountJ is not None:
            discount_value = discountJ 
            cart_has_specified_item = True
        else :
            total_discounted_price += round(line.actualPrice * line.quantity, 4) 
            continue
    
        if isPercentageDiscount:
            discount_value = round((line.actualPrice*discount_value)/100, 4)
        if max_discount is not None:
            discount_value = min(max_discount, discount_value)
            
        line.discountedPrice = round(line.actualPrice - discount_value, 4)
        total_discounted_price += round(line.discountedPrice * 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 None