Subversion Repositories SmartDukaan

Rev

Go to most recent revision | Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
777 rajveer 1 
#!/usr/bin/env python
2 
#
3 
# Autogenerated by Thrift
4 
#
5 
# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
6 
#
7 
 
8 
import sys
9 
import pprint
10 
from urlparse import urlparse
11 
from thrift.transport import TTransport
12 
from thrift.transport import TSocket
13 
from thrift.transport import THttpClient
14 
from thrift.protocol import TBinaryProtocol
15 
 
16 
import TestService
17 
from ttypes import *
18 
 
19 
if len(sys.argv) <= 1 or sys.argv[1] == '--help':
20 
  print ''
21 
  print 'Usage: ' + sys.argv[0] + ' [-h host:port] [-u url] [-f[ramed]] function [arg1 [arg2...]]'
22 
  print ''
23 
  print 'Functions:'
24 
  print '  void sayHello(string message)'
25 
  print '  string getUser(i64 id)'
26 
  print '  void createUser(string name)'
27 
  print '  void closeSession()'
28 
  print ''
29 
  sys.exit(0)
30 
 
31 
pp = pprint.PrettyPrinter(indent = 2)
32 
host = 'localhost'
33 
port = 9090
34 
uri = ''
35 
framed = False
36 
http = False
37 
argi = 1
38 
 
39 
if sys.argv[argi] == '-h':
40 
  parts = sys.argv[argi+1].split(':')
41 
  host = parts[0]
42 
  port = int(parts[1])
43 
  argi += 2
44 
 
45 
if sys.argv[argi] == '-u':
46 
  url = urlparse(sys.argv[argi+1])
47 
  parts = url[1].split(':')
48 
  host = parts[0]
49 
  if len(parts) > 1:
50 
    port = int(parts[1])
51 
  else:
52 
    port = 80
53 
  uri = url[2]
54 
  http = True
55 
  argi += 2
56 
 
57 
if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':
58 
  framed = True
59 
  argi += 1
60 
 
61 
cmd = sys.argv[argi]
62 
args = sys.argv[argi+1:]
63 
 
64 
if http:
65 
  transport = THttpClient.THttpClient(host, port, uri)
66 
else:
67 
  socket = TSocket.TSocket(host, port)
68 
  if framed:
69 
    transport = TTransport.TFramedTransport(socket)
70 
  else:
71 
    transport = TTransport.TBufferedTransport(socket)
72 
protocol = TBinaryProtocol.TBinaryProtocol(transport)
73 
client = TestService.Client(protocol)
74 
transport.open()
75 
 
76 
if cmd == 'sayHello':
77 
  if len(args) != 1:
78 
    print 'sayHello requires 1 args'
79 
    sys.exit(1)
80 
  pp.pprint(client.sayHello(args[0],))
81 
 
82 
elif cmd == 'getUser':
83 
  if len(args) != 1:
84 
    print 'getUser requires 1 args'
85 
    sys.exit(1)
86 
  pp.pprint(client.getUser(eval(args[0]),))
87 
 
88 
elif cmd == 'createUser':
89 
  if len(args) != 1:
90 
    print 'createUser requires 1 args'
91 
    sys.exit(1)
92 
  pp.pprint(client.createUser(args[0],))
93 
 
94 
elif cmd == 'closeSession':
95 
  if len(args) != 0:
96 
    print 'closeSession requires 0 args'
97 
    sys.exit(1)
98 
  pp.pprint(client.closeSession())
99 
 
100 
transport.close()