#!/usr/bin/env python # # Autogenerated by Thrift Compiler (0.9.0) # # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING # # options string: py:new_style # 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 ImpalaService 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 ' TStatus Cancel(QueryHandle query_id)' print ' TStatus ResetCatalog()' print ' TInsertResult CloseInsert(QueryHandle handle)' print ' void PingImpalaService()' 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] if len(parts) > 1: 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 = ImpalaService.Client(protocol) transport.open() if cmd == 'Cancel': if len(args) != 1: print 'Cancel requires 1 args' sys.exit(1) pp.pprint(client.Cancel(eval(args[0]),)) elif cmd == 'ResetCatalog': if len(args) != 0: print 'ResetCatalog requires 0 args' sys.exit(1) pp.pprint(client.ResetCatalog()) elif cmd == 'CloseInsert': if len(args) != 1: print 'CloseInsert requires 1 args' sys.exit(1) pp.pprint(client.CloseInsert(eval(args[0]),)) elif cmd == 'PingImpalaService': if len(args) != 0: print 'PingImpalaService requires 0 args' sys.exit(1) pp.pprint(client.PingImpalaService()) else: print 'Unrecognized method %s' % cmd sys.exit(1) transport.close()