Subversion Repositories SmartDukaan

'''

Created on May 24, 2010

@author: Varun Gupta

'''

from shop2020.clients.UserClient import UserClient

from shop2020.clients.InventoryClient import InventoryClient

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

from shop2020.model.v1.user.impl import Dataservice

from shop2020.model.v1.user.impl.Dataservice import Promotion, Coupon,\

    PromotionTracker

from shop2020.utils.Utils import to_py_date

from elixir import session

import datetime

def initialize(dbname = 'user'):

    Dataservice.initialize(dbname)

def create_promotion(name, rule_execution_src, start_on, end_on):

    promotion = Promotion()

    promotion.name = name

    promotion.rule_execution_src = rule_execution_src

    promotion.start_on = to_py_date(start_on)

    promotion.end_on = to_py_date(end_on)

    promotion.created_on = datetime.datetime.now()

    session.commit()

def get_all_promotions():

    return Promotion.query.all()

def generate_coupons_for_promotion(promotion_id, coupon_code):

    promotion = Promotion.get_by(id = promotion_id)

    coupon = Coupon()

    coupon.promotion = promotion

    coupon.coupon_code = coupon_code

    coupon.arguments = ""

    session.commit()

def apply_coupon(coupon_code, cart_id):

    print  coupon_code, cart_id

    coupon = Coupon.get_by(coupon_code = coupon_code)

    if coupon:

        user_client = UserClient().get_client()

        cart = user_client.getCart(cart_id)

        #TODO: Use rule_name to updates the prices in the cart.

        rule_name = coupon.promotion.rule_execution_src

        updated_cart = rule_for_5perc_discount_on_5k_purchase(cart, coupon_code, {})

        user_client.applyCouponToCart(cart_id, coupon_code, updated_cart.totalPrice, updated_cart.discountedPrice)

        return updated_cart

    else:

        print 'Invalid Coupon Code'

        raise PromotionException(101, 'Invalid Coupon Code')

def rule_for_5perc_discount_on_5k_purchase(cart, coupon_code, args):

    #Allow a user to use the coupon only once.

    count_users_usage = get_coupon_usage_count_by_user(coupon_code, cart.userId)

    if count_users_usage > 0:

        raise PromotionException(111, 'You have already used this Coupon.')

    #Allow only first 100 users to use the coupon

    count_coupon_usage = get_coupon_usage_count(coupon_code)

    if count_coupon_usage >= 100:

        raise PromotionException(112, 'This Coupon is now exhausted.')

    if cart.lines:

        total_selling_price = 0

        for line in cart.lines:

            if line.actualPrice:

                total_selling_price +=  line.actualPrice * line.quantity

        #MAX Discount 1500

        if total_selling_price * 0.05 >= 1500:

            discount_perc = 1500 / total_selling_price

        elif total_selling_price >= 5000:

            discount_perc = 0.05

        else:

            discount_perc = 0.03

        cart.totalPrice = total_selling_price

        cart.discountedPrice = (1 - discount_perc) * total_selling_price

        for line in cart.lines:

            if line.actualPrice:

                line.discountedPrice = (1 - discount_perc) * line.actualPrice

                print "The final price, after discount is " + str(cart.discountedPrice)

    return cart

def track_coupon_usage(coupon_code, transaction_id, user_id):

    promotion_tracker = PromotionTracker()

    promotion_tracker.coupon_code = coupon_code

    promotion_tracker.transaction_id = transaction_id

    promotion_tracker.user_id = user_id

    promotion_tracker.applied_on = datetime.datetime.now()

    session.commit()

def get_coupon_usage_count_by_user(coupon_code, user_id):

    return PromotionTracker.query.filter_by(coupon_code = coupon_code).filter_by(user_id = user_id).count()

def get_coupon_usage_count(coupon_code):

    return PromotionTracker.query.filter_by(coupon_code = coupon_code).count()

def close_session():

    if session.is_active:

        session.close()