#!/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 Hbase
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 '   getTableNames()'
  print '   getColumnDescriptors(Text tableName)'
  print '   getTableRegions(Text tableName)'
  print '  void createTable(Text tableName,  columnFamilies)'
  print '  void deleteTable(Text tableName)'
  print '  Bytes get(Text tableName, Text row, Text column)'
  print '   getVer(Text tableName, Text row, Text column, i32 numVersions)'
  print '   getVerTs(Text tableName, Text row, Text column, i64 timestamp, i32 numVersions)'
  print '   getRow(Text tableName, Text row)'
  print '   getRowTs(Text tableName, Text row, i64 timestamp)'
  print '  void put(Text tableName, Text row, Text column, Bytes value)'
  print '  void mutateRow(Text tableName, Text row,  mutations)'
  print '  void mutateRowTs(Text tableName, Text row,  mutations, i64 timestamp)'
  print '  void mutateRows(Text tableName,  rowBatches)'
  print '  void mutateRowsTs(Text tableName,  rowBatches, i64 timestamp)'
  print '  void deleteAll(Text tableName, Text row, Text column)'
  print '  void deleteAllTs(Text tableName, Text row, Text column, i64 timestamp)'
  print '  void deleteAllRow(Text tableName, Text row)'
  print '  void deleteAllRowTs(Text tableName, Text row, i64 timestamp)'
  print '  ScannerID scannerOpen(Text tableName, Text startRow,  columns)'
  print '  ScannerID scannerOpenWithStop(Text tableName, Text startRow, Text stopRow,  columns)'
  print '  ScannerID scannerOpenTs(Text tableName, Text startRow,  columns, i64 timestamp)'
  print '  ScannerID scannerOpenWithStopTs(Text tableName, Text startRow, Text stopRow,  columns, i64 timestamp)'
  print '  ScanEntry scannerGet(ScannerID id)'
  print '  void scannerClose(ScannerID 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 = Hbase.Client(protocol)
transport.open()

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

transport.close()
