def _get_args_from_signature(function):
if getattr(function, ATTR_EXPECTS_NAMESPACE_OBJECT, False):
return
spec = compat.getargspec(function)
names = get_arg_names(function)
kwargs = dict(zip(*[reversed(x) for x in (names, spec.defaults or [])]))
if sys.version_info < (3, 0):
annotations = {}
else:
annotations = dict((k, v) for k, v in function.__annotations__.items()
if isinstance(v, str))
# define the list of conflicting option strings
# (short forms, i.e. single-character ones)
chars = [a[0] for a in names]
char_counts = dict((char, chars.count(char)) for char in set(chars))
conflicting_opts = tuple(char for char in char_counts
if 1 < char_counts[char])
for name in names:
flags = [] # name_or_flags
akwargs = {} # keyword arguments for add_argument()
if name in annotations:
# help message: func(a : "b") -> add_argument("a", help="b")
akwargs.update(help=annotations.get(name))
if name in kwargs:
akwargs.update(default=kwargs.get(name))
flags = ('-{0}'.format(name[0]), '--{0}'.format(name))
if name.startswith(conflicting_opts):
# remove short name
flags = flags[1:]
else:
# positional argument
flags = (name, )
# cmd(foo_bar) -> add_argument('foo-bar')
flags = tuple(x.replace('_', '-') for x in flags)
yield dict(option_strings=flags, **akwargs)
if spec.varargs:
# *args
yield dict(option_strings=[spec.varargs], nargs='*')
def _get_args_from_signature(function):
if getattr(function, ATTR_EXPECTS_NAMESPACE_OBJECT, False):
return
spec = compat.getargspec(function)
names = get_arg_names(function)
kwargs = dict(zip(*[reversed(x) for x in (names, spec.defaults or [])]))
if sys.version_info < (3,0):
annotations = {}
else:
annotations = dict((k,v) for k,v in function.__annotations__.items()
if isinstance(v, str))
# define the list of conflicting option strings
# (short forms, i.e. single-character ones)
chars = [a[0] for a in names]
char_counts = dict((char, chars.count(char)) for char in set(chars))
conflicting_opts = tuple(char for char in char_counts
if 1 < char_counts[char])
for name in names:
flags = [] # name_or_flags
akwargs = {} # keyword arguments for add_argument()
if name in annotations:
# help message: func(a : "b") -> add_argument("a", help="b")
akwargs.update(help=annotations.get(name))
if name in kwargs:
akwargs.update(default=kwargs.get(name))
flags = ('-{0}'.format(name[0]), '--{0}'.format(name))
if name.startswith(conflicting_opts):
# remove short name
flags = flags[1:]
else:
# positional argument
flags = (name,)
# cmd(foo_bar) -> add_argument('foo-bar')
flags = tuple(x.replace('_', '-') for x in flags)
yield dict(option_strings=flags, **akwargs)
if spec.varargs:
# *args
yield dict(option_strings=[spec.varargs], nargs='*')
def _call():
# Actually call the function
if getattr(args.function, ATTR_EXPECTS_NAMESPACE_OBJECT, False):
result = args.function(args)
else:
# namespace -> dictionary
_flat_key = lambda key: key.replace('-', '_')
all_input = dict((_flat_key(k), v) for k,v in vars(args).items())
# filter the namespace variables so that only those expected by the
# actual function will pass
spec = compat.getargspec(args.function)
names = get_arg_names(args.function)
positional = [all_input[k] for k in names]
keywords = {}
# *args
if spec.varargs:
positional += getattr(args, spec.varargs)
# **kwargs
varkw = getattr(spec, 'varkw', getattr(spec, 'keywords', []))
if varkw:
not_kwargs = ['function'] + spec.args + [spec.varargs]
extra = [k for k in vars(args) if k not in not_kwargs]
for k in extra:
keywords[k] = getattr(args, k)
result = args.function(*positional, **keywords)
# Yield the results
if isinstance(result, (GeneratorType, list, tuple)):
# yield each line ASAP, convert CommandError message to a line
for line in result:
yield line
else:
# yield non-empty non-iterable result as a single line
if result is not None:
yield result
def _call():
# Actually call the function
if getattr(args.function, ATTR_EXPECTS_NAMESPACE_OBJECT, False):
result = args.function(args)
else:
# namespace -> dictionary
_flat_key = lambda key: key.replace('-', '_')
all_input = dict((_flat_key(k), v) for k, v in vars(args).items())
# filter the namespace variables so that only those expected by the
# actual function will pass
spec = compat.getargspec(args.function)
names = get_arg_names(args.function)
positional = [all_input[k] for k in names]
keywords = {}
# *args
if spec.varargs:
positional += getattr(args, spec.varargs)
# **kwargs
varkw = getattr(spec, 'varkw', getattr(spec, 'keywords', []))
if varkw:
not_kwargs = ['function'] + spec.args + [spec.varargs]
extra = [k for k in vars(args) if k not in not_kwargs]
for k in extra:
keywords[k] = getattr(args, k)
result = args.function(*positional, **keywords)
# Yield the results
if isinstance(result, (GeneratorType, list, tuple)):
# yield each line ASAP, convert CommandError message to a line
for line in result:
yield line
else:
# yield non-empty non-iterable result as a single line
if result is not None:
yield result
