-# Copyright 2011-2012 GRNET S.A. All rights reserved.
+# Copyright 2011-2013 GRNET S.A. All rights reserved.
#
# Redistribution and use in source and binary forms, with or
# without modification, are permitted provided that the following
# interpreted as representing official policies, either expressed
# or implied, of GRNET S.A.command
-from kamaki.logger import get_logger
+from kamaki.cli.logger import get_logger
+from kamaki.cli.utils import (
+ print_list, print_dict, print_json, print_items, ask_user,
+ filter_dicts_by_dict)
+from kamaki.cli.argument import FlagArgument, ValueArgument
+from kamaki.cli.errors import CLIInvalidArgument
+from sys import stdin, stdout, stderr
log = get_logger(__name__)
+def DontRaiseKeyError(func):
+ def wrap(*args, **kwargs):
+ try:
+ return func(*args, **kwargs)
+ except KeyError:
+ return None
+ return wrap
+
+
+def addLogSettings(func):
+ def wrap(self, *args, **kwargs):
+ try:
+ return func(self, *args, **kwargs)
+ finally:
+ self._set_log_params()
+ return wrap
+
+
class _command_init(object):
- def __init__(self, arguments={}):
+ # self.arguments (dict) contains all non-positional arguments
+ # self.required (list or tuple) contains required argument keys
+ # if it is a list, at least one of these arguments is required
+ # if it is a tuple, all arguments are required
+ # Lists and tuples can nest other lists and/or tuples
+
+ def __init__(
+ self,
+ arguments={}, auth_base=None, cloud=None,
+ _in=None, _out=None, _err=None):
+ self._in, self._out, self._err = (
+ _in or stdin, _out or stdout, _err or stderr)
+ self.required = getattr(self, 'required', None)
if hasattr(self, 'arguments'):
arguments.update(self.arguments)
+ if isinstance(self, _optional_output_cmd):
+ arguments.update(self.oo_arguments)
+ if isinstance(self, _optional_json):
+ arguments.update(self.oj_arguments)
+ if isinstance(self, _name_filter):
+ arguments.update(self.nf_arguments)
+ if isinstance(self, _id_filter):
+ arguments.update(self.if_arguments)
+ try:
+ arguments.update(self.wait_arguments)
+ except AttributeError:
+ pass
self.arguments = dict(arguments)
try:
self.config = self['config']
- #self.config = self.get_argument('config')
except KeyError:
pass
+ self.auth_base = auth_base or getattr(self, 'auth_base', None)
+ self.cloud = cloud or getattr(self, 'cloud', None)
+
+ def write(self, s):
+ self._out.write('%s' % s)
+ self._out.flush()
+
+ def writeln(self, s=''):
+ self.write('%s\n' % s)
+
+ def error(self, s=''):
+ self._err.write('%s\n' % s)
+ self._err.flush()
+
+ def print_list(self, *args, **kwargs):
+ kwargs.setdefault('out', self._out)
+ return print_list(*args, **kwargs)
+
+ def print_dict(self, *args, **kwargs):
+ kwargs.setdefault('out', self._out)
+ return print_dict(*args, **kwargs)
+
+ def print_json(self, *args, **kwargs):
+ kwargs.setdefault('out', self._out)
+ return print_json(*args, **kwargs)
+
+ def print_items(self, *args, **kwargs):
+ kwargs.setdefault('out', self._out)
+ return print_items(*args, **kwargs)
+
+ def ask_user(self, *args, **kwargs):
+ kwargs.setdefault('user_in', self._in)
+ kwargs.setdefault('out', self._out)
+ return ask_user(*args, **kwargs)
+
+ @DontRaiseKeyError
+ def _custom_url(self, service):
+ return self.config.get_cloud(self.cloud, '%s_url' % service)
+
+ @DontRaiseKeyError
+ def _custom_token(self, service):
+ return self.config.get_cloud(self.cloud, '%s_token' % service)
+
+ @DontRaiseKeyError
+ def _custom_type(self, service):
+ return self.config.get_cloud(self.cloud, '%s_type' % service)
+
+ @DontRaiseKeyError
+ def _custom_version(self, service):
+ return self.config.get_cloud(self.cloud, '%s_version' % service)
+
+ def _uuids2usernames(self, uuids):
+ return self.auth_base.post_user_catalogs(uuids)
+
+ def _usernames2uuids(self, username):
+ return self.auth_base.post_user_catalogs(displaynames=username)
+
+ def _uuid2username(self, uuid):
+ return self._uuids2usernames([uuid]).get(uuid, None)
+
+ def _username2uuid(self, username):
+ return self._usernames2uuids([username]).get(username, None)
def _set_log_params(self):
try:
- self.client.LOG_TOKEN, self.client.LOG_DATA = (
- self['config'].get('global', 'log_token') == 'on',
- self['config'].get('global', 'log_data') == 'on')
+ self.client.LOG_TOKEN = (
+ self['config'].get('global', 'log_token').lower() == 'on')
except Exception as e:
- log.warning('Failed to read custom log settings:'
- '%s\n defaults for token and data logging are off' % e)
-
- def _update_max_threads(self):
+ log.debug('Failed to read custom log_token setting:'
+ '%s\n default for log_token is off' % e)
+ try:
+ self.client.LOG_DATA = (
+ self['config'].get('global', 'log_data').lower() == 'on')
+ except Exception as e:
+ log.debug('Failed to read custom log_data setting:'
+ '%s\n default for log_data is off' % e)
try:
- max_threads = int(self['config'].get('global', 'max_threads'))
- assert max_threads > 0
- self.client.MAX_THREADS = max_threads
+ self.client.LOG_PID = (
+ self['config'].get('global', 'log_pid').lower() == 'on')
except Exception as e:
- log.warning('Failed to read custom thread settings: '
- '%s, use default max threads (%s)' % (
- e, self.client.MAX_THREADS))
+ log.debug('Failed to read custom log_pid setting:'
+ '%s\n default for log_pid is off' % e)
- def _safe_progress_bar(self, msg, arg='progress_bar'):
+ def _safe_progress_bar(
+ self, msg, arg='progress_bar', countdown=False, timeout=100):
"""Try to get a progress bar, but do not raise errors"""
try:
progress_bar = self.arguments[arg]
- gen = progress_bar.get_generator(msg)
+ progress_bar.file = self._err
+ gen = progress_bar.get_generator(
+ msg, countdown=countdown, timeout=timeout)
except Exception:
return (None, None)
return (progress_bar, gen)
:raises KeyError: if argterm not in self.arguments of this object
"""
return self[argterm]
+
+
+# feature classes - inherit them to get special features for your commands
+
+
+class OutputFormatArgument(ValueArgument):
+ """Accepted output formats: json (default)"""
+
+ formats = ('json', )
+
+ def ___init__(self, *args, **kwargs):
+ super(OutputFormatArgument, self).___init__(*args, **kwargs)
+
+ @property
+ def value(self):
+ return getattr(self, '_value', None)
+
+ @value.setter
+ def value(self, newvalue):
+ if not newvalue:
+ self._value = self.default
+ elif newvalue.lower() in self.formats:
+ self._value = newvalue.lower
+ else:
+ raise CLIInvalidArgument(
+ 'Invalid value %s for argument %s' % (
+ newvalue, self.lvalue),
+ details=['Valid output formats: %s' % ', '.join(self.formats)])
+
+
+class _optional_output_cmd(object):
+
+ oo_arguments = dict(
+ with_output=FlagArgument('show response headers', ('--with-output')),
+ json_output=FlagArgument(
+ 'show headers in json (DEPRECATED from v0.12,'
+ ' please use --output-format=json instead)', ('-j', '--json'))
+ )
+
+ def _optional_output(self, r):
+ if self['json_output']:
+ print_json(r, out=self._out)
+ elif self['with_output']:
+ print_items([r] if isinstance(r, dict) else r, out=self._out)
+
+
+class _optional_json(object):
+
+ oj_arguments = dict(
+ output_format=OutputFormatArgument(
+ 'Show output in chosen output format (%s)' % ', '.join(
+ OutputFormatArgument.formats),
+ '--output-format'),
+ json_output=FlagArgument(
+ 'show output in json (DEPRECATED from v0.12,'
+ ' please use --output-format instead)', ('-j', '--json'))
+ )
+
+ def _print(self, output, print_method=print_items, **print_method_kwargs):
+ if self['json_output'] or self['output_format']:
+ print_json(output, out=self._out)
+ else:
+ print_method_kwargs.setdefault('out', self._out)
+ print_method(output, **print_method_kwargs)
+
+
+class _name_filter(object):
+
+ nf_arguments = dict(
+ name=ValueArgument('filter by name', '--name'),
+ name_pref=ValueArgument(
+ 'filter by name prefix (case insensitive)', '--name-prefix'),
+ name_suff=ValueArgument(
+ 'filter by name suffix (case insensitive)', '--name-suffix'),
+ name_like=ValueArgument(
+ 'print only if name contains this (case insensitive)',
+ '--name-like')
+ )
+
+ def _non_exact_name_filter(self, items):
+ np, ns, nl = self['name_pref'], self['name_suff'], self['name_like']
+ return [item for item in items if (
+ (not np) or (item['name'] or '').lower().startswith(
+ np.lower())) and (
+ (not ns) or (item['name'] or '').lower().endswith(
+ ns.lower())) and (
+ (not nl) or nl.lower() in (item['name'] or '').lower())]
+
+ def _exact_name_filter(self, items):
+ return filter_dicts_by_dict(items, dict(name=self['name'] or '')) if (
+ self['name']) else items
+
+ def _filter_by_name(self, items):
+ return self._non_exact_name_filter(self._exact_name_filter(items))
+
+
+class _id_filter(object):
+
+ if_arguments = dict(
+ id=ValueArgument('filter by id', '--id'),
+ id_pref=ValueArgument(
+ 'filter by id prefix (case insensitive)', '--id-prefix'),
+ id_suff=ValueArgument(
+ 'filter by id suffix (case insensitive)', '--id-suffix'),
+ id_like=ValueArgument(
+ 'print only if id contains this (case insensitive)', '--id-like')
+ )
+
+ def _non_exact_id_filter(self, items):
+ np, ns, nl = self['id_pref'], self['id_suff'], self['id_like']
+ return [item for item in items if (
+ (not np) or (
+ '%s' % item['id']).lower().startswith(np.lower())) and (
+ (not ns) or ('%s' % item['id']).lower().endswith(ns.lower())) and (
+ (not nl) or nl.lower() in ('%s' % item['id']).lower())]
+
+ def _exact_id_filter(self, items):
+ return filter_dicts_by_dict(items, dict(id=self['id'])) if (
+ self['id']) else items
+
+ def _filter_by_id(self, items):
+ return self._non_exact_id_filter(self._exact_id_filter(items))