"""
Useful functions used for implementing constraints and predicates.
"""
from typing import Any, Dict, Iterable, List
from click import Argument, Context, Option, Parameter
[docs]def param_value_is_set(param: Parameter, value: Any) -> bool:
"""Defines what it means for a parameter of a specific kind to be "set".
All cases are obvious besides that of boolean options:
- (common rule) if the value is ``None``, the parameter is unset;
- a parameter that takes multiple values is set if at least one argument is provided;
- a boolean **flag** is set only if True;
- a boolean option is set if not None, even if it's False.
"""
if value is None:
return False
elif isinstance(param, Option) and param.is_bool_flag:
return value
elif param.nargs != 1 or param.multiple:
return len(value) > 0
return True
[docs]def get_params_whose_value_is_set(
params: Iterable[Parameter], values: Dict[str, Any]
) -> List[Parameter]:
"""Filters ``params`` returning only the parameters that have a value.
Boolean flags are considered "set" if their value is ``True``."""
return [p for p in params if param_value_is_set(p, values[p.name])]
[docs]def get_required_params(params: Iterable[Parameter]) -> List[Parameter]:
return [p for p in params if p.required]
[docs]def get_param_label(param: Parameter) -> str:
if param.param_type_name == 'argument':
return param.human_readable_name
return sorted(param.opts, key=len)[-1]
[docs]def join_param_labels(params: Iterable[Parameter], sep: str = ', ') -> str:
return sep.join(get_param_label(p) for p in params)
[docs]def param_label_by_name(ctx, name: str) -> str:
return get_param_label(ctx.command.get_param_by_name(name))
[docs]def param_value_by_name(ctx: Context, name: str) -> Any:
try:
return ctx.params[name]
except KeyError:
raise KeyError(f'"{name}" is not the name of a CLI parameter')