Subversion Repositories SmartDukaan

#!/usr/bin/env python
#
# Autogenerated by Thrift
#
# 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 CatalogService
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 addCategory(Category category)'
  print '  void addItemToCategory(Item item, i64 category_id)'
  print '  void addItemAttribsToItem(ItemAttribs itemattribs, i64 item_id)'
  print '  void addItemInventoryToItem(ItemInventory iteminventory, i64 item_id)'
  print '  void addWarehouse(Warehouse warehouse)'
  print '  void addVendor(Vendor vendor)'
  print '  void updateInventory(i64 item_id, i64 warehouse_id, i64 quantity, i64 timestamp)'
  print '  Category getCategory(i64 id)'
  print '   getAllCategories()'
  print '  Item getItem(i64 item_id)'
  print '   getAllItemsForCategory(i64 category_id)'
  print '  ItemAttribs getItemAtttribs(i64 item_id)'
  print '  ItemInventory getItemInventory(i64 item_id)'
  print '  ItemInventoryHistory getItemInventoryHistory(i64 id, i64 item_id, i64 warehouse_id, i64 from_date, i64 to_date)'
  print '  Warehouse getWarehouse(i64 warehouse_id)'
  print '   getAllWarehousesForItem(i64 item_is)'
  print '  i64 getExpectedDeliveryTime(i64 item_id)'
  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]
  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 = CatalogService.Client(protocol)
transport.open()

if cmd == 'addCategory':
  if len(args) != 1:
    print 'addCategory requires 1 args'
    sys.exit(1)
  pp.pprint(client.addCategory(eval(args[0]),))

elif cmd == 'addItemToCategory':
  if len(args) != 2:
    print 'addItemToCategory requires 2 args'
    sys.exit(1)
  pp.pprint(client.addItemToCategory(eval(args[0]),eval(args[1]),))

elif cmd == 'addItemAttribsToItem':
  if len(args) != 2:
    print 'addItemAttribsToItem requires 2 args'
    sys.exit(1)
  pp.pprint(client.addItemAttribsToItem(eval(args[0]),eval(args[1]),))

elif cmd == 'addItemInventoryToItem':
  if len(args) != 2:
    print 'addItemInventoryToItem requires 2 args'
    sys.exit(1)
  pp.pprint(client.addItemInventoryToItem(eval(args[0]),eval(args[1]),))

elif cmd == 'addWarehouse':
  if len(args) != 1:
    print 'addWarehouse requires 1 args'
    sys.exit(1)
  pp.pprint(client.addWarehouse(eval(args[0]),))

elif cmd == 'addVendor':
  if len(args) != 1:
    print 'addVendor requires 1 args'
    sys.exit(1)
  pp.pprint(client.addVendor(eval(args[0]),))

elif cmd == 'updateInventory':
  if len(args) != 4:
    print 'updateInventory requires 4 args'
    sys.exit(1)
  pp.pprint(client.updateInventory(eval(args[0]),eval(args[1]),eval(args[2]),eval(args[3]),))

elif cmd == 'getCategory':
  if len(args) != 1:
    print 'getCategory requires 1 args'
    sys.exit(1)
  pp.pprint(client.getCategory(eval(args[0]),))

elif cmd == 'getAllCategories':
  if len(args) != 0:
    print 'getAllCategories requires 0 args'
    sys.exit(1)
  pp.pprint(client.getAllCategories())

elif cmd == 'getItem':
  if len(args) != 1:
    print 'getItem requires 1 args'
    sys.exit(1)
  pp.pprint(client.getItem(eval(args[0]),))

elif cmd == 'getAllItemsForCategory':
  if len(args) != 1:
    print 'getAllItemsForCategory requires 1 args'
    sys.exit(1)
  pp.pprint(client.getAllItemsForCategory(eval(args[0]),))

elif cmd == 'getItemAtttribs':
  if len(args) != 1:
    print 'getItemAtttribs requires 1 args'
    sys.exit(1)
  pp.pprint(client.getItemAtttribs(eval(args[0]),))

elif cmd == 'getItemInventory':
  if len(args) != 1:
    print 'getItemInventory requires 1 args'
    sys.exit(1)
  pp.pprint(client.getItemInventory(eval(args[0]),))

elif cmd == 'getItemInventoryHistory':
  if len(args) != 5:
    print 'getItemInventoryHistory requires 5 args'
    sys.exit(1)
  pp.pprint(client.getItemInventoryHistory(eval(args[0]),eval(args[1]),eval(args[2]),eval(args[3]),eval(args[4]),))

elif cmd == 'getWarehouse':
  if len(args) != 1:
    print 'getWarehouse requires 1 args'
    sys.exit(1)
  pp.pprint(client.getWarehouse(eval(args[0]),))

elif cmd == 'getAllWarehousesForItem':
  if len(args) != 1:
    print 'getAllWarehousesForItem requires 1 args'
    sys.exit(1)
  pp.pprint(client.getAllWarehousesForItem(eval(args[0]),))

elif cmd == 'getExpectedDeliveryTime':
  if len(args) != 1:
    print 'getExpectedDeliveryTime requires 1 args'
    sys.exit(1)
  pp.pprint(client.getExpectedDeliveryTime(eval(args[0]),))

transport.close()