class BMemcachedCli(cmd.Cmd, object):
def __init__(self, host_list):
super(BMemcachedCli, self).__init__()
# look hosts for username/password for SASL authentication - when found, assume all servers use the same
auth = [None, None]
for idx in xrange(len(host_list)):
if host_list[idx].find('@') > -1:
auth = host_list[idx].split('@')[0].split(':')
host_list[idx] = host_list[idx].split('@')[
1] # remove auth from server entry
self.memcache = Client(host_list, auth[0], auth[1])
self._check_connection(
host_list
) # Make sure we actually have connections, since our library doesn't make this apparent
if auth[0]:
self.prompt = '([B]memcached) '
else:
self.prompt = '(memcached) '
# Time for some hardcore introspection and dynamic method creation
for name in dir(self.memcache):
# List our methods, ignoring private methods and ones that should be hidden to our CLI
if not name.startswith('_') and not self._is_hidden(name):
attr = getattr(self.memcache, name)
if callable(attr):
# Make sure we only keep callable methods, let's pin the doc strings to them while we are at it
setattr(self.__class__, 'do_' + name, self._make_cmd(name))
doc = (getattr(attr, '__doc__') or '').strip()
if doc: # Not everything has a docstring
setattr(self.__class__, 'help_' + name,
self._make_help(doc))
@staticmethod
def _make_cmd(name):
# Whats a little functional passing between friends?
# This is our core factory for creating dynamic methods
def _get_stats(self, line):
try:
# Rabbit Holes can get pretty deep, better to keep a lid on it
pp = pprint.PrettyPrinter(depth=4)
pp.pprint(self.memcache.stats(line))
except Exception, e:
print_error(e)
def handler(self, line):
parts = line.split()
try:
# This is where the magic happens, get our function by name, pass the arguments,
# then pretty-print the results
pprint.pprint(getattr(self.memcache, name)(*parts))
except Exception, e:
print_error(e)
class BMemcachedCli(cmd.Cmd, object):
def __init__(self, host_list):
super(BMemcachedCli, self).__init__()
# look hosts for username/password for SASL authentication - when found, assume all servers use the same
auth = [None, None]
for idx in xrange(len(host_list)):
if host_list[idx].find('@') > -1:
auth = host_list[idx].split('@')[0].split(':')
host_list[idx] = host_list[idx].split('@')[1] # remove auth from server entry
self.memcache = Client(host_list, auth[0], auth[1])
self._check_connection(
host_list) # Make sure we actually have connections, since our library doesn't make this apparent
if auth[0]:
self.prompt = '([B]memcached) '
else:
self.prompt = '(memcached) '
# Time for some hardcore introspection and dynamic method creation
for name in dir(self.memcache):
# List our methods, ignoring private methods and ones that should be hidden to our CLI
if not name.startswith('_') and not self._is_hidden(name):
attr = getattr(self.memcache, name)
if callable(attr):
# Make sure we only keep callable methods, let's pin the doc strings to them while we are at it
setattr(self.__class__, 'do_' + name, self._make_cmd(name))
doc = (getattr(attr, '__doc__') or '').strip()
if doc: # Not everything has a docstring
setattr(self.__class__, 'help_' + name, self._make_help(doc))
@staticmethod
def _make_cmd(name):
# Whats a little functional passing between friends?
# This is our core factory for creating dynamic methods
def _get_stats(self, line):
try:
# Rabbit Holes can get pretty deep, better to keep a lid on it
pp = pprint.PrettyPrinter(depth=4)
pp.pprint(self.memcache.stats(line))
except Exception, e:
print_error(e)
def handler(self, line):
parts = line.split()
try:
# This is where the magic happens, get our function by name, pass the arguments,
# then pretty-print the results
pprint.pprint(getattr(self.memcache, name)(*parts))
except Exception, e:
print_error(e)
class BMemcachedCli(cmd.Cmd, object):
def __init__(self, host_list):
super(BMemcachedCli, self).__init__()
# look hosts for username/password for SASL authentication - when found, assume all servers use the same
auth = [None, None]
for idx in range(len(host_list)):
if host_list[idx].find('@') > -1:
auth = host_list[idx].split('@')[0].split(':')
host_list[idx] = host_list[idx].split('@')[
1] # remove auth from server entry
self.memcache = Client(host_list, auth[0], auth[1])
self._check_connection(
host_list
) # Make sure we actually have connections, since our library doesn't make this apparent
if auth[0]:
self.prompt = '([B]memcached) '
else:
self.prompt = '(memcached) '
# Time for some hardcore introspection and dynamic method creation
for name in dir(self.memcache):
# List our methods, ignoring private methods and ones that should be hidden to our CLI
if not name.startswith('_') and not self._is_hidden(name):
attr = getattr(self.memcache, name)
if callable(attr):
# Make sure we only keep callable methods, let's pin the doc strings to them while we are at it
setattr(self.__class__, 'do_' + name, self._make_cmd(name))
doc = (getattr(attr, '__doc__') or '').strip()
if doc: # Not everything has a docstring
setattr(self.__class__, 'help_' + name,
self._make_help(doc))
@staticmethod
def _make_cmd(name):
# Whats a little functional passing between friends?
# This is our core factory for creating dynamic methods
def _get_stats(self, line):
try:
# Rabbit Holes can get pretty deep, better to keep a lid on it
pp = pprint.PrettyPrinter(depth=4)
pp.pprint(self.memcache.stats(line))
except Exception as e:
print_error(e)
def handler(self, line):
parts = line.split()
try:
# This is where the magic happens, get our function by name, pass the arguments,
# then pretty-print the results
pprint.pprint(getattr(self.memcache, name)(*parts))
except Exception as e:
print_error(e)
return handler
@staticmethod
# This just lets cmd know how to print help commands for arbitrary methods
def _make_help(doc):
def help(self):
print(doc)
return help
def _check_connection(self, host_list):
# Get Stats for all hosts, make sure we connect to all of them
unreachable_hosts = []
reachable_hosts = []
stats = self.memcache.stats()
# Stats returns a list with each host getting a tuple with two elements, a string and a
# dictionary (We just hit 3 of the python data structures in one return, wtf)
# The string contains the hostname and some other junk
# The dictionary contains all of the stats for that host
for stat in stats:
reachable_hosts.append(stat)
# Compare our list of reachable hosts with the ones we were passed
unreachable_hosts = [
host for host in host_list if host not in reachable_hosts
]
if unreachable_hosts:
for host in unreachable_hosts:
print_error("Unable to connect to memcache server: %s" % host)
sys.exit(1)
# Helper method for trimming out arbitrary commands from our library
def _is_hidden(self, name):
hidden_commands = ['get_multi', 'set_multi']
for command in hidden_commands:
if command in name:
return True
return False
# Make it so we can exit easily
def do_exit(self, line):
return True
# Alias EOF to exit
do_EOF = do_exit
# We don't want to repeat the last command if a blank line is entered (default behavior), so we pass
def emptyline(self):
pass
