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 308 lines (253 sloc) 11.002 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."
try:
import IPython
if hasattr(IPython, "embed"):
def runshell():
IPython.embed(banner1="")
else:
import IPython.Shell
def runshell():
IPython.Shell.IPShellEmbed([])()
except ImportError:
print "[I]: IPython not found, falling back to default interpreter."
def runshell():
import os
os.environ['PYTHONINSPECT'] = '1'
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'])
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
if hasattr(cfdef, 'memtable_throughput_in_mb'):
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
if getattr(cfdef, 'row_cache_save_period_in_seconds', None) is not None:
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 getattr(cfdef, 'key_cache_save_period_in_seconds', None) is not None:
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 sorted(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.