| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- #!/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 ImpalaHiveServer2Service
- 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 ResetCatalog()'
- 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 = ImpalaHiveServer2Service.Client(protocol)
- transport.open()
- if cmd == 'ResetCatalog':
- if len(args) != 0:
- print 'ResetCatalog requires 0 args'
- sys.exit(1)
- pp.pprint(client.ResetCatalog())
- else:
- print 'Unrecognized method %s' % cmd
- sys.exit(1)
- transport.close()
|