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 InventoryService

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 '  i64 addWarehouse(Warehouse warehouse)'

  print '  i64 addVendor(Vendor vendor)'

  print '  void updateInventoryHistory(i64 warehouse_id, string timestamp,  availability)'

  print '  void updateInventory(i64 warehouse_id, string timestamp,  availability)'

  print '  void addInventory(i64 itemId, i64 warehouseId, i64 quantity)'

  print '  void retireWarehouse(i64 warehouse_id)'

  print '  ItemInventory getItemInventoryByItemId(i64 item_id)'

  print '  i64 getItemAvailibilityAtWarehouse(i64 warehouse_id, i64 item_id)'

  print '   getItemAvailabilityAtLocation(i64 itemId)'

  print '   getAllWarehouses(bool isActive)'

  print '  Warehouse getWarehouse(i64 warehouse_id)'

  print '   getAllItemsForWarehouse(i64 warehouse_id)'

  print '  bool reserveItemInWarehouse(i64 itemId, i64 warehouseId, double quantity)'

  print '  bool reduceReservationCount(i64 itemId, i64 warehouseId, double quantity)'

  print '  VendorItemPricing getItemPricing(i64 itemId, i64 vendorId)'

  print '   getAllItemPricing(i64 itemId)'

  print '  void addVendorItemPricing(VendorItemPricing vendorItemPricing)'

  print '  Vendor getVendor(i64 vendorId)'

  print '   getAllVendors()'

  print '  void addVendorItemMapping(string key, VendorItemMapping vendorItemMapping)'

  print '   getVendorItemMappings(i64 itemId)'

  print '   getPendingOrdersInventory(i64 vendorid)'

  print '   getWarehouses(WarehouseType warehouseType, InventoryType inventoryType, i64 vendorId, i64 billingWarehouseId, i64 shippingWarehouseId)'

  print '  void resetAvailability(string itemKey, i64 vendorId, i64 quantity, i64 warehouseId)'

  print '  void resetAvailabilityForWarehouse(i64 warehouseId)'

  print '   getItemKeysToBeProcessed(i64 warehouseId)'

  print '  void markMissedInventoryUpdatesAsProcessed(string itemKey, i64 warehouseId)'

  print '   getIgnoredItemKeys()'

  print '  void addBadInventory(i64 itemId, i64 warehouseId, i64 quantity)'

  print '   getShippingLocations()'

  print '   getAllVendorItemMappings()'

  print '   getInventorySnapshot(i64 warehouseId)'

  print '  void clearItemAvailabilityCache()'

  print '  void updateVendorString(i64 warehouseId, string vendorString)'

  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 = InventoryService.Client(protocol)

transport.open()

if 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 == 'updateInventoryHistory':

  if len(args) != 3:

    print 'updateInventoryHistory requires 3 args'

    sys.exit(1)

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

elif cmd == 'updateInventory':

  if len(args) != 3:

    print 'updateInventory requires 3 args'

    sys.exit(1)

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

elif cmd == 'addInventory':

  if len(args) != 3:

    print 'addInventory requires 3 args'

    sys.exit(1)

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

elif cmd == 'retireWarehouse':

  if len(args) != 1:

    print 'retireWarehouse requires 1 args'

    sys.exit(1)

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

elif cmd == 'getItemInventoryByItemId':

  if len(args) != 1:

    print 'getItemInventoryByItemId requires 1 args'

    sys.exit(1)

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

elif cmd == 'getItemAvailibilityAtWarehouse':

  if len(args) != 2:

    print 'getItemAvailibilityAtWarehouse requires 2 args'

    sys.exit(1)

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

elif cmd == 'getItemAvailabilityAtLocation':

  if len(args) != 1:

    print 'getItemAvailabilityAtLocation requires 1 args'

    sys.exit(1)

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

elif cmd == 'getAllWarehouses':

  if len(args) != 1:

    print 'getAllWarehouses requires 1 args'

    sys.exit(1)

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

elif cmd == 'getWarehouse':

  if len(args) != 1:

    print 'getWarehouse requires 1 args'

    sys.exit(1)

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

elif cmd == 'getAllItemsForWarehouse':

  if len(args) != 1:

    print 'getAllItemsForWarehouse requires 1 args'

    sys.exit(1)

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

elif cmd == 'reserveItemInWarehouse':

  if len(args) != 3:

    print 'reserveItemInWarehouse requires 3 args'

    sys.exit(1)

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

elif cmd == 'reduceReservationCount':

  if len(args) != 3:

    print 'reduceReservationCount requires 3 args'

    sys.exit(1)

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

elif cmd == 'getItemPricing':

  if len(args) != 2:

    print 'getItemPricing requires 2 args'

    sys.exit(1)

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

elif cmd == 'getAllItemPricing':

  if len(args) != 1:

    print 'getAllItemPricing requires 1 args'

    sys.exit(1)

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

elif cmd == 'addVendorItemPricing':

  if len(args) != 1:

    print 'addVendorItemPricing requires 1 args'

    sys.exit(1)

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

elif cmd == 'getVendor':

  if len(args) != 1:

    print 'getVendor requires 1 args'

    sys.exit(1)

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

elif cmd == 'getAllVendors':

  if len(args) != 0:

    print 'getAllVendors requires 0 args'

    sys.exit(1)

  pp.pprint(client.getAllVendors())

elif cmd == 'addVendorItemMapping':

  if len(args) != 2:

    print 'addVendorItemMapping requires 2 args'

    sys.exit(1)

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

elif cmd == 'getVendorItemMappings':

  if len(args) != 1:

    print 'getVendorItemMappings requires 1 args'

    sys.exit(1)

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

elif cmd == 'getPendingOrdersInventory':

  if len(args) != 1:

    print 'getPendingOrdersInventory requires 1 args'

    sys.exit(1)

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

elif cmd == 'getWarehouses':

  if len(args) != 5:

    print 'getWarehouses requires 5 args'

    sys.exit(1)

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

elif cmd == 'resetAvailability':

  if len(args) != 4:

    print 'resetAvailability requires 4 args'

    sys.exit(1)

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

elif cmd == 'resetAvailabilityForWarehouse':

  if len(args) != 1:

    print 'resetAvailabilityForWarehouse requires 1 args'

    sys.exit(1)

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

elif cmd == 'getItemKeysToBeProcessed':

  if len(args) != 1:

    print 'getItemKeysToBeProcessed requires 1 args'

    sys.exit(1)

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

elif cmd == 'markMissedInventoryUpdatesAsProcessed':

  if len(args) != 2:

    print 'markMissedInventoryUpdatesAsProcessed requires 2 args'

    sys.exit(1)

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

elif cmd == 'getIgnoredItemKeys':

  if len(args) != 0:

    print 'getIgnoredItemKeys requires 0 args'

    sys.exit(1)

  pp.pprint(client.getIgnoredItemKeys())

elif cmd == 'addBadInventory':

  if len(args) != 3:

    print 'addBadInventory requires 3 args'

    sys.exit(1)

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

elif cmd == 'getShippingLocations':

  if len(args) != 0:

    print 'getShippingLocations requires 0 args'

    sys.exit(1)

  pp.pprint(client.getShippingLocations())

elif cmd == 'getAllVendorItemMappings':

  if len(args) != 0:

    print 'getAllVendorItemMappings requires 0 args'

    sys.exit(1)

  pp.pprint(client.getAllVendorItemMappings())

elif cmd == 'getInventorySnapshot':

  if len(args) != 1:

    print 'getInventorySnapshot requires 1 args'

    sys.exit(1)

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

elif cmd == 'clearItemAvailabilityCache':

  if len(args) != 0:

    print 'clearItemAvailabilityCache requires 0 args'

    sys.exit(1)

  pp.pprint(client.clearItemAvailabilityCache())

elif cmd == 'updateVendorString':

  if len(args) != 2:

    print 'updateVendorString requires 2 args'

    sys.exit(1)

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

else:

  print 'Unrecognized method %s' % cmd

  sys.exit(1)

transport.close()