Subversion Repositories SmartDukaan

'''
Created on 06-Sept-2011

@author: Varun Gupta
'''
from shop2020.thriftpy.model.v1.user.ttypes import PromotionException
from shop2020.model.v1.user.impl.PromotionRuleDataUtilities import get_coupon_usage_count
from shop2020.clients.CatalogClient import CatalogClient
from shop2020.thriftpy.model.v1.catalog.ttypes import Item

def execute(cart, coupon_code, args):
    
    #Allow only first 1000 users to use the coupon
    count_coupon_usage = get_coupon_usage_count(coupon_code)
    
    if count_coupon_usage >= 1000:
        raise PromotionException(112, 'This promotion is over.')
    
    if cart.lines:
        total_selling_price = 0
        total_discounted_price = 0
        
        has_qualified_model = False
        
        for line in cart.lines:
            '''
            Blackberry    9900    Bold 4    32490    OFF Rs.2000    30490
            Samsung    i9100    Galaxy S II    29900    OFF Rs.1000    28900
            Spice    QT-58    Blueberry Mini    2849    OFF 20%    2279
            Spice    QT-61    Blueberry Exp    3261    OFF 20%    2609
            Nokia    All Models    Below Rs.5000 SP    OFF Rs.200/-
            Nokia    All Models    Above Rs.5001 SP    OFF Rs.500/-
            '''
            
            line.discountedPrice = line.actualPrice
            
            if line.itemId in (159, 160, 161, 1557, 1558):  #Spice QT-58/61
                
                has_qualified_model = True
                
                if line.actualPrice:
                    line.discountedPrice = round(line.actualPrice * 0.8)
            
            elif line.itemId == 1591:  #Blackberry 9900 Bold 4
                
                has_qualified_model = True
                
                if line.actualPrice:
                    line.discountedPrice = line.actualPrice - 2000
            
            elif line.itemId == 1502:  #Galaxy SII
                
                has_qualified_model = True
                
                if line.actualPrice:
                    line.discountedPrice = line.actualPrice - 1000
            else:
                print 'Can be Nokia'
                catalog_client = CatalogClient().get_client()
                item = catalog_client.getItem(line.itemId)
                print 'Item: %d, Brand: *%s*' % (line.itemId, item.brand.lower())
                if item.brand.lower() == 'nokia':
                    has_qualified_model = True
                    
                    print 'Brand matched'
                    discount_value = 200 if line.actualPrice < 5001 else 500
                    line.discountedPrice = line.actualPrice - discount_value
                else:
                    print 'Brand not matched'
            
            total_selling_price += line.actualPrice * line.quantity
            total_discounted_price += line.discountedPrice * line.quantity
        
        if has_qualified_model is False:
            raise PromotionException(115, 'This coupon is applicable only for selected handsets.')
        
        cart.totalPrice = total_selling_price
        cart.discountedPrice = total_discounted_price
    
    return cart