Subversion Repositories SmartDukaan

#!/usr/bin/env python

#

# Autogenerated by Thrift Compiler (0.7.0)

#

# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING

#

import sys

import pprint

from urlparse import urlparse

from thrift.transport import TTransport

from thrift.transport import TSocket

from thrift.transport import THttpClient

from thrift.protocol import TBinaryProtocol

import PromotionService

from ttypes import *

if len(sys.argv) <= 1 or sys.argv[1] == '--help':

  print ''

  print 'Usage: ' + sys.argv[0] + ' [-h host:port] [-u url] [-f[ramed]] function [arg1 [arg2...]]'

  print ''

  print 'Functions:'

  print '  void createPromotion(string name, string ruleExecutionSrc, i64 startOn, i64 endOn)'

  print '   getAllPromotions()'

  print '  Promotion getPromotionById(i64 promotionId)'

  print '  void generateCouponsForPromotion(i64 promotionId, string couponCode)'

  print '  Cart applyCoupon(string couponCode, i64 cartId)'

  print '  void trackCouponUsage(string couponCode, i64 transactionId, i64 userId)'

  print '  i64 getCouponUsageCountByUser(string couponCode, i64 userId)'

  print '   getActiveCoupons()'

  print '  string createCoupon(i64 promotionId, i64 endOn, string email, i64 amount, i64 usage)'

  print '  i64 getSuccessfulPaymentCountForCoupon(string couponCode)'

  print '  string getRuleDocString(string ruleName)'

  print '   getItemDiscountMap( itemIds)'

  print '   getDiscountsForEntity(i64 entityId)'

  print '  void addVoucher(Voucher voucher)'

  print '  Voucher assignVoucher(i64 userId, string userEmail, VoucherType voucherType, i64 amount)'

  print '  bool markVoucherAsRedeemed(string voucherCode, i64 redeemedOn)'

  print ''

  sys.exit(0)

pp = pprint.PrettyPrinter(indent = 2)

host = 'localhost'

port = 9090

uri = ''

framed = False

http = False

argi = 1

if sys.argv[argi] == '-h':

  parts = sys.argv[argi+1].split(':')

  host = parts[0]

  port = int(parts[1])

  argi += 2

if sys.argv[argi] == '-u':

  url = urlparse(sys.argv[argi+1])

  parts = url[1].split(':')

  host = parts[0]

  if len(parts) > 1:

    port = int(parts[1])

  else:

    port = 80

  uri = url[2]

  if url[4]:

    uri += '?%s' % url[4]

  http = True

  argi += 2

if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':

  framed = True

  argi += 1

cmd = sys.argv[argi]

args = sys.argv[argi+1:]

if http:

  transport = THttpClient.THttpClient(host, port, uri)

else:

  socket = TSocket.TSocket(host, port)

  if framed:

    transport = TTransport.TFramedTransport(socket)

  else:

    transport = TTransport.TBufferedTransport(socket)

protocol = TBinaryProtocol.TBinaryProtocol(transport)

client = PromotionService.Client(protocol)

transport.open()

if cmd == 'createPromotion':

  if len(args) != 4:

    print 'createPromotion requires 4 args'

    sys.exit(1)

  pp.pprint(client.createPromotion(args[0],args[1],eval(args[2]),eval(args[3]),))

elif cmd == 'getAllPromotions':

  if len(args) != 0:

    print 'getAllPromotions requires 0 args'

    sys.exit(1)

  pp.pprint(client.getAllPromotions())

elif cmd == 'getPromotionById':

  if len(args) != 1:

    print 'getPromotionById requires 1 args'

    sys.exit(1)

  pp.pprint(client.getPromotionById(eval(args[0]),))

elif cmd == 'generateCouponsForPromotion':

  if len(args) != 2:

    print 'generateCouponsForPromotion requires 2 args'

    sys.exit(1)

  pp.pprint(client.generateCouponsForPromotion(eval(args[0]),args[1],))

elif cmd == 'applyCoupon':

  if len(args) != 2:

    print 'applyCoupon requires 2 args'

    sys.exit(1)

  pp.pprint(client.applyCoupon(args[0],eval(args[1]),))

elif cmd == 'trackCouponUsage':

  if len(args) != 3:

    print 'trackCouponUsage requires 3 args'

    sys.exit(1)

  pp.pprint(client.trackCouponUsage(args[0],eval(args[1]),eval(args[2]),))

elif cmd == 'getCouponUsageCountByUser':

  if len(args) != 2:

    print 'getCouponUsageCountByUser requires 2 args'

    sys.exit(1)

  pp.pprint(client.getCouponUsageCountByUser(args[0],eval(args[1]),))

elif cmd == 'getActiveCoupons':

  if len(args) != 0:

    print 'getActiveCoupons requires 0 args'

    sys.exit(1)

  pp.pprint(client.getActiveCoupons())

elif cmd == 'createCoupon':

  if len(args) != 5:

    print 'createCoupon requires 5 args'

    sys.exit(1)

  pp.pprint(client.createCoupon(eval(args[0]),eval(args[1]),args[2],eval(args[3]),eval(args[4]),))

elif cmd == 'getSuccessfulPaymentCountForCoupon':

  if len(args) != 1:

    print 'getSuccessfulPaymentCountForCoupon requires 1 args'

    sys.exit(1)

  pp.pprint(client.getSuccessfulPaymentCountForCoupon(args[0],))

elif cmd == 'getRuleDocString':

  if len(args) != 1:

    print 'getRuleDocString requires 1 args'

    sys.exit(1)

  pp.pprint(client.getRuleDocString(args[0],))

elif cmd == 'getItemDiscountMap':

  if len(args) != 1:

    print 'getItemDiscountMap requires 1 args'

    sys.exit(1)

  pp.pprint(client.getItemDiscountMap(eval(args[0]),))

elif cmd == 'getDiscountsForEntity':

  if len(args) != 1:

    print 'getDiscountsForEntity requires 1 args'

    sys.exit(1)

  pp.pprint(client.getDiscountsForEntity(eval(args[0]),))

elif cmd == 'addVoucher':

  if len(args) != 1:

    print 'addVoucher requires 1 args'

    sys.exit(1)

  pp.pprint(client.addVoucher(eval(args[0]),))

elif cmd == 'assignVoucher':

  if len(args) != 4:

    print 'assignVoucher requires 4 args'

    sys.exit(1)

  pp.pprint(client.assignVoucher(eval(args[0]),args[1],eval(args[2]),eval(args[3]),))

elif cmd == 'markVoucherAsRedeemed':

  if len(args) != 2:

    print 'markVoucherAsRedeemed requires 2 args'

    sys.exit(1)

  pp.pprint(client.markVoucherAsRedeemed(args[0],eval(args[1]),))

else:

  print 'Unrecognized method %s' % cmd

  sys.exit(1)

transport.close()