| #!/usr/bin/env python |
| # |
| # Autogenerated by Thrift Compiler (0.10.0) |
| # |
| # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING |
| # |
| # options string: py |
| # |
| |
| import sys |
| import pprint |
| if sys.version_info[0] > 2: |
| from urllib.parse import urlparse |
| else: |
| from urlparse import urlparse |
| from thrift.transport import TTransport, TSocket, TSSLSocket, THttpClient |
| from thrift.protocol.TBinaryProtocol import TBinaryProtocol |
| |
| from cassandra import Cassandra |
| from cassandra.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]] [-s[sl]] [-novalidate] [-ca_certs certs] [-keyfile keyfile] [-certfile certfile] function [arg1 [arg2...]]') |
| print('') |
| print('Functions:') |
| print(' void login(AuthenticationRequest auth_request)') |
| print(' void set_keyspace(string keyspace)') |
| print(' ColumnOrSuperColumn get(string key, ColumnPath column_path, ConsistencyLevel consistency_level)') |
| print(' get_slice(string key, ColumnParent column_parent, SlicePredicate predicate, ConsistencyLevel consistency_level)') |
| print(' i32 get_count(string key, ColumnParent column_parent, SlicePredicate predicate, ConsistencyLevel consistency_level)') |
| print(' multiget_slice( keys, ColumnParent column_parent, SlicePredicate predicate, ConsistencyLevel consistency_level)') |
| print(' multiget_count( keys, ColumnParent column_parent, SlicePredicate predicate, ConsistencyLevel consistency_level)') |
| print(' get_range_slices(ColumnParent column_parent, SlicePredicate predicate, KeyRange range, ConsistencyLevel consistency_level)') |
| print(' get_paged_slice(string column_family, KeyRange range, string start_column, ConsistencyLevel consistency_level)') |
| print(' get_indexed_slices(ColumnParent column_parent, IndexClause index_clause, SlicePredicate column_predicate, ConsistencyLevel consistency_level)') |
| print(' void insert(string key, ColumnParent column_parent, Column column, ConsistencyLevel consistency_level)') |
| print(' void add(string key, ColumnParent column_parent, CounterColumn column, ConsistencyLevel consistency_level)') |
| print(' CASResult cas(string key, string column_family, expected, updates, ConsistencyLevel serial_consistency_level, ConsistencyLevel commit_consistency_level)') |
| print(' void remove(string key, ColumnPath column_path, i64 timestamp, ConsistencyLevel consistency_level)') |
| print(' void remove_counter(string key, ColumnPath path, ConsistencyLevel consistency_level)') |
| print(' void batch_mutate( mutation_map, ConsistencyLevel consistency_level)') |
| print(' void atomic_batch_mutate( mutation_map, ConsistencyLevel consistency_level)') |
| print(' void truncate(string cfname)') |
| print(' get_multi_slice(MultiSliceRequest request)') |
| print(' describe_schema_versions()') |
| print(' describe_keyspaces()') |
| print(' string describe_cluster_name()') |
| print(' string describe_version()') |
| print(' describe_ring(string keyspace)') |
| print(' describe_local_ring(string keyspace)') |
| print(' describe_token_map()') |
| print(' string describe_partitioner()') |
| print(' string describe_snitch()') |
| print(' KsDef describe_keyspace(string keyspace)') |
| print(' describe_splits(string cfName, string start_token, string end_token, i32 keys_per_split)') |
| print(' string trace_next_query()') |
| print(' describe_splits_ex(string cfName, string start_token, string end_token, i32 keys_per_split)') |
| print(' string system_add_column_family(CfDef cf_def)') |
| print(' string system_drop_column_family(string column_family)') |
| print(' string system_add_keyspace(KsDef ks_def)') |
| print(' string system_drop_keyspace(string keyspace)') |
| print(' string system_update_keyspace(KsDef ks_def)') |
| print(' string system_update_column_family(CfDef cf_def)') |
| print(' CqlResult execute_cql_query(string query, Compression compression)') |
| print(' CqlResult execute_cql3_query(string query, Compression compression, ConsistencyLevel consistency)') |
| print(' CqlPreparedResult prepare_cql_query(string query, Compression compression)') |
| print(' CqlPreparedResult prepare_cql3_query(string query, Compression compression)') |
| print(' CqlResult execute_prepared_cql_query(i32 itemId, values)') |
| print(' CqlResult execute_prepared_cql3_query(i32 itemId, values, ConsistencyLevel consistency)') |
| print(' void set_cql_version(string version)') |
| print('') |
| sys.exit(0) |
| |
| pp = pprint.PrettyPrinter(indent=2) |
| host = 'localhost' |
| port = 9090 |
| uri = '' |
| framed = False |
| ssl = False |
| validate = True |
| ca_certs = None |
| keyfile = None |
| certfile = None |
| 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 |
| |
| if sys.argv[argi] == '-s' or sys.argv[argi] == '-ssl': |
| ssl = True |
| argi += 1 |
| |
| if sys.argv[argi] == '-novalidate': |
| validate = False |
| argi += 1 |
| |
| if sys.argv[argi] == '-ca_certs': |
| ca_certs = sys.argv[argi+1] |
| argi += 2 |
| |
| if sys.argv[argi] == '-keyfile': |
| keyfile = sys.argv[argi+1] |
| argi += 2 |
| |
| if sys.argv[argi] == '-certfile': |
| certfile = sys.argv[argi+1] |
| argi += 2 |
| |
| cmd = sys.argv[argi] |
| args = sys.argv[argi + 1:] |
| |
| if http: |
| transport = THttpClient.THttpClient(host, port, uri) |
| else: |
| if ssl: |
| socket = TSSLSocket.TSSLSocket(host, port, validate=validate, ca_certs=ca_certs, keyfile=keyfile, certfile=certfile) |
| else: |
| socket = TSocket.TSocket(host, port) |
| if framed: |
| transport = TTransport.TFramedTransport(socket) |
| else: |
| transport = TTransport.TBufferedTransport(socket) |
| protocol = TBinaryProtocol(transport) |
| client = Cassandra.Client(protocol) |
| transport.open() |
| |
| if cmd == 'login': |
| if len(args) != 1: |
| print('login requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.login(eval(args[0]),)) |
| |
| elif cmd == 'set_keyspace': |
| if len(args) != 1: |
| print('set_keyspace requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.set_keyspace(args[0],)) |
| |
| elif cmd == 'get': |
| if len(args) != 3: |
| print('get requires 3 args') |
| sys.exit(1) |
| pp.pprint(client.get(args[0], eval(args[1]), eval(args[2]),)) |
| |
| elif cmd == 'get_slice': |
| if len(args) != 4: |
| print('get_slice requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.get_slice(args[0], eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'get_count': |
| if len(args) != 4: |
| print('get_count requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.get_count(args[0], eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'multiget_slice': |
| if len(args) != 4: |
| print('multiget_slice requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.multiget_slice(eval(args[0]), eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'multiget_count': |
| if len(args) != 4: |
| print('multiget_count requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.multiget_count(eval(args[0]), eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'get_range_slices': |
| if len(args) != 4: |
| print('get_range_slices requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.get_range_slices(eval(args[0]), eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'get_paged_slice': |
| if len(args) != 4: |
| print('get_paged_slice requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.get_paged_slice(args[0], eval(args[1]), args[2], eval(args[3]),)) |
| |
| elif cmd == 'get_indexed_slices': |
| if len(args) != 4: |
| print('get_indexed_slices requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.get_indexed_slices(eval(args[0]), eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'insert': |
| if len(args) != 4: |
| print('insert requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.insert(args[0], eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'add': |
| if len(args) != 4: |
| print('add requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.add(args[0], eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'cas': |
| if len(args) != 6: |
| print('cas requires 6 args') |
| sys.exit(1) |
| pp.pprint(client.cas(args[0], args[1], eval(args[2]), eval(args[3]), eval(args[4]), eval(args[5]),)) |
| |
| elif cmd == 'remove': |
| if len(args) != 4: |
| print('remove requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.remove(args[0], eval(args[1]), eval(args[2]), eval(args[3]),)) |
| |
| elif cmd == 'remove_counter': |
| if len(args) != 3: |
| print('remove_counter requires 3 args') |
| sys.exit(1) |
| pp.pprint(client.remove_counter(args[0], eval(args[1]), eval(args[2]),)) |
| |
| elif cmd == 'batch_mutate': |
| if len(args) != 2: |
| print('batch_mutate requires 2 args') |
| sys.exit(1) |
| pp.pprint(client.batch_mutate(eval(args[0]), eval(args[1]),)) |
| |
| elif cmd == 'atomic_batch_mutate': |
| if len(args) != 2: |
| print('atomic_batch_mutate requires 2 args') |
| sys.exit(1) |
| pp.pprint(client.atomic_batch_mutate(eval(args[0]), eval(args[1]),)) |
| |
| elif cmd == 'truncate': |
| if len(args) != 1: |
| print('truncate requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.truncate(args[0],)) |
| |
| elif cmd == 'get_multi_slice': |
| if len(args) != 1: |
| print('get_multi_slice requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.get_multi_slice(eval(args[0]),)) |
| |
| elif cmd == 'describe_schema_versions': |
| if len(args) != 0: |
| print('describe_schema_versions requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.describe_schema_versions()) |
| |
| elif cmd == 'describe_keyspaces': |
| if len(args) != 0: |
| print('describe_keyspaces requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.describe_keyspaces()) |
| |
| elif cmd == 'describe_cluster_name': |
| if len(args) != 0: |
| print('describe_cluster_name requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.describe_cluster_name()) |
| |
| elif cmd == 'describe_version': |
| if len(args) != 0: |
| print('describe_version requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.describe_version()) |
| |
| elif cmd == 'describe_ring': |
| if len(args) != 1: |
| print('describe_ring requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.describe_ring(args[0],)) |
| |
| elif cmd == 'describe_local_ring': |
| if len(args) != 1: |
| print('describe_local_ring requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.describe_local_ring(args[0],)) |
| |
| elif cmd == 'describe_token_map': |
| if len(args) != 0: |
| print('describe_token_map requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.describe_token_map()) |
| |
| elif cmd == 'describe_partitioner': |
| if len(args) != 0: |
| print('describe_partitioner requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.describe_partitioner()) |
| |
| elif cmd == 'describe_snitch': |
| if len(args) != 0: |
| print('describe_snitch requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.describe_snitch()) |
| |
| elif cmd == 'describe_keyspace': |
| if len(args) != 1: |
| print('describe_keyspace requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.describe_keyspace(args[0],)) |
| |
| elif cmd == 'describe_splits': |
| if len(args) != 4: |
| print('describe_splits requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.describe_splits(args[0], args[1], args[2], eval(args[3]),)) |
| |
| elif cmd == 'trace_next_query': |
| if len(args) != 0: |
| print('trace_next_query requires 0 args') |
| sys.exit(1) |
| pp.pprint(client.trace_next_query()) |
| |
| elif cmd == 'describe_splits_ex': |
| if len(args) != 4: |
| print('describe_splits_ex requires 4 args') |
| sys.exit(1) |
| pp.pprint(client.describe_splits_ex(args[0], args[1], args[2], eval(args[3]),)) |
| |
| elif cmd == 'system_add_column_family': |
| if len(args) != 1: |
| print('system_add_column_family requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.system_add_column_family(eval(args[0]),)) |
| |
| elif cmd == 'system_drop_column_family': |
| if len(args) != 1: |
| print('system_drop_column_family requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.system_drop_column_family(args[0],)) |
| |
| elif cmd == 'system_add_keyspace': |
| if len(args) != 1: |
| print('system_add_keyspace requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.system_add_keyspace(eval(args[0]),)) |
| |
| elif cmd == 'system_drop_keyspace': |
| if len(args) != 1: |
| print('system_drop_keyspace requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.system_drop_keyspace(args[0],)) |
| |
| elif cmd == 'system_update_keyspace': |
| if len(args) != 1: |
| print('system_update_keyspace requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.system_update_keyspace(eval(args[0]),)) |
| |
| elif cmd == 'system_update_column_family': |
| if len(args) != 1: |
| print('system_update_column_family requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.system_update_column_family(eval(args[0]),)) |
| |
| elif cmd == 'execute_cql_query': |
| if len(args) != 2: |
| print('execute_cql_query requires 2 args') |
| sys.exit(1) |
| pp.pprint(client.execute_cql_query(args[0], eval(args[1]),)) |
| |
| elif cmd == 'execute_cql3_query': |
| if len(args) != 3: |
| print('execute_cql3_query requires 3 args') |
| sys.exit(1) |
| pp.pprint(client.execute_cql3_query(args[0], eval(args[1]), eval(args[2]),)) |
| |
| elif cmd == 'prepare_cql_query': |
| if len(args) != 2: |
| print('prepare_cql_query requires 2 args') |
| sys.exit(1) |
| pp.pprint(client.prepare_cql_query(args[0], eval(args[1]),)) |
| |
| elif cmd == 'prepare_cql3_query': |
| if len(args) != 2: |
| print('prepare_cql3_query requires 2 args') |
| sys.exit(1) |
| pp.pprint(client.prepare_cql3_query(args[0], eval(args[1]),)) |
| |
| elif cmd == 'execute_prepared_cql_query': |
| if len(args) != 2: |
| print('execute_prepared_cql_query requires 2 args') |
| sys.exit(1) |
| pp.pprint(client.execute_prepared_cql_query(eval(args[0]), eval(args[1]),)) |
| |
| elif cmd == 'execute_prepared_cql3_query': |
| if len(args) != 3: |
| print('execute_prepared_cql3_query requires 3 args') |
| sys.exit(1) |
| pp.pprint(client.execute_prepared_cql3_query(eval(args[0]), eval(args[1]), eval(args[2]),)) |
| |
| elif cmd == 'set_cql_version': |
| if len(args) != 1: |
| print('set_cql_version requires 1 args') |
| sys.exit(1) |
| pp.pprint(client.set_cql_version(args[0],)) |
| |
| else: |
| print('Unrecognized method %s' % cmd) |
| sys.exit(1) |
| |
| transport.close() |