def usage(*, globalvars=globals()):
"""Prints this usage message and exits"""
print("Usage: {} <command> [arguments]".format(argv[0]))
doc = _get_calling_module().__doc__
if doc:
print(doc.strip())
print("\nCommands:")
fs = {}
for k, v in globalvars.items():
l = fs.setdefault(v, [])
l.append(k)
l.sort() # TODO hmm...
for func, cmds in sorted(fs.items(), key=lambda tpl: tpl[1][0]):
specs = _getargs(func)
variables = []
for v in specs.args:
vv = v
if v in specs.annotations:
vv += ":" + specs.annotations[v].__name__
# TODO missing kw[only]args
variables.append(vv)
print(' | '.join(sorted(cmds)), ' '.join(variables))
docs = func.__doc__
if docs:
docs = docs.strip()
print(" ", docs)
exit(0)
def args(arguments: _split=argv[1:], stdin_char: str='-',
**mappings):
"""Parses the cmd-arguments to execute the requested command.
:param arguments: The arguments to use (default: argv[1:]).
:param stdin_char: The char to denote stdin-input. Use
None to disable (default: '-').
:param mappings: Additional/custom mappings
(i.e. {'command': function}).
"""
# getting arguments
try:
if stdin_char is not None and arguments[-1] == stdin_char:
del arguments[-1]
while True:
try:
i = input()
arguments.append(i)
except EOFError:
break
f, *a = arguments
except (ValueError, IndexError):
print("ERROR: Requires function or --help!", file=stderr)
exit(1)
# get caller
module = _get_calling_module()
# getting functions
funcs = {k: v for k, v in vars(module).items()
if not k.startswith('_') and _isfunction(v)}
funcs['--help'] = funcs['-h'] = funcs['help'] = usage
funcs['--list-commands'] = commands
funcs.update(mappings) # use defined mappings
if f not in funcs:
raise Exception("Invalid function '{}'.".format(f))
# getting this function
ff = funcs[f]
# converting arguments
fspecs = _getargs(ff)
for i, (val, name) in enumerate(zip(a, fspecs.args)):
if name in fspecs.annotations:
a[i] = fspecs.annotations[name](val)
if fspecs.varargs:
l = len(fspecs.args)
vals = a[l:]
if fspecs.varargs in fspecs.annotations:
a[l:] = fspecs.annotations[fspecs.varargs](vals)
# executing
if ff in {usage, commands}:
ff(globalvars=funcs)
else:
return ff(*a)
