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 shop2020.model.v1.user.promotionrules import rule_for_5perc_discount_on_5k_purchase, rule_500_off_on_MI310

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):
    coupon = Coupon.get_by(coupon_code = coupon_code)
    
    if coupon:
        user_client = UserClient().get_client()
        cart = user_client.getCart(cart_id)

        updated_cart = eval(coupon.promotion.rule_execution_src.strip() + ".execute(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 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 close_session():
    if session.is_active:
        session.close()