Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

executable file 300 lines (245 sloc) 10.526 kb
#!/usr/bin/env python
"""
interactive Cassandra Python shell
"""
try:
from IPython.Shell import IPShellEmbed
except ImportError:
print "[I]: IPython not found, falling back to default interpreter."
def runshell():
import os
os.environ['PYTHONINSPECT'] = '1'
try:
runshell = IPShellEmbed([])
except NameError:
pass
import pycassa, optparse
from pycassa.system_manager import *
from sys import stdout, stderr, exit
def _print_line(label, value):
spaces = " " * (35 - len(label + ':'))
print "%s:%s%s" % (label, spaces, str(value))
def _make_line(label, value):
spaces = " " * (35 - len(label + ':'))
return "%s:%s%s\n" % (label, spaces, str(value))
def describe_keyspace(keyspace):
try:
ks = SYSTEM_MANAGER.get_keyspace_properties(keyspace)
except pycassa.NotFoundException:
print "\nKeyspace %s does not exist." % keyspace
print
_print_line("Name", keyspace)
print
s = ks['replication_strategy']
_print_line('Replication Strategy', s[s.rfind('.') + 1: ])
print
if ks['strategy_options']:
_print_line("Strategy Options", ks['strategy_options'])
_print_line("Replication Factor", str(ks['replication_factor']))
print
cfs = SYSTEM_MANAGER.get_keyspace_column_families(keyspace).keys()
cfs.sort()
print "Column Families:"
for cf in cfs:
print " " + cf
def describe_column_family(arg1, arg2=None):
"""
Prints a description of a column family.
A keyspace and column family name may be passed in, or a
:class:`~pycassa.columnfamil.ColumnFamily` object may be passed in.
"""
if arg2 is None:
# This is a ColumnFamily object
column_family = arg1.column_family
keyspace = arg1.pool.keyspace
else:
keyspace, column_family = (arg1, arg2)
try:
cfdef = SYSTEM_MANAGER.get_keyspace_column_families(keyspace)[column_family]
except KeyError:
print "Column family %s does not exist in keyspace %s" % (column_family, keyspace)
print
_print_line('Name', cfdef.name)
_print_line('Description', cfdef.comment)
_print_line('Column Type', cfdef.column_type)
print
s = cfdef.comparator_type
_print_line('Comparator Type', s[s.rfind('.') + 1:])
if cfdef.column_type == 'Super':
s = cfdef.subcomparator_type
_print_line('Subcomparator Type', s[s.rfind('.') + 1:])
s = cfdef.default_validation_class
_print_line('Default Validation Class', s[s.rfind('.') + 1:])
print
print "Cache Sizes"
if cfdef.row_cache_size == 0:
s = 'Disabled'
elif cfdef.row_cache_size >= 1:
s = str(int(cfdef.row_cache_size)) + " rows"
else:
s = str(cfdef.key_cache_size) + "%"
_print_line(" Row Cache", s)
if cfdef.key_cache_size == 0:
s = 'Disabled'
elif cfdef.key_cache_size >= 1:
s = str(int(cfdef.key_cache_size)) + " keys"
else:
s = str(cfdef.key_cache_size) + "%"
_print_line(" Key Cache", s)
print
if cfdef.read_repair_chance == 0:
s = 'Disabled'
else:
s = str(cfdef.read_repair_chance * 100) + '%'
_print_line("Read Repair Chance", s)
print
_print_line("GC Grace Seconds", cfdef.gc_grace_seconds)
print
compact_disabled = cfdef.min_compaction_threshold == 0 or cfdef.max_compaction_threshold == 0
print "Compaction Thresholds"
if compact_disabled:
_print_line(" Min", "Minor Compactions Disabled")
else:
_print_line(" Min", cfdef.min_compaction_threshold)
if compact_disabled:
_print_line(" Max", "Minor Compactions Disabled")
else:
_print_line(" Max", cfdef.max_compaction_threshold)
print
print 'Memtable Flush After Thresholds'
_print_line(" Throughput", str(cfdef.memtable_throughput_in_mb) + " MiB")
s = str(int(cfdef.memtable_operations_in_millions * 1000000))
_print_line(" Operations", s + " operations")
_print_line(" Time", str(cfdef.memtable_flush_after_mins) + " minutes")
print
print "Cache Save Periods"
if cfdef.row_cache_save_period_in_seconds == 0:
s = 'Disabled'
else:
s = str(cfdef.row_cache_save_period_in_seconds) + ' seconds'
_print_line(" Row Cache", s)
if cfdef.key_cache_save_period_in_seconds == 0:
s = 'Disabled'
else:
s = str(cfdef.key_cache_save_period_in_seconds) + ' seconds'
_print_line(" Key Cache", s)
if cfdef.column_metadata:
print "\nColumn Metadata"
for coldef in cfdef.column_metadata:
print
_print_line(" - Name", coldef.name)
s = coldef.validation_class
_print_line(" Value Type", s[s.rfind('.') + 1: ])
if coldef.index_type is not None:
s = IndexType._VALUES_TO_NAMES[coldef.index_type]
_print_line(" Index Type", s[s.rfind('.') + 1: ])
_print_line(" Index Name", coldef.index_name)
_pool = None
def _update_cf(ks, cf, delete=False):
if ks == options.keyspace and _pool is not None:
if not delete:
existed = cf.upper() in globals()
globals()[cf.upper()] = pycassa.ColumnFamily(_pool, cf)
if not existed:
print "\nLoaded %s as %s" % (cf, cf.upper())
else:
print "\nReloaded %s" % cf.upper()
else:
globals().pop(cf.upper())
print "\nDropped %s" % cf.upper()
class InteractiveSystemManager(pycassa.SystemManager):
"""
Allows ColumnFamily instances to be passed directly to SystemManager
methods instead of specifying a keyspace and column family as strings.
"""
def create_column_family(self, keyspace, name, *args, **kwargs):
super(InteractiveSystemManager, self).create_column_family(keyspace, name, *args, **kwargs)
_update_cf(keyspace, name)
def alter_column_family(self, arg1, arg2, *args, **kwargs):
if isinstance(arg1, pycassa.ColumnFamily):
keyspace = arg1.pool.keyspace
column_family = arg1.column_family
super(InteractiveSystemManager, self).alter_column_family(keyspace, column_family, arg2, *args, **kwargs)
_update_cf(keyspace, name)
else:
super(InteractiveSystemManager, self).alter_column_family(arg1, arg2, *args, **kwargs)
_update_cf(arg1, arg2)
def drop_column_family(self, arg1, arg2=None):
if isinstance(arg1, pycassa.ColumnFamily):
keyspace = arg1.pool.keyspace
column_family = arg1.column_family
else:
keyspace = arg1
column_family = arg2
super(InteractiveSystemManager, self).drop_column_family(keyspace, column_family)
_update_cf(keyspace, column_family, delete=True)
def alter_column(self, arg1, arg2, *args, **kwargs):
if isinstance(arg1, pycassa.ColumnFamily):
keyspace = arg1.pool.keyspace
column_family = arg1.column_family
super(InteractiveSystemManager, self).alter_column(keyspace, column_family, arg2, *args, **kwargs)
_update_cf(keyspace, column_family)
else:
super(InteractiveSystemManager, self).alter_column(arg1, arg2, *args, **kwargs)
_update_cf(arg1, arg2)
def create_index(self, arg1, arg2, *args, **kwargs):
if isinstance(arg1, pycassa.ColumnFamily):
keyspace = arg1.pool.keyspace
column_family = arg1.column_family
super(InteractiveSystemManager, self).create_index(keyspace, column_family, arg2, *args, **kwargs)
_update_cf(keyspace, column_family)
else:
super(InteractiveSystemManager, self).create_index(arg1, arg2, *args, **kwargs)
_update_cf(arg1, arg2)
parser = optparse.OptionParser(usage='Usage: %prog [OPTIONS]')
parser.add_option('-k', '--keyspace', help='Cassandra keyspace name.')
parser.add_option('-H', '--host', help='Hostname.')
parser.add_option('-p', '--port', type="int", help='Thrift port number.')
parser.add_option('-u', '--user', help='Username (for simple auth).')
parser.add_option('-P', '--passwd', help='Password (for simple auth).')
parser.add_option('-S', '--streaming', help='Using streaming transport.',
action="store_false", dest='framed')
parser.add_option('-F', '--framed',
help='Use framed transport. Default transport.',
action="store_true", dest='framed')
parser.add_option('-f', '--file', help='Run a script after startup')
(options, args) = parser.parse_args()
hostname = options.host and options.host or 'localhost'
port = options.port and options.port or 9160
framed = True if options.framed is None else options.framed
credentials = None
if options.user or options.passwd:
if options.user and (not options.passwd):
print >>stderr, "You must supply a password for username", options.user
exit(1)
if options.passwd and (not options.user):
print >>stderr, "You need a user to go with that password!"
exit(1)
credentials = {'username': options.user, 'password': options.passwd}
SYSTEM_MANAGER = InteractiveSystemManager('%s:%d' % (hostname, port),
credentials, framed)
print "----------------------------------"
print "Cassandra Interactive Python Shell"
print "----------------------------------"
print "Keyspace: %s" % options.keyspace
print "Host: %s:%d" % (hostname, port)
if options.keyspace:
_pool = pycassa.QueuePool(keyspace=options.keyspace,
server_list=['%s:%d' % (hostname, port)],
credentials=credentials,
framed_transport=framed,
timeout=5)
print "\nAvailable ColumnFamily instances:"
for cfname in SYSTEM_MANAGER.get_keyspace_column_families(options.keyspace).keys():
cfinstance = pycassa.ColumnFamily(_pool, cfname)
exec('%s = cfinstance' % cfname.upper())
spaces = " " * (25 - len(cfname))
print " *", cfname.upper(), spaces, "(", cfname, ")"
else:
print "\nColumnFamily instances are only available if a keyspace is specified with -k/--keyspace"
print "\nSchema definition tools and cluster information are available through SYSTEM_MANAGER."
if (options.file):
print "\nExecuting script ...",
execfile(options.file)
print " done."
runshell()
Jump to Line
Something went wrong with that request. Please try again.