Subversion Repositories SmartDukaan

#!/usr/bin/python 

'''

If a order has been marked as delivered for more than

5 days and it has spicedeck voucher issuance penning, voucher should be issued.

@author: Rajveer

'''

import datetime

import optparse

import sys

import urllib2

from string import Template

from shop2020.clients.HelperClient import HelperClient

if __name__ == '__main__' and __package__ is None:

    import os

    sys.path.insert(0, os.getcwd())

from shop2020.clients.PromotionClient import PromotionClient

from shop2020.thriftpy.model.v1.order.ttypes import OrderStatus

from shop2020.model.v1.order.impl import DataService

from shop2020.model.v1.order.impl.DataService import RechargeVoucherTracker

from shop2020.thriftpy.model.v1.user.ttypes import Voucher, VoucherType

from elixir import session

voucherGenerationUrl = "http://www.spicedeck.com/couponregister.sdesk"

voucherActivationUrl = "http://www.spicedeck.com/couponassign.sdesk"

voucherDeactivationUrl = "http://www.spicedeck.com/coupondeact.sdesk"

voucherStatusUrl = "http://www.spicedeck.com/couponstatus.sdesk"

username = "saholic"

password = "abc1234"

def issue_voucher_to_orders(days, db_hostname):

    DataService.initialize(db_hostname=db_hostname, echoOn=True)

    current_time = datetime.datetime.now()

    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)

    to_datetime = to_datetime - datetime.timedelta(days-1)

    vouchers = RechargeVoucherTracker.query.filter_by(voucherIssued=False).all()

    for voucher in vouchers:

        order = voucher.order

        if order.status == OrderStatus.DELIVERY_SUCCESS and order.delivery_timestamp <= to_datetime:

            storeVoucher(voucher.amount)

            coupon = issueVoucher(order.customer_id, order.customer_email, voucher.amount)

            voucher.issuedOn = current_time

            voucher.voucherIssued = True

            voucher.voucherCode = coupon

            #sendEmail(order.customer_email, voucher.amount, order.id, coupon)

    session.commit()

def storeVoucher(amount):

    coupon = generateVoucher(amount)

    voucher = Voucher()

    voucher.voucherCode = coupon

    voucher.voucherType = VoucherType.SPICEDECK_MOBILE

    voucher.amount = amount

    pc = PromotionClient().get_client()

    pc.addVoucher(voucher)

def generateVoucher(amount):

    params = "userid=" + username + "&password=" + password + "&amount=" + str(amount) 

    returnValue = urllib2.urlopen(voucherGenerationUrl + "?" + params).read()

    if "-1" == returnValue:

        raise Exception("Credit balance low.")

    elif "0" == returnValue:

        raise Exception("Network or some other failure.")

    return returnValue;

def issueVoucher(userId, userEmail, amount):

    pc = PromotionClient().get_client()

    voucher = pc.assignVoucher(userId, userEmail, VoucherType.SPICEDECK_MOBILE, amount)

    isActivated = activateVoucher(voucher.voucherCode, voucher.userEmail);

    if not isActivated:

        raise Exception("Voucher could not get activated.");

    return voucher.voucherCode

def activateVoucher(voucherCode, userEmail):

    params = "userid=" + username + "&password=" + password + "&cid=" + voucherCode + "&email=" + userEmail + "&expiry=" + "30-AUG-12" 

    returnValue = urllib2.urlopen(voucherActivationUrl + "?" + params).read()

    if "1" == returnValue:

        return True;

    elif "-1" == returnValue:

        raise Exception("Invalid voucher.")

    elif "0" == returnValue:

        raise Exception("Network or some other failure.")

    return False

def deactivateVoucher(voucherCode, reason):

    params = "userid=" + username + "&password=" + password + "&cid=" + voucherCode + "&comment=" + reason 

    returnValue = urllib2.urlopen(voucherDeactivationUrl + "?" + params).read()

    print returnValue

    ## 0#2#2#pankaj.kankar@shop2020.in

    tokens = returnValue.split('#')

    status = tokens[0]

    actualAmount = int(tokens[1])

    availableAmount = int(tokens[2])

    email = tokens[3] 

    if "1" == status:

        return True;

    elif "0" == status:

        raise Exception("Voucher could not be deactivated")

    elif "-1" == status:

        raise Exception("Network or some other failure.")

    return False

def checkVoucherStatus(voucherCode):

    params = "userid=" + username + "&password=" + password + "&cid=" + voucherCode

    returnValue = urllib2.urlopen(voucherStatusUrl + "?" + params).read()

    print returnValue

    ## ACTIVATE#2#2#pankaj.kankar@shop2020.in

    ## DEACTIVATED#2#2#pankaj.kankar@shop2020.in

    ## CONSUMED

    ## EXPIRED

    if "-1" == returnValue:

        raise Exception("Network or some other failure.")

    else:

        tokens = returnValue.split('#')

        status = tokens[0]

        actualAmount = int(tokens[1])

        availableAmount = int(tokens[2])

        email = tokens[3]

def sendEmail(email, amount, orderId, voucherCode):

    html = """

        <html>

        <body>

        <div>

        <p>

            Hi,<br /><br />

            Recharge voucher of $amount against order id $orderId is mentioned below. You can use it to recharge mobile through www.spicedeck.com. 

        </p>

        <p>    

        <strong>Product: $voucherCode </strong>

        </p>

        <p>

        Regards,<br/>

        Saholic Customer Support Team<br/>

        www.saholic.com<br/>

        Email: help@saholic.com<br/>

        </p>

        </div>

        </body>

        </html>

        """

    html = Template(html).substitute(dict(orderId=orderId,amount=amount,voucherCode=voucherCode))

    try:

        helper_client = HelperClient().get_client()

        helper_client.saveUserEmailForSending(email, "", "Recharge voucher against order id " + str(orderId), html, str(orderId), "RechargeVoucher")

    except Exception as e:

        print e

def main():

    parser = optparse.OptionParser()

    parser.add_option("-d", "--days", dest="days",

                   default=5, type="int",

                   help="issue voucher to the delivered orders before NUM days",

                   metavar="NUM")

    parser.add_option("-H", "--host", dest="hostname",

                      default="localhost",

                      type="string", help="The HOST where the DB server is running",

                      metavar="HOST")

    (options, args) = parser.parse_args()

    if len(args) != 0:

        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")

    issue_voucher_to_orders(options.days, options.hostname)

    #checkVoucherStatus('SHL1340973067027')

    #deactivateVoucher('SHL1340973067027', 'Order not delivered')

if __name__ == '__main__':

    main()